能否通过Python向VM实例启动脚本传递环境变量？如何实现？

Google Cloud Compute Engine 没有在start接口直接提供传递自定义环境变量的参数，但可以通过实例自定义元数据的方式间接实现，这也是官方推荐的参数传递方案。

实现逻辑 #

VM的启动脚本可以直接从实例元数据服务中读取自定义配置，我们只需要在调用启动接口前，先把要传递的参数写入实例的自定义元数据即可。

操作步骤 #

1. 修改云函数侧的Python代码 #

在原有启动VM的逻辑之前，增加元数据更新的逻辑，示例代码如下：

# 先获取实例当前的元数据，避免覆盖原有启动脚本、标签等配置
instance_info = service.instances().get(project=project, zone=zone, instance=instance_name).execute()
current_metadata = instance_info["metadata"]

# 自定义要传递的参数，可根据云函数的触发逻辑动态赋值
target_run_type = "x"

# 更新元数据中的自定义参数
meta_items = current_metadata.get("items", [])
key_found = False
for item in meta_items:
    if item["key"] == "RUN_TYPE":
        item["value"] = target_run_type
        key_found = True
        break
if not key_found:
    meta_items.append({"key": "RUN_TYPE", "value": target_run_type})
current_metadata["items"] = meta_items

# 提交元数据更新请求
service.instances().setMetadata(
    project=project,
    zone=zone,
    instance=instance_name,
    body=current_metadata
).execute()

# 执行原有VM启动逻辑
request = service.instances().start(project=project, zone=zone, instance=instance_name)
response = request.execute()

2. 修改VM内部的启动脚本 #

在启动脚本中通过元数据服务器读取我们写入的参数，做分支判断即可，示例代码如下：

# 读取元数据中的RUN_TYPE参数，固定请求头Metadata-Flavor必须携带
RUN_TYPE=$(curl -s -H "Metadata-Flavor: Google" http://metadata.google.internal/computeMetadata/v1/instance/attributes/RUN_TYPE)

# 分支判断逻辑
if [ "$RUN_TYPE" = "x" ]; then
    python3 /你的脚本路径/x.py
else
    python3 /你的脚本路径/y.py
fi

注意事项 #

• 元数据更新完成后再调用启动接口，即可保证启动脚本读取到最新的参数值
• 如果存在并发触发同一个VM启动的场景，建议增加分布式锁避免元数据被互相覆盖

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