Go-INI - 超赞的Go语言INI文件操作库

INI 文件(Initialization File)是十分常用的配置文件格式,其由节(section)、键(key)和值(value)组成,编写方便,表达性强,并能实现基本的配置分组功能,被各类软件框架和项目广泛使用。然而,日渐受到关注的 Go 语言,其官方库并没有对 INI 文件操作的库,而进行 INI 文件的解析也并不是几行就能完成的简单工作。Go-INI,就为 Go 语言添加了对 INI 文件进行读取、解析和写入等操作,使得 Go 项目也能充分利用 INI 文件的便利性。

简介

Go-INI,是 go-ini 在 Github 上开源的 Go 语言 INI 文件操作库,项目位于 https://github.com/go-ini/ini,目前版本为 1.61.0。Go-INIT 功能强大,支持丰富的 INI 语法,功能特性包括:

支持覆盖加载多个数据源(file, []byte, io.Reader and - io.ReadCloser)

  • 支持递归读取键值
  • 支持读取父子分区
  • 支持读取自增键名
  • 支持读取多行的键值
  • 支持大量辅助方法
  • 支持在读取时直接转换为 Go 语言类型
  • 支持读取和 写入 分区和键的注释
  • 轻松操作分区、键值和注释
  • 在保存文件时分区和键值会保持原有的顺序
    ……

安装

Go-INI 要求 Go 1.6+,使用 go get 安装:

$ go get gopkg.in/ini.v1

示例
我们来看一个简单的使用例子。首先,创建一个 INI 文件 my.ini:

# possible values : production, development
app_mode = development

[paths]
# Path to where grafana can store temp files, sessions, and the sqlite3 db (if that is used)
data = /home/git/grafana

[server]
# Protocol (http or https)
protocol = http
​
# The http port  to use
http_port = 9999
​
# Redirect to correct domain if host header does not match domain
# Prevents DNS rebinding attacks
enforce_domain = true

这是一个十分常见的 INI 配置文件。然后,我们使用 Go-INIT 来进行操作:

package main
​
import (
    "fmt"
    "os""gopkg.in/ini.v1"
)
​
func main() {
    cfg, err := ini.Load("my.ini")
    if err != nil {
        fmt.Printf("Fail to read file: %v", err)
        os.Exit(1)
    }
​
    // 典型读取操作,默认分区可以使用空字符串表示
    fmt.Println("App Mode:", cfg.Section("").Key("app_mode").String())
    fmt.Println("Data Path:", cfg.Section("paths").Key("data").String())
​
    // 我们可以做一些候选值限制的操作
    fmt.Println("Server Protocol:",
        cfg.Section("server").Key("protocol").In("http", []string{"http", "https"}))
    // 如果读取的值不在候选列表内,则会回退使用提供的默认值
    fmt.Println("Email Protocol:",
        cfg.Section("server").Key("protocol").In("smtp", []string{"imap", "smtp"}))
​
    // 试一试自动类型转换
    fmt.Printf("Port Number: (%[1]T) %[1]d\n", cfg.Section("server").Key("http_port").MustInt(9999))
    fmt.Printf("Enforce Domain: (%[1]T) %[1]v\n", cfg.Section("server").Key("enforce_domain").MustBool(false))
​
    // 修改某个值然后进行保存
    cfg.Section("").Key("app_mode").SetValue("production")
    cfg.SaveTo("my.ini.local")
}

我们来看一看这段代码做了什么。首先,进行了 Go-INI 的引入:

import "gopkg.in/ini.v1"

然后,使用 Load 接口,进行 INI 文件的打开、加载和解析。Load 的函数定义如下:

func Load(source interface{}, others ...interface{}) (*File, error) {

Load 的 source 参数使用 interface{} 类型,允许多种类型的数据源,包括字符串类型的文件名、[]byte 类型的原数据等。Load 返回一个 *File 类型的文件指针,和一个错误信息。 然后,可以使用 Go-INI 提供的 Section 接口,获取 INI 文件的节:

func (f *File) Section(name string) *Section

再利用 Section 的 Key 接口,实现对于参数值的获取:

func (s *Section) Key(name string) *Key

返回一个 Key 结构体,然后再利用 Key 的 String 接口,获取对应的字符串类型的值:

func (k *Key) String() string

有时候,我们需要对于配置值的值进行验证,当值不在预选列表里时,需要返回一个默认值,而不是一个无效的值。这时,可以使用 Key 的 In 接口实现:

func (k *Key) In(defaultVal string, candidates []string) string

Go-INI 还可以进行值的类型转换:

func (k *Key) MustInt(defaultVal ...int) int

最后,使用 Key 的 SetValue 进行值得设置,然后使用 SaveTo 重新写入文件:

func (k *Key) SetValue(v string)
func (f *File) SaveTo(filename string) error

运行上述代码,我们可以得到以下输出:

$ go run main.go
App Mode: development
Data Path: /home/git/grafana
Server Protocol: http
Email Protocol: smtp
Port Number: (int) 9999
Enforce Domain: (bool) true

此外,Go-INI 还提供了 INI 文件和结构体之间的映射。当配置文件是固定的时候,在代码中定义一个结构体,可以使得对配置的操作更为方便。使用前缀为 ini 的结构体标签,就可以实现 INI 文件和结构体的双向映射:

type Embeded struct {
    Dates  []time.Time `delim:"|" comment:"Time data"`
    Places []string    `ini:"places,omitempty"`
    None   []int       `ini:",omitempty"`
}
​
type Author struct {
    Name      string `ini:"NAME"`
    Male      bool
    Age       int `comment:"Author's age"`
    GPA       float64
    NeverMind string `ini:"-"`
    *Embeded `comment:"Embeded section"`
}
​
func main() {
    a := &Author{"Unknwon", true, 21, 2.8, "",
        &Embeded{
            []time.Time{time.Now(), time.Now()},
            []string{"HangZhou", "Boston"},
            []int{},
        }}
    cfg := ini.Empty()
    err = ini.ReflectFrom(cfg, a)
    // ...
}

总结

Go-INI 作为 Go 语言的 INI 文件操作库,提供了丰富的 INI 文件操作,使得 Go 项目也能应用 INI 配置文件,为大量的已有项目提供了极大的便利,是使用 INI 文件的 Go 项目必备的依赖库。

转自https://m.toutiao.com/is/eTxD9oK/