溫馨提示×

Go語言配置文件讀取如何適應不同環(huán)境

小樊
81
2024-11-07 11:59:40
欄目: 編程語言

在Go語言中，可以使用viper庫來處理不同環(huán)境的配置文件。viper支持多種配置文件格式（如JSON、YAML、TOML等），并允許您根據(jù)環(huán)境變量或命令行參數(shù)選擇使用哪個配置文件。以下是一個簡單的示例，說明如何使用viper適應不同環(huán)境的配置文件。

首先，安裝viper庫：

go get github.com/spf13/viper

然后，創(chuàng)建一個名為config.go的文件，其中包含以下內(nèi)容：

package main

import (
	"fmt"
	"github.com/spf13/viper"
)

func init() {
	// 設置默認配置文件路徑和環(huán)境變量前綴
	viper.SetConfigName("config") // 配置文件名（不帶擴展名）
	viper.AddConfigPath(".")      // 配置文件路徑
	viper.SetConfigType("yaml")   // 配置文件類型
	viper.SetEnvPrefix("APP")    // 環(huán)境變量前綴

	// 讀取默認配置文件
	if err := viper.ReadInConfig(); err != nil {
		fmt.Println("Error reading config file:", err)
	}

	// 將環(huán)境變量覆蓋到配置中
	viper.AutomaticEnv()
}

func getConfigValue(key string) string {
	return viper.GetString(key)
}

func main() {
	// 獲取不同環(huán)境的配置值
	devDBHost := getConfigValue("dev.db.host")
	prodDBHost := getConfigValue("prod.db.host")

	fmt.Printf("Development DB Host: %s\n", devDBHost)
	fmt.Printf("Production DB Host: %s\n", prodDBHost)
}

在這個示例中，我們創(chuàng)建了一個名為config.yaml的默認配置文件：

# config.yaml
dev:
  db:
    host: localhost
prod:
  db:
    host: production-server

根據(jù)不同的環(huán)境，您可以創(chuàng)建相應的配置文件（如config_dev.yaml和config_prod.yaml），并在運行時選擇使用哪個配置文件。例如，設置環(huán)境變量APP_ENV=dev將導致viper讀取config_dev.yaml文件。

要運行此示例，請將代碼保存到名為main.go的文件中，并確保配置文件和代碼位于同一目錄中。然后，使用以下命令運行程序：

# 使用默認配置文件（config.yaml）
go run main.go

# 使用開發(fā)環(huán)境配置文件（config_dev.yaml）
export APP_ENV=dev && go run main.go

# 使用生產(chǎn)環(huán)境配置文件（config_prod.yaml）
export APP_ENV=prod && go run main.go

這將分別輸出開發(fā)環(huán)境和生產(chǎn)環(huán)境的數(shù)據(jù)庫主機。

0 踩