如何按环境条件选用components目录下的Dapr PubSub YAML文件

1. 环境变量指定配置路径 #

在服务启动的主配置或启动命令中，用环境变量直接指定要加载的PubSub配置文件路径：

• 主配置示例（以YAML为例）：

# main-config.yaml
pubsub:
  config_path: ${PUBSUB_CONFIG:./components/rabbitmq-pubsub.yaml}

• 启动时根据环境传参：
  本地环境：PUBSUB_CONFIG=./components/rabbitmq-pubsub.yaml ./start-service
  其他环境：PUBSUB_CONFIG=./components/other-broker-pubsub.yaml ./start-service

2. 分层配置合并 #

采用「基础通用配置 + 环境专属配置」的分层模式：

• 编写base-pubsub.yaml，存放所有环境共用的配置项
• 每个环境单独维护专属配置文件，比如local-pubsub.yaml（仅RabbitMQ特有配置）、prod-pubsub.yaml（其他消息代理配置）
• 启动时用配置工具（如Kustomize、Helm，或自定义脚本）自动合并基础配置与对应环境的专属配置，生成最终生效的配置文件

3. 代码逻辑动态加载 #

在服务启动代码中，根据环境标识动态读取对应配置文件：

// 示例Go代码
package main

import (
	"os"
	"gopkg.in/yaml.v3"
)

type PubSubConfig struct {
	// 配置结构定义
}

func loadPubSubConfig() PubSubConfig {
	env := os.Getenv("ENV")
	var configPath string
	if env == "local" {
		configPath = "./components/rabbitmq-pubsub.yaml"
	} else {
		configPath = "./components/other-broker-pubsub.yaml"
	}

	data, _ := os.ReadFile(configPath)
	var config PubSubConfig
	yaml.Unmarshal(data, &config)
	return config
}

4. 模板引擎渲染配置 #

用模板引擎编写统一的配置模板，根据环境变量渲染出对应环境的配置：

• 模板示例（Jinja2语法）：

# pubsub-template.j2
{% if ENV == "local" %}
publisher:
  type: rabbitmq
  host: localhost
  port: 5672
{% else %}
publisher:
  type: other-broker
  endpoint: "https://prod-broker.example.com"
{% endif %}

• 启动前渲染模板：

# 本地环境渲染
ENV=local jinja2 pubsub-template.j2 > ./components/active-pubsub.yaml
# 其他环境渲染
ENV=prod jinja2 pubsub-template.j2 > ./components/active-pubsub.yaml

内容的提问来源于stack exchange，提问作者Santos Kumar Balla