Java Code Examples for org.keycloak.models.KeycloakSession#close()

The following examples show how to use org.keycloak.models.KeycloakSession#close() . You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.

Example 1

Source File: EmbeddedKeycloakApplication.java From spring-security-oauth with MIT License

6 votes

private void createBaeldungRealm() {
	KeycloakSession session = getSessionFactory().create();

	try {
		session.getTransactionManager().begin();

		RealmManager manager = new RealmManager(session);
		Resource lessonRealmImportFile = new ClassPathResource(keycloakServerProperties.getRealmImportFile());

		manager.importRealm(
				JsonSerialization.readValue(lessonRealmImportFile.getInputStream(), RealmRepresentation.class));

		session.getTransactionManager().commit();
	} catch (Exception ex) {
		LOG.warn("Failed to import Realm json file: {}", ex.getMessage());
		session.getTransactionManager().rollback();
	}

	session.close();
}

Example 2

Source File: EmbeddedKeycloakApplication.java From spring-security-oauth with MIT License

6 votes

private void createMasterRealmAdminUser() {

        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) {
            LOG.warn("Couldn't create keycloak master admin user: {}", ex.getMessage());
            session.getTransactionManager()
                .rollback();
        }

        session.close();
    }

Example 3

Source File: EmbeddedKeycloakApplication.java From spring-security-oauth with MIT License

6 votes

private void createBaeldungRealm() {
    KeycloakSession session = getSessionFactory().create();

    try {
        session.getTransactionManager()
            .begin();

        RealmManager manager = new RealmManager(session);
        Resource lessonRealmImportFile = new ClassPathResource(keycloakServerProperties.getRealmImportFile());

        manager.importRealm(JsonSerialization.readValue(lessonRealmImportFile.getInputStream(), RealmRepresentation.class));

        session.getTransactionManager()
            .commit();
    } catch (Exception ex) {
        LOG.warn("Failed to import Realm json file: {}", ex.getMessage());
        session.getTransactionManager()
            .rollback();
    }

    session.close();
}

Example 4

Source File: EmbeddedKeycloakApplication.java From spring-security-oauth with MIT License

6 votes

private void createMasterRealmAdminUser() {

		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) {
			LOG.warn("Couldn't create keycloak master admin user: {}", ex.getMessage());
			session.getTransactionManager().rollback();
		}

		session.close();
	}

Example 5

Source File: EmbeddedKeycloakApplication.java From spring-security-oauth with MIT License

6 votes

private void createBaeldungRealm() {
	KeycloakSession session = getSessionFactory().create();

	try {
		session.getTransactionManager().begin();

		RealmManager manager = new RealmManager(session);
		Resource lessonRealmImportFile = new ClassPathResource(keycloakServerProperties.getRealmImportFile());

		manager.importRealm(
				JsonSerialization.readValue(lessonRealmImportFile.getInputStream(), RealmRepresentation.class));

		session.getTransactionManager().commit();
	} catch (Exception ex) {
		LOG.warn("Failed to import Realm json file: {}", ex.getMessage());
		session.getTransactionManager().rollback();
	}

	session.close();
}

Example 6

Source File: EmbeddedKeycloakApplication.java From spring-boot-keycloak-server-example with Apache License 2.0

6 votes

private void tryCreateMasterRealmAdminUser() {

        KeycloakSession session = getSessionFactory().create();

        ApplianceBootstrap applianceBootstrap = new ApplianceBootstrap(session);

        AdminUser admin = customProperties.getAdminUser();

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

        session.close();
    }

Example 7

Source File: EmbeddedKeycloakApplication.java From spring-security-oauth with MIT License

6 votes

private void createMasterRealmAdminUser() {

		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) {
			LOG.warn("Couldn't create keycloak master admin user: {}", ex.getMessage());
			session.getTransactionManager().rollback();
		}

		session.close();
	}

Example 8

Source File: KeycloakModelUtils.java From keycloak with Apache License 2.0

6 votes

/**
 * Wrap given runnable job into KeycloakTransaction.
 *
 * @param factory
 * @param task
 */
public static void runJobInTransaction(KeycloakSessionFactory factory, KeycloakSessionTask task) {
    KeycloakSession session = factory.create();
    KeycloakTransaction tx = session.getTransactionManager();
    try {
        tx.begin();
        task.run(session);

        if (tx.isActive()) {
            if (tx.getRollbackOnly()) {
                tx.rollback();
            } else {
                tx.commit();
            }
        }
    } catch (RuntimeException re) {
        if (tx.isActive()) {
            tx.rollback();
        }
        throw re;
    } finally {
        session.close();
    }
}

Example 9

Source File: QuarkusLifecycleObserver.java From keycloak with Apache License 2.0

5 votes

