com.emily.infrastructure.datasource.context.DataSourceContextHolder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of emily-spring-boot-datasource Show documentation

The newest version!

package com.emily.infrastructure.datasource.context;

import org.springframework.core.NamedInheritableThreadLocal;

/**
 * 线程持有数据源上线文
 *
 * @author Emily
 * @since 1.0
 */
public class DataSourceContextHolder {
    /**
     * 当前线程对应的数据源
     */
    private static final ThreadLocal CONTEXT = new NamedInheritableThreadLocal<>("Multi data source switching context");

    /**
     * 设置当前线程持有的数据源
     *
     * @param dataSource 数据源标识
     */
    public static void bind(String dataSource) {
        CONTEXT.set(dataSource);
    }

    /**
     * 获取当前线程持有的数据源
     *
     * @return 数据源标识
     */
    public static String current() {
        return CONTEXT.get();
    }

    /**
     * 删除当前线程持有的数据源
     */
    public static void unbind() {
        CONTEXT.remove();
    }

}