Spring Boot集成MongoDB时Team实体ID未自动生成问题求助

问题重现 #

你在Spring Boot中使用MongoDB时，保存Rider文档时，Rider的ID能自动生成，但嵌套的Team对象ID为null。相关代码和数据如下：

Rider 实体类 #

package com.app.enities;

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

import java.util.List;

@Document(collection = "Rider")
@Getter
@Setter
@AllArgsConstructor
@NoArgsConstructor
@ToString(callSuper = true)
public class Rider {
    @Id
    private String id;
    private String firstName;
    private String lastName;
    private String shortCred;
    private Team teamName;
    private List<Sponsor> sponsorList; //Enum
}

Team 实体类 #

package com.app.enities;

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

@Document
@Getter
@Setter
@AllArgsConstructor
@NoArgsConstructor
@ToString(callSuper = true)
public class Team {
    @Id
    private String id;
    private String name;
    private Bikes bike; //Enum
}

请求JSON #

{
    "firstName": "Marc",
    "lastName": "Marquez",
    "shortCred": "MM93",
    "teamName": {
        "name": "Gresini Racing",
        "bike": "DUCATI"
    },
    "sponsorList": [
        "REDBULL",
        "SHOEI",
        "ALPINESTAR",
        "ESTRELLA_GALICIA",
        "SAMSUNG"
    ]
}

保存后返回结果 #

{
    "id": "65c731de83a0e43d7526c880",
    "firstName": "Marc",
    "lastName": "Marquez",
    "shortCred": "MM93",
    "teamName": {
        "id": null, <--- null值
        "name": "Gresini Racing",
        "bike": "DUCATI"
    },
    "sponsorList": [
        "REDBULL",
        "SHOEI",
        "ALPINESTAR",
        "ESTRELLA_GALICIA",
        "SAMSUNG"
    ]
}

原因分析 #

MongoDB是文档型数据库，和关系型SQL的关联逻辑完全不同：

• 默认情况下，直接把Team作为Rider的属性时，Spring Data MongoDB会将Team嵌入到Rider文档中，不会把它当成独立的集合文档处理，因此不会自动生成ID。
• SQL中的@OneToOne/@ManyToOne是关系型关联，而MongoDB没有原生外键概念，仅支持嵌入式存储或**引用型关联（@DBRef）**两种方式。

解决方案 #

根据业务需求，分两种场景处理：

场景1：Team作为Rider的嵌入式文档（不需要独立存储） #

如果Team不需要单独维护成集合，只是Rider的附属信息：

• 可移除Team类上的@Document注解（明确它不是独立集合）
• 若一定要给嵌入式Team生成ID，需手动在保存前设置：

  // 在Service层保存Rider前，手动生成Team的ID
  newRider.getTeamName().setId(ObjectId.get().toString());
  riderRepository.save(newRider);
  

场景2：Team作为独立集合存储（需要单独维护） #

如果Team需要独立存在（比如多个Rider可能关联同一个Team），需使用@DBRef实现引用关联：

1. 修改Rider实体类，给teamName字段添加@DBRef注解：

   @Document(collection = "Rider")
   @Getter
   @Setter
   @AllArgsConstructor
   @NoArgsConstructor
   @ToString(callSuper = true)
   public class Rider {
       @Id
       private String id;
       private String firstName;
       private String lastName;
       private String shortCred;
       @DBRef // 添加引用注解
       private Team teamName;
       private List<Sponsor> sponsorList; //Enum
   }
   

2. 调整保存逻辑，先保存Team获取自动生成的ID，再关联到Rider保存：

   // 1. 先保存Team，获取带ID的Team对象
   Team savedTeam = teamRepository.save(newRider.getTeamName());
   // 2. 将保存后的Team关联到Rider
   newRider.setTeamName(savedTeam);
   // 3. 最后保存Rider
   Rider savedRider = riderRepository.save(newRider);
   

   保存后，Rider中的teamName会包含自动生成的ID，同时Team会单独存储在team集合中。

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