org.casbin.spring.boot.autoconfigure.TxWatcher Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of casbin-spring-boot-starter Show documentation

Casbin SpringBoot Starter

The newest version!

package org.casbin.spring.boot.autoconfigure;

import org.casbin.jcasbin.persist.Watcher;
import org.springframework.transaction.support.TransactionSynchronization;

import java.util.function.Consumer;

import static org.springframework.transaction.support.TransactionSynchronizationManager.isActualTransactionActive;
import static org.springframework.transaction.support.TransactionSynchronizationManager.registerSynchronization;

public class TxWatcher implements Watcher {
    private final Watcher watcher;
    private final AfterCommitExecute afterCommitExecute;

    public TxWatcher(Watcher watcher) {
        this.watcher = watcher;
        this.afterCommitExecute = new AfterCommitExecute(watcher);
    }

    @Override
    public void setUpdateCallback(Runnable runnable) {
        this.watcher.setUpdateCallback(runnable);
    }

    @Override
    public void setUpdateCallback(Consumer consumer) {
        this.watcher.setUpdateCallback(consumer);
    }

    @Override
    public void update() {
        if (isActualTransactionActive()) {
            registerSynchronization(afterCommitExecute);
        } else {
            watcher.update();
        }
    }

    public static class AfterCommitExecute implements TransactionSynchronization {
        private final Watcher watcher;

        public AfterCommitExecute(Watcher watcher) {
            this.watcher = watcher;
        }

        @Override
        public void afterCommit() {
            watcher.update();
        }
    }
}