Golang如何从JSON或命令行获取配置文件

Golang如何从JSON或命令行获取配置文件 我编写了这段代码来从命令行界面获取参数，但我想修改它，使其也能够从JSON文件中获取这些参数。

type Options struct {
	one *string
	two  *string
	three *string
}

func ParseOptions() (Options) {
	opt := Options {
		one : flag.String(...),
		two : flag.String(....),
		three : flag.String(...),
	}

	flag.Parse()

	return opt
}

我该如何实现呢？是否应该先询问用户，例如输入1表示从命令行获取输入，输入2表示从JSON文件获取？那么，我应该如何解码/解析JSON呢？ 谢谢。


更多关于Golang如何从JSON或命令行获取配置文件的实战教程也可以访问 https://www.itying.com/category-94-b0.html

4 回复

我的意思是，当我运行 go run main.go -configuration.json 时，它会将该文件中的所有配置存储在变量 one、two、three 中。

更多关于Golang如何从JSON或命令行获取配置文件的实战系列教程也可以访问 https://www.itying.com/category-94-b0.html


wecandoit:

-configuration.json

这个 JSON 文件的名称是哪一部分？文件是叫 configuration.json 吗？如果是，请按照我上面的建议操作。

你所说的“从JSON”是什么意思？你是在解析一个JSON文件吗？那么你应该将这个文件的路径作为 flag.String 参数传入，读取文件并解析它。

你可以通过组合使用 flag 包和 encoding/json 包来实现从命令行或 JSON 文件获取配置。以下是一个示例实现，它首先检查命令行是否指定了配置文件路径，如果指定了则从 JSON 文件读取配置，否则使用命令行参数。

package main

import (
    "encoding/json"
    "flag"
    "fmt"
    "io/ioutil"
    "os"
)

type Options struct {
    One   *string `json:"one,omitempty"`
    Two   *string `json:"two,omitempty"`
    Three *string `string:"three,omitempty"`
}

func ParseOptions() Options {
    // 定义命令行参数
    configFile := flag.String("config", "", "Path to JSON config file")
    one := flag.String("one", "", "Option one")
    two := flag.String("two", "", "Option two")
    three := flag.String("three", "", "Option three")

    flag.Parse()

    // 如果指定了配置文件，则从 JSON 文件读取
    if *configFile != "" {
        return loadOptionsFromJSON(*configFile)
    }

    // 否则使用命令行参数
    return Options{
        One:   one,
        Two:   two,
        Three: three,
    }
}

func loadOptionsFromJSON(filename string) Options {
    file, err := ioutil.ReadFile(filename)
    if err != nil {
        fmt.Printf("Error reading config file: %v\n", err)
        os.Exit(1)
    }

    var opts Options
    err = json.Unmarshal(file, &opts)
    if err != nil {
        fmt.Printf("Error parsing JSON: %v\n", err)
        os.Exit(1)
    }

    return opts
}

func main() {
    opts := ParseOptions()
    fmt.Printf("One: %v\n", *opts.One)
    fmt.Printf("Two: %v\n", *opts.Two)
    fmt.Printf("Three: %v\n", *opts.Three)
}

JSON 配置文件示例（例如 config.json）：

{
    "one": "value1",
    "two": "value2",
    "three": "value3"
}

使用方式：

  1. 从命令行获取参数：
    go run main.go -one=value1 -two=value2 -three=value3
  2. 从 JSON 文件获取参数：
    go run main.go -config=config.json

这个实现通过 -config 标志来指定 JSON 配置文件路径。如果提供了该标志，程序会从 JSON 文件加载配置；否则，它会解析其他命令行参数。结构体字段使用 JSON 标签来映射 JSON 键。

回到顶部