【MongoDB整合】Springboot3.x整合MongoDB

1 MongoDB整合

• 1.1 依赖引入

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

---

• 1.2 yaml配置

spring:
  data:
    mongodb:
      # mongodb://username:password@ip:port/database
      uri: mongodb://admin:123456@127.0.0.1:27017/demo
      # 是否自动创建索引
      auto-index-creation: true

---

• 1.3 自定义工具类(用于update操作)

public class MongoUtil {

    /**
     * 根据bean中非空值构建成mongo更新类
     * @param bean 类对象
     * @return mongo更新类
     * @param <T> 泛型
     */
    public static <T> Update getUpdate(T bean) {
        Update update = new Update();
        Map<String, Object> paramMap = getParamMap(bean);
        if (CollectionUtils.isEmpty(paramMap)) {
            return update;
        }
        for (Map.Entry<String, Object> entry : paramMap.entrySet()) {
            update.set(entry.getKey(), entry.getValue());
        }
        return update;
    }

    /**
     * 获取类中属性值
     * @param bean 类对象
     * @return key: 属性名, value: 属性值
     * @param <T> 泛型
     */
    private static <T> Map<String, Object> getParamMap(T bean) {
        Class<?> beanClass = bean.getClass();
        Map<String, Method> methodMap = new HashMap<>(16);
        int nameIndex;
        String filedName;
        // 记录get方法
        for (Method method : beanClass.getMethods()) {
            if (method.getParameterCount() == 0 && method.getReturnType() != Void.class) {
                if (method.getName().startsWith("get")) {
                    nameIndex = 3;
                } else if (method.getName().startsWith("is")) {
                    nameIndex = 2;
                } else {
                    continue;
                }
                filedName = String.valueOf(method.getName().charAt(nameIndex)).toLowerCase() + method.getName().substring(nameIndex+1);
                methodMap.put(filedName, method);
            }
        }
        // 调用get方法取值，非空则放入paramMap中
        Map<String, Object> paramMap = new HashMap<>(16);
        for (Field field : beanClass.getDeclaredFields()) {
            Method method = methodMap.get(field.getName());
            if (method != null) {
                try {
                    Object value = method.invoke(bean);
                    if (Objects.isNull(value)) {
                        continue;
                    }
                    paramMap.put(field.getName(), value);
                } catch (Exception e) {
                    throw new RuntimeException(e);
                }
            }
        }
        return paramMap;
    }
}

---

2 使用演示

• 2.1 创建索引类

@Data
@Document("demo")
public class MongoDemo {

    @Id
    private Integer demoId;

    private String demoName;

    private LocalDateTime createTime;
}

---

• 2.2 增删查改

@RequestMapping("/mongo/demo")
@RestController
public class MongoDemoController {

    @Autowired
    private MongoOperations mongoOperations;

    @PostMapping
    public void save(@RequestBody MongoDemo mongoDemo) {
        mongoDemo.setCreateTime(LocalDateTime.now());
        mongoOperations.save(mongoDemo);
    }

    @PutMapping
    public void update(@RequestBody MongoDemo mongoDemo) {
        Update update = MongoUtil.getUpdate(mongoDemo);
        Criteria criteria = new Criteria("demoId").is(mongoDemo.getDemoId());
        Query query = new Query(criteria);
        mongoOperations.updateFirst(query, update, MongoDemo.class);
    }

    @DeleteMapping("/{demoId}")
    public void delete(@PathVariable("demoId") Integer demoId) {
        Criteria criteria = new Criteria("demoId").is(demoId);
        Query query = new Query(criteria);
        mongoOperations.remove(query, MongoDemo.class);
    }

    @GetMapping("/{demoId}")
    public MongoDemo getInfo(@PathVariable("demoId") Integer demoId) {
        Criteria criteria = new Criteria("demoId").is(demoId);
        Query query = new Query(criteria);
        return mongoOperations.findOne(query, MongoDemo.class);
    }

    @GetMapping("/list")
    public List<MongoDemo> list() {
        Criteria criteria = new Criteria();
        Query query = new Query(criteria);
        return mongoOperations.find(query, MongoDemo.class);
    }

    @GetMapping("/page")
    public Page<MongoDemo> page(int pageNum, int pageSize) {
        Criteria criteria = new Criteria();
        Query query = new Query(criteria);
        long total = mongoOperations.count(query, MongoDemo.class);
        PageRequest pageRequest = PageRequest.of(pageNum-1, pageSize);
        query.with(pageRequest);
        List<MongoDemo> mongoDemos = mongoOperations.find(query, MongoDemo.class);
        return new PageImpl<>(mongoDemos, pageRequest, total);
    }
}