在将它们添加到 MongoDB 之前，如何在 Spring Data 中处理插入请求？

Question

在遵循https://spring.io/guides/gs/accessing-mongodb-data-rest/ 的非常有用的教程之后，我正在尝试创建一个链接缩短器应用程序。我编写了一个 URL 类（带有 id、longURL 和 hash 属性）和 URLRepository 类（目前，只有 findByHash 方法，如演示中所示）。请注意，“散列”是指由尚未实现的方法生成的随机短字符串，而不是像 MD5 这样的实际散列函数。例如，只需致电独立的generateHash(String URL)

我可以使用 Spring 提供的 REST 接口将新条目添加到数据库中，也可以获取它们。但是，我想让应用程序自己生成哈希并在存储之前进行一些检查和处理 - 主要检查 URL 是否尚未存储，在这种情况下它只会返回现有的哈希。

我想我必须在我的URLRepository 类下扩展SimpleMongoRepository 的<S extends T> List<S> save(Iterable<S> entites); 方法（由MongoRepository 扩展），但我不确定该怎么做。我也看到了insert方法，不知道应该用哪个

URL.java

public class URL {

    @Id private String id;

    private String longURL;
    private String hash;

    public String getLongURL() {
        return longURL;
    }

    public void setLongURL(String longURL) {
        this.longURL = longURL;
    }

    public String getHash() {
        return hash;
    }

    public void setHash(String Hash) {
        this.hash = Hash;
    }

}

URLRepository.java

import org.springframework.data.mongodb.repository.MongoRepository;
import org.springframework.data.repository.query.Param;
import org.springframework.data.rest.core.annotation.RepositoryRestResource;
import org.springframework.util.Assert;

@RepositoryRestResource(collectionResourceRel = "urls", path = "urls")
public interface URLRepository extends MongoRepository<URL, String> {

    List<URL> findByHash(@Param("hash") String hash);

}

Answer 1

感觉你最好编写一个自定义控制器而不是在这里使用 Spring Data REST，因为你基本上需要两个资源：一个用于添加链接或返回现有的，第二个用于通过其哈希检索原始 URI .

在第一种方法中，您只需调用存储库方法findByLongURL(…) 并使用获得的URL 实例，如果您有结果或采取第二步实际创建哈希并保存URL 实例思想存储库.第二个资源基本上只是调用你已经存在的方法。

这很简单，很容易消化。

如果您需要将前一种方法实现为原子操作，则需要手动实现存储库查询方法（有关一般说明，请阅读reference documentation 中的相关部分）：

class UrlRepositoryImpl implements UrlRepositoryCustom {

  private final MongoOperations operations;

  public UrlRepositoryImpl(MongoOperations operations) {
    this.operations = operations;
  }

  @Override
  public URL findOrInsert(String source) {

    // What to find?
    Query query = Query.query(Criteria.where("longURL").is(source);

    // What to write if nothing can be found
    Update update = new Update()
      .setOnInsert("longURL", source)
      .setOnInsert("hash", calculatedHash);

    FindAndModifyOptions options = new FindAndModifyOptions.options()
      .returnNew(true) // returns the document insert (if so)
      .upsert(true); // insert document if it doesn't exist

    return operations.findAndModify(query, update, options, URL.class);
  }
}

如您所见，这涉及处理一些较低级别的细节（尽管可以通过使用静态导入来减少冗长），但基本上为您提供了原子操作。