libre.sh/keycloak-scim
3
0
Fork 0
Code Issues 15 Pull requests Releases Packages Activity Actions

Merge pull request #507 from stianst/master

Browse source 
Removed model methods from KeycloakSession
This commit is contained in:
Stian Thorgersen 2014-07-08 15:28:13 +01:00
commit 20bf888d73
40 changed files with 196 additions and 436 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
audit/email/src/main/java/org/keycloak/audit/email/EmailAuditListener.java
View file

@ -7,6 +7,7 @@ import org.keycloak.audit.EventType;
import org.keycloak.email.EmailException;
import org.keycloak.email.EmailProvider;
import org.keycloak.models.KeycloakSession;
import org.keycloak.models.ModelProvider;
import org.keycloak.models.RealmModel;
import org.keycloak.models.UserModel;
@ -20,11 +21,13 @@ public class EmailAuditListener implements AuditListener {
private static final Logger log = Logger.getLogger(EmailAuditListener.class);
private KeycloakSession session;
private ModelProvider model;
private EmailProvider emailProvider;
private Set<EventType> includedEvents;
public EmailAuditListener(KeycloakSession session, EmailProvider emailProvider, Set<EventType> includedEvents) {
this.session = session;
this.model = session.getModel();
this.emailProvider = emailProvider;
this.includedEvents = includedEvents;
}
@ -33,7 +36,7 @@ public class EmailAuditListener implements AuditListener {
public void onEvent(Event event) {
if (includedEvents.contains(event.getEvent())) {
if (event.getRealmId() != null && event.getUserId() != null) {
RealmModel realm = session.getRealm(event.getRealmId());
RealmModel realm = model.getRealm(event.getRealmId());
UserModel user = realm.getUserById(event.getUserId());
if (user != null && user.getEmail() != null && user.isEmailVerified()) {
try {

15
authentication/authentication-model/src/main/java/org/keycloak/authentication/model/ExternalModelAuthenticationProvider.java
View file

@ -1,9 +1,9 @@
package org.keycloak.authentication.model;
import org.jboss.resteasy.spi.ResteasyProviderFactory;
import org.keycloak.authentication.AuthProviderConstants;
import org.keycloak.authentication.AuthenticationProviderException;
import org.keycloak.models.KeycloakSession;
import org.keycloak.models.ModelProvider;
import org.keycloak.models.RealmModel;
import java.util.Arrays;
@ -17,7 +17,10 @@ import java.util.Map;
*/
public class ExternalModelAuthenticationProvider extends AbstractModelAuthenticationProvider {
public ExternalModelAuthenticationProvider() {
private ModelProvider model;
public ExternalModelAuthenticationProvider(KeycloakSession session) {
this.model = session.getModel();
}
@Override
@ -37,13 +40,7 @@ public class ExternalModelAuthenticationProvider extends AbstractModelAuthentica
throw new AuthenticationProviderException("Option '" + AuthProviderConstants.EXTERNAL_REALM_ID + "' not specified in configuration");
}
// TODO: This won't be needed when KeycloakSession is available from ProviderSession
KeycloakSession session = ResteasyProviderFactory.getContextData(KeycloakSession.class);
if (session == null) {
throw new AuthenticationProviderException("KeycloakSession not available");
}
RealmModel realm = session.getRealm(realmId);
RealmModel realm = model.getRealm(realmId);
if (realm == null) {
throw new AuthenticationProviderException("Realm with id '" + realmId + "' doesn't exists");
}

2
authentication/authentication-model/src/main/java/org/keycloak/authentication/model/ExternalModelAuthenticationProviderFactory.java
View file

@ -13,7 +13,7 @@ public class ExternalModelAuthenticationProviderFactory implements Authenticatio
@Override
public AuthenticationProvider create(KeycloakSession session) {
return new ExternalModelAuthenticationProvider();
return new ExternalModelAuthenticationProvider(session);
}
@Override

4
export-import/export-import-impl/src/main/java/org/keycloak/exportimport/ExportImportProviderImpl.java
View file

@ -41,11 +41,11 @@ public class ExportImportProviderImpl implements ExportImportProvider {
if (export) {
ExportWriter exportWriter = getProvider().getExportWriter();
new ModelExporter().exportModel(session, exportWriter);
new ModelExporter().exportModel(session.getModel(), exportWriter);
logger.infof("Export finished successfully");
} else {
ImportReader importReader = getProvider().getImportReader();
new ModelImporter().importModel(session, importReader);
new ModelImporter().importModel(session.getModel(), importReader);
logger.infof("Import finished successfully");
}

49
export-import/export-import-impl/src/main/java/org/keycloak/exportimport/ModelExporter.java
View file

@ -12,6 +12,7 @@ import org.keycloak.models.AuthenticationLinkModel;
import org.keycloak.models.AuthenticationProviderModel;
import org.keycloak.models.ClientModel;
import org.keycloak.models.KeycloakSession;
import org.keycloak.models.ModelProvider;
import org.keycloak.models.OAuthClientModel;
import org.keycloak.models.RealmModel;
import org.keycloak.models.RequiredCredentialModel;
@ -43,24 +44,24 @@ public class ModelExporter {
private ExportWriter exportWriter;
private ExportImportPropertiesManager propertiesManager;
public void exportModel(KeycloakSession session, ExportWriter exportWriter) {
public void exportModel(ModelProvider model, ExportWriter exportWriter) {
// Initialize needed objects
this.exportWriter = exportWriter;
this.propertiesManager = new ExportImportPropertiesManager();
// Create separate files for "realms", "applications", "oauthClients", "roles" and finally "users". Users may be done in more files (pagination)
exportRealms(session, "realms.json");
exportApplications(session, "applications.json");
exportOAuthClients(session, "oauthClients.json");
exportRoles(session, "roles.json");
exportUsers(session, "users.json");
exportUserFailures(session, "userFailures.json");
exportRealms(model, "realms.json");
exportApplications(model, "applications.json");
exportOAuthClients(model, "oauthClients.json");
exportRoles(model, "roles.json");
exportUsers(model, "users.json");
exportUserFailures(model, "userFailures.json");
this.exportWriter.closeExportWriter();
}
protected void exportRealms(KeycloakSession session, String fileName) {
List<RealmModel> realms = session.getRealms();
protected void exportRealms(ModelProvider model, String fileName) {
List<RealmModel> realms = model.getRealms();
// Convert models to entities, which will be written into JSON file
List<RealmEntity> result = new LinkedList<RealmEntity>();
@ -106,8 +107,8 @@ public class ModelExporter {
logger.infof("Realms exported: " + result);
}
protected void exportApplications(KeycloakSession session, String fileName) {
List<ApplicationModel> allApplications = getAllApplications(session);
protected void exportApplications(ModelProvider model, String fileName) {
List<ApplicationModel> allApplications = getAllApplications(model);
List<ApplicationEntity> result = new LinkedList<ApplicationEntity>();
for (ApplicationModel appModel : allApplications) {
@ -129,8 +130,8 @@ public class ModelExporter {
logger.infof("Applications exported: " + result);
}
protected void exportOAuthClients(KeycloakSession session, String fileName) {
List<RealmModel> realms = session.getRealms();
protected void exportOAuthClients(ModelProvider model, String fileName) {
List<RealmModel> realms = model.getRealms();
List<OAuthClientModel> allClients = new ArrayList<OAuthClientModel>();
for (RealmModel realmModel : realms) {
allClients.addAll(realmModel.getOAuthClients());
@ -156,8 +157,8 @@ public class ModelExporter {
logger.infof("OAuth clients exported: " + result);
}
protected void exportRoles(KeycloakSession session, String fileName) {
List<RoleModel> allRoles = getAllRoles(session);
protected void exportRoles(ModelProvider model, String fileName) {
List<RoleModel> allRoles = getAllRoles(model);
List<RoleEntity> result = new LinkedList<RoleEntity>();
for (RoleModel roleModel : allRoles) {
@ -198,8 +199,8 @@ public class ModelExporter {
}
}
protected void exportUsers(KeycloakSession session, String fileName) {
List<RealmModel> realms = session.getRealms();
protected void exportUsers(ModelProvider model, String fileName) {
List<RealmModel> realms = model.getRealms();
List<UserEntity> result = new LinkedList<UserEntity>();
for (RealmModel realm : realms) {
@ -277,8 +278,8 @@ public class ModelExporter {
// Does it makes sense to export user failures ?
protected void exportUserFailures(KeycloakSession session, String fileName) {
List<RealmModel> realms = session.getRealms();
protected void exportUserFailures(ModelProvider model, String fileName) {
List<RealmModel> realms = model.getRealms();
List<UsernameLoginFailureModel> allFailures = new ArrayList<UsernameLoginFailureModel>();
for (RealmModel realmModel : realms) {
allFailures.addAll(realmModel.getAllUserLoginFailures());
@ -306,8 +307,8 @@ public class ModelExporter {
return scopeIds;
}
private List<ApplicationModel> getAllApplications(KeycloakSession session) {
List<RealmModel> realms = session.getRealms();
private List<ApplicationModel> getAllApplications(ModelProvider model) {
List<RealmModel> realms = model.getRealms();
List<ApplicationModel> allApplications = new ArrayList<ApplicationModel>();
for (RealmModel realmModel : realms) {
allApplications.addAll(realmModel.getApplications());
@ -315,15 +316,15 @@ public class ModelExporter {
return allApplications;
}
private List<RoleModel> getAllRoles(KeycloakSession session) {
private List<RoleModel> getAllRoles(ModelProvider model) {
List<RoleModel> allRoles = new ArrayList<RoleModel>();
List<RealmModel> realms = session.getRealms();
List<RealmModel> realms = model.getRealms();
for (RealmModel realmModel : realms) {
allRoles.addAll(realmModel.getRoles());
}
List<ApplicationModel> allApplications = getAllApplications(session);
List<ApplicationModel> allApplications = getAllApplications(model);
for (ApplicationModel appModel : allApplications) {
allRoles.addAll(appModel.getRoles());
}

70
export-import/export-import-impl/src/main/java/org/keycloak/exportimport/ModelImporter.java
View file

@ -15,6 +15,7 @@ import org.keycloak.models.AuthenticationProviderModel;
import org.keycloak.models.ClientModel;
import org.keycloak.Config;
import org.keycloak.models.KeycloakSession;
import org.keycloak.models.ModelProvider;
import org.keycloak.models.OAuthClientModel;
import org.keycloak.models.PasswordPolicy;
import org.keycloak.models.RealmModel;
@ -47,34 +48,31 @@ public class ModelImporter {
private ImportReader importReader;
private ExportImportPropertiesManager propertiesManager;
public void importModel(KeycloakSession session, ImportReader importReader) {
public void importModel(ModelProvider model, ImportReader importReader) {
// Initialize needed objects
this.importReader = importReader;
this.propertiesManager = new ExportImportPropertiesManager();
// Delete all the data from current model
session.removeAllData();
importRealms(session, "realms.json");
importApplications(session, "applications.json");
importRoles(session, "roles.json");
importRealms(model, "realms.json");
importApplications(model, "applications.json");
importRoles(model, "roles.json");
// Now we have all realms,applications and roles filled. So fill other objects (default roles, scopes etc)
importRealmsStep2(session, "realms.json");
importApplicationsStep2(session, "applications.json");
importRealmsStep2(model, "realms.json");
importApplicationsStep2(model, "applications.json");
importOAuthClients(session, "oauthClients.json");
importUsers(session, "users.json");
importUserFailures(session, "userFailures.json");
importOAuthClients(model, "oauthClients.json");
importUsers(model, "users.json");
importUserFailures(model, "userFailures.json");
this.importReader.closeImportReader();
}
protected void importRealms(KeycloakSession session, String fileName) {
protected void importRealms(ModelProvider model, String fileName) {
List<RealmEntity> realms = this.importReader.readEntities(fileName, RealmEntity.class);
for (RealmEntity realmEntity : realms) {
RealmModel realm = session.createRealm(realmEntity.getId(), realmEntity.getName());
RealmModel realm = model.createRealm(realmEntity.getId(), realmEntity.getName());
this.propertiesManager.setBasicPropertiesToModel(realm, realmEntity);
@ -103,10 +101,10 @@ public class ModelImporter {
logger.infof("Realms imported: " + realms);
}
protected void importApplications(KeycloakSession session, String fileName) {
protected void importApplications(ModelProvider model, String fileName) {
List<ApplicationEntity> apps = this.importReader.readEntities(fileName, ApplicationEntity.class);
for (ApplicationEntity appEntity : apps) {
RealmModel realm = session.getRealm(appEntity.getRealmId());
RealmModel realm = model.getRealm(appEntity.getRealmId());
ApplicationModel app = realm.addApplication(appEntity.getId(), appEntity.getName());
this.propertiesManager.setBasicPropertiesToModel(app , appEntity);
@ -117,7 +115,7 @@ public class ModelImporter {
logger.infof("Applications imported: " + apps);
}
protected void importRoles(KeycloakSession session, String fileName) {
protected void importRoles(ModelProvider model, String fileName) {
// helper map for composite roles
Map<String, RoleEntity> rolesMap = new HashMap<String, RoleEntity>();
@ -125,10 +123,10 @@ public class ModelImporter {
for (RoleEntity roleEntity : roles) {
RoleModel role = null;
if (roleEntity.getRealmId() != null) {
RealmModel realm = session.getRealm(roleEntity.getRealmId());
RealmModel realm = model.getRealm(roleEntity.getRealmId());
role = realm.addRole(roleEntity.getId(), roleEntity.getName());
} else if (roleEntity.getApplicationId() != null) {
ApplicationModel app = findApplicationById(session, roleEntity.getApplicationId());
ApplicationModel app = findApplicationById(model, roleEntity.getApplicationId());
role = app.addRole(roleEntity.getId(), roleEntity.getName());
} else {
throw new IllegalStateException("Role " + roleEntity.getId() + " doesn't have realmId nor applicationId");
@ -140,7 +138,7 @@ public class ModelImporter {
}
// All roles were added. Fill composite roles now
for (RealmModel realm : session.getRealms()) {
for (RealmModel realm : model.getRealms()) {
// realm roles
fillCompositeRoles(rolesMap, realm, realm);
@ -172,12 +170,12 @@ public class ModelImporter {
}
}
protected void importRealmsStep2(KeycloakSession session, String fileName) {
protected void importRealmsStep2(ModelProvider model, String fileName) {
List<RealmEntity> realms = this.importReader.readEntities(fileName, RealmEntity.class);
RealmModel adminRealm = session.getRealm(Config.getAdminRealm());
RealmModel adminRealm = model.getRealm(Config.getAdminRealm());
for (RealmEntity realmEntity : realms) {
RealmModel realm = session.getRealm(realmEntity.getId());
RealmModel realm = model.getRealm(realmEntity.getId());
// admin app
String adminAppId = realmEntity.getAdminAppId();
@ -190,10 +188,10 @@ public class ModelImporter {
}
}
protected void importApplicationsStep2(KeycloakSession session, String fileName) {
protected void importApplicationsStep2(ModelProvider model, String fileName) {
List<ApplicationEntity> apps = this.importReader.readEntities(fileName, ApplicationEntity.class);
for (ApplicationEntity appEntity : apps) {
RealmModel realm = session.getRealm(appEntity.getRealmId());
RealmModel realm = model.getRealm(appEntity.getRealmId());
ApplicationModel application = realm.getApplicationById(appEntity.getId());
// Default roles
@ -211,10 +209,10 @@ public class ModelImporter {
}
}
protected void importOAuthClients(KeycloakSession session, String fileName) {
protected void importOAuthClients(ModelProvider model, String fileName) {
List<OAuthClientEntity> clients = this.importReader.readEntities(fileName, OAuthClientEntity.class);
for (OAuthClientEntity clientEntity : clients) {
RealmModel realm = session.getRealm(clientEntity.getRealmId());
RealmModel realm = model.getRealm(clientEntity.getRealmId());
OAuthClientModel client = realm.addOAuthClient(clientEntity.getId(), clientEntity.getName());
this.propertiesManager.setBasicPropertiesToModel(client, clientEntity);
@ -228,8 +226,8 @@ public class ModelImporter {
logger.info("OAuth clients imported: " + clients);
}
protected ApplicationModel findApplicationById(KeycloakSession session, String applicationId) {
for (RealmModel realm : session.getRealms()) {
protected ApplicationModel findApplicationById(ModelProvider model, String applicationId) {
for (RealmModel realm : model.getRealms()) {
ApplicationModel appModel = realm.getApplicationById(applicationId);
if (appModel != null) {
return appModel;
@ -239,10 +237,10 @@ public class ModelImporter {
return null;
}
public void importUsers(KeycloakSession session, String fileName) {
public void importUsers(ModelProvider model, String fileName) {
List<UserEntity> users = this.importReader.readEntities(fileName, UserEntity.class);
for (UserEntity userEntity : users) {
RealmModel realm = session.getRealm(userEntity.getRealmId());
RealmModel realm = model.getRealm(userEntity.getRealmId());
UserModel user = realm.addUser(userEntity.getId(), userEntity.getUsername(), false);
// We need to remove defaultRoles here as realm.addUser is automatically adding them. We may add them later during roles mapping processing
@ -313,16 +311,16 @@ public class ModelImporter {
}
}
public void importUserFailures(KeycloakSession session, String fileName) {
public void importUserFailures(ModelProvider model, String fileName) {
List<UsernameLoginFailureEntity> userFailures = this.importReader.readEntities(fileName, UsernameLoginFailureEntity.class);
for (UsernameLoginFailureEntity entity : userFailures) {
RealmModel realm = session.getRealm(entity.getRealmId());
UsernameLoginFailureModel model = realm.addUserLoginFailure(entity.getUsername());
RealmModel realm = model.getRealm(entity.getRealmId());
UsernameLoginFailureModel loginFailureModel = realm.addUserLoginFailure(entity.getUsername());
this.propertiesManager.setBasicPropertiesToModel(model , entity);
this.propertiesManager.setBasicPropertiesToModel(loginFailureModel , entity);
for (int i=0 ; i<entity.getNumFailures() ; i++) {
model.incrementFailures();
loginFailureModel.incrementFailures();
}
}
}

4
export-import/export-import-impl/src/test/java/org/keycloak/exportimport/ExportImportTestBase.java
View file

@ -55,7 +55,7 @@ public abstract class ExportImportTestBase {
exportModel(factory);
beginTransaction();
realm = session.getRealm("demo");
realm = session.getModel().getRealm("demo");
String wburkeId = realm.getUser("wburke").getId();
String appId = realm.getApplicationByName("Application").getId();
@ -72,7 +72,7 @@ public abstract class ExportImportTestBase {
// Verify it's imported in mongo (reusing ImportTest)
beginTransaction();
RealmModel importedRealm = session.getRealm("demo");
RealmModel importedRealm = session.getModel().getRealm("demo");
System.out.println("Exported realm: " + realm + ", Imported realm: " + importedRealm);
Assert.assertEquals(wburkeId, importedRealm.getUser("wburke").getId());

41
model/api/src/main/java/org/keycloak/models/KeycloakSession.java
View file

@ -23,45 +23,8 @@ public interface KeycloakSession {
<T extends Provider> Set<T> getAllProviders(Class<T> clazz);
RealmModel createRealm(String name);
RealmModel createRealm(String id, String name);
RealmModel getRealm(String id);
RealmModel getRealmByName(String name);
UserModel getUserById(String id, RealmModel realm);
UserModel getUserByUsername(String username, RealmModel realm);
UserModel getUserByEmail(String email, RealmModel realm);
UserModel getUserBySocialLink(SocialLinkModel socialLink, RealmModel realm);
List<UserModel> getUsers(RealmModel realm);
List<UserModel> searchForUser(String search, RealmModel realm);
List<UserModel> searchForUserByAttributes(Map<String, String> attributes, RealmModel realm);
Set<SocialLinkModel> getSocialLinks(UserModel user, RealmModel realm);
SocialLinkModel getSocialLink(UserModel user, String socialProvider, RealmModel realm);
RoleModel getRoleById(String id, RealmModel realm);
ApplicationModel getApplicationById(String id, RealmModel realm);
OAuthClientModel getOAuthClientById(String id, RealmModel realm);
List<RealmModel> getRealms();
boolean removeRealm(String id);
UsernameLoginFailureModel getUserLoginFailure(String username, RealmModel realm);
UsernameLoginFailureModel addUserLoginFailure(String username, RealmModel realm);
List<UsernameLoginFailureModel> getAllUserLoginFailures(RealmModel realm);
UserSessionModel createUserSession(RealmModel realm, UserModel user, String ipAddress);
UserSessionModel getUserSession(String id, RealmModel realm);
List<UserSessionModel> getUserSessions(UserModel user, RealmModel realm);
Set<UserSessionModel> getUserSessions(RealmModel realm, ClientModel client);
int getActiveUserSessions(RealmModel realm, ClientModel client);
void removeUserSession(UserSessionModel session);
void removeUserSessions(RealmModel realm, UserModel user);
void removeExpiredUserSessions(RealmModel realm);
void removeUserSessions(RealmModel realm);
void removeAllData();
ModelProvider getModel();
void close();
}

3
model/api/src/main/java/org/keycloak/models/ModelProvider.java
View file

@ -52,8 +52,5 @@ public interface ModelProvider extends Provider {
void removeExpiredUserSessions(RealmModel realm);
void removeUserSessions(RealmModel realm);
void removeAllData();
void close();
}

4
model/hybrid/src/main/java/org/keycloak/models/hybrid/HybridModelProvider.java
View file

@ -245,10 +245,6 @@ public class HybridModelProvider implements ModelProvider {
sessions().removeUserSessions(realm.getId());
}
@Override
public void removeAllData() {
}
@Override
public void close() {
}

7
model/invalidation-cache/model-adapters/src/main/java/org/keycloak/models/cache/DefaultCacheModelProvider.java vendored
View file

@ -291,13 +291,6 @@ public class DefaultCacheModelProvider implements CacheModelProvider {
return didIt;
}
@Override
public void removeAllData() {
cache.clear();
getDelegate().removeAllData();
clearAll = true;
}
@Override
public void close() {
if (delegate != null) delegate.close();

5
model/invalidation-cache/model-adapters/src/main/java/org/keycloak/models/cache/NoCacheModelProvider.java vendored
View file

@ -156,11 +156,6 @@ public class NoCacheModelProvider implements CacheModelProvider {
return getDelegate().removeRealm(id);
}
@Override
public void removeAllData() {
getDelegate().removeAllData();
}
@Override
public void close() {
if (delegate != null) delegate.close();

9
model/jpa/src/main/java/org/keycloak/models/jpa/JpaModelProvider.java
View file

@ -144,15 +144,6 @@ public class JpaModelProvider implements ModelProvider {
if (em.isOpen()) em.close();
}
@Override
public void removeAllData() {
// Should be sufficient to delete all realms. Rest data should be removed in cascade
List<RealmModel> realms = getRealms();
for (RealmModel realm : realms) {
removeRealm(realm.getId());
}
}
@Override
public UserModel getUserBySocialLink(SocialLinkModel socialLink, RealmModel realm) {
TypedQuery<UserEntity> query = em.createNamedQuery("findUserByLinkAndRealm", UserEntity.class);

47
model/jpa/src/main/java/org/keycloak/models/jpa/RealmAdapter.java
View file

@ -6,6 +6,7 @@ import org.keycloak.models.ClientModel;
import org.keycloak.models.CredentialValidation;
import org.keycloak.models.KeycloakSession;
import org.keycloak.models.ModelDuplicateException;
import org.keycloak.models.ModelProvider;
import org.keycloak.models.RoleContainerModel;
import org.keycloak.models.UserCredentialValueModel;
import org.keycloak.models.UserSessionModel;
@ -66,10 +67,12 @@ public class RealmAdapter implements RealmModel {
protected volatile transient PublicKey publicKey;
protected volatile transient PrivateKey privateKey;
protected KeycloakSession session;
protected ModelProvider model;
private PasswordPolicy passwordPolicy;
public RealmAdapter(KeycloakSession session, EntityManager em, RealmEntity realm) {
this.session = session;
this.model = session.getModel();
this.em = em;
this.realm = realm;
}
@ -426,32 +429,32 @@ public class RealmAdapter implements RealmModel {
@Override
public UserModel getUser(String name) {
return session.getUserByUsername(name, this);
return model.getUserByUsername(name, this);
}
@Override
public UsernameLoginFailureModel getUserLoginFailure(String username) {
return session.getUserLoginFailure(username, this);
return model.getUserLoginFailure(username, this);
}
@Override
public UsernameLoginFailureModel addUserLoginFailure(String username) {
return session.addUserLoginFailure(username, this);
return model.addUserLoginFailure(username, this);
}
@Override
public List<UsernameLoginFailureModel> getAllUserLoginFailures() {
return session.getAllUserLoginFailures(this);
return model.getAllUserLoginFailures(this);
}
@Override
public UserModel getUserByEmail(String email) {
return session.getUserByEmail(email, this);
return model.getUserByEmail(email, this);
}
@Override
public UserModel getUserById(String id) {
return session.getUserById(id, this);
return model.getUserById(id, this);
}
@Override
@ -661,7 +664,7 @@ public class RealmAdapter implements RealmModel {
@Override
public ApplicationModel getApplicationById(String id) {
return session.getApplicationById(id, this);
return model.getApplicationById(id, this);
}
@Override
@ -671,17 +674,17 @@ public class RealmAdapter implements RealmModel {
@Override
public UserModel getUserBySocialLink(SocialLinkModel socialLink) {
return session.getUserBySocialLink(socialLink, this);
return model.getUserBySocialLink(socialLink, this);
}
@Override
public Set<SocialLinkModel> getSocialLinks(UserModel user) {
return session.getSocialLinks(user, this);
return model.getSocialLinks(user, this);
}
@Override
public SocialLinkModel getSocialLink(UserModel user, String socialProvider) {
return session.getSocialLink(user, socialProvider, this);
return model.getSocialLink(user, socialProvider, this);
}
@Override
@ -744,17 +747,17 @@ public class RealmAdapter implements RealmModel {
@Override
public List<UserModel> getUsers() {
return session.getUsers(this);
return model.getUsers(this);
}
@Override
public List<UserModel> searchForUser(String search) {
return session.searchForUser(search, this);
return model.searchForUser(search, this);
}
@Override
public List<UserModel> searchForUserByAttributes(Map<String, String> attributes) {
return session.searchForUserByAttributes(attributes, this);
return model.searchForUserByAttributes(attributes, this);
}
@Override
@ -798,7 +801,7 @@ public class RealmAdapter implements RealmModel {
@Override
public OAuthClientModel getOAuthClientById(String id) {
return session.getOAuthClientById(id, this);
return model.getOAuthClientById(id, this);
}
@ -960,7 +963,7 @@ public class RealmAdapter implements RealmModel {
@Override
public RoleModel getRoleById(String id) {
return session.getRoleById(id, this);
return model.getRoleById(id, this);
}
@Override
@ -1105,38 +1108,38 @@ public class RealmAdapter implements RealmModel {
@Override
public UserSessionModel createUserSession(UserModel user, String ipAddress) {
return session.createUserSession(this, user, ipAddress);
return model.createUserSession(this, user, ipAddress);
}
@Override
public UserSessionModel getUserSession(String id) {
return session.getUserSession(id, this);
return model.getUserSession(id, this);
}
@Override
public List<UserSessionModel> getUserSessions(UserModel user) {
return session.getUserSessions(user, this);
return model.getUserSessions(user, this);
}
@Override
public void removeUserSession(UserSessionModel session) {
this.session.removeUserSession(session);
this.model.removeUserSession(session);
}
@Override
public void removeUserSessions() {
session.removeUserSessions(this);
model.removeUserSessions(this);
}
@Override
public void removeUserSessions(UserModel user) {
session.removeUserSessions(this, user);
model.removeUserSessions(this, user);
}
@Override
public void removeExpiredUserSessions() {
session.removeExpiredUserSessions(this);
model.removeExpiredUserSessions(this);
}
}

4
model/mongo/src/main/java/org/keycloak/models/mongo/api/MongoStore.java
View file

@ -41,8 +41,4 @@ public interface MongoStore {
<S> boolean pullItemFromList(MongoIdentifiableEntity entity, String listPropertyName, S itemToPull, MongoStoreInvocationContext context);
/**
* Completely remove all data from DB
*/
void removeAllEntities();
}

7
model/mongo/src/main/java/org/keycloak/models/mongo/impl/MongoStoreImpl.java
View file

@ -417,13 +417,6 @@ public class MongoStoreImpl implements MongoStore {
}
}
@Override
public void removeAllEntities() {
Set<Class<?>> managedTypes = this.entityInfoCache.keySet();
Class<? extends MongoEntity>[] arrayTemplate = (Class<? extends MongoEntity>[])new Class<?>[0];
this.clearManagedCollections(managedTypes.toArray(arrayTemplate));
}
// Possibility to add user-defined mappers
public void addAppObjectConverter(Mapper<?, ?> mapper) {
mapperRegistry.addAppObjectMapper(mapper);

2
model/mongo/src/main/java/org/keycloak/models/mongo/keycloak/adapters/ApplicationAdapter.java
View file

@ -7,8 +7,6 @@ import org.keycloak.models.ClientModel;
import org.keycloak.models.KeycloakSession;
import org.keycloak.models.RealmModel;
import org.keycloak.models.RoleModel;
import org.keycloak.models.UserModel;
import org.keycloak.models.UserSessionModel;
import org.keycloak.models.mongo.api.context.MongoStoreInvocationContext;
import org.keycloak.models.mongo.keycloak.entities.MongoApplicationEntity;
import org.keycloak.models.mongo.keycloak.entities.MongoRoleEntity;

20
model/mongo/src/main/java/org/keycloak/models/mongo/keycloak/adapters/ClientAdapter.java
View file

@ -1,14 +1,8 @@
package org.keycloak.models.mongo.keycloak.adapters;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import com.mongodb.DBObject;
import com.mongodb.QueryBuilder;
import org.keycloak.models.ClientModel;
import org.keycloak.models.KeycloakSession;
import org.keycloak.models.ModelProvider;
import org.keycloak.models.RealmModel;
import org.keycloak.models.RoleModel;
import org.keycloak.models.UserSessionModel;
@ -16,9 +10,13 @@ import org.keycloak.models.entities.ClientEntity;
import org.keycloak.models.mongo.api.MongoIdentifiableEntity;
import org.keycloak.models.mongo.api.context.MongoStoreInvocationContext;
import org.keycloak.models.mongo.keycloak.entities.MongoRoleEntity;
import org.keycloak.models.mongo.keycloak.entities.MongoUserSessionEntity;
import org.keycloak.models.mongo.utils.MongoModelUtils;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
/**
* @author <a href="mailto:mposolda@redhat.com">Marek Posolda</a>
*/
@ -27,12 +25,14 @@ public abstract class ClientAdapter<T extends MongoIdentifiableEntity> extends A
protected final T clientEntity;
private final RealmModel realm;
protected KeycloakSession session;
private final ModelProvider model;
public ClientAdapter(KeycloakSession session, RealmModel realm, T clientEntity, MongoStoreInvocationContext invContext) {
super(invContext);
this.clientEntity = clientEntity;
this.realm = realm;
this.session = session;
this.model = session.getModel();
}
@Override
@ -176,12 +176,12 @@ public abstract class ClientAdapter<T extends MongoIdentifiableEntity> extends A
@Override
public Set<UserSessionModel> getUserSessions() {
return session.getUserSessions(realm, this);
return model.getUserSessions(realm, this);
}
@Override
public int getActiveUserSessions() {
return session.getActiveUserSessions(realm, this);
return model.getActiveUserSessions(realm, this);
}
@Override

5
model/mongo/src/main/java/org/keycloak/models/mongo/keycloak/adapters/MongoModelProvider.java
View file

@ -67,11 +67,6 @@ public class MongoModelProvider implements ModelProvider {
// TODO
}
@Override
public void removeAllData() {
getMongoStore().removeAllEntities();
}
@Override
public RealmModel createRealm(String name) {
return createRealm(KeycloakModelUtils.generateId(), name);

58
model/mongo/src/main/java/org/keycloak/models/mongo/keycloak/adapters/RealmAdapter.java
View file

@ -1,15 +1,14 @@
package org.keycloak.models.mongo.keycloak.adapters;
import com.mongodb.BasicDBObject;
import com.mongodb.DBObject;
import com.mongodb.QueryBuilder;
import org.jboss.logging.Logger;
import org.keycloak.models.ApplicationModel;
import org.keycloak.models.AuthenticationLinkModel;
import org.keycloak.models.AuthenticationProviderModel;
import org.keycloak.models.ClientModel;
import org.keycloak.models.CredentialValidation;
import org.keycloak.models.KeycloakSession;
import org.keycloak.models.ModelProvider;
import org.keycloak.models.OAuthClientModel;
import org.keycloak.models.PasswordPolicy;
import org.keycloak.models.RealmModel;
@ -21,9 +20,7 @@ import org.keycloak.models.UserCredentialValueModel;
import org.keycloak.models.UserModel;
import org.keycloak.models.UserSessionModel;
import org.keycloak.models.UsernameLoginFailureModel;
import org.keycloak.models.entities.AuthenticationLinkEntity;
import org.keycloak.models.entities.AuthenticationProviderEntity;
import org.keycloak.models.entities.CredentialEntity;
import org.keycloak.models.entities.RequiredCredentialEntity;
import org.keycloak.models.entities.SocialLinkEntity;
import org.keycloak.models.mongo.api.context.MongoStoreInvocationContext;
@ -32,13 +29,8 @@ import org.keycloak.models.mongo.keycloak.entities.MongoOAuthClientEntity;
import org.keycloak.models.mongo.keycloak.entities.MongoRealmEntity;
import org.keycloak.models.mongo.keycloak.entities.MongoRoleEntity;
import org.keycloak.models.mongo.keycloak.entities.MongoUserEntity;
import org.keycloak.models.mongo.keycloak.entities.MongoUserSessionEntity;
import org.keycloak.models.mongo.keycloak.entities.MongoUsernameLoginFailureEntity;
import org.keycloak.models.mongo.utils.MongoModelUtils;
import org.keycloak.models.utils.KeycloakModelUtils;
import org.keycloak.models.utils.Pbkdf2PasswordEncoder;
import org.keycloak.models.utils.TimeBasedOTP;
import org.keycloak.util.Time;
import java.security.PrivateKey;
import java.security.PublicKey;
@ -47,11 +39,9 @@ import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.regex.Pattern;
/**
* @author <a href="mailto:mposolda@redhat.com">Marek Posolda</a>
@ -61,6 +51,7 @@ public class RealmAdapter extends AbstractMongoAdapter<MongoRealmEntity> impleme
private static final Logger logger = Logger.getLogger(RealmAdapter.class);
private final MongoRealmEntity realm;
private final ModelProvider model;
protected volatile transient PublicKey publicKey;
protected volatile transient PrivateKey privateKey;
@ -72,6 +63,7 @@ public class RealmAdapter extends AbstractMongoAdapter<MongoRealmEntity> impleme
super(invocationContext);
this.realm = realmEntity;
this.session = session;
this.model = session.getModel();
}
@Override
@ -450,33 +442,33 @@ public class RealmAdapter extends AbstractMongoAdapter<MongoRealmEntity> impleme
@Override
public UserModel getUser(String name) {
return session.getUserByUsername(name, this);
return model.getUserByUsername(name, this);
}
@Override
public UsernameLoginFailureModel getUserLoginFailure(String name) {
return session.getUserLoginFailure(name, this);
return model.getUserLoginFailure(name, this);
}
@Override
public UsernameLoginFailureModel addUserLoginFailure(String username) {
return session.addUserLoginFailure(username, this);
return model.addUserLoginFailure(username, this);
}
@Override
public List<UsernameLoginFailureModel> getAllUserLoginFailures() {
return session.getAllUserLoginFailures(this);
return model.getAllUserLoginFailures(this);
}
@Override
public UserModel getUserByEmail(String email) {
return session.getUserByEmail(email, this);
return model.getUserByEmail(email, this);
}
@Override
public UserModel getUserById(String id) {
return session.getUserById(id, this);
return model.getUserById(id, this);
}
@Override
@ -584,7 +576,7 @@ public class RealmAdapter extends AbstractMongoAdapter<MongoRealmEntity> impleme
@Override
public RoleModel getRoleById(String id) {
return session.getRoleById(id, this);
return model.getRoleById(id, this);
}
@Override
@ -636,7 +628,7 @@ public class RealmAdapter extends AbstractMongoAdapter<MongoRealmEntity> impleme
@Override
public ApplicationModel getApplicationById(String id) {
return session.getApplicationById(id, this);
return model.getApplicationById(id, this);
}
@Override
@ -727,7 +719,7 @@ public class RealmAdapter extends AbstractMongoAdapter<MongoRealmEntity> impleme
@Override
public OAuthClientModel getOAuthClientById(String id) {
return session.getOAuthClientById(id, this);
return model.getOAuthClientById(id, this);
}
@Override
@ -830,17 +822,17 @@ public class RealmAdapter extends AbstractMongoAdapter<MongoRealmEntity> impleme
@Override
public UserModel getUserBySocialLink(SocialLinkModel socialLink) {
return session.getUserBySocialLink(socialLink, this);
return model.getUserBySocialLink(socialLink, this);
}
@Override
public Set<SocialLinkModel> getSocialLinks(UserModel user) {
return session.getSocialLinks(user, this);
return model.getSocialLinks(user, this);
}
@Override
public SocialLinkModel getSocialLink(UserModel user, String socialProvider) {
return session.getSocialLink(user, socialProvider, this);
return model.getSocialLink(user, socialProvider, this);
}
@Override
@ -893,18 +885,18 @@ public class RealmAdapter extends AbstractMongoAdapter<MongoRealmEntity> impleme
@Override
public List<UserModel> getUsers() {
return session.getUsers(this);
return model.getUsers(this);
}
@Override
public List<UserModel> searchForUser(String search) {
return session.searchForUser(search, this);
return model.searchForUser(search, this);
}
@Override
public List<UserModel> searchForUserByAttributes(Map<String, String> attributes) {
return session.searchForUserByAttributes(attributes, this);
return model.searchForUserByAttributes(attributes, this);
}
@ -1023,37 +1015,37 @@ public class RealmAdapter extends AbstractMongoAdapter<MongoRealmEntity> impleme
@Override
public UserSessionModel createUserSession(UserModel user, String ipAddress) {
return session.createUserSession(this, user, ipAddress);
return model.createUserSession(this, user, ipAddress);
}
@Override
public UserSessionModel getUserSession(String id) {
return session.getUserSession(id, this);
return model.getUserSession(id, this);
}
@Override
public List<UserSessionModel> getUserSessions(UserModel user) {
return session.getUserSessions(user, this);
return model.getUserSessions(user, this);
}
@Override
public void removeUserSession(UserSessionModel session) {
this.session.removeUserSession(session);
this.model.removeUserSession(session);
}
@Override
public void removeUserSessions(UserModel user) {
this.session.removeUserSessions(this, user);
this.model.removeUserSessions(this, user);
}
@Override
public void removeUserSessions() {
this.session.removeUserSessions(this);
this.model.removeUserSessions(this);
}
@Override
public void removeExpiredUserSessions() {
this.session.removeExpiredUserSessions(this);
this.model.removeExpiredUserSessions(this);
}
@Override

7
model/tests/src/test/java/org/keycloak/model/test/AbstractModelTest.java
View file

@ -8,6 +8,7 @@ import org.junit.BeforeClass;
import org.keycloak.Config;
import org.keycloak.models.KeycloakSession;
import org.keycloak.models.KeycloakSessionFactory;
import org.keycloak.models.ModelProvider;
import org.keycloak.models.RealmModel;
import org.keycloak.models.RoleModel;
import org.keycloak.representations.idm.RealmRepresentation;
@ -30,6 +31,7 @@ public class AbstractModelTest {
protected KeycloakSession session;
protected RealmManager realmManager;
protected ModelProvider model;
@BeforeClass
public static void beforeClass() {
@ -54,6 +56,7 @@ public class AbstractModelTest {
public void before() throws Exception {
session = sessionFactory.create();
session.getTransaction().begin();
model = session.getModel();
realmManager = new RealmManager(session);
}
@ -65,9 +68,10 @@ public class AbstractModelTest {
session = sessionFactory.create();
try {
session.getTransaction().begin();
model = session.getModel();
RealmManager rm = new RealmManager(session);
for (RealmModel realm : session.getRealms()) {
for (RealmModel realm : model.getRealms()) {
if (!realm.getName().equals(Config.getAdminRealm())) {
rm.removeRealm(realm);
}
@ -98,6 +102,7 @@ public class AbstractModelTest {
session = sessionFactory.create();
session.getTransaction().begin();
model = session.getModel();
realmManager = new RealmManager(session);
}

10
model/tests/src/test/java/org/keycloak/model/test/AdapterTest.java
View file

@ -94,7 +94,7 @@ public class AdapterTest extends AbstractModelTest {
realmModel.getId();
commit();
List<RealmModel> realms = session.getRealms();
List<RealmModel> realms = model.getRealms();
Assert.assertEquals(realms.size(), 2);
}
@ -179,7 +179,7 @@ public class AdapterTest extends AbstractModelTest {
commit();
realmModel = session.getRealm("JUGGLER");
realmModel = model.getRealm("JUGGLER");
Assert.assertTrue(realmModel.removeUser("bburke"));
Assert.assertFalse(realmModel.removeUser("bburke"));
assertNull(realmModel.getUser("bburke"));
@ -235,7 +235,7 @@ public class AdapterTest extends AbstractModelTest {
app.addScopeMapping(realmRole);
commit();
realmModel = session.getRealm("JUGGLER");
realmModel = model.getRealm("JUGGLER");
Assert.assertTrue(realmManager.removeRealm(realmModel));
Assert.assertFalse(realmManager.removeRealm(realmModel));
@ -261,7 +261,7 @@ public class AdapterTest extends AbstractModelTest {
app.addScopeMapping(realmRole);
commit();
realmModel = session.getRealm("JUGGLER");
realmModel = model.getRealm("JUGGLER");
app = realmModel.getApplicationByName("test-app");
Assert.assertTrue(realmModel.removeRoleById(realmRole.getId()));
@ -513,7 +513,7 @@ public class AdapterTest extends AbstractModelTest {
commit();
realmModel = session.getRealmByName("JUGGLER");
realmModel = model.getRealmByName("JUGGLER");
app1 = realmModel.getApplicationByName("app1");
app2 = realmModel.getApplicationByName("app2");
client = realmModel.getOAuthClient("client");

12
model/tests/src/test/java/org/keycloak/model/test/MultipleRealmsTest.java
View file

@ -49,8 +49,8 @@ public class MultipleRealmsTest extends AbstractModelTest {
Assert.assertEquals(2, realm1.searchForUser("user").size());
commit();
realm1 = session.getRealm("id1");
realm2 = session.getRealm("id2");
realm1 = model.getRealm("id1");
realm2 = model.getRealm("id2");
realm1.removeUser("user1");
realm1.removeUser("user2");
@ -60,10 +60,10 @@ public class MultipleRealmsTest extends AbstractModelTest {
@Test
public void testGetById() {
Assert.assertEquals(realm1, session.getRealm("id1"));
Assert.assertEquals(realm1, session.getRealmByName("realm1"));
Assert.assertEquals(realm2, session.getRealm("id2"));
Assert.assertEquals(realm2, session.getRealmByName("realm2"));
Assert.assertEquals(realm1, model.getRealm("id1"));
Assert.assertEquals(realm1, model.getRealmByName("realm1"));
Assert.assertEquals(realm2, model.getRealm("id2"));
Assert.assertEquals(realm2, model.getRealmByName("realm2"));
ApplicationModel r1app1 = realm1.getApplicationByName("app1");
ApplicationModel r1app2 = realm1.getApplicationByName("app2");

169
services/src/main/java/org/keycloak/services/DefaultKeycloakSession.java
View file

@ -1,24 +1,14 @@
package org.keycloak.services;
import org.keycloak.models.ApplicationModel;
import org.keycloak.models.ClientModel;
import org.keycloak.models.KeycloakSession;
import org.keycloak.models.KeycloakTransaction;
import org.keycloak.models.ModelProvider;
import org.keycloak.models.OAuthClientModel;
import org.keycloak.models.RealmModel;
import org.keycloak.models.RoleModel;
import org.keycloak.models.SocialLinkModel;
import org.keycloak.models.UserModel;
import org.keycloak.models.UserSessionModel;
import org.keycloak.models.UsernameLoginFailureModel;
import org.keycloak.models.cache.CacheModelProvider;
import org.keycloak.provider.Provider;
import org.keycloak.provider.ProviderFactory;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
@ -27,8 +17,8 @@ import java.util.Set;
*/
public class DefaultKeycloakSession implements KeycloakSession {
private DefaultKeycloakSessionFactory factory;
private Map<Integer, Provider> providers = new HashMap<Integer, Provider>();
private final DefaultKeycloakSessionFactory factory;
private final Map<Integer, Provider> providers = new HashMap<Integer, Provider>();
private final ModelProvider model;
public DefaultKeycloakSession(DefaultKeycloakSessionFactory factory) {
@ -85,159 +75,8 @@ public class DefaultKeycloakSession implements KeycloakSession {
return providers;
}
@Override
public RealmModel createRealm(String name) {
return model.createRealm(name);
}
@Override
public RealmModel createRealm(String id, String name) {
return model.createRealm(id, name);
}
@Override
public RealmModel getRealm(String id) {
return model.getRealm(id);
}
@Override
public RealmModel getRealmByName(String name) {
return model.getRealmByName(name);
}
@Override
public UserModel getUserById(String id, RealmModel realm) {
return model.getUserById(id, realm);
}
@Override
public UserModel getUserByUsername(String username, RealmModel realm) {
return model.getUserByUsername(username, realm);
}
@Override
public UserModel getUserByEmail(String email, RealmModel realm) {
return model.getUserByEmail(email, realm);
}
@Override
public UserModel getUserBySocialLink(SocialLinkModel socialLink, RealmModel realm) {
return model.getUserBySocialLink(socialLink, realm);
}
@Override
public List<UserModel> getUsers(RealmModel realm) {
return model.getUsers(realm);
}
@Override
public List<UserModel> searchForUser(String search, RealmModel realm) {
return model.searchForUser(search, realm);
}
@Override
public List<UserModel> searchForUserByAttributes(Map<String, String> attributes, RealmModel realm) {
return model.searchForUserByAttributes(attributes, realm);
}
@Override
public Set<SocialLinkModel> getSocialLinks(UserModel user, RealmModel realm) {
return model.getSocialLinks(user, realm);
}
@Override
public SocialLinkModel getSocialLink(UserModel user, String socialProvider, RealmModel realm) {
return model.getSocialLink(user, socialProvider, realm);
}
@Override
public RoleModel getRoleById(String id, RealmModel realm) {
return model.getRoleById(id, realm);
}
@Override
public ApplicationModel getApplicationById(String id, RealmModel realm) {
return model.getApplicationById(id, realm);
}
@Override
public OAuthClientModel getOAuthClientById(String id, RealmModel realm) {
return model.getOAuthClientById(id, realm);
}
@Override
public List<RealmModel> getRealms() {
return model.getRealms();
}
@Override
public boolean removeRealm(String id) {
return model.removeRealm(id);
}
@Override
public UsernameLoginFailureModel getUserLoginFailure(String username, RealmModel realm) {
return model.getUserLoginFailure(username, realm);
}
@Override
public UsernameLoginFailureModel addUserLoginFailure(String username, RealmModel realm) {
return model.addUserLoginFailure(username, realm);
}
@Override
public List<UsernameLoginFailureModel> getAllUserLoginFailures(RealmModel realm) {
return model.getAllUserLoginFailures(realm);
}
@Override
public UserSessionModel createUserSession(RealmModel realm, UserModel user, String ipAddress) {
return model.createUserSession(realm, user, ipAddress);
}
@Override
public UserSessionModel getUserSession(String id, RealmModel realm) {
return model.getUserSession(id, realm);
}
@Override
public List<UserSessionModel> getUserSessions(UserModel user, RealmModel realm) {
return model.getUserSessions(user, realm);
}
@Override
public Set<UserSessionModel> getUserSessions(RealmModel realm, ClientModel client) {
return model.getUserSessions(realm, client);
}
@Override
public int getActiveUserSessions(RealmModel realm, ClientModel client) {
return model.getActiveUserSessions(realm, client);
}
@Override
public void removeUserSession(UserSessionModel session) {
model.removeUserSession(session);
}
@Override
public void removeUserSessions(RealmModel realm, UserModel user) {
model.removeUserSessions(realm, user);
}
@Override
public void removeExpiredUserSessions(RealmModel realm) {
model.removeExpiredUserSessions(realm);
}
@Override
public void removeUserSessions(RealmModel realm) {
model.removeUserSessions(realm);
}
@Override
public void removeAllData() {
model.removeAllData();
public ModelProvider getModel() {
return model;
}
public void close() {

3
services/src/main/java/org/keycloak/services/managers/ApplianceBootstrap.java
View file

@ -8,6 +8,7 @@ import org.keycloak.models.AuthenticationProviderModel;
import org.keycloak.models.Constants;
import org.keycloak.models.KeycloakSession;
import org.keycloak.models.KeycloakSessionFactory;
import org.keycloak.models.ModelProvider;
import org.keycloak.models.RealmModel;
import org.keycloak.models.RoleModel;
import org.keycloak.models.UserCredentialModel;
@ -39,7 +40,7 @@ public class ApplianceBootstrap {
public void bootstrap(KeycloakSession session, String contextPath) {
String adminRealmName = Config.getAdminRealm();
if (session.getRealm(adminRealmName) != null) {
if (session.getModel().getRealm(adminRealmName) != null) {
return;
}

2
services/src/main/java/org/keycloak/services/managers/BruteForceProtector.java
View file

@ -128,7 +128,7 @@ public class BruteForceProtector implements Runnable {
}
protected RealmModel getRealmModel(KeycloakSession session, LoginEvent event) {
RealmModel realm = session.getRealm(event.realmId);
RealmModel realm = session.getModel().getRealm(event.realmId);
if (realm == null) return null;
return realm;
}

13
services/src/main/java/org/keycloak/services/managers/RealmManager.java
View file

@ -10,6 +10,7 @@ import org.keycloak.models.ClientModel;
import org.keycloak.Config;
import org.keycloak.models.Constants;
import org.keycloak.models.KeycloakSession;
import org.keycloak.models.ModelProvider;
import org.keycloak.models.OAuthClientModel;
import org.keycloak.models.PasswordPolicy;
import org.keycloak.models.RealmModel;
@ -53,6 +54,7 @@ public class RealmManager {
protected static final Logger logger = Logger.getLogger(RealmManager.class);
protected KeycloakSession session;
protected ModelProvider model;
protected String contextPath = "";
public String getContextPath() {
@ -65,6 +67,7 @@ public class RealmManager {
public RealmManager(KeycloakSession session) {
this.session = session;
this.model = session.getModel();
}
public RealmModel getKeycloakAdminstrationRealm() {
@ -72,11 +75,11 @@ public class RealmManager {
}
public RealmModel getRealm(String id) {
return session.getRealm(id);
return model.getRealm(id);
}
public RealmModel getRealmByName(String name) {
return session.getRealmByName(name);
return model.getRealmByName(name);
}
public RealmModel createRealm(String name) {
@ -85,7 +88,7 @@ public class RealmManager {
public RealmModel createRealm(String id, String name) {
if (id == null) id = KeycloakModelUtils.generateId();
RealmModel realm = session.createRealm(id, name);
RealmModel realm = model.createRealm(id, name);
realm.setName(name);
// setup defaults
@ -143,7 +146,7 @@ public class RealmManager {
}
public boolean removeRealm(RealmModel realm) {
boolean removed = session.removeRealm(realm.getId());
boolean removed = model.removeRealm(realm.getId());
if (removed) {
getKeycloakAdminstrationRealm().removeApplication(realm.getMasterAdminApp().getId());
}
@ -244,7 +247,7 @@ public class RealmManager {
RoleModel createRealmRole = realm.addRole(AdminRoles.CREATE_REALM);
adminRole.addCompositeRole(createRealmRole);
} else {
adminRealm = session.getRealmByName(Config.getAdminRealm());
adminRealm = model.getRealmByName(Config.getAdminRealm());
adminRole = adminRealm.getRole(AdminRoles.ADMIN);
}

3
services/src/main/java/org/keycloak/services/resources/admin/AdminConsole.java
View file

@ -12,6 +12,7 @@ import org.keycloak.models.AdminRoles;
import org.keycloak.models.ApplicationModel;
import org.keycloak.models.Constants;
import org.keycloak.models.KeycloakSession;
import org.keycloak.models.ModelProvider;
import org.keycloak.models.RealmModel;
import org.keycloak.models.RoleModel;
import org.keycloak.models.UserModel;
@ -224,7 +225,7 @@ public class AdminConsole {
}
private void addMasterRealmAccess(RealmModel masterRealm, UserModel user, Map<String, Set<String>> realmAdminAccess) {
List<RealmModel> realms = session.getRealms();
List<RealmModel> realms = session.getModel().getRealms();
for (RealmModel realm : realms) {
ApplicationModel realmAdminApp = realm.getMasterAdminApp();
Set<RoleModel> roles = realmAdminApp.getRoles();

2
services/src/main/java/org/keycloak/services/resources/admin/RealmsAdminResource.java
View file

@ -87,7 +87,7 @@ public class RealmsAdminResource {
RealmManager realmManager = new RealmManager(session);
List<RealmRepresentation> reps = new ArrayList<RealmRepresentation>();
if (auth.getRealm().equals(realmManager.getKeycloakAdminstrationRealm())) {
List<RealmModel> realms = session.getRealms();
List<RealmModel> realms = session.getModel().getRealms();
for (RealmModel realm : realms) {
addRealmRep(reps, realm, realm.getMasterAdminApp());
}

2
services/src/main/java/org/keycloak/services/scheduled/ClearExpiredAuditEvents.java
View file

@ -13,7 +13,7 @@ public class ClearExpiredAuditEvents implements ScheduledTask {
public void run(KeycloakSession session) {
AuditProvider audit = session.getProvider(AuditProvider.class);
if (audit != null) {
for (RealmModel realm : session.getRealms()) {
for (RealmModel realm : session.getModel().getRealms()) {
if (realm.isAuditEnabled() && realm.getAuditExpiration() > 0) {
long olderThan = System.currentTimeMillis() - realm.getAuditExpiration() * 1000;
audit.clear(realm.getId(), olderThan);

2
services/src/main/java/org/keycloak/services/scheduled/ClearExpiredUserSessions.java
View file

@ -10,7 +10,7 @@ public class ClearExpiredUserSessions implements ScheduledTask {
@Override
public void run(KeycloakSession session) {
for (RealmModel realm : session.getRealms()) {
for (RealmModel realm : session.getModel().getRealms()) {
realm.removeExpiredUserSessions();
}
}

14
testsuite/integration/src/test/java/org/keycloak/testsuite/adapter/AdapterTest.java
View file

@ -231,7 +231,7 @@ public class AdapterTest {
Assert.assertTrue(pageSource.contains("Bill Burke") && pageSource.contains("Stian Thorgersen"));
KeycloakSession session = keycloakRule.startSession();
RealmModel realm = session.getRealmByName("demo");
RealmModel realm = session.getModel().getRealmByName("demo");
int originalIdle = realm.getSsoSessionIdleTimeout();
realm.setSsoSessionIdleTimeout(1);
session.getTransaction().commit();
@ -245,7 +245,7 @@ public class AdapterTest {
Assert.assertTrue(driver.getCurrentUrl().startsWith(LOGIN_URL));
session = keycloakRule.startSession();
realm = session.getRealmByName("demo");
realm = session.getModel().getRealmByName("demo");
realm.setSsoSessionIdleTimeout(originalIdle);
session.getTransaction().commit();
session.close();
@ -265,7 +265,7 @@ public class AdapterTest {
Assert.assertTrue(pageSource.contains("Bill Burke") && pageSource.contains("Stian Thorgersen"));
KeycloakSession session = keycloakRule.startSession();
RealmModel realm = session.getRealmByName("demo");
RealmModel realm = session.getModel().getRealmByName("demo");
int originalIdle = realm.getSsoSessionIdleTimeout();
realm.setSsoSessionIdleTimeout(1);
session.getTransaction().commit();
@ -274,7 +274,7 @@ public class AdapterTest {
Thread.sleep(2000);
session = keycloakRule.startSession();
realm = session.getRealmByName("demo");
realm = session.getModel().getRealmByName("demo");
realm.removeExpiredUserSessions();
session.getTransaction().commit();
session.close();
@ -284,7 +284,7 @@ public class AdapterTest {
Assert.assertTrue(driver.getCurrentUrl().startsWith(LOGIN_URL));
session = keycloakRule.startSession();
realm = session.getRealmByName("demo");
realm = session.getModel().getRealmByName("demo");
realm.setSsoSessionIdleTimeout(originalIdle);
session.getTransaction().commit();
session.close();
@ -304,7 +304,7 @@ public class AdapterTest {
Assert.assertTrue(pageSource.contains("Bill Burke") && pageSource.contains("Stian Thorgersen"));
KeycloakSession session = keycloakRule.startSession();
RealmModel realm = session.getRealmByName("demo");
RealmModel realm = session.getModel().getRealmByName("demo");
int original = realm.getSsoSessionMaxLifespan();
realm.setSsoSessionMaxLifespan(1);
session.getTransaction().commit();
@ -318,7 +318,7 @@ public class AdapterTest {
Assert.assertTrue(driver.getCurrentUrl().startsWith(LOGIN_URL));
session = keycloakRule.startSession();
realm = session.getRealmByName("demo");
realm = session.getModel().getRealmByName("demo");
realm.setSsoSessionMaxLifespan(original);
session.getTransaction().commit();
session.close();

16
testsuite/integration/src/test/java/org/keycloak/testsuite/oauth/RefreshTokenTest.java
View file

@ -183,7 +183,7 @@ public class RefreshTokenTest {
String refreshId = oauth.verifyRefreshToken(tokenResponse.getRefreshToken()).getId();
KeycloakSession session = keycloakRule.startSession();
RealmModel realm = session.getRealmByName("test");
RealmModel realm = session.getModel().getRealmByName("test");
UserSessionModel userSession = realm.getUserSession(sessionId);
int last = userSession.getLastSessionRefresh();
session.getTransaction().commit();
@ -199,7 +199,7 @@ public class RefreshTokenTest {
Assert.assertEquals(200, tokenResponse.getStatusCode());
session = keycloakRule.startSession();
realm = session.getRealmByName("test");
realm = session.getModel().getRealmByName("test");
userSession = realm.getUserSession(sessionId);
int next = userSession.getLastSessionRefresh();
session.getTransaction().commit();
@ -211,7 +211,7 @@ public class RefreshTokenTest {
session = keycloakRule.startSession();
realm = session.getRealmByName("test");
realm = session.getModel().getRealmByName("test");
int lastAccessTokenLifespan = realm.getAccessTokenLifespan();
realm.setAccessTokenLifespan(100000);
session.getTransaction().commit();
@ -221,7 +221,7 @@ public class RefreshTokenTest {
tokenResponse = oauth.doRefreshTokenRequest(tokenResponse.getRefreshToken(), "password");
session = keycloakRule.startSession();
realm = session.getRealmByName("test");
realm = session.getModel().getRealmByName("test");
userSession = realm.getUserSession(sessionId);
next = userSession.getLastSessionRefresh();
session.getTransaction().commit();
@ -231,7 +231,7 @@ public class RefreshTokenTest {
Assert.assertThat(next, allOf(greaterThan(last), lessThan(last + 6)));
session = keycloakRule.startSession();
realm = session.getRealmByName("test");
realm = session.getModel().getRealmByName("test");
int originalIdle = realm.getSsoSessionIdleTimeout();
realm.setSsoSessionIdleTimeout(1);
session.getTransaction().commit();
@ -249,7 +249,7 @@ public class RefreshTokenTest {
events.expectRefresh(refreshId, sessionId).error(Errors.INVALID_TOKEN);
session = keycloakRule.startSession();
realm = session.getRealmByName("test");
realm = session.getModel().getRealmByName("test");
realm.setSsoSessionIdleTimeout(originalIdle);
realm.setAccessTokenLifespan(lastAccessTokenLifespan);
session.getTransaction().commit();
@ -274,7 +274,7 @@ public class RefreshTokenTest {
String refreshId = oauth.verifyRefreshToken(tokenResponse.getRefreshToken()).getId();
KeycloakSession session = keycloakRule.startSession();
RealmModel realm = session.getRealmByName("test");
RealmModel realm = session.getModel().getRealmByName("test");
int maxLifespan = realm.getSsoSessionMaxLifespan();
realm.setSsoSessionMaxLifespan(1);
session.getTransaction().commit();
@ -289,7 +289,7 @@ public class RefreshTokenTest {
assertNull(tokenResponse.getRefreshToken());
session = keycloakRule.startSession();
realm = session.getRealmByName("test");
realm = session.getModel().getRealmByName("test");
realm.setSsoSessionMaxLifespan(maxLifespan);
session.getTransaction().commit();
session.close();

4
testsuite/integration/src/test/java/org/keycloak/testsuite/rule/AbstractKeycloakRule.java
View file

@ -40,7 +40,7 @@ public abstract class AbstractKeycloakRule extends ExternalResource {
KeycloakSession session = server.getSessionFactory().create();
session.getTransaction().begin();
try {
UserModel user = session.getRealmByName(realm).getUser(name);
UserModel user = session.getModel().getRealmByName(realm).getUser(name);
return user != null ? ModelToRepresentation.toRepresentation(user) : null;
} finally {
session.close();
@ -51,7 +51,7 @@ public abstract class AbstractKeycloakRule extends ExternalResource {
KeycloakSession session = server.getSessionFactory().create();
session.getTransaction().begin();
try {
return ModelToRepresentation.toRepresentation(session.getRealmByName(realm).getUserById(id));
return ModelToRepresentation.toRepresentation(session.getModel().getRealmByName(realm).getUserById(id));
} finally {
session.close();
}

2
testsuite/integration/src/test/java/org/keycloak/testsuite/rule/KeycloakRule.java
View file

@ -100,7 +100,7 @@ public class KeycloakRule extends AbstractKeycloakRule {
public void removeUserSession(String sessionId) {
KeycloakSession session = startSession();
RealmModel realm = session.getRealm("test");
RealmModel realm = session.getModel().getRealm("test");
UserSessionModel userSession = realm.getUserSession(sessionId);
assertNotNull(userSession);
realm.removeUserSession(userSession);

2
testsuite/performance/src/test/java/org/keycloak/testsuite/performance/BaseJMeterPerformanceTest.java
View file

@ -40,7 +40,7 @@ public class BaseJMeterPerformanceTest extends AbstractJavaSamplerClient {
try {
String adminRealmName = Config.getAdminRealm();
if (keycloakSession.getRealm(adminRealmName) == null) {
if (keycloakSession.getModel().getRealm(adminRealmName) == null) {
RealmManager manager = new RealmManager(keycloakSession);
manager.setContextPath(contextPath);

2
testsuite/performance/src/test/java/org/keycloak/testsuite/performance/CreateUsersWorker.java
View file

@ -59,7 +59,7 @@ public class CreateUsersWorker implements Worker {
@Override
public void run(SampleResult result, KeycloakSession session) {
// We need to obtain realm first
RealmModel realm = session.getRealm(realmId);
RealmModel realm = session.getModel().getRealm(realmId);
if (realm == null) {
throw new IllegalStateException("Realm '" + realmId + "' not found");
}

2
testsuite/performance/src/test/java/org/keycloak/testsuite/performance/ReadUsersWorker.java
View file

@ -68,7 +68,7 @@ public class ReadUsersWorker implements Worker {
@Override
public void run(SampleResult result, KeycloakSession session) {
// We need to obtain realm first
RealmModel realm = session.getRealm(realmId);
RealmModel realm = session.getModel().getRealm(realmId);
if (realm == null) {
throw new IllegalStateException("Realm '" + realmId + "' not found");
}

2
testsuite/performance/src/test/java/org/keycloak/testsuite/performance/RemoveUsersWorker.java
View file

@ -32,7 +32,7 @@ public class RemoveUsersWorker implements Worker {
int realmNumber = realmsOffset + workerId;
String realmId = PerfTestUtils.getRealmName(realmNumber);
realm = session.getRealm(realmId);
realm = session.getModel().getRealm(realmId);
if (realm == null) {
throw new IllegalStateException("Realm '" + realmId + "' not found");
}

2
testsuite/tools/src/main/java/org/keycloak/test/tools/PerfTools.java
View file

@ -170,7 +170,7 @@ public class PerfTools {
}
private int getUsersCount(String realmName, String prefix) {
RealmModel realm = session.getRealmByName(realmName);
RealmModel realm = session.getModel().getRealmByName(realmName);
// TODO: method for count on model
if (prefix == null) {