JPA根据请求参数更改实体表名称

Question

目前我们的数据库有两种模式，一种用于生产，一种用于测试，表相同但数据不同。

对于服务器端，我们使用 RoutingDatasource 来管理数据。每个请求都有一个参数来知道要访问哪个数据库，例如“prod”或“test”。然后将参数存储在 ThreadLocal 中。我们的 RoutingDatasource 将获取此参数，取决于值并返回正确的数据源。所以我们只需要一个映射实体和一个 CrudRepository 用于两个模式中的同一张表。

但出于某种（愚蠢的）原因，我们被要求将两种模式合并为一个模式，在每个表前都有一个前缀，如 PROD_TABLE1、TEST_TABLE1、PROD_TABLE2、TEST_TABLE2 等。

所以我想问一下，我们如何才能为具有此要求的每个表只保留 1 个实体、1 个存储库？我们不想克隆代码，创建太多类继承现有的只是为了映射到第二个表。 @SecondaryTable 似乎不是我们想要的。

=================================

更新 1：

我正在查看 PhysicalNamingStrategy，能够为表名添加前缀。但这只会在他们初始化 EntityManager 时运行一次。那么有没有其他方法可以为每个请求的表添加前缀？

=================================

更新 2： 我尝试这样的事情：

@Table(name = "##schema##TABLE1")

然后在我们的项目中添加一个 Hibernate 拦截器。在准备语句时，我们从 ThreadLocal 获取参数，然后将“##schema##”部分替换为正确的前缀。使用这种方式，一切正常，但我不太喜欢这种解决方案。

谁有更好的方法？

Answer 1

如果需要，可以在这里为任何人提供我自己的解决方法：

首先给每个实体对象加上前缀“##prefix##”：

@Table(name = "##prefix##TEST_TABLE_1")

创建一个休眠拦截器类扩展 EmptyInterceptor 并覆盖方法 onPrepareStatement。这里从 ThreadLocal 中获取前缀（预先收到请求时设置）并将其替换为表名中的“##prefix##”。

public class HibernateInterceptor extends EmptyInterceptor {

    @Override
    public String onPrepareStatement(String sql) {
        String prefix = ThreadLocal.getDbPrefix();
        sql = sql.replaceAll("##schema##", prefix + "_");
        return super.onPrepareStatement(sql);
    }
}

这样，查询将指向正确的表。

并配置到EntityManagerFactory：（你也可以在xml文件中设置）

properties.put("hibernate.ejb.interceptor", "mypackage.HibernateInterceptor");

Answer 2

从安全性的角度来看，我绝不建议在一个架构中混合生产数据和测试数据，但无论如何，如果您打算这样做，这里有一个简单的方法来假装您使用不同的所有者访问数据。

假设您有一个应用程序架构 APP，其中包含两个表 TEST_TAB1 和 PROD_TAB1。

-- connect as APP
create table app.prod_tab1 as
select 'prod' col1 from dual;

create table app.test_tab1 as
select 'test' col1 from dual;

您还有两个用户，例如 PROD_USR 和 TEST_USR，他们访问数据的方式如下：

--- connect with test_usr
select * from app.test_tab1;
COL1
----
test 

--- connect with prod_usr
select * from app.prod_tab1;
COL1
----
prod 

所以，现在你遇到了问题，表在不同的环境中具有不同的名称。

您需要为用户 test_usr 和 prod_usr 提供特权 create synonym 并定义以下同义词：

-- test_usr
create synonym test_usr.tab1 for app.test_tab1;
-- prod_usr
create synonym prod_usr.tab1 for app.prod_tab1;

现在访问具有唯一的名称和不同的架构：

--- connect with test_usr
select * from test_usr.tab1;
COL1
----
test

--- connect with prod_usr
select * from prod_usr.tab1;
COL1
----
prod 

如果您说，不，我们没有两个不同的访问用户，我们使用架构所有者APP 访问两个环境，请重新阅读我的第一句话。