MtcpContext.java

package org.n3r.eql.mtcp;

import org.slf4j.MDC;

public class MtcpContext {
    private static ThreadLocal<String> tenantIdLocal = new InheritableThreadLocal<String>();
    private static ThreadLocal<String> tenantCodeLocal = new InheritableThreadLocal<String>();

    public static void setTenantId(String tenantId) {
        tenantIdLocal.set(tenantId);
        MDC.put("tenantId", tenantId);
    }

    public static void setTenantCode(String tenantCode) {
        tenantCodeLocal.set(tenantCode);
        MDC.put("tenantCode", tenantCode);
    }

    public static String getTenantId() {
        return tenantIdLocal.get();
    }
    public static String getTenantCode() {
        return tenantCodeLocal.get();
    }

    public static void clear() {
        String tenantId = getTenantId();
        if (tenantId != null) MDC.remove(tenantId);
        tenantIdLocal.remove();


        String tenantCode = tenantCodeLocal.get();
        if (tenantCode != null) MDC.remove(tenantCode);
        tenantCodeLocal.remove();
    }
}