Keycloak自定义SPI-自定义交易超时

时间:2020-05-28 12:33:52

标签: java transactions timeout keycloak spi

我正在尝试通过 keycloak ImportSynchronization 接口执行 sync 过程。 (疯狂18.0.1。决赛) 它提供了以下方法来覆盖:

    @Override
    public SynchronizationResult sync(
        final KeycloakSessionFactory sessionFactory,
        final String realmId,
        final UserStorageProviderModel model) {
    ...}

此同步过程花费的时间比我们的默认交易超时时间长得多(300秒= 5分钟)。

我知道这个可能的机会:

            <core-environment node-identifier="${jboss.tx.node.id:1}">
                <process-id>
                    <uuid/>
                </process-id>
            </core-environment>
            <recovery-environment socket-binding="txn-recovery-environment" status-socket-binding="txn-status-manager"/>
            <coordinator-environment statistics-enabled="${wildfly.transactions.statistics-enabled:${wildfly.statistics-enabled:false}}"
                                     default-timeout="${env.TRANSACTION_TIMEOUT_SEC:300}"/>
            <object-store path="tx-object-store" relative-to="jboss.server.data.dir"/>
        </subsystem>

不幸的是,这不是我想要的。

不想要 ,只是为了这个,我们要增加所有进程的超时时间。这意味着我正在搜索自定义事务超时解决方案。 (类似于@TransactionTimeout ...)


我尝试了以下


@Override
    public SynchronizationResult doCleanup(final KeycloakSessionFactory keycloakSessionFactory, final String realmId)  {
        final SynchronizationResult synchronizationResult = new SynchronizationResult();

        try {

            final RealmModel realmModel = this.getRealmModelInTx(realmId, keycloakSessionFactory);

            Thread.sleep(<more than tx timeout>);

            final int userEntityCountBeforeCleanup = this.getUserCountInTx(realmModel, keycloakSessionFactory);
...}

在这两种方法中,我都创建了一个新的keycloakSession和一个新的transactionManager ,并 closed

    private CleanupTransactionHandler createSession(final KeycloakSessionFactory  keycloakSessionFactory) {
        KeycloakTransactionManager transactionManager = null;
        final KeycloakSession keycloakSession = keycloakSessionFactory.create();

        log.tracef("KeycloakSession has been created [%s].", keycloakSession.hashCode());
        transactionManager = keycloakSession.getTransactionManager();
        transactionManager.begin();
        log.tracef("KeycloakTransactionManager's transaction has been begun [%s].", transactionManager.hashCode());
        return new CleanupTransactionHandler(keycloakSession,transactionManager);
    }

    private void closeSession(final KeycloakSession keycloakSession) {
        keycloakSession.close();
        log.tracef("KeycloakSession has been closed [%s].", keycloakSession.hashCode());
    }

    private void rollback(final KeycloakTransactionManager keycloakTransactionManager) {
        keycloakTransactionManager.rollback();
        log.tracef("KeycloakTransactionManager's transaction has been rolled back [%s].", keycloakTransactionManager.hashCode());
    }

1 个答案:

答案 0 :(得分:0)

我可以回答我自己的答案。

由于自KeycloakModelUtils.runJobInTransaction调用ImportSynchronization.sync,因此无法为此过程自定义事务超时,如上所述,只能通过默认超时进行配置。