Spring Boot多存储树结构API：MongoDB创建子节点报父节点未找到

背景 #

我正在构建支持内存、PostgreSQL和MongoDB三种存储实现的Spring Boot树结构API，可通过配置属性（如app.storage=postgres或app.storage=mongo）切换存储方式。

API契约 #

创建子节点的端点为POST /nodes/{parentId}/children，控制器接收Long类型的parentId：

@Override
public NodeResponse nodesParentIdChildrenPost(Long parentId, NodeRequest nodeRequest) {
    Object child = persistenceService.addChild(parentId, nodeRequest.getName());
    return toNodeResponse(child);
}

现有实现情况 #

PostgreSQL实现 #

该实现使用数值ID，运行正常：

@Entity
@Table(name = "nodes")
public class NodeEntity {

    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;

    private String name;

    private Long parentId;
}

示例请求：POST /nodes/1/children，请求体：

{
  "name": "1.1 Caja"
}

MongoDB实现 #

文档结构如下：

@Document(collection = "nodes")
public class MongoNodeDocument {

    @Id
    private String id;

    private String value;

    private String parentId;
}

仓库方法：

public MongoNodeDocument addChild(String parentId, String value) {
    mongoRepository.findById(parentId)
            .orElseThrow(() ->
                    new RuntimeException("Nodo padre no encontrado: " + parentId));

    MongoNodeDocument child = new MongoNodeDocument(value, parentId);
    return mongoRepository.save(child);
}

持久化服务：

@Override
public Object addChild(Long parentId, String name) {
    return mongoTreeRepository.addChild(String.valueOf(parentId), name);
}

问题描述 #

MongoDB生成的_id是665f2a0e9d7...这类字符串ID，而非1、2、3这样的数值ID。调用POST /nodes/1/children时，MongoDB会尝试查找_id="1"的文档，导致返回“父节点未找到”错误。

解决方案（保持API契约一致） #

方案1：统一使用字符串类型ID #

• 修改API契约中的parentId为String类型，控制器接收String参数：

  @Override
  public NodeResponse nodesParentIdChildrenPost(String parentId, NodeRequest nodeRequest) {
      Object child = persistenceService.addChild(parentId, nodeRequest.getName());
      return toNodeResponse(child);
  }
  

• 调整PostgreSQL实体类的ID类型为String，使用UUID生成策略：

  @Entity
  @Table(name = "nodes")
  public class NodeEntity {
  
      @Id
      @GeneratedValue(strategy = GenerationType.UUID)
      private String id;
  
      private String name;
  
      private String parentId;
  }
  

• MongoDB实现无需修改，直接使用原生字符串ID。
• 优势：彻底消除ID类型差异，API契约统一适配所有存储；无需额外转换逻辑。
• 劣势：需要迁移现有PostgreSQL数据的ID类型；原数值ID请求需做兼容（可在控制器层添加类型转换逻辑）。

方案2：MongoDB使用自增数值ID #

• 为MongoDB实现自增数值ID生成逻辑，模拟PostgreSQL的IDENTITY策略：
  1. 创建序列集合存储自增计数器：

     @Document(collection = "counters")
     public class CounterDocument {
         @Id
         private String id;
         private Long seq;
     }
     

  2. 实现ID生成器，获取下一个自增ID：

     public Long getNextSequence(String seqName) {
         Query query = new Query(Criteria.where("_id").is(seqName));
         Update update = new Update().inc("seq", 1);
         FindAndModifyOptions options = new FindAndModifyOptions().returnNew(true).upsert(true);
         CounterDocument counter = mongoTemplate.findAndModify(query, update, options, CounterDocument.class);
         return counter.getSeq();
     }
     

  3. 修改MongoNodeDocument的ID类型为Long：

     @Document(collection = "nodes")
     public class MongoNodeDocument {
     
         @Id
         private Long id;
     
         private String value;
     
         private Long parentId;
     }
     

  4. 调整仓库方法，使用自增ID创建子节点：

     public MongoNodeDocument addChild(Long parentId, String value) {
         mongoRepository.findById(parentId)
                 .orElseThrow(() -> new RuntimeException("父节点未找到: " + parentId));
     
         MongoNodeDocument child = new MongoNodeDocument(value, parentId);
         child.setId(getNextSequence("node_seq"));
         return mongoRepository.save(child);
     }
     

• 优势：API契约无需修改，保持Long类型ID；PostgreSQL实现无需改动。
• 劣势：MongoDB需额外维护自增序列，增加复杂度；分布式场景下需处理并发问题（可通过原子操作解决）。

方案3：控制器层做ID类型适配 #

• 保持API契约的Long类型parentId不变，在MongoDB持久化层做反向适配：
  1. 为MongoDB节点添加numericId字段存储数值ID，保留原生字符串_id：

     @Document(collection = "nodes")
     public class MongoNodeDocument {
     
         @Id
         private String id;
         private Long numericId; // 新增数值ID字段
         private String value;
         private Long parentNumericId; // 用数值类型存储父ID
     }
     

  2. 修改MongoDB仓库方法，通过parentNumericId查找父节点：

     public MongoNodeDocument addChild(Long parentId, String value) {
         mongoRepository.findByParentNumericId(parentId)
                 .orElseThrow(() -> new RuntimeException("父节点未找到: " + parentId));
     
         MongoNodeDocument child = new MongoNodeDocument(value, parentId);
         child.setNumericId(getNextSequence("node_seq")); // 用自增逻辑生成数值ID
         return mongoRepository.save(child);
     }
     

• 优势：API契约完全不变；PostgreSQL实现无需修改。
• 劣势：MongoDB文档冗余，需维护两个ID字段；增加ID转换和维护的复杂度。

方案选择建议 #

如果是新项目或数据量不大，优先选方案1，统一使用字符串ID（UUID），长期维护成本最低；如果必须保留数值ID的API契约且不想改动PostgreSQL实现，选方案2；如果需要兼容现有大量数值ID请求且不想修改MongoDB原生ID生成逻辑，可考虑方案3。

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