如何在Elasticsearch中实现字段自增以生成特定格式的ARN

嘿，这个需求我刚好有实践经验，咱们一步步来搞定！Elasticsearch本身没有原生的自增字段，但我们可以通过两种可靠的方式实现你的需求，同时生成DD/MM/YYYY/序号格式的ARN：

方法一：用update_by_query加脚本（适合低并发场景） #

这种方法直接在目标索引里处理自增逻辑，不用额外维护其他索引。核心思路是：每次新增文档时，先捞取当前最大的id值，自增后作为新文档的id，再结合日期生成ARN。

具体操作步骤： #

1. 先创建目标索引（提前定义映射更规范，可选但推荐）：

PUT /arn_index
{
  "mappings": {
    "properties": {
      "id": {"type": "integer"},
      "arn": {"type": "keyword"}
    }
  }
}

2. 执行自增并生成ARN：
   用update_by_query运行Painless脚本，自动计算新ID并生成格式正确的ARN。这里默认用当前系统日期，你也可以换成固定日期或者文档里的其他日期字段：

POST /arn_index/_update_by_query
{
  "script": {
    "source": """
      // 先拿到当前文档的最大id，没有的话默认0
      def maxId = ctx._source.id ?: 0;
      def newId = maxId + 1;
      // 把当前日期格式化成DD/MM/YYYY
      def df = new SimpleDateFormat("dd/MM/yyyy");
      def currentDate = df.format(new Date());
      // 生成ARN，序号补成两位（比如01、02）
      def arn = currentDate + "/" + String.format("%02d", newId);
      // 更新文档里的字段
      ctx._source.id = newId;
      ctx._source.arn = arn;
    """,
    "lang": "painless"
  },
  "query": {
    "match_all": {}
  },
  "conflicts": "proceed"
}

小贴士：如果是第一个文档，你可以先插一个初始文档（比如{"id":0}），或者直接用上面的脚本——它已经处理了maxId为0的情况。

方法二：维护独立的序列索引（适合高并发场景） #

如果你的系统有高并发写入需求，方法一可能会因为并发更新导致ID重复。这时候最好单独搞一个序列索引，专门存每个日期对应的最大序号，每次取号都是原子性自增，稳得一批。

具体操作步骤： #

1. 创建序列索引：

PUT /sequence_index
{
  "mappings": {
    "properties": {
      "date_key": {"type": "keyword"},
      "current_max": {"type": "integer"}
    }
  }
}

2. 初始化当天的序列：

PUT /sequence_index/_doc/20240520
{
  "date_key": "20240520",
  "current_max": 0
}

3. 原子性获取自增序号：
   用update API的upsert功能——如果当天的序列不存在就自动初始化，存在就直接自增：

POST /sequence_index/_doc/20240520/_update
{
  "script": {
    "source": "ctx._source.current_max += 1",
    "lang": "painless"
  },
  "upsert": {
    "date_key": "20240520",
    "current_max": 1
  }
}

执行完这个请求，你会拿到更新后的current_max值，接着就可以用它结合日期生成ARN，再插入到目标索引里：

PUT /arn_index/_doc/new_doc
{
  "id": 1,
  "arn": "20/05/2024/01"
}

划重点：如果你的ARN需要每天重置序号（比如每天从01开始），这种方法简直完美，因为可以按日期单独维护序列。

一些额外提醒 #

• 并发冲突：方法一在高并发下可能出现ID重复，建议加上版本控制；方法二的update是原子操作，冲突概率极低。
• 序号补位：要让序号是两位（比如01、02），用String.format("%02d", newId)就搞定了，格式统一还好看。
• 时区问题：脚本里的SimpleDateFormat要注意时区，比如加上df.setTimeZone(TimeZone.getTimeZone("UTC"))，避免日期显示错误。

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