使用 Spring 数据 MongoRepository 更新查询的自定义方法

Question

我正在使用 org.springframework.data.mongodb.repository.MongoRepository。我写了一些像下面这样的自定义方法，

public interface DocRepository extends MongoRepository<Doc, String> {
     Doc findByDocIdAndAssignmentId(final String docId, final String assignemtId);
}

如何编写自定义方法，在满足条件时更新所有条目。

例如，如果分配 ID 为“xyz”，则将文档倾斜字段设置为“abc”？

Answer 1

1) 您需要创建接口，例如 CustomDocRepository 并将此接口添加为您的 DocRepository 的基础：

public interface DocRepository extends MongoRepository<Doc, String>, CustomDocRepository {

    void updateDocumentTitle(String id, String title);

}

2) 您需要为 DocRepository 添加实现：

@Repository
public class CustomDocRepositoryImpl implements DocRepository {
  @Autowired
  private MongoTemplate mongoTemplate;


@Override
public void updateDocumentTitle(String id, String title) {
    Query query = new Query().addCriteria(where("_id").is(id));

    Update update = new Update();
    update.set("title", title);
    mongoTemplate.update(Doc.class).matching(query).apply(update).first();
}
}

这就是你需要做的一切