Spring Data Elasticsearch Repository如何实现Upsert功能？

默认情况下，Spring Data Elasticsearch的ElasticsearchRepository.save()方法仅支持全量替换已有文档，无法直接实现你需要的「不存在则插入、存在则更新指定字段」的upsert逻辑（对应原RestHighLevelClient中UpdateRequest.docAsUpsert(true)的效果）。不过可以通过以下两种方案实现需求，尽量贴合Repository的低代码风格：

方案一：自定义Repository片段（推荐） #

通过自定义Repository片段结合ElasticsearchOperations实现通用的upsert逻辑，复用性强，适合复杂实体：

1. 定义自定义片段接口

public interface CustomUpsertOperations<T> {
    void upsert(T entity);
}

2. 实现片段逻辑

public class CustomUpsertOperationsImpl<T> implements CustomUpsertOperations<T> {

    private final ElasticsearchOperations elasticsearchOperations;

    public CustomUpsertOperationsImpl(ElasticsearchOperations elasticsearchOperations) {
        this.elasticsearchOperations = elasticsearchOperations;
    }

    @Override
    public void upsert(T entity) {
        // 获取实体对应的索引信息
        IndexCoordinates index = elasticsearchOperations.getIndexCoordinatesFor(entity.getClass());
        // 获取实体ID
        String entityId = elasticsearchOperations.getEntityId(entity);
        
        // 构建UpdateQuery，同时设置更新文档和upsert文档
        UpdateQuery updateQuery = UpdateQuery.builder(entityId)
                .withDocument(elasticsearchOperations.getDocumentMapper().map(entity))
                .withUpsert(elasticsearchOperations.getDocumentMapper().map(entity))
                .build();
                
        // 执行upsert操作
        elasticsearchOperations.update(updateQuery, index);
    }
}

3. 让主Repository继承自定义片段

public interface UserRepository extends ElasticsearchRepository<User, String>, CustomUpsertOperations<User> {
}

之后直接调用userRepository.upsert(user)即可实现upsert，底层会对应ES的_update API并启用upsert参数，和原RestHighLevelClient的逻辑完全一致。

方案二：使用@Query注解（适合简单场景） #

对于字段较少的简单场景，可以通过@Query注解直接编写ES的更新DSL，嵌入upsert逻辑：

public interface UserRepository extends ElasticsearchRepository<User, String> {

    @Query("""
        {"script": {
            "source": "ctx._source.username = params.username; ctx._source.email = params.email",
            "params": {"username": ?0, "email": ?1}
        },
        "upsert": {"username": ?0, "email": ?1}}
        """)
    void upsertUser(String username, String email);
}

这种方式无需编写额外实现类，但需要手动维护DSL，字段较多时容易出错，灵活性不如方案一。

关键说明 #

Spring Data Elasticsearch 5.x（对应Spring Boot 3+）的原生Repository方法中没有直接提供upsert的封装，因为upsert属于带有条件的写操作，不属于标准的CRUD方法范畴。因此必须通过自定义片段或@Query注解扩展实现，无法仅通过Repository的自动生成查询方法完成。

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