package com.dy.common.multiDataSource;

/**
 * 线程安全的数据源持有者，持有当前访问所应用的数据源名称
 */
public class DataSourceContext {
    private final static ThreadLocal<String> LOCAL_DATASOURCE = new ThreadLocal<>();

    public static void set(String name) {
        LOCAL_DATASOURCE.set(name);
    }

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

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