Spring Boot Hibernate复用实体操作多数据库忽略不存在关联表的方法

方案一：自定义注解+Hibernate拦截器（推荐） #

这个方法通过自定义标记注解配合拦截器，动态检测当前数据源的表是否存在，自动移除不存在表对应的关联映射，完全复用原有实体类。

1. 定义用于标记可忽略关联的注解：

@Target({ElementType.FIELD})
@Retention(RetentionPolicy.RUNTIME)
public @interface IgnoreWhenTableMissing {
    // 指定关联对应的数据库表名
    String tableName();
}

2. 在Parent实体的ChildEntity2关联字段上添加该注解：

@Entity
public class Parent {
    // 基础字段和主键...
    
    // DB2中不存在的关联，标记注解
    @OneToMany(mappedBy = "parent", cascade = CascadeType.ALL)
    @IgnoreWhenTableMissing(tableName = "Child_Entity2")
    private List<ChildEntity2> childEntity2List;
    
    // DB1和DB2都存在的关联，保持原样
    @OneToMany(mappedBy = "parent", cascade = CascadeType.ALL)
    private List<ChildEntity1> childEntity1List;
    
    // getter、setter方法...
}

3. 编写Hibernate拦截器，在元数据初始化时处理关联映射：

public class TableMissingIgnoreInterceptor extends EmptyInterceptor {
    private final DataSource dataSource;

    public TableMissingIgnoreInterceptor(DataSource dataSource) {
        this.dataSource = dataSource;
    }

    // 核心逻辑：检查表是否存在，不存在则移除对应关联
    public static void processMetadata(Metadata metadata, DataSource dataSource) throws SQLException {
        try (Connection conn = dataSource.getConnection()) {
            DatabaseMetaData dbMeta = conn.getMetaData();
            for (PersistentClass entity : metadata.getEntityBindings()) {
                Iterator<Property> propIterator = entity.getProperties().iterator();
                while (propIterator.hasNext()) {
                    Property prop = propIterator.next();
                    IgnoreWhenTableMissing anno = prop.getMember().getAnnotation(IgnoreWhenTableMissing.class);
                    if (anno != null) {
                        String targetTable = anno.tableName().toUpperCase();
                        ResultSet tableRs = dbMeta.getTables(null, null, targetTable, new String[]{"TABLE"});
                        if (!tableRs.next()) {
                            propIterator.remove();
                        }
                        tableRs.close();
                    }
                }
            }
        }
    }
}

4. 在DB2的数据源配置中注入拦截器并处理元数据：

@Configuration
@EnableJpaRepositories(
        basePackages = "com.your.repo.db2",
        entityManagerFactoryRef = "db2EntityManagerFactory"
)
public class DB2Config {
    // DB2数据源配置...

    @Bean
    public LocalContainerEntityManagerFactoryBean db2EntityManagerFactory(
            EntityManagerFactoryBuilder builder,
            @Qualifier("db2DataSource") DataSource db2DataSource
    ) throws SQLException {
        LocalContainerEntityManagerFactoryBean em = builder
                .dataSource(db2DataSource)
                .packages("com.your.entity") // 直接复用DB1的实体包
                .persistenceUnit("db2PU")
                .build();
        
        // 处理元数据，移除不存在表的关联
        TableMissingIgnoreInterceptor.processMetadata(em.getMetadata(), db2DataSource);
        
        // 注册拦截器确保后续操作生效
        em.getJpaPropertyMap().put(
                "hibernate.ejb.interceptor",
                new TableMissingIgnoreInterceptor(db2DataSource)
        );
        return em;
    }
}

方案二：动态映射条件（适合只读场景） #

如果以只读操作为主，可以直接在关联字段上添加条件判断，让Hibernate在不存在表时跳过该关联：

@OneToMany(mappedBy = "parent", cascade = CascadeType.ALL)
@Where(clause = "(select count(*) from information_schema.tables where table_name='Child_Entity2') > 0")
private List<ChildEntity2> childEntity2List;

注意：写入DB2时需先清空childEntity2List，否则仍可能触发关联操作报错。

方案三：条件化实体加载 #

若可接受拆分实体类，用Spring条件注解实现不同数据源加载不同实体：

1. 创建两个Parent实体类：ParentWithChild2（包含ChildEntity2关联）和ParentWithoutChild2（不含该关联）
2. 用@ConditionalOnProperty或自定义条件类，在DB1配置中加载ParentWithChild2，DB2配置中加载ParentWithoutChild2
   此方式直观但会增加实体类数量，复用性不如方案一。

内容的提问来源于stack exchange，提问作者Hacke