java - 使用 spring data 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) 您需要创建接口(interface)，例如 CustomDocRepository 并将此接口(interface)添加为 DocRepository 的 Base:

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();
}
}

这就是你需要做的一切