Spring Boot集成MongoDB如何将MongoRepository保存结果转自定义类返回JSON

根因说明 #

1. 语法错误：new customerRepository.save(customer) 写法非法，customerRepository是Spring注入的Bean实例，不需要加new关键字；且save方法返回的是Customer实体类型，和你要返回的CustomerResponse DTO类型不匹配，无法直接传入ResponseEntity构造参数。
2. 注解配置错误：Customer类上的@Document注解属性使用错误，指定MongoDB集合名称需要用collection属性，collation是用于配置排序字符集规则的属性，会导致集合映射异常。
3. 缺少类型转换逻辑：Spring不会自动完成MongoDB实体到自定义响应DTO的字段拷贝，需要手动实现转换逻辑。
4. 异常逻辑错误：原有try-finally配置不管接口调用成功失败都会打印数据库保存失败的日志，不符合业务逻辑。

修复步骤 #

1. 修正Customer实体的@Document注解 #

package com.customerapi.model;

import org.springframework.data.annotation.Id;
import org.springframework.data.mongodb.core.mapping.Document;

import lombok.Getter;
import lombok.Setter;
import lombok.ToString;

@Getter
@Setter
@ToString
// 替换collation属性为collection
@Document(collection = "customer")
public class Customer {
    @Id
    private String custRefId;
    
    // 其余原有字段保持不变
}

2. 修正Controller层方法逻辑 #

package com.customerapi.resource;

import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RestController;

import com.customerapi.repository.CustomerRepository;
import com.customerapi.model.Customer;
import com.customerapi.dto.CustomerResponse;

@RestController
public class CustomerController {
    
    @Autowired
    private CustomerRepository customerRepository;
    
    // 移除冗余的@ResponseStatus注解，ResponseEntity已携带状态码
    @PostMapping(value="/addCustomer", produces = MediaType.APPLICATION_JSON_VALUE)
    public ResponseEntity<CustomerResponse> saveCustomer(@RequestBody Customer customer) {
        try {
            // 保存实体到MongoDB
            Customer savedCustomer = customerRepository.save(customer);
            // 转换实体到响应DTO
            CustomerResponse response = new CustomerResponse();
            // 手动赋值字段，字段较多时也可以用BeanUtils自动拷贝同名字段
            // BeanUtils.copyProperties(savedCustomer, response);
            response.setCustRefId(savedCustomer.getCustRefId());
            // 返回JSON响应
            return new ResponseEntity<>(response, HttpStatus.CREATED);
        } catch (Exception e) {
            // 异常时打印错误日志，返回500状态码
            e.printStackTrace();
            return new ResponseEntity<>(HttpStatus.INTERNAL_SERVER_ERROR);
        }
    }
}

补充说明 #

如果后续需要在响应中返回更多字段，只需要在CustomerResponse中定义对应字段，然后在转换逻辑中补充赋值即可，使用BeanUtils.copyProperties可以自动匹配两个类的同名字段完成拷贝，减少手写赋值的代码量。

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