uas_database/src/main/java/com/uas/erp/database/datasource/DynamicDataSourceContextHolder.java

package com.uas.erp.database.datasource;

/**
 * Created by Pro1 on 2017/7/27.
 */
public class DynamicDataSourceContextHolder {

    private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();

    public static void set(String dataSource) {
        contextHolder.set(dataSource);
    }

    public static void set(Connectable connectable) {
        contextHolder.set(connectable.qualifier());
    }

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

    public static void clear() {
        contextHolder.remove();
    }
}