private void createAdminUser() {
    String adminUserName = System.getenv(KEYCLOAK_ADMIN_ENV_VAR);
    String adminPassword = System.getenv(KEYCLOAK_ADMIN_PASSWORD_ENV_VAR);

    if ((adminUserName == null || adminUserName.trim().length() == 0)
            || (adminPassword == null || adminPassword.trim().length() == 0)) {
        return;
    }

    KeycloakSessionFactory sessionFactory = application.getSessionFactory();
    KeycloakSession session = sessionFactory.create();
    KeycloakTransactionManager transaction = session.getTransactionManager();

    try {
        transaction.begin();

        new ApplianceBootstrap(session).createMasterRealmUser(adminUserName, adminPassword);
        ServicesLogger.LOGGER.addUserSuccess(adminUserName, Config.getAdminRealm());

        transaction.commit();
    } catch (IllegalStateException e) {
        session.getTransactionManager().rollback();
        ServicesLogger.LOGGER.addUserFailedUserExists(adminUserName, Config.getAdminRealm());
    } catch (Throwable t) {
        session.getTransactionManager().rollback();
        ServicesLogger.LOGGER.addUserFailed(t, adminUserName, Config.getAdminRealm());
    } finally {
        session.close();
    }
}

Example 10

Source File: QuarkusJpaConnectionProviderFactory.java From keycloak with Apache License 2.0

5 votes

public void importRealm(RealmRepresentation rep, String from) {
    KeycloakSession session = factory.create();
    boolean exists = false;
    try {
        session.getTransactionManager().begin();

        try {
            RealmManager manager = new RealmManager(session);

            if (rep.getId() != null && manager.getRealm(rep.getId()) != null) {
                ServicesLogger.LOGGER.realmExists(rep.getRealm(), from);
                exists = true;
            }

            if (manager.getRealmByName(rep.getRealm()) != null) {
                ServicesLogger.LOGGER.realmExists(rep.getRealm(), from);
                exists = true;
            }
            if (!exists) {
                RealmModel realm = manager.importRealm(rep);
                ServicesLogger.LOGGER.importedRealm(realm.getName(), from);
            }
            session.getTransactionManager().commit();
        } catch (Throwable t) {
            session.getTransactionManager().rollback();
            if (!exists) {
                ServicesLogger.LOGGER.unableToImportRealm(t, rep.getRealm(), from);
            }
        }
    } finally {
        session.close();
    }
}

Example 11

Source File: KeycloakSessionServletFilter.java From keycloak with Apache License 2.0

5 votes

private void closeSession(KeycloakSession session) {
    // KeycloakTransactionCommitter is responsible for committing the transaction, but if an exception is thrown it's not invoked and transaction
    // should be rolled back
    if (session.getTransactionManager() != null && session.getTransactionManager().isActive()) {
        session.getTransactionManager().rollback();
    }

    session.close();
    Resteasy.clearContextData();
}

Example 12

Source File: KeycloakApplication.java From keycloak with Apache License 2.0

5 votes

public void importRealm(RealmRepresentation rep, String from) {
    KeycloakSession session = sessionFactory.create();
    boolean exists = false;
    try {
        session.getTransactionManager().begin();

        try {
            RealmManager manager = new RealmManager(session);

            if (rep.getId() != null && manager.getRealm(rep.getId()) != null) {
                ServicesLogger.LOGGER.realmExists(rep.getRealm(), from);
                exists = true;
            }

            if (manager.getRealmByName(rep.getRealm()) != null) {
                ServicesLogger.LOGGER.realmExists(rep.getRealm(), from);
                exists = true;
            }
            if (!exists) {
                RealmModel realm = manager.importRealm(rep);
                ServicesLogger.LOGGER.importedRealm(realm.getName(), from);
            }
            session.getTransactionManager().commit();
        } catch (Throwable t) {
            session.getTransactionManager().rollback();
            if (!exists) {
                ServicesLogger.LOGGER.unableToImportRealm(t, rep.getRealm(), from);
            }
        }
    } finally {
        session.close();
    }
}

Example 13

Source File: KeycloakApplication.java From keycloak with Apache License 2.0

5 votes

public static void setupScheduledTasks(final KeycloakSessionFactory sessionFactory) {
    long interval = Config.scope("scheduled").getLong("interval", 900L) * 1000;

    KeycloakSession session = sessionFactory.create();
    try {
        TimerProvider timer = session.getProvider(TimerProvider.class);
        timer.schedule(new ClusterAwareScheduledTaskRunner(sessionFactory, new ClearExpiredEvents(), interval), interval, "ClearExpiredEvents");
        timer.schedule(new ClusterAwareScheduledTaskRunner(sessionFactory, new ClearExpiredClientInitialAccessTokens(), interval), interval, "ClearExpiredClientInitialAccessTokens");
        timer.schedule(new ScheduledTaskRunner(sessionFactory, new ClearExpiredUserSessions()), interval, ClearExpiredUserSessions.TASK_NAME);
        new UserStorageSyncManager().bootstrapPeriodic(sessionFactory, timer);
    } finally {
        session.close();
    }
}

Example 14

Source File: KeycloakApplication.java From keycloak with Apache License 2.0

5 votes

protected void migrateModel() {
    KeycloakSession session = sessionFactory.create();
    try {
        session.getTransactionManager().begin();
        MigrationModelManager.migrate(session);
        session.getTransactionManager().commit();
    } catch (Exception e) {
        session.getTransactionManager().rollback();
        throw e;
    } finally {
        session.close();
    }
}

