libre.sh/keycloak-scim
2
0
Fork 0
Code Issues 14 Pull requests Releases Packages Activity Actions 6

Support for realm-less entities in providers (login failures)

Browse source 
Closes: #19818
This commit is contained in:
Hynek Mlnarik 2023-03-15 17:34:03 +01:00 committed by Hynek Mlnařík
parent 6b3eb46314
commit 68b3c87666
1 changed files with 18 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
model/map/src/main/java/org/keycloak/models/map/loginFailure/MapUserLoginFailureProvider.java
View file

@ -22,6 +22,7 @@ import org.keycloak.models.UserLoginFailureProvider;
import org.keycloak.models.RealmModel; import org.keycloak.models.RealmModel;
import org.keycloak.models.UserLoginFailureModel; import org.keycloak.models.UserLoginFailureModel;
import org.keycloak.models.map.common.DeepCloner; import org.keycloak.models.map.common.DeepCloner;
import org.keycloak.models.map.common.HasRealmId;
import org.keycloak.models.map.storage.MapStorage; import org.keycloak.models.map.storage.MapStorage;
import org.keycloak.models.map.storage.ModelCriteriaBuilder.Operator; import org.keycloak.models.map.storage.ModelCriteriaBuilder.Operator;
@ -39,11 +40,21 @@ public class MapUserLoginFailureProvider implements UserLoginFailureProvider {
private static final Logger LOG = Logger.getLogger(MapUserLoginFailureProvider.class); private static final Logger LOG = Logger.getLogger(MapUserLoginFailureProvider.class);
private final KeycloakSession session; private final KeycloakSession session;
protected final MapStorage<MapUserLoginFailureEntity, UserLoginFailureModel> userLoginFailureTx; protected final MapStorage<MapUserLoginFailureEntity, UserLoginFailureModel> store;
private final boolean storeHasRealmId;
public MapUserLoginFailureProvider(KeycloakSession session, MapStorage<MapUserLoginFailureEntity, UserLoginFailureModel> userLoginFailureStore) { public MapUserLoginFailureProvider(KeycloakSession session, MapStorage<MapUserLoginFailureEntity, UserLoginFailureModel> userLoginFailureStore) {
this.session = session; this.session = session;
this.userLoginFailureTx = userLoginFailureStore;
this.store = userLoginFailureStore;
this.storeHasRealmId = store instanceof HasRealmId;
}
private MapStorage<MapUserLoginFailureEntity, UserLoginFailureModel> storeWithRealm(RealmModel realm) {
if (storeHasRealmId) {
((HasRealmId) store).setRealmId(realm == null ? null : realm.getId());
}
return store;
} }
private Function<MapUserLoginFailureEntity, UserLoginFailureModel> userLoginFailureEntityToAdapterFunc(RealmModel realm) { private Function<MapUserLoginFailureEntity, UserLoginFailureModel> userLoginFailureEntityToAdapterFunc(RealmModel realm) {
@ -59,7 +70,7 @@ public class MapUserLoginFailureProvider implements UserLoginFailureProvider {
LOG.tracef("getUserLoginFailure(%s, %s)%s", realm, userId, getShortStackTrace()); LOG.tracef("getUserLoginFailure(%s, %s)%s", realm, userId, getShortStackTrace());
return userLoginFailureTx.read(withCriteria(mcb)) return storeWithRealm(realm).read(withCriteria(mcb))
.findFirst() .findFirst()
.map(userLoginFailureEntityToAdapterFunc(realm)) .map(userLoginFailureEntityToAdapterFunc(realm))
.orElse(null); .orElse(null);
@ -73,14 +84,14 @@ public class MapUserLoginFailureProvider implements UserLoginFailureProvider {
LOG.tracef("addUserLoginFailure(%s, %s)%s", realm, userId, getShortStackTrace()); LOG.tracef("addUserLoginFailure(%s, %s)%s", realm, userId, getShortStackTrace());
MapUserLoginFailureEntity userLoginFailureEntity = userLoginFailureTx.read(withCriteria(mcb)).findFirst().orElse(null); MapUserLoginFailureEntity userLoginFailureEntity = storeWithRealm(realm).read(withCriteria(mcb)).findFirst().orElse(null);
if (userLoginFailureEntity == null) { if (userLoginFailureEntity == null) {
userLoginFailureEntity = DeepCloner.DUMB_CLONER.newInstance(MapUserLoginFailureEntity.class); userLoginFailureEntity = DeepCloner.DUMB_CLONER.newInstance(MapUserLoginFailureEntity.class);
userLoginFailureEntity.setRealmId(realm.getId()); userLoginFailureEntity.setRealmId(realm.getId());
userLoginFailureEntity.setUserId(userId); userLoginFailureEntity.setUserId(userId);
userLoginFailureEntity = userLoginFailureTx.create(userLoginFailureEntity); userLoginFailureEntity = storeWithRealm(realm).create(userLoginFailureEntity);
} }
return userLoginFailureEntityToAdapterFunc(realm).apply(userLoginFailureEntity); return userLoginFailureEntityToAdapterFunc(realm).apply(userLoginFailureEntity);
@ -94,7 +105,7 @@ public class MapUserLoginFailureProvider implements UserLoginFailureProvider {
LOG.tracef("removeUserLoginFailure(%s, %s)%s", realm, userId, getShortStackTrace()); LOG.tracef("removeUserLoginFailure(%s, %s)%s", realm, userId, getShortStackTrace());
userLoginFailureTx.delete(withCriteria(mcb)); storeWithRealm(realm).delete(withCriteria(mcb));
} }
@Override @Override
@ -104,7 +115,7 @@ public class MapUserLoginFailureProvider implements UserLoginFailureProvider {
LOG.tracef("removeAllUserLoginFailures(%s)%s", realm, getShortStackTrace()); LOG.tracef("removeAllUserLoginFailures(%s)%s", realm, getShortStackTrace());
userLoginFailureTx.delete(withCriteria(mcb)); storeWithRealm(realm).delete(withCriteria(mcb));
} }
@Override @Override