java 给es id赋值

原创

mob64ca12d16caa 2023-12-30 08:54:59 ©著作权

文章标签 elasticsearch 赋值 java 文章分类 Java 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12d16caa的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java给ES ID赋值

在使用Java与Elasticsearch（简称ES）进行数据交互时，通常需要给每个文档分配一个唯一的标识符，也就是ID。ES提供了多种方式来生成ID，本文将介绍如何使用Java给ES ID赋值。

使用UUID生成ID

UUID（Universally Unique Identifier）是一种128位的唯一标识符，可以通过Java的java.util.UUID类生成。在ES中，可以使用UUID生成ID，并将其赋值给文档。

下面是一个使用UUID生成ID并将其赋值给ES文档的示例代码：

import org.elasticsearch.action.index.IndexRequest;
import org.elasticsearch.client.RequestOptions;
import org.elasticsearch.client.RestHighLevelClient;
import org.elasticsearch.common.xcontent.XContentType;
import org.elasticsearch.index.Index;

import java.io.IOException;
import java.util.UUID;

public class ESExample {
    private RestHighLevelClient client;

    public void indexDocument(String indexName, String documentJson) throws IOException {
        IndexRequest request = new IndexRequest(indexName);
        request.id(UUID.randomUUID().toString());
        request.source(documentJson, XContentType.JSON);

        client.index(request, RequestOptions.DEFAULT);
    }

    // 其他代码...
}

在上述代码中，UUID.randomUUID().toString()用于生成一个随机的UUID，并将其转换为字符串形式。然后，使用request.id()方法将生成的ID赋值给ES文档。

使用自定义ID生成策略

除了使用UUID，还可以自定义ID生成策略。例如，可以使用时间戳、业务主键等信息生成ID。下面是一个使用时间戳生成ID并赋值给ES文档的示例代码：

import org.elasticsearch.action.index.IndexRequest;
import org.elasticsearch.client.RequestOptions;
import org.elasticsearch.client.RestHighLevelClient;
import org.elasticsearch.common.xcontent.XContentType;
import org.elasticsearch.index.Index;

import java.io.IOException;
import java.time.Instant;

public class ESExample {
    private RestHighLevelClient client;

    public void indexDocument(String indexName, String documentJson) throws IOException {
        IndexRequest request = new IndexRequest(indexName);
        request.id(generateCustomId());
        request.source(documentJson, XContentType.JSON);

        client.index(request, RequestOptions.DEFAULT);
    }

    private String generateCustomId() {
        long timestamp = Instant.now().toEpochMilli();
        String businessKey = "123456";
        return timestamp + "-" + businessKey;
    }

    // 其他代码...
}

在上述代码中，generateCustomId()方法使用当前时间戳和业务主键生成一个自定义的ID，并将其赋值给ES文档。