Ansible InfluxDB桶创建模块多主机并发运行仅单节点获取ID问题求解

故障根因 #

你的模块存在冲突处理逻辑漏洞，并发请求时会触发该问题：

1. 多台主机几乎同时调用GET /api/v2/buckets接口查询桶，此时目标桶尚未创建，所有请求都未查到匹配的桶，全部进入创建逻辑
2. 第一个请求创建桶成功，返回201状态码，正确拿到桶ID
3. 剩余两个请求创建时返回409冲突错误，你的代码在冲突分支直接返回了初始值为0的bucket_id，没有重新查询已创建的桶信息，导致无法拿到正确ID

可行解决方案 #

方案1：修复模块逻辑 （优先推荐，无额外依赖） #

直接修改bucket_present函数中冲突处理的分支，冲突时重新查询一次桶ID即可，修改后的代码片段如下：

# Create new bucket
r = requests.post(url, headers=headers, json=payload, verify=ssl_verify)
if r.status_code != 201:
    if r.json()['code'] == 'conflict':
        # 冲突时重新查询已创建的桶ID
        get_r = requests.get(url, headers=headers, verify=ssl_verify)
        if get_r.status_code == 200:
            for bucket in get_r.json()['buckets']:
                if bucket['name'] == bucket_name:
                    bucket_id = bucket['id']
                    break
        meta = {"present": "Bucket: " + bucket_name + " already exists.", "bucket_id": bucket_id}
        has_changed = False
        return has_changed, has_failed, meta
    else:
        has_failed = True
        return has_changed, has_failed, r.json()
else:
    has_changed = True
    meta = {"created": r.json(), "bucket_id": r.json()['id']}
    return has_changed, has_failed, meta

修复后不管哪个请求先执行，所有请求最终都能拿到正确的桶ID。

方案2：用Ansible原生能力 （无需改模块代码） #

如果不想修改模块代码，可以直接在任务上加run_once: true参数，让该任务只在第一台主机上执行一次，执行结果会自动同步给所有主机，所有主机都能拿到桶ID，修改后的任务代码如下：

- name: Create new bucket for tenant
  influxdb_bucket:
    url: "https://{{ influxdb_host }}:{{ influxdb_port }}"
    auth_token: "{{ influxdb_admin_token }}"
    org_id: "{{ organization_id }}"
    name: "{{ client_bucket }}"
    description: "For client {{ client_bucket }} by Ansible"
    retention_policy: 365d
    state: "{{ state }}"
    selfsigned_ssl: "yes"
  register: bucket
  run_once: true

如果你的场景不允许用run_once，也可以给play设置serial: 1参数，让主机串行执行任务，第一台创建完成后，后面的主机查询就能拿到桶ID，不过该方案会拉长整个play的执行时间。

方案3：补充优化方案 #

可以在模块的查询桶逻辑前加随机休眠，错开并发请求时间，进一步降低冲突概率，代码如下：

import random
import time

# Check if the bucket already exists in influxdb
time.sleep(random.uniform(0, 3)) # 随机休眠0-3秒
r = requests.get(url, headers=headers, verify=ssl_verify)

该方案不需要单独使用，配合方案1或方案2使用效果更好，不需要引入复杂的分布式锁机制，该场景下完全没必要用这么重的方案。

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