springboot多市场数据源注入

因为公司业务原因，有多市场多数据源。所以常用方法是上层抽象，下层注入：

数据库： mongodb

但是因为api层传递的countryCode是不一样的，所以通过一个一个注入方式然后equals判断选择哪个还是太low了。

想了一种方法，没次在使用的时候进行setCountryCode，但是具体还没测试。 后续更新。。。

/**
 * @author hope 2022年03月04日 上午10:22:46
 */
@Service
public class ExecuteOrderRealDao extends AbstractExecuteOrderHisDao {

    public static volatile String tableName = StringUtils.EMPTY;

    public ExecuteOrderRealDao() {
        super(tableName);
    }

    public static void setCountryCode(String countryCode) {
        ExecuteOrderRealDao.tableName = ExecOrderTable.getTableName(countryCode);
    }

    public static void setTableName(String tableName) {
        ExecuteOrderRealDao.tableName = tableName;
    }

    @Getter
    @AllArgsConstructor
    public enum ExecOrderTable {

        US("US", "polaroid_country_us_primary_t_execute_order"),
        HK("HK", "polaroid_country_hk_primary_t_execute_order"),
        SG("SG", "polaroid_country_sg_primary_t_execute_order"),
        ACCOUNT("ACCOUNT", "polaroid_account_primary_test_t_execute_order");

        private String countryCode;
        private String tableName;

        /**
         * 默认返回账户表
         */
        public static String getTableName(String countryCode) {
            for (ExecuteOrderRealDao.ExecOrderTable value : values()) {
                if (value.getCountryCode().equals(countryCode)) {
                    return value.getTableName();
                }
            }
            return ExecOrderTable.ACCOUNT.getTableName();
        }

    }

}

复制代码

欢迎大佬评论，私聊。

• [ 萱儿AXW ]