package cn.jsxwsl.www.config.datasource;

import cn.jsxwsl.www.nums.DBTypeEnum;
import lombok.extern.slf4j.Slf4j;

import java.util.concurrent.atomic.AtomicInteger;

/**
 * @Author: 佚名
 * @Date: 2020/3/25 18:30
 */
@Slf4j
public class DataSourceTheadLocal {
    private static final ThreadLocal<DBTypeEnum> dbTypeEnumThreadLocal = new ThreadLocal<>();

    private static final AtomicInteger counter = new AtomicInteger(-1);

    /**
     * 设置当前线程中的数据源 key, 由 aop 切面调用
     *
     * @param dbType DBTypeEnum
     */
    public static void set(DBTypeEnum dbType) {
        dbTypeEnumThreadLocal.set(dbType);
    }

    /**
     * 获取当前线程的数据库key
     *
     * @return DBTypeEnum
     */
    public static DBTypeEnum get() {
        return dbTypeEnumThreadLocal.get();
    }

    public static void master() {
        set(DBTypeEnum.MASTER);
        log.info(" ========> this is master");
    }

    public static void slave() {
        //  轮询
        int index = counter.getAndIncrement() % 2;
        if (counter.get() > 9999) {
            counter.set(-1);
        }
        if (index == 0) {
            set(DBTypeEnum.SLAVE1);
            log.info(" ========> this is slave1");
        } else {
            set(DBTypeEnum.SLAVE2);
            log.info(" ========> this is slave2");
        }
    }


}