springboot 简单配置mongodb多数据源
准备工作：
- 本地mongodb一个
- 创建两个数据库 student 和 student-two
所需jar包：
# springboot基于的版本
org.springframework.boot
spring-boot-starter-parent
2.2.1.RELEASE
# maven 关键jar包
org.springframework.boot
spring-boot-starter-web
org.projectlombok
lombok
true
org.springframework.boot
spring-boot-starter-data-mongodb
org.springframework.boot
spring-boot-configuration-processor
true
配置文件：
# uri 配置样例: mongodb://myDBReader:D1fficultP%[email protected]:27017/admin
# 如果用户名或密码包含at符号@，冒号：，斜杠/或百分号％字符，请使用百分比编码方式消除歧义
# uri 集群配置样例： mongodb://myDBReader:D1fficultP%[email protected]:27017,mongodb1.example.com:27017,mongodb2.example.com:27017/admin?replicaSet=myRepl
douzi:
mongo:
primary:
uri: mongodb://127.0.0.1:27017/student
repository-package: com.douzi.mongo.dao.primary
secondary:
uri: mongodb://127.0.0.1:27017/student-two
repository-package: com.douzi.mongo.dao.secondary多数据源配置类 MultipleMongoConfig：
package com.douzi.mongo.config;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.data.mongodb.MongoDbFactory;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.SimpleMongoDbFactory;
import org.springframework.util.StringUtils;
import com.mongodb.MongoClientURI;
/**
* @Author douzi
* @Title: MultipleMongoConfig
* @Description: 读取对应的配置信息并且构造对应的MongoTemplate
* @Date 2023-09-27
*/
@Configuration
public class MultipleMongoConfig {
@Value("${douzi.mongo.primary.uri}")
private String primaryUri;
@Value("${douzi.mongo.secondary.uri}")
private String secondaryUri;
/**
* 配置主数据源template
* @return 主数据源template
*/
@Primary
@Bean(name = PrimaryMongoConfig.MONGO_TEMPLATE)
public MongoTemplate primaryMongoTemplate() {
return new MongoTemplate(primaryFactory());
}
/**
* 配置从数据源template
* @return 从数据源template
*/
@Bean
@Qualifier(SecondaryMongoConfig.MONGO_TEMPLATE)
public MongoTemplate secondaryMongoTemplate() {
return new MongoTemplate(secondaryFactory());
}
/**
* 配置主数据源db工厂
* @param mongo 属性配置信息
* @return 主数据源db工厂
*/
@Bean
@Primary
public MongoDbFactory primaryFactory() {
if (StringUtils.isEmpty(primaryUri)) {
throw new RuntimeException("必须配置mongo primary Uri");
}
return new SimpleMongoDbFactory(new MongoClientURI(primaryUri));
}
/**
* 配置从数据源db工厂
* @param mongo 属性配置信息
* @return 从数据源db工厂
*/
@Bean
public MongoDbFactory secondaryFactory() {
return new SimpleMongoDbFactory(new MongoClientURI(secondaryUri));
}
}
主数据源配置 PrimaryMongoConfig：
package com.douzi.mongo.config;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.mongodb.repository.config.EnableMongoRepositories;
/**
* @Author douzi
* @Title: PrimaryMongoConfig
* @Description: 主数据源配置
* @date 2023-09-27
*/
@Configuration
@EnableMongoRepositories(basePackages = "${douzi.mongo.primary.repository-package}", mongoTemplateRef = PrimaryMongoConfig.MONGO_TEMPLATE)
public class PrimaryMongoConfig {
protected static final String MONGO_TEMPLATE = "primaryMongoTemplate";
}
副数据源配置 SecondaryMongoConfig：
package com.douzi.mongo.config;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.mongodb.repository.config.EnableMongoRepositories;
/**
* @Author douzi
* @Title: SecondaryMongoConfig
* @Description: 从数据源配置
* @Date 2023-09-27
*/
@Configuration
@EnableMongoRepositories(basePackages = "${douzi.mongo.secondary.repository-package}", mongoTemplateRef = SecondaryMongoConfig.MONGO_TEMPLATE)
public class SecondaryMongoConfig {
protected static final String MONGO_TEMPLATE = "secondaryMongoTemplate";
}
以下是辅助测试类 高手可以自行测试：
dao层：
package com.douzi.mongo.dao.primary;
import lombok.Data;
import org.springframework.data.annotation.Id;
import org.springframework.data.mongodb.core.mapping.Document;
import java.io.Serializable;
import java.math.BigDecimal;
import java.util.Date;
/**
* @Author douzi
* @Title: PrimaryStudent
* @Description: 主数据源学生对象
* @Date 2023/09/24 13:33
*/
@Data
@Document(collection = "first_students")
public class PrimaryStudent implements Serializable {
/**
* id
*/
@Id
private Integer id;
/**
* 姓名
*/
private String name;
/**
* 生活费
*/
private BigDecimal salary;
/**
* 生日
*/
private Date birth;
}
##################################分隔符########################################
package com.douzi.mongo.dao.primary;
import org.springframework.data.mongodb.repository.MongoRepository;
/**
* @Author douzi
* @Title: PrimaryRepository
* @Description: 主数据源dao层
* @Date 2023/09/24 16:01
*/
public interface PrimaryRepository extends MongoRepository {
/**
* 通过名字查找学生
*
* @param name 名字
* @return 学生信息
*/
PrimaryStudent findByName(String name);
/**
* 通过名字删除学生
*
* @param name 名字
* @return 学生信息
*/
PrimaryStudent removeStudentByName(String name);
}
##################################分隔符########################################
package com.douzi.mongo.dao.secondary;
import lombok.Data;
import org.springframework.data.annotation.Id;
import org.springframework.data.mongodb.core.mapping.Document;
import java.io.Serializable;
import java.math.BigDecimal;
import java.util.Date;
/**
*
* @Author douzi
* @Title: PrimaryStudent
* @Description: 副数据源学生对象
* @Date 2023/09/24 13:33
*/
@Data
@Document(collection = "secondary_students")
public class SecondaryStudent implements Serializable {
/**
* id
*/
@Id
private Integer id;
/**
* 姓名
*/
private String name;
/**
* 生活费
*/
private BigDecimal salary;
/**
* 生日
*/
private Date birth;
}
##################################分隔符########################################
package com.douzi.mongo.dao.secondary;
import org.springframework.data.mongodb.repository.MongoRepository;
/**
* @Author douzi
* @Title: SecondaryRepository
* @Description: 从数据源dao层
* @Date 2023/09/24 16:02
*/
public interface SecondaryRepository extends MongoRepository {
/**
* 通过名字查找学生
*
* @param name 名字
* @return 学生信息
*/
SecondaryStudent findByName(String name);
/**
* 通过名字删除学生
*
* @param name 名字
* @return 学生信息
*/
SecondaryStudent removeStudentByName(String name);
}
Service层：
package com.douzi.mongo.service;
import com.douzi.mongo.dao.primary.PrimaryRepository;
import com.douzi.mongo.dao.primary.PrimaryStudent;
import com.douzi.mongo.dao.secondary.SecondaryRepository;
import com.douzi.mongo.dao.secondary.SecondaryStudent;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import java.util.List;
/**
*
* @Author douzi
* @Title: StudentService
* @Description:
* @Date 2023/09/24 14:32
*/
@Service
@Slf4j
public class StudentService {
@Autowired
private PrimaryRepository primaryRepository;
@Autowired
private SecondaryRepository secondaryRepository;
/**
* 保存学生
*
* @param primaryStudent
* @param secondaryStudent
*/
public void save(PrimaryStudent primaryStudent, SecondaryStudent secondaryStudent) {
PrimaryStudent student1 = primaryRepository.insert(primaryStudent);
log.info(student1.toString());
SecondaryStudent student2 = secondaryRepository.insert(secondaryStudent);
log.info(student2.toString());
}
/**
* 删除学生
*
* @param name
* @return
*/
public void delete(String name) {
primaryRepository.deleteById(1);
secondaryRepository.deleteById(1);
}
/**
* 修改学生
*
* @param primaryStudent
* @param secondaryStudent
*/
public void update(PrimaryStudent primaryStudent, SecondaryStudent secondaryStudent) {
PrimaryStudent student1 = primaryRepository.save(primaryStudent);
log.info(student1.toString());
SecondaryStudent student2 = secondaryRepository.save(secondaryStudent);
log.info(student2.toString());
}
/**
* 查找学生
*
* @param name 学生姓名
* @return
*/
public void find(String name) {
PrimaryStudent student1 = primaryRepository.findByName(name);
log.info(student1.toString());
SecondaryStudent student2 = secondaryRepository.findByName(name);
log.info(student2.toString());
}
/**
* 查找学生集合
*
* @return 学生集合
*/
public void findAll() {
List primaryRepositoryAll = primaryRepository.findAll();
log.info(primaryRepositoryAll.toString());
List secondaryRepositoryAll = secondaryRepository.findAll();
log.info(secondaryRepositoryAll.toString());
}
}
Controller层：
package com.douzi.mongo.controller;
import com.douzi.mongo.dao.primary.PrimaryStudent;
import com.douzi.mongo.dao.secondary.SecondaryStudent;
import com.douzi.mongo.service.StudentService;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import java.math.BigDecimal;
import java.util.Date;
/**
* @Author douzi
* @Title: StudentController
* @Description:
* @Date 2023/09/24 14:53
*/
@RestController
@RequestMapping("/student")
public class StudentController {
@Autowired
private StudentService studentService;
/**
* 保存学生
*
* @return 保存的学生
*/
@RequestMapping("/save")
public void save() {
PrimaryStudent primaryStudent = new PrimaryStudent();
SecondaryStudent secondaryStudent = new SecondaryStudent();
primaryStudent.setId(1);
primaryStudent.setName("mongo");
primaryStudent.setSalary(BigDecimal.valueOf(1000));
primaryStudent.setBirth(new Date());
BeanUtils.copyProperties(primaryStudent, secondaryStudent);
studentService.save(primaryStudent, secondaryStudent);
}
/**
* 修改学生
*
* @return 修改结果
*/
@RequestMapping("/update")
public void update() {
PrimaryStudent primaryStudent = new PrimaryStudent();
SecondaryStudent secondaryStudent = new SecondaryStudent();
primaryStudent.setId(1);
primaryStudent.setName("mongo");
primaryStudent.setSalary(BigDecimal.valueOf(2000));
primaryStudent.setBirth(new Date());
BeanUtils.copyProperties(primaryStudent, secondaryStudent);
studentService.update(primaryStudent, secondaryStudent);
}
/**
* 根据姓名删除学生
*
* @param name 姓名
* @return 删除结果
*/
@RequestMapping("/delete")
public void delete(String name) {
studentService.delete(name);
}
/**
* 通过名字查找学生
*
* @param name 学生名
* @return 学生信息
*/
@RequestMapping("/find")
public void find(String name) {
studentService.find(name);
}
/**
* 查找所有学生
*
* @return 学生集合
*/
@RequestMapping("/all")
public void findAll() {
studentService.findAll();
}
}
测试：
http://localhost:8080/student/save
结果：
