golang 环境变量|极客教程

golang 环境变量

在操作系统中，环境变量是指在操作系统中存储的动态值，这些值用于控制应用程序的行为。在 Golang 中，我们可以通过操作系统的环境变量来控制程序的行为，比如配置文件的路径、数据库连接信息等。本文将详细介绍如何在 Golang 中使用环境变量。

1. 设置环境变量

在 Golang 中，我们可以使用 os 包来获取和设置环境变量。下面我们来看如何设置环境变量。

package main

import (
    "fmt"
    "os"
)

func main() {
    // 设置环境变量
    os.Setenv("DB_HOST", "localhost")
    os.Setenv("DB_PORT", "3306")

    // 获取环境变量
    dbHost := os.Getenv("DB_HOST")
    dbPort := os.Getenv("DB_PORT")

    fmt.Printf("DB_HOST: %s\n", dbHost)
    fmt.Printf("DB_PORT: %s\n", dbPort)
}

在上面的示例中，我们使用 os.Setenv 函数来设置环境变量，并使用 os.Getenv 函数来获取环境变量的值。在这个示例中，我们设置了两个环境变量 DB_HOST 和 DB_PORT，并获取它们的值并打印出来。

运行结果

DB_HOST: localhost
DB_PORT: 3306

2. 读取环境变量

除了使用 os.Getenv 函数来获取环境变量的值外，我们还可以使用 os.LookupEnv 函数来获取环境变量的值，并判断环境变量是否存在。

package main

import (
    "fmt"
    "os"
)

func main() {
    // 获取环境变量的值
    dbHost, ok := os.LookupEnv("DB_HOST")
    if !ok {
        fmt.Println("DB_HOST not set")
    } else {
        fmt.Printf("DB_HOST: %s\n", dbHost)
    }

    // 判断环境变量是否存在
    _, ok = os.LookupEnv("DB_PORT")
    if !ok {
        fmt.Println("DB_PORT not set")
    }
}

在上面的示例中，我们使用 os.LookupEnv 函数来获取环境变量的值，并判断环境变量是否存在。如果环境变量存在，则打印出环境变量的值；如果环境变量不存在，则打印出提示信息。

运行结果

DB_HOST: localhost
DB_PORT not set

3. 使用 flag 包获取命令行参数

除了通过环境变量来控制程序的行为外，我们还可以使用 flag 包来获取命令行参数。下面我们来看如何使用 flag 包获取命令行参数。

package main

import (
    "flag"
    "fmt"
)

func main() {
    // 定义命令行参数
    dbHost := flag.String("db-host", "localhost", "database host")
    dbPort := flag.Int("db-port", 3306, "database port")

    // 解析命令行参数
    flag.Parse()

    fmt.Printf("DB_HOST: %s\n", *dbHost)
    fmt.Printf("DB_PORT: %d\n", *dbPort)
}

在上面的示例中，我们使用 flag 包来定义两个命令行参数 db-host 和 db-port，并使用 flag.Parse 函数来解析命令行参数。如果用户在命令行中没有提供这两个参数的值，则使用默认值。最后打印出这两个参数的值。

运行结果

假设我们在命令行中运行该程序并提供了 db-host 和 db-port 参数的值：

go run main.go -db-host example.com -db-port 5432

则运行结果为：

DB_HOST: example.com
DB_PORT: 5432

4. 使用 viper 包读取配置文件

在实际开发中，我们通常会将配置信息保存在配置文件中，使用 viper 包可以很方便地读取配置文件中的信息。下面我们来看如何使用 viper 包读取配置文件中的信息。

首先，我们需要安装 viper 包：

go get github.com/spf13/viper

接下来，我们创建一个名为 config.yaml 的配置文件，内容如下：

db:
  host: localhost
  port: 3306

然后，我们使用 viper 包来读取配置文件中的信息：

package main

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

func main() {
    // 读取配置文件
    viper.SetConfigFile("config.yaml")
    err := viper.ReadInConfig()
    if err != nil {
        panic(fmt.Errorf("Fatal error config file: %s \n", err))
    }

    // 获取配置信息
    dbHost := viper.GetString("db.host")
    dbPort := viper.GetInt("db.port")

    fmt.Printf("DB_HOST: %s\n", dbHost)
    fmt.Printf("DB_PORT: %d\n", dbPort)
}

在上面的示例中，我们使用 viper 包来读取配置文件 config.yaml 中 db.host 和 db.port 的值，并打印出这两个值。