Java KeycloakSession类代码示例

import org.keycloak.models.KeycloakSession; //导入依赖的package包/类
private void tryCreateMasterRealmAdminUser() {

    KeycloakSession session = getSessionFactory().create();

    ApplianceBootstrap applianceBootstrap = new ApplianceBootstrap(session);

    AdminUser admin = keycloakServerProperties.getAdminUser();

    try {
      session.getTransactionManager().begin();
      applianceBootstrap.createMasterRealmUser(admin.getUsername(), admin.getPassword());
      session.getTransactionManager().commit();
    } catch (Exception ex) {
      System.out.println("Couldn't create keycloak master admin user: " + ex.getMessage());
      session.getTransactionManager().rollback();
    }

    session.close();
  }
 

import org.keycloak.models.KeycloakSession; //导入依赖的package包/类
@Override
public PropertyFileUserStorageProvider create(KeycloakSession session, ComponentModel model) {
    String path = model.getConfig().getFirst("path");
    path = EnvUtil.replace(path);

    Properties props = new Properties();
    try {
        InputStream is = new FileInputStream(path);
        props.load(is);
        is.close();
    } catch (IOException e) {
        throw new RuntimeException(e);
    }

    return new PropertyFileUserStorageProvider(session, model, props);
}
 

import org.keycloak.models.KeycloakSession; //导入依赖的package包/类
@Before
public void setup() {
    keycloakSession = mock(KeycloakSession.class);
    KeycloakTransactionManager txnManager = mock(KeycloakTransactionManager.class);
    when(keycloakSession.getTransactionManager()).thenReturn(txnManager);
    RealmProvider realms = mock(RealmProvider.class);
    when(keycloakSession.realms()).thenReturn(realms);
    RealmModel realm = mock(RealmModel.class);
    when(realms.getRealmByName(eq("realm"))).thenReturn(realm);
    UserProvider userProvider = mock(UserProvider.class);
    UserModel user = mock(UserModel.class);
    when(userProvider.getUserByUsername(eq("user"), eq(realm))).thenReturn(user);
    when(keycloakSession.userStorageManager()).thenReturn(userProvider);
    UserCredentialManager userCredentialManager = mock(UserCredentialManager.class);
    when(keycloakSession.userCredentialManager()).thenReturn(userCredentialManager);
    when(userCredentialManager.isValid(eq(realm), eq(user), argThat(new PasswordCredentialMatcher("password")))).thenReturn(true);

    config = mock(Config.Scope.class);
}
 

import org.keycloak.models.KeycloakSession; //导入依赖的package包/类
public static void postWithRedirect(KeycloakSession session, String url, HttpEntity postBody) throws IOException {
    HttpClient httpClient = session.getProvider(HttpClientProvider.class).getHttpClient();
    for (int i = 0; i < 2; i++) { // follow redirects once
        HttpPost post = new HttpPost(url);
        post.setEntity(postBody);
        HttpResponse response = httpClient.execute(post);
        try {
            int status = response.getStatusLine().getStatusCode();
            if (status == 302 && !url.endsWith("/")) {
                String redirect = response.getFirstHeader(HttpHeaders.LOCATION).getValue();
                String withSlash = url + "/";
                if (withSlash.equals(redirect)) {
                    url = withSlash;
                    continue;
                }
            }
        } finally {
            HttpEntity entity = response.getEntity();
            if (entity != null) {
                InputStream is = entity.getContent();
                if (is != null)
                    is.close();
            }

        }
        break;
    }
}
 

import org.keycloak.models.KeycloakSession; //导入依赖的package包/类
@Override
public Response generateInstallation(KeycloakSession session, RealmModel realm, ClientModel client, URI baseUri) {
    UriBuilder bindingUrlBuilder = UriBuilder.fromUri(baseUri);
    String bindingUrl = RealmsResource.protocolUrl(bindingUrlBuilder)
            .build(realm.getName(), CASLoginProtocol.LOGIN_PROTOCOL).toString();
    String description = "CAS Server URL: " + bindingUrl + "\n" +
            "CAS Protocol: CAS 2.0/3.0 (SAML 1.1 is not supported)\n" +
            "Use CAS REST API: false (unsupported)";
    return Response.ok(description, MediaType.TEXT_PLAIN_TYPE).build();
}
 

import org.keycloak.models.KeycloakSession; //导入依赖的package包/类
@Override
public void validateConfiguration(KeycloakSession session, RealmModel realm, ComponentModel config) throws ComponentValidationException {
    String fp = config.getConfig().getFirst("path");
    if (fp == null) throw new ComponentValidationException("user property file does not exist");
    fp = EnvUtil.replace(fp);
    File file = new File(fp);
    if (!file.exists()) {
        throw new ComponentValidationException("user property file does not exist");
    }
}
 

import org.keycloak.models.KeycloakSession; //导入依赖的package包/类
@Override
public EjbExampleUserStorageProvider create(KeycloakSession session, ComponentModel model) {
    try {
        InitialContext ctx = new InitialContext();
        EjbExampleUserStorageProvider provider = (EjbExampleUserStorageProvider)ctx.lookup("java:global/user-storage-jpa-example/" + EjbExampleUserStorageProvider.class.getSimpleName());
        provider.setModel(model);
        provider.setSession(session);
        return provider;
    } catch (Exception e) {
        throw new RuntimeException(e);
    }
}
 

import org.keycloak.models.KeycloakSession; //导入依赖的package包/类
@Override
public AitEjbUserStorageProvider create(KeycloakSession session, ComponentModel model) {
    try {
        InitialContext ctx = new InitialContext();
        AitEjbUserStorageProvider provider = (AitEjbUserStorageProvider)ctx.lookup("java:global/keycloak-spi-sqlserver/" + AitEjbUserStorageProvider.class.getSimpleName());
        provider.setModel(model);
        provider.setSession(session);
        return provider;
    } catch (Exception e) {
        throw new RuntimeException(e);
    }
}
 

import org.keycloak.models.KeycloakSession; //导入依赖的package包/类
/**
 * Creates a new {@link ForwardingEventListenerProvider}
 * 
 * @param keycloakSession
 * @param forwardedContextActions
 */
public ForwardingEventListenerProvider(KeycloakSession keycloakSession, Set<String> forwardedContextActions) {

	if (forwardedContextActions == null) {
		throw new NullPointerException("includedKeycloakContextActions must not be null!");
	}

	this.keycloakSession = keycloakSession;
	this.forwardedContextActions = forwardedContextActions;
}
 

import org.keycloak.models.KeycloakSession; //导入依赖的package包/类
@Override
public EventListenerProvider create(KeycloakSession keycloakSession) {

	ForwardingEventListenerProvider provider = new ForwardingEventListenerProvider(keycloakSession,
			unmodifiableSet(includedContextActions));
	keycloakSession.enlistForClose(provider);

	return provider;
}
 

import org.keycloak.models.KeycloakSession; //导入依赖的package包/类
public ScramSaslAuthenticator(final KeycloakSession keycloakSession,
                              final String hostname,
                              String digestName,
                              String hmacName,
                              Map<String, Function<CredentialModel, StoredAndServerKey>> keyRetrievalFunctions) {
    this.keycloakSession = keycloakSession;
    this.hostname = hostname;
    this.randomBytes = new byte[32];
    (new SecureRandom()).nextBytes(this.randomBytes);
    this.digestName = digestName;
    this.hmacName = hmacName;
    this.keyRetrievalFunctions = new HashMap<>(keyRetrievalFunctions);
}
 

import org.keycloak.models.KeycloakSession; //导入依赖的package包/类
@Override
protected void setupDefaultRealm(String contextPath) {
    super.setupDefaultRealm(contextPath);

    KeycloakSession session = getSessionFactory().create();
    session.getTransaction().begin();
    try {
        configureLiveOakConsole(session);
        session.getTransaction().commit();
    } finally {
        session.close();
    }
}
 

import org.keycloak.models.KeycloakSession; //导入依赖的package包/类
@Override
public final ProtocolMapper create(KeycloakSession session) {
    throw new RuntimeException("UNSUPPORTED METHOD");
}
 

import org.keycloak.models.KeycloakSession; //导入依赖的package包/类
@Override
public ClientInstallationProvider create(KeycloakSession session) {
    return this;
}
 

import org.keycloak.models.KeycloakSession; //导入依赖的package包/类
@Override
public EventListenerProvider create(KeycloakSession session) {
    return new MonitoringEventListenerProvider(eventsDirectory);
}