com.dream.mate.share.holder.DataSourceHolder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dream-orm-mate Show documentation

The newest version!

package com.dream.mate.share.holder;

import java.util.function.Supplier;

public final class DataSourceHolder {

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

    private DataSourceHolder() {
    }

    public static String get() {
        return LOOKUP_KEY_HOLDER.get();
    }


    public static void set(String dataSourceName) {
        LOOKUP_KEY_HOLDER.set(dataSourceName);
    }


    public static void remove() {
        LOOKUP_KEY_HOLDER.remove();
    }

    public static  T use(String dataSourceName, Supplier supplier) {
        try {
            set(dataSourceName);
            return supplier.get();
        } finally {
            remove();
        }
    }
}