介绍

本文将介绍如何在 gRPC 微服务中根据环境区分配置文件。也就是如何在【测试】，【线上】等环境中，读取不同的配置文件。

我们将会使用 rk-boot 来启动 gRPC 服务。

（福利推荐：阿里云、腾讯云、华为云服务器最新限时优惠活动，云服务器1核2G仅88元/年、2核4G仅698元/3年，点击这里立即抢购>>>）

请访问如下地址获取完整教程：

• https://rkdev.info/cn
• https://rkdocs.netlify.app/cn (备用)

安装

go get github.com/rookie-ninja/rk-boot

快速开始

我们会创建 config/beijing.yaml, config/shanghai.yaml, config/default.yaml 三个配置文件，然后根据不同的环境变量读取不同的文件。

rk-boot 使用 REALM，REGION，AZ，DOMAIN 环境变量来区分不同的环境。这也是我们推荐的云原生环境分辨法。 比如，REALM=”你的业务”，REGION=”北京”，AZ=”北京一区”，DOMAIN=”测试环境”。

rk-boot 集成了 viper 来处理配置文件。

1.创建配置文件

• config/beijing.yaml

--- region: beijing

• config/shanghai.yaml

--- region: shanghai

• config/default.yaml

--- region: default

1.创建 boot.yaml

我们使用 config 作为配置文件的入口，可以提供多个 Config 文件路径。

locale 代表 Config 的环境，我们使用 locale 来区分不同的 Config。

为什么 config.name 使用同一个名字？

我们希望使用同一套代码，但是读取不同的文件，并且希望文件的名字也不一样。 所以通过 locale 来区分不同文件。我们在后面具体介绍 locale 的逻辑。

config:   # 默认   - name: my-config     locale: "*::*::*::*"     path: config/default.yaml   # 如果环境变量 REGION=beijing，读取此文件   - name: my-config     locale: "*::beijing::*::*"     path: config/beijing.yaml   # 如果环境变量 REGION=shanghai，读取此文件   - name: my-config     locale: "*::shanghai::*::*"     path: config/shanghai.yaml grpc:   - name: greeter     port: 8080     enabled: true

2.创建 main.go

设置环境变量：REGION=”beijing”，然后读取配置文件，config/beijing.yaml 会被读取。

package main  import (     "context"     "fmt"     "github.com/rookie-ninja/rk-boot"     "os" )  // Application entrance. func main() {     // Set REGION=beijing     os.Setenv("REGION", "beijing")      // Create a new boot instance.     boot := rkboot.NewBoot()      // Load config which is config/beijing.yaml     fmt.Println(boot.GetConfigEntry("my-config").GetViper().GetString("region"))      // Bootstrap     boot.Bootstrap(context.Background())      // Wait for shutdown sig     boot.WaitForShutdownSig(context.Background()) }

3.文件夹结构

$ tree . ├── boot.yaml ├── config │   ├── beijing.yaml │   ├── default.yaml │   └── shanghai.yaml ├── go.mod ├── go.sum └── main.go

5.验证

$ go run main.go

我们会得到如下的输出：

beijing

6.未找到匹配的环境变量

如果 REGION=”not-matched”，即未找到匹配的环境变量，则会读取默认的配置文件（config/default.yaml）。因为 config/default.yaml 的 locale 属性为 ::_::_::

// Application entrance. func main() {     // Set REGION=not-matched     os.Setenv("REGION", "not-matched")          ...     // Load config which is config/beijing.yaml     fmt.Println(boot.GetConfigEntry("my-config").GetViper().GetString("region"))     ... }

default

7.环境变量未配置

如果我们没有配置 REGION 环境变量，则会读取 config/default.yaml 文件。

// Application entrance. func main() {     ...     // Load config which is config/beijing.yaml     fmt.Println(boot.GetConfigEntry("my-config").GetViper().GetString("region"))     ... }

default

概念

rk-boot 使用 REALM，REGION，AZ，DOMAIN 四个环境变量来区分配置文件。

这四个环境变量可以是任意的值。

最佳实践

举个例子，如果我们希望【云相册】业务，不同环境里使用的 MySQL 的 IP 地址不一样，则可以这么配置。

架构

假定，我们的业务为【云相册】，并且在【北京】，【上海】都有服务器，同时为了服务可用性，在【北京】和【上海】又各开了2个区。

这时候，我们可以在每一个不同的环境的机器上，配置如下的环境变量，可以通过 Ansible 等工具来批量设置。

同时，如果我们不使用配置文件管理服务，可以直接在微服务中，添加如下文件。每个文件都有不同的 MySQL IP 地址。

文件夹结构

. ├── boot.yaml ├── config │   ├── bj-1-test.yaml │   ├── bj-1-prod.yaml │   ├── bj-2-test.yaml │   ├── bj-2-prod.yaml │   ├── sh-1-test.yaml │   ├── sh-1-prod.yaml │   ├── sh-2-test.yaml │   ├── sh-2-prod.yaml │   └── default.yaml ├── go.mod ├── go.sum └── main.go

boot.yaml

接下来，我们在 boot.yaml 里添加如下 config 入口。

config:   # 默认入口   - name: my-config     locale: "*::*::*::*"     path: config/default.yaml   # 北京，一区，测试环境   - name: my-config     locale: "cloud-album::bj::bj-1::test"     path: config/bj-1-test.yaml   # 北京，一区，线上环境   - name: my-config     locale: "cloud-album::bj::bj-1::prod"     path: config/bj-1-prod.yaml   # 北京，二区，测试环境   - name: my-config     locale: "cloud-album::bj::bj-2::test"     path: config/bj-2-test.yaml   # 北京，二区，线上环境   - name: my-config     locale: "cloud-album::bj::bj-2::prod"     path: config/bj-2-prod.yaml   # 上海，一区，测试环境   - name: my-config     locale: "cloud-album::sh::sh-1::test"     path: config/sh-1-test.yaml   # 上海，一区，线上环境   - name: my-config     locale: "cloud-album::sh::sh-1::prod"     path: config/sh-1-prod.yaml   # 上海，二区，测试环境   - name: my-config     locale: "cloud-album::sh::sh-2::test"     path: config/sh-2-test.yaml   # 上海，二区，线上环境   - name: my-config     locale: "cloud-album::sh::sh-2::prod"     path: config/sh-2-prod.yaml grpc:   - name: greeter     port: 8080     enabled: true

main.go 中读取配置文件。

因为，所有的 Config 都命名为 my-config，在 main.go 中读取的时候，我们可以使用 my-config 获取 ConfigEntry。

package main  import (     "context"     "fmt"     "github.com/rookie-ninja/rk-boot"     "os" )  // Application entrance. func main() {     // Create a new boot instance.     boot := rkboot.NewBoot()      // Get viper instance based on environment variable     boot.GetConfigEntry("my-config").GetViper()      // Bootstrap     boot.Bootstrap(context.Background())      // Wait for shutdown sig     boot.WaitForShutdownSig(context.Background()) }

你还在原价购买阿里云、腾讯云、华为云、天翼云产品？那就亏大啦！现在申请成为四大品牌云厂商VIP用户，可以3折优惠价购买云服务器等云产品，并且可享四大云服务商产品终身VIP优惠价，还等什么？赶紧点击下面对应链接免费申请VIP客户吧：

1、点击这里立即申请成为腾讯云VIP客户

2、点击这里立即注册成为天翼云VIP客户

3、点击这里立即申请成为华为云VIP客户

4、点击这里立享阿里云产品终身VIP优惠价

喜欢 (0)赏

[[email protected]]

分享 (0)