se.fortnox.reactivewizard.db.transactions.StatementContext Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of reactivewizard-dao Show documentation

There is a newer version: 24.6.0

package se.fortnox.reactivewizard.db.transactions;

import se.fortnox.reactivewizard.db.statement.Statement;

import java.util.function.Supplier;

public final class StatementContext {
    private final Supplier statementSupplier;
    private final ConnectionScheduler connectionScheduler;
    private Runnable transactionCompletedAction;

    public StatementContext(Supplier statementSupplier, ConnectionScheduler connectionScheduler) {
        this.statementSupplier = statementSupplier;
        this.connectionScheduler = connectionScheduler;
    }

    public Statement getStatement() {
        return statementSupplier.get();
    }

    public ConnectionScheduler getConnectionScheduler() {
        return connectionScheduler;
    }

    public void onTransactionCompleted(Runnable action) {
        this.transactionCompletedAction = action;
    }

    void transactionCompleted() {
        if (this.transactionCompletedAction != null) {
            this.transactionCompletedAction.run();
        }
    }
}