SpringBoot（12） - NoSQL（3） - mogodb多数据源配置

一时失言乱红尘 2022-04-22 13:14 47阅读 0赞

参考：[https://www.cnblogs.com/ityouknow/p/6828919.html][https_www.cnblogs.com_ityouknow_p_6828919.html]

1. 添加mongodb starter

<dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-data-mongodb</artifactId>
    </dependency>

2. 配置数据源

application.yml中添加以下配置：

mongodb: 
      primary: 
        host: localhost
        port: 27017
        database: test
      secondary: 
        host: localhost
        port: 27017
        database: test2

3. 获取属性配置

import org.springframework.boot.autoconfigure.mongo.MongoProperties;
    import org.springframework.boot.context.properties.ConfigurationProperties;
    import org.springframework.stereotype.Component;
    
    @Component
    @ConfigurationProperties(prefix = "mongodb")
    public class MongoPropertiesWrapper {
        private MongoProperties primary = new MongoProperties();
        private MongoProperties secondary = new MongoProperties();
       // getter setter略
    }

4. 配置不同包路径下使用不同的数据源

（1）primary包

import org.springframework.context.annotation.Configuration;
    import org.springframework.data.mongodb.repository.config.EnableMongoRepositories;
    
    @Configuration
    @EnableMongoRepositories(basePackages = "com.momo.springbootmongodb.multi.dao.primary",
            mongoTemplateRef = PrimaryMongoConfig.MONGO_TEMPLATE)
    public class PrimaryMongoConfig {
        public static final String MONGO_TEMPLATE = "primaryMongoTemplate";
    }

（2）secondary包

import org.springframework.context.annotation.Configuration;
    import org.springframework.data.mongodb.repository.config.EnableMongoRepositories;
    
    @Configuration
    @EnableMongoRepositories(basePackages = "com.momo.springbootmongodb.multi.dao.sencodary",
            mongoTemplateRef = SecondaryMongoConfig.MONGO_TEMPLATE)
    public class SecondaryMongoConfig {
        public static final String MONGO_TEMPLATE = "secondaryMongoTemplate";
    }

（3）构造MongoTemplate

@Configuration
    public class MultipleMongoConfig {
    
        @Autowired
        private MongoPropertiesWrapper mongoPropertiesWrapper;
        
    
        @Bean(PrimaryMongoConfig.MONGO_TEMPLATE)
        @Primary
        public MongoTemplate primaryMongoTemplate() throws Exception {
            return new MongoTemplate(this.primaryFactory(this.mongoPropertiesWrapper.getPrimary()));
        }
    
        @Bean(SecondaryMongoConfig.MONGO_TEMPLATE)
        public MongoTemplate secondaryMongoTemplate() throws Exception {
            return new MongoTemplate(this.secondaryFactory(this.mongoPropertiesWrapper.getSecondary()));
        }
    
        @Bean
        @Primary
        public MongoDbFactory primaryFactory(MongoProperties mongoProperties) throws Exception {
            return new SimpleMongoDbFactory(new MongoClient(mongoProperties.getHost(), mongoProperties.getPort()), 
                    mongoProperties.getDatabase());
        }
    
        @Bean
        public MongoDbFactory secondaryFactory(MongoProperties mongoProperties) throws Exception {
            return new SimpleMongoDbFactory(new MongoClient(mongoProperties.getHost(), mongoProperties.getPort()),
                    mongoProperties.getDatabase());
        }
    }

5. 创建实体

import org.springframework.data.annotation.Id;
    import org.springframework.data.mongodb.core.mapping.Document;
    import org.springframework.data.mongodb.core.mapping.Field;
    import java.io.Serializable;
    
    @Document(collection = "t_user")
    public class PrimaryUser implements Serializable {
        public PrimaryUser() {
        }
        public PrimaryUser(Long id, String userName) {
            this.id = id;
            this.userName = userName;
        }
    
        @Id
        @Field("id")
        private Long id;
    
        @Field("user_name")
        private String userName;
        // getter setter略
    }

6. 创建Dao

import org.springframework.data.mongodb.repository.MongoRepository;
    
    public interface PrimaryUserDao extends MongoRepository<PrimaryUser, Long> {
    }

另外一个与之类似

7. 测试

import org.junit.Test;
    import org.junit.runner.RunWith;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.boot.test.context.SpringBootTest;
    import org.springframework.test.context.junit4.SpringRunner;
    import java.util.List;
    
    @RunWith(SpringRunner.class)
    @SpringBootTest
    public class MultiApplicationTests {
    
        @Autowired
        private PrimaryUserDao primaryUserDao;
        @Autowired
        private SecondaryUserDao secondaryUserDao;
    
        @Test
        public void testSave() {
            this.primaryUserDao.save(new PrimaryUser(1L, "p-小明"));
    
            this.secondaryUserDao.save(new SecondaryUser(2L, "sec-小明"));
    
            List<PrimaryUser> primaries = this.primaryUserDao.findAll();
            for (PrimaryUser primary : primaries) {
                System.out.println(primary.toString());
            }
    
            List<SecondaryUser> secondaries = this.secondaryUserDao.findAll();
            for (SecondaryUser secondary : secondaries) {
                System.out.println(secondary.toString());
            }
        }
    }

demo代码：[https://github.com/mytt-10566/springboot-mongodb][https_github.com_mytt-10566_springboot-mongodb]

[https_www.cnblogs.com_ityouknow_p_6828919.html]: https://www.cnblogs.com/ityouknow/p/6828919.html
[https_github.com_mytt-10566_springboot-mongodb]: https://github.com/mytt-10566/springboot-mongodb