erp / dili-boss · Files

package com.diligrp.boss.shared.datasource;

public class DataSourceNameContext {
    private static final ThreadLocal<DataSourceName> CONTEXT = new ThreadLocal<>();

    public static void setDataSourceName(DataSourceName name){
        CONTEXT.set(name);
    }

    public static DataSourceName getDataSourceName() {
        DataSourceName name = CONTEXT.get();
        return name != null ? name : DataSourceName.MASTER;
    }

    public static void removeDataSourceName() {
        CONTEXT.remove();
    }
}