使用fig库读取配置文件和环境变量

fig是一个轻量级的Go配置库，支持从配置文件和环境变量中读取配置，具有简洁的API和良好的默认值支持。下面我将详细介绍fig的使用方法。

安装fig

首先安装fig库：

go get github.com/kkyr/fig

基本用法

1. 从配置文件加载配置

假设我们有一个配置文件config.yaml：

server:
  host: "localhost"
  port: 8080
  timeout: 30s
database:
  url: "postgres://user:pass@localhost:5432/db"
  pool: 10
debug: true

对应的Go结构体：

type Config struct {
    Server struct {
        Host    string        `fig:"host" default:"0.0.0.0"`
        Port    int           `fig:"port" validate:"required"`
        Timeout time.Duration `fig:"timeout"`
    }
    Database struct {
        URL  string `fig:"url" validate:"required"`
        Pool int    `fig:"pool" default:"5"`
    }
    Debug bool `fig:"debug"`
}

加载配置的代码：

package main

import (
	"fmt"
	"log"
	"time"

	"github.com/kkyr/fig"
)

func main() {
	var cfg Config
	
	// 从当前目录加载config.yaml
	err := fig.Load(&cfg)
	if err != nil {
		log.Fatal(err)
	}

	fmt.Printf("%+v\n", cfg)
}

2. 自定义配置选项

fig提供了多种自定义选项：

err := fig.Load(&cfg,
    fig.File("app.json"),          // 指定文件名
    fig.Dirs(".", "config", "/etc/myapp"), // 搜索目录
    fig.UseEnv("MYAPP"),           // 环境变量前缀
    fig.Tag("json"),               // 使用json标签
)

3. 环境变量支持

fig可以自动从环境变量读取配置，环境变量名遵循以下规则：

• 对于嵌套字段：PREFIX_PARENT_CHILD
• 数组/切片：PREFIX_FIELD_0, PREFIX_FIELD_1等

type Config struct {
    Host string `fig:"host" env:"HOST"` // 显式指定环境变量名
    Port int    `fig:"port"`           // 自动转换为PORT
}

设置环境变量后加载：

export MYAPP_HOST=example.com
export MYAPP_PORT=3000

err := fig.Load(&cfg, fig.UseEnv("MYAPP"))

4. 验证配置

fig支持简单的验证：

type Config struct {
    Port int `fig:"port" validate:"required,min=1,max=65535"`
}

完整示例

下面是一个完整的示例，展示fig的各种功能：

package main

import (
	"fmt"
	"log"
	"time"

	"github.com/kkyr/fig"
)

type Config struct {
	App struct {
		Name    string `fig:"name" default:"myapp"`
		Version string `fig:"version" default:"1.0.0"`
	}
	Server struct {
		Host    string        `fig:"host" default:"0.0.0.0"`
		Port    int           `fig:"port" validate:"required"`
		Timeout time.Duration `fig:"timeout" default:"30s"`
	}
	Database struct {
		URL  string `fig:"url" validate:"required"`
		Pool int    `fig:"pool" default:"5"`
	}
	Debug bool `fig:"debug" default:"false"`
}

func main() {
	var cfg Config

	// 从config.yaml加载配置，支持环境变量
	err := fig.Load(&cfg,
		fig.File("config.yaml"),
		fig.Dirs(".", "config"),
		fig.UseEnv("APP"),
	)
	if err != nil {
		log.Fatal(err)
	}

	fmt.Printf("App Name: %s\n", cfg.App.Name)
	fmt.Printf("Server Host: %s\n", cfg.Server.Host)
	fmt.Printf("Server Port: %d\n", cfg.Server.Port)
	fmt.Printf("Database URL: %s\n", cfg.Database.URL)
	fmt.Printf("Debug Mode: %v\n", cfg.Debug)
}

支持的配置文件格式

fig支持多种配置文件格式：

• YAML (.yml, .yaml)
• JSON (.json)
• TOML (.toml)
• HCL (.hcl)

最佳实践

1. 配置结构设计：使用嵌套结构体组织配置，保持与配置文件结构一致
2. 默认值：为所有字段设置合理的默认值
3. 环境变量：生产环境优先使用环境变量
4. 验证：添加必要的验证规则
5. 多环境配置：使用不同的配置文件或环境变量区分环境

fig是一个简单但功能齐全的配置库，适合大多数Go应用程序的配置管理需求。它的轻量级设计和灵活的选项使其成为许多项目的理想选择。