Example 15

Source File: KeycloakSessionServletFilter.java From keycloak with Apache License 2.0

5 votes

private void closeSession(KeycloakSession session) {
    // KeycloakTransactionCommitter is responsible for committing the transaction, but if an exception is thrown it's not invoked and transaction
    // should be rolled back
    if (session.getTransactionManager() != null && session.getTransactionManager().isActive()) {
        session.getTransactionManager().rollback();
    }

    session.close();
    Resteasy.clearContextData();
}

Example 16

Source File: KeycloakServer.java From keycloak with Apache License 2.0

5 votes

protected void setupDevConfig() {
    if (System.getProperty("keycloak.createAdminUser", "true").equals("true")) {
        KeycloakSession session = sessionFactory.create();
        try {
            session.getTransactionManager().begin();
            if (new ApplianceBootstrap(session).isNoMasterUser()) {
                new ApplianceBootstrap(session).createMasterRealmUser("admin", "admin");
                log.info("Created master user with credentials admin:admin");
            }
            session.getTransactionManager().commit();
        } finally {
            session.close();
        }
    }
}

Example 17

Source File: QuarkusFilter.java From keycloak with Apache License 2.0

5 votes

@Override
public void filter(ContainerRequestContext containerRequestContext, ContainerResponseContext containerResponseContext) throws IOException {
    //End the session and clear context
    KeycloakSession session = Resteasy.getContextData(KeycloakSession.class);

    // KeycloakTransactionCommitter is responsible for committing the transaction, but if an exception is thrown it's not invoked and transaction
    // should be rolled back
    if (session.getTransactionManager() != null && session.getTransactionManager().isActive()) {
        session.getTransactionManager().rollback();
    }

    session.close();
    Resteasy.clearContextData();
}

Example 18

Source File: KeycloakOnUndertow.java From keycloak with Apache License 2.0

5 votes

protected void setupDevConfig() {
    KeycloakSession session = sessionFactory.create();
    try {
        session.getTransactionManager().begin();
        if (new ApplianceBootstrap(session).isNoMasterUser()) {
            new ApplianceBootstrap(session).createMasterRealmUser("admin", "admin");
        }
        session.getTransactionManager().commit();
    } finally {
        session.close();
    }
}

Example 19

Source File: EmbeddedKeycloakApplication.java From spring-boot-keycloak-server-example with Apache License 2.0

5 votes

private void tryImportRealm() {

        KeycloakCustomProperties.Migration imex = customProperties.getMigration();
        Resource importLocation = imex.getImportLocation();

        if (!importLocation.exists()) {
            LOG.info("Could not find keycloak import file %s", importLocation);
            return;
        }

        File file;
        try {
            file = importLocation.getFile();
        } catch (IOException e) {
            LOG.error("Could not read keycloak import file %s", importLocation, e);
            return;
        }

        LOG.info("Starting Keycloak realm configuration import from location: %s", importLocation);

        KeycloakSession session = getSessionFactory().create();

        ExportImportConfig.setAction("import");
        ExportImportConfig.setProvider(imex.getImportProvider());
        ExportImportConfig.setFile(file.getAbsolutePath());

        ExportImportManager manager = new ExportImportManager(session);
        manager.runImport();

        session.close();

        LOG.info("Keycloak realm configuration import finished.");
    }

Example 20

Source File: KeycloakApplication.java From keycloak with Apache License 2.0

4 votes

protected ExportImportManager migrateAndBootstrap() {
    ExportImportManager exportImportManager;
    logger.debug("Calling migrateModel");
    migrateModel();

    logger.debug("bootstrap");
    KeycloakSession session = sessionFactory.create();
    try {
        session.getTransactionManager().begin();
        JtaTransactionManagerLookup lookup = (JtaTransactionManagerLookup) sessionFactory.getProviderFactory(JtaTransactionManagerLookup.class);
        if (lookup != null) {
            if (lookup.getTransactionManager() != null) {
                try {
                    Transaction transaction = lookup.getTransactionManager().getTransaction();
                    logger.debugv("bootstrap current transaction? {0}", transaction != null);
                    if (transaction != null) {
                        logger.debugv("bootstrap current transaction status? {0}", transaction.getStatus());
                    }
                } catch (SystemException e) {
                    throw new RuntimeException(e);
                }
            }
        }


        ApplianceBootstrap applianceBootstrap = new ApplianceBootstrap(session);
        exportImportManager = new ExportImportManager(session);

        boolean createMasterRealm = applianceBootstrap.isNewInstall();
        if (exportImportManager.isRunImport() && exportImportManager.isImportMasterIncluded()) {
            createMasterRealm = false;
        }

        if (createMasterRealm) {
            applianceBootstrap.createMasterRealm();
        }
        session.getTransactionManager().commit();
    } catch (RuntimeException re) {
        if (session.getTransactionManager().isActive()) {
            session.getTransactionManager().rollback();
        }
        throw re;
    } finally {
        session.close();
    }

    if (exportImportManager.isRunImport()) {
        exportImportManager.runImport();
    } else {
        importRealms();
    }

    importAddUser();

    return exportImportManager;
}