JPA保存数据存入不同的表

背景:
使用JPA保存数据时,需要根据单位将数据保存到不同的表中

单位缩写
北京BJUSERBJ
山东SDUSERSD
 jpa:
    properties:
      hibernate:
        ejb:
          # 注册自定义的插件类
          interceptor: com.xxx.Interceptor
保存实际信息类

/**
 *@description 线程安全的类,保存实际信息
 **/
public class ReqContextHolder {
    private ReqContextHolder() {
    }

    protected static ThreadLocal<String> bhThreadLocal = new ThreadLocal<>();

    public static void setbh(String bh) {
        bhThreadLocal.set(bh);
    }

    public static String getbh() {
        return bhThreadLocal.get();
    }

    public static void remove() {
        bhThreadLocal.remove();
    }
}
动态表存入(拦截器),用来在 Hibernate 处理实体时插入额外的逻辑
public class Interceptor extends EmptyInterceptor {
    @Override
    public String onPrepareStatement(String sql) {
        String bh = ReqContextHolder.getbh();

        // 对表名进行替换
        if (!StringUtil.isNullOrEmpty(bh)) {
            sql = sql.toLowerCase(Locale.ROOT);
            if (sql.contains("user")) {
                //表名增加个性化后缀
                sql = sql.replaceFirst("user", "user" + bh);
            }
            return sql;
        }

        return super.onPrepareStatement(sql);
    }

调用:

@Resource
private UserRepository userRepository;
User user = new User;
user.setName("123");
user.setCode("456");

ReqContextHolder.setbh("BJ");
userRepository.save(user);  //userBJ
ReqContextHolder.remove();
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值