Spring Boot配置PostgreSQL与Cassandra双数据库及关联对应Repository咨询

一、独立配置双数据源 #

在application.properties中保留原有PostgreSQL配置，并新增Cassandra的独立配置项：

# PostgreSQL 原有配置（修正规范格式）
spring.datasource.url=jdbc:postgresql://localhost:5432/your_postgres_db
spring.datasource.username=username
spring.datasource.password=password
spring.datasource.driver-class-name=org.postgresql.Driver

# Cassandra 新增独立配置
spring.data.cassandra.contact-points=localhost
spring.data.cassandra.port=9042
spring.data.cassandra.keyspace-name=your_cassandra_keyspace
spring.data.cassandra.username=cassandra_user
spring.data.cassandra.password=cassandra_pass

二、数据源与Repository关联配置（核心：不改动现有JPA代码） #

通过包路径隔离实现不同Repository绑定对应数据源，无需修改原有JPA Repository代码。

1. PostgreSQL（JPA）数据源配置类 #

创建配置类指定原有JPA Repository和实体类的扫描范围，标记为主数据源保证兼容性：

@Configuration
@EnableJpaRepositories(
        basePackages = "com.yourproject.repository.postgres", // 现有JPA Repository所在包
        entityManagerFactoryRef = "postgresEntityManagerFactory",
        transactionManagerRef = "postgresTransactionManager"
)
@EnableTransactionManagement
public class PostgresJpaConfig {

    @Primary // 标记为默认数据源，避免多数据源冲突
    @Bean
    @ConfigurationProperties(prefix = "spring.datasource")
    public DataSource postgresDataSource() {
        return DataSourceBuilder.create().build();
    }

    @Primary
    @Bean
    public LocalContainerEntityManagerFactoryBean postgresEntityManagerFactory(EntityManagerFactoryBuilder builder) {
        return builder
                .dataSource(postgresDataSource())
                .packages("com.yourproject.entity.postgres") // PostgreSQL实体类所在包
                .persistenceUnit("postgresPU")
                .build();
    }

    @Primary
    @Bean
    public PlatformTransactionManager postgresTransactionManager(EntityManagerFactory postgresEntityManagerFactory) {
        return new JpaTransactionManager(postgresEntityManagerFactory);
    }
}

2. Cassandra数据源配置类 #

创建配置类指定Cassandra Repository和实体类的扫描范围：

@Configuration
@EnableCassandraRepositories(
        basePackages = "com.yourproject.repository.cassandra", // 新建Cassandra Repository所在包
        cassandraTemplateRef = "cassandraTemplate"
)
public class CassandraConfig extends AbstractCassandraConfiguration {

    @Value("${spring.data.cassandra.contact-points}")
    private String contactPoints;

    @Value("${spring.data.cassandra.port}")
    private int port;

    @Value("${spring.data.cassandra.keyspace-name}")
    private String keyspaceName;

    @Value("${spring.data.cassandra.username}")
    private String username;

    @Value("${spring.data.cassandra.password}")
    private String password;

    @Override
    protected String getKeyspaceName() {
        return keyspaceName;
    }

    @Override
    protected String getContactPoints() {
        return contactPoints;
    }

    @Override
    protected int getPort() {
        return port;
    }

    @Override
    public CassandraClusterFactoryBean cluster() {
        CassandraClusterFactoryBean cluster = super.cluster();
        cluster.setUsername(username);
        cluster.setPassword(password);
        return cluster;
    }
}

三、Repository与实体类的组织 #

• 现有JPA Repository和@Entity标记的实体类，保持在原包路径（如com.yourproject.repository.postgres、com.yourproject.entity.postgres），无需任何修改，自动绑定PostgreSQL数据源。
• 新建Cassandra相关代码：
  • Repository放在com.yourproject.repository.cassandra包下，继承CassandraRepository：

    package com.yourproject.repository.cassandra;
    
    import com.yourproject.entity.cassandra.UserCassandra;
    import org.springframework.data.cassandra.repository.CassandraRepository;
    
    public interface UserCassandraRepository extends CassandraRepository<UserCassandra, String> {
    }
    

  • 实体类放在com.yourproject.entity.cassandra包下，用@Table标记：

    package com.yourproject.entity.cassandra;
    
    import org.springframework.data.cassandra.core.mapping.PrimaryKey;
    import org.springframework.data.cassandra.core.mapping.Table;
    
    @Table("user")
    public class UserCassandra {
        @PrimaryKey
        private String id;
        private String name;
        // getter、setter方法
    }
    

四、关键注意事项 #

• 严格按照包路径隔离两类Repository和实体类，避免Spring扫描时混淆。
• @Primary注解必须加在PostgreSQL的配置Bean上，确保原有依赖JPA的业务代码默认使用PostgreSQL数据源。
• 启动类无需添加额外扫描注解，两个配置类已分别指定扫描范围。

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