pl.setblack.airomem.core.WriteChecker Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of airomem-core Show documentation

Prevayler based persistence engine

The newest version!

/* Copyright (c) Jarek Ratajski, Licensed under the Apache License, Version 2.0
 * http://www.apache.org/licenses/LICENSE-2.0 */
package pl.setblack.airomem.core;

/**
 * Extra helper method check if PrevalanceContext was initialized.
 *
 * @author jarekr
 */
public class WriteChecker {

    private static final ThreadLocal CONTEXT_STORE = new ThreadLocal<>();

    private WriteChecker() {}

    public static void setContext(final PrevalanceContext ctx) {
        assert !hasPrevalanceContext() : "Context was initialized before";
        CONTEXT_STORE.set(ctx);
    }

    public static PrevalanceContext getContext() {
        return CONTEXT_STORE.get();
    }

    public static void clearContext() {
        assert hasPrevalanceContext() : "Context has to be initialized first";
        CONTEXT_STORE.remove();
    }

    public static boolean hasPrevalanceContext() {
        return CONTEXT_STORE.get() != null;
    }

    public static void enterSafe() {
        assert hasPrevalanceContext() : "Context has to be initialized first";
        assert !getContext().safe : "context cannot be safe before";
        CONTEXT_STORE.set(getContext().safe());
    }

    public static void leaveSafe() {
        assert hasPrevalanceContext() : "Context has to be initialized first";
        assert getContext().safe : "context must be safe before";

        CONTEXT_STORE.set(getContext().unsafe());
    }

}