org.bonitasoft.engine.tenant.TenantRestarter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of bonita-server Show documentation
There is a newer version: 10.2.0
/**
 * Copyright (C) 2019 Bonitasoft S.A.
 * Bonitasoft, 32 rue Gustave Eiffel - 38000 Grenoble
 * This library is free software; you can redistribute it and/or modify it under the terms
 * of the GNU Lesser General Public License as published by the Free Software Foundation
 * version 2.1 of the License.
 * This library is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU Lesser General Public License for more details.
 * You should have received a copy of the GNU Lesser General Public License along with this
 * program; if not, write to the Free Software Foundation, Inc., 51 Franklin Street, Fifth
 * Floor, Boston, MA 02110-1301, USA.
 **/
package org.bonitasoft.engine.tenant;

import java.util.List;

import org.bonitasoft.engine.api.impl.StarterThread;
import org.bonitasoft.engine.execution.work.RestartException;
import org.bonitasoft.engine.platform.PlatformService;
import org.bonitasoft.engine.session.SessionService;
import org.bonitasoft.engine.sessionaccessor.SessionAccessor;
import org.bonitasoft.engine.tenant.restart.TenantRestartHandler;
import org.bonitasoft.engine.transaction.BonitaTransactionSynchronization;
import org.bonitasoft.engine.transaction.STransactionNotFoundException;
import org.bonitasoft.engine.transaction.TransactionState;
import org.bonitasoft.engine.transaction.UserTransactionService;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

/**
 * @author Baptiste Mesta
 */
@Component
public class TenantRestarter {

    private UserTransactionService transactionService;
    private List tenantRestartHandlers;
    private Long tenantId;
    private SessionAccessor sessionAccessor;
    private SessionService sessionService;
    private PlatformService platformService;

    public TenantRestarter(@Value("${tenantId}") Long tenantId, UserTransactionService transactionService,
            SessionAccessor sessionAccessor, SessionService sessionService,
            PlatformService platformService, List tenantRestartHandlers) {
        this.tenantId = tenantId;
        this.transactionService = transactionService;
        this.sessionAccessor = sessionAccessor;
        this.sessionService = sessionService;
        this.platformService = platformService;
        this.tenantRestartHandlers = tenantRestartHandlers;
    }

    public List executeBeforeServicesStart() throws Exception {
        if (transactionService.isTransactionActive()) {
            return beforeServicesStart();
        }
        return transactionService.executeInTransaction(this::beforeServicesStart);
    }

    private List beforeServicesStart() throws RestartException {
        for (TenantRestartHandler tenantRestartHandler : tenantRestartHandlers) {
            tenantRestartHandler.beforeServicesStart();
        }
        return tenantRestartHandlers;
    }

    public void executeAfterServicesStart(List tenantRestartHandlers)
            throws STransactionNotFoundException {
        if (transactionService.isTransactionActive()) {
            executeAfterServicesStartAfterCurrentTransaction(tenantRestartHandlers);
        } else {
            afterServicesStart(tenantRestartHandlers);
        }
    }

    private void afterServicesStart(List tenantRestartHandlers) {
        new StarterThread(tenantId, sessionAccessor, sessionService, transactionService, platformService,
                tenantRestartHandlers).start();
    }

    private void executeAfterServicesStartAfterCurrentTransaction(
            final List tenantRestartHandlers) throws STransactionNotFoundException {
        transactionService.registerBonitaSynchronization(new BonitaTransactionSynchronization() {

            @Override
            public void beforeCommit() {

            }

            @Override
            public void afterCompletion(TransactionState txState) {
                if (txState.equals(TransactionState.COMMITTED)) {
                    afterServicesStart(tenantRestartHandlers);
                }
            }
        });
    }
}