【Go】-viper库的使用

于 2024-09-25 20:53:42 发布
阅读量1k 收藏 12
点赞数 10
分类专栏： Go语言学习 文章标签： golang 开发语言 后端 学习 viper
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/knoci/article/details/142532343
版权

目录

viper简介

viper使用

通过viper.Set设置值

读取配置文件说明

读取配置文件

读取多个配置文件

读取配置项的值

读取命令行的值

io.Reader中读取值

写配置文件

WriteConfig() 和 SafeWriteConfig() 区别:

viper简介

         配置管理解析库，是由大神 Steve Francia 开发，他在google领导着 golang 的产品开发，他也是 gohugo.io 的创始人之一，命令行解析库 cobra 开发者。总之，他在golang领域是专家，很牛的一个人。

        viper是一个配置管理的解决方案，它能够从 json，toml，ini，yaml，hcl，env 等多种格式文件中，读取配置内容，它还能从一些远程配置中心读取配置文件，如consul，etcd等；它还能够监听文件的内容变化。

功能介绍

  • 读取 json，toml，ini，yaml，hcl，env 等格式的文件内容
  • 读取远程配置文件，如 consul，etcd 等和监控配置文件变化
  • 读取命令行 flag 的值
  • 从 buffer 中读取值

        配置文件又可以分为不同的环境，比如dev，test，prod等。读取配置文件的优先顺序，从高到低，如下：

  • 显式设置的Set函数
  • 命令行参数
  • 环境变量
  • 配置文件
  • 远程k-v 存储系统，如consul，etcd等
  • 默认值

Viper 配置key是不区分大小写的。

其实，上面的每一种文件格式，都有一些比较有名的解析库，如：

        但是为什么要用viper，因为它是一个综合文件解析库，包含了上面所有的文件格式解析，是一个集合体，少了配置多个库的烦恼。

viper使用

安装viper命令：

go get github.com/spf13/viper

文档: viper/README.md at master · spf13/viper · GitHub

通过viper.Set设置值

        如果某个键通过viper.Set设置了值，那么这个值读取的优先级最高

viper.Set("mysql.info", "this is mysql info")

        同时还可以设置默认值

viper.SetDefault("ContentDir", "content") 
viper.SetDefault("LayoutDir", "layouts") 
viper.SetDefault("Taxonomies", map[string]string{"tag": "tags", "category": "categories"})

读取配置文件说明

        读取配置文件要求：最少要知道从哪个位置查找配置文件。用户一定要设置这个路径。viper可以从多个路径搜索配置文件，单个viper实例只支持单个配置文件。viper本身没有设置默认的搜索路径，需要用户自己设置默认路径。

viper.SetConfigName("config") // 配置文件的文件名，没有扩展名，如 .yaml, .toml 这样的扩展名 viper.SetConfigType("yaml") // 设置扩展名。在这里设置文件的扩展名。另外，如果配置文件的名称没有扩展名，则需要配置这个选项 
viper.AddConfigPath("/etc/appname/") // 查找配置文件所在路径 viper.AddConfigPath("$HOME/.appname") // 多次调用AddConfigPath，可以添加多个搜索路径 viper.AddConfigPath(".") // 还可以在工作目录中搜索配置文件 
err := viper.ReadInConfig() // 搜索并读取配置文件 
if err != nil { // 处理错误 
    panic(fmt.Errorf("Fatal error config file: %s \n", err)) 
}

说明：
        这里执行viper.ReadInConfig()之后，viper才能确定到底用哪个文件，viper按照上面的AddConfigPath() 进行搜索，找到第一个名为 config.ext (这里的ext代表扩展名： 如 json,toml,yaml,yml,ini,prop 等扩展名) 的文件后即停止搜索。

如果有多个名称为config的配置文件，viper怎么搜索呢？它会按照如下顺序搜索

  • config.json
  • config.toml
  • config.yaml
  • config.yml
  • config.properties (这种一般是java中的配置文件名)
  • config.props (这种一般是java中的配置文件名)

        你还可以处理一些特殊情况：

if err := viper.ReadInConfig(); err != nil { 
    if _, ok := err.(viper.ConfigFileNotFoundError); ok { 
    // 配置文件没有找到; 如果需要可以忽略 
    } else { 
        // 查找到了配置文件但是产生了其它的错误 
    } 
} // 查找到配置文件并解析成功

        注意[自1.6起]：  你也可以有不带扩展名的文件，并以编程方式指定其格式。对于位于用户$HOME目录中的配置文件没有任何扩展名，如.bashrc。

读取配置文件

        config.toml 配置文件：

# this is a toml 
title = "toml exaples" 
redis = "127.0.0.1:3300" # redis 
[mysql] 
    host = "192.168.1.1" 
    ports = 3306 
    username = "root" 
    password = "root123456"
package main 
import( "fmt" "github.com/spf13/viper" ) 
// 读取配置文件
config type Config struct { 
    Redis string 
    MySQL MySQLConfig 
} 

type MySQLConfig struct { 
    Port int 
    Host string 
    Username string 
    Password string 
} 

func main() { 
    // 把配置文件读取到结构体上 
    var config Config 
    viper.SetConfigName("config") 
    viper.AddConfigPath(".") 
    err := viper.ReadInConfig() 
    if err != nil { 
        fmt.Println(err) 
        return 
    } 
    viper.Unmarshal(&config) //将配置文件绑定到config上 
    fmt.Println("config: ", config, "redis: ", config.Redis) 
}

读取多个配置文件

        在例子1基础上多增加一个json的配置文件，config3.json 配置文件：

{ 
    "redis": "127.0.0.1:33000", 
    "mysql": { "port": 3306, "host": "127.0.0.1", "username": "root", "password": "123456"                     
    } 
}
package main 
import ( "fmt" "github.com/spf13/viper" ) 
type Config struct { 
    Redis string 
    MySQL MySQLConfig 
} 

type MySQLConfig struct { 
    Port int 
    Host string 
    Username string 
    Password string 
} 

func main() { 
    // 读取 toml 配置文件 
    var config1 Config 
    vtoml := viper.New() 
    vtoml.SetConfigName("config") 
    vtoml.SetConfigType("toml") 
    vtoml.AddConfigPath(".") 
    if err := vtoml.ReadInConfig(); err != nil { 
        fmt.Println(err) 
        return 
    } 
    vtoml.Unmarshal(&config1) 
    fmt.Println("read config.toml") 
    fmt.Println("config: ", config1, "redis: ", config1.Redis) 

    // 读取 json 配置文件 
    var config2 Config 
    vjson := viper.New() 
    vjson.SetConfigName("config3") 
    vjson.SetConfigType("json") 
    vjson.AddConfigPath(".") 
    if err := vjson.ReadInConfig(); err != nil { 
        fmt.Println(err) 
        return 
    } 
    vjson.Unmarshal(&config2) 
    fmt.Println("read config3.json") 
    fmt.Println("config: ", config1, "redis: ", config1.Redis) 
}

运行：

$ go run viper_multi.go

read config.toml
config:  {127.0.0.1:33000 {0 192.168.1.1 root 123456}} redis:  127.0.0.1:33000
read config3.json
config:  {127.0.0.1:33000 {0 192.168.1.1 root 123456}} redis:  127.0.0.1:33000

读取配置项的值

        新建文件夹 item， 在里面创建文件 config.json，内容如下：

{ 
    "redis": "127.0.0.1:33000", 
    "mysql": { "port": 3306, "host": "127.0.0.1", "username": "root", "password": "123456", "ports": [ 5799, 6029 ], "metric": { "host": "127.0.0.1", "port": 2112 } 
    } 
}
package main

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

func main() {
	viper.SetConfigName("config")   // 配置文件名称（不包括扩展名）
	viper.SetConfigType("json")    // 配置文件类型
	viper.AddConfigPath(".")       // 配置文件搜索路径

	// 根据配置加载文件
	err := viper.ReadInConfig()
	if err != nil {
		fmt.Println(err)
		return
	}

	// 读取配置项
	host := viper.Get("mysql.host")
	username := viper.GetString("mysql.username")
	port := viper.GetInt("mysql.port")
	portsSlice := viper.GetIntSlice("mysql.ports")
	metricPort := viper.GetInt("mysql.metric.port")
	redis := viper.Get("redis")
	mysqlMap := viper.GetStringMapString("mysql")

	// 检查配置项是否设置
	if viper.IsSet("mysql.host") {
		fmt.Println("[IsSet()] mysql.host is set")
	} else {
		fmt.Println("[IsSet()] mysql.host is not set")
	}

	// 打印配置信息
	fmt.Println("mysql - host:", host, ", username:", username, ", port:", port)
	fmt.Println("mysql ports:", portsSlice)
	fmt.Println("metric port:", metricPort)
	fmt.Println("redis -", redis)
	fmt.Println("mysqlmap -", mysqlMap, ", username:", mysqlMap["username"])
}

运行：

$ go run viper_get_item.go

[IsSet()]mysql.host is set
mysql - host:  127.0.0.1 , username:  root , port:  3306
mysql ports : [5799 6029]
metric port:  2112
redis -  127.0.0.1:33000
mysqlmap -  map[host:127.0.0.1 metric: password:123456 port:3306 ports: username:root] , username:  root

读取命令行的值

viper获取值的方法：

  • Get(key string) : interface{}
  • GetBool(key string) : bool
  • GetFloat64(key string) : float64
  • GetInt(key string) : int
  • GetIntSlice(key string) : []int
  • GetString(key string) : string
  • GetStringMap(key string) : map[string]interface{}
  • GetStringMapString(key string) : map[string]string
  • GetStringSlice(key string) : []string
  • GetTime(key string) : time.Time
  • GetDuration(key string) : time.Duration
  • IsSet(key string) : bool
  • AllSettings() : map[string]interface{}

        新建文件夹 cmd，然后cmd文件夹里新建config.json文件：

{ 
"redis":{ "port": 3301, "host": "127.0.0.1" }, 
"mysql": { "port": 3306, "host": "127.0.0.1", "username": "root", "password": "123456" } 
}
package main

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

func main() {
	// 使用pflag定义命令行参数
	pflag.Int("redis.port", 3302, "redis port")
	viper.BindPFlags(pflag.CommandLine) // 将pflag绑定到viper
	pflag.Parse()                     // 解析命令行参数

	// 使用viper加载配置文件
	viper.SetConfigName("config")   // 配置文件名称（不包括扩展名）
	viper.SetConfigType("json")    // 配置文件类型
	viper.AddConfigPath(".")       // 配置文件搜索路径

	// 根据配置加载文件
	err := viper.ReadInConfig()
	if err != nil {
		fmt.Println(err)
		return
	}

	// 读取配置项
	host := viper.Get("mysql.host")
	username := viper.GetString("mysql.username")
	port := viper.GetInt("mysql.port")
	redisHost := viper.GetString("redis.host")
	redisPort := viper.GetInt("redis.port")

	// 打印配置信息
	fmt.Println("mysql - host:", host, ", username:", username, ", port:", port)
	fmt.Println("redis - host:", redisHost, ", port:", redisPort)
}

1.不加命令行参数运行：

$ go run viper_pflag.go

mysql - host:  127.0.0.1 , username:  root , port:  3306
redis - host:  127.0.0.1 , port:  3301

        说明：redis.port 的值是 3301，是 config.json 配置文件里的值。

2.加命令行参数运行

$ go run viper_pflag.go --redis.port 6666

mysql - host:  127.0.0.1 , username:  root , port:  3306
redis - host:  127.0.0.1 , port:  6666

        说明：加了命令行参数 --redis.port 6666，这时候redis.port输出的值为 6666，读取的是cmd命令行的值

io.Reader中读取值

package main

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

func main() {
	viper.SetConfigType("yaml") // 设置配置文件类型为yaml

	// 定义YAML配置内容
	var yaml = []byte(`Hacker: true
name: steve
hobbies:
- skateboarding
- snowboarding
- go
clothing:
  jacket: leather
  trousers: denim
age: 35
eyes: brown
beard: true`)

	// 从字节切片中读取配置
	err := viper.ReadConfig(bytes.NewBuffer(yaml))
	if err != nil {
		fmt.Println(err)
		return
	}

	// 读取配置项
	hacker := viper.GetBool("Hacker")
	hobbies := viper.GetStringSlice("hobbies")
	jacket := viper.Get("clothing.jacket")
	age := viper.GetInt("age")

	// 打印配置信息
	fmt.Println("Hacker:", hacker, ", hobbies:", hobbies, ", jacket:", jacket, ", age:", age)
}

写配置文件

package main

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

func main() {
	viper.SetConfigName("config") // 设置配置文件名称（不包括扩展名）
	viper.SetConfigType("yaml")  // 设置配置文件类型为yaml
	viper.AddConfigPath(".")     // 添加配置文件搜索路径

	// 直接设置配置项
	viper.Set("yaml", "this is an example of yaml")
	viper.Set("redis.port", 4405)
	viper.Set("redis.host", "127.0.0.1")
	viper.Set("mysql.port", 3306)
	viper.Set("mysql.host", "192.168.1.0")
	viper.Set("mysql.username", "root123")
	viper.Set("mysql.password", "root123")

	// 将配置写入文件
	if err := viper.WriteConfig(); err != nil {
		fmt.Println(err)
	}
}

运行：

$ go run viper_write_config.go

        没有任何输出表示生成配置文件成功

WriteConfig() 和 SafeWriteConfig() 区别:

        如果待生成的文件已经存在，那么SafeWriteConfig()就会报错，Config File "config.yaml" Already Exists， 而WriteConfig()则会直接覆盖同名文件。

knoci
关注
专栏目录
[Golang] Viper原理以及详细使用案例
Jing_Hua
07-26 2806
介绍：用于处理配置文件中解析和读取配置文件优点：支持多种配置格式，json，yaml 等等作用：配置文件解析和读取 默认值支持 环境变量支持 命令行标志支持 配置监听和热加载。
[Golang]每日一库-viper
Jing_Hua
04-18 494
😚一个不甘平凡的普通人，致力于为Golang社区和算法学习做出贡献，期待您的关注和认可，陪您一起学习打卡！！！😘😘😘。
viper 读取 环境变量_Go语言配置管理神器——Viper中文教程
weixin_39619481的博客
12-14 3881
Viper是适用于Go应用程序的完整配置解决方案。它被设计用于在应用程序中工作，并且可以处理所有类型的配置需求和格式。ViperViper是适用于Go应用程序的完整配置解决方案。它被设计用于在应用程序中工作，并且可以处理所有类型的配置需求和格式。鉴于viper库本身的README已经写得十分详细，这里就将其翻译成中文，并在最后附上两个项目中使用viper的示例代码以供参考。安装go get git...
golang使用viper库读取配置文件
u012559967的专栏
09-08 657
golang中用viper读取配置文件的方法
viper4android io错误,Go viper包处理JSON笔记
weixin_34239718的博客
05-26 278
第一部分 前景提要一、背景用Go搭一个小项目, 按传统的套路把开发环境和生产环境配置信息写在配置文件config.json中, 初始化加载配置，解析出数据。看着是解析一个JSON的小事，过程中却发生了3次演进：1、定义一个配置文件对应的结构体。2、递归加载配置的全部节点。3、使用viper思路1: 是最初的想法，考虑到动态加配置的话，还需要改结构体，这就有点low了，故而并没有实施，仅停留在思考层...
go语言中的配置管理神器 --viper 一
个人博客，记录了本人学习的各种有趣的技术。
09-29 679
Viper 是 Go 应用程序的完整配置解决方案，包括 12-Factor 应用程序。它旨在在应用程序中工作，并可以处理所有类型的配置需求和格式。Viper 可以被认为是所有应用程序配置需求的注册表。Viper 使用以下优先级顺序。
golang常用库之配置文件解析库-viper使用详解
01-19
golang常用库：配置文件解析库-viper使用 golang常用库：操作数据库的orm框架-gorm基本使用 golang常用库：字段参数验证库-validator使用 一、viper简介 viper 配置管理解析库，是由大神 Steve Francia 开发，他在...
Go-viper-一个Go应用程序的完整配置解决方案
08-13
Go-viper是一个专门为Go语言应用程序设计的全面配置管理库，由知名Go开发者spf13创建。这个库的出现解决了Go应用在处理配置文件时的复杂性和多样性问题，支持从命令行参数、环境变量、配置文件（JSON, TOML, YAML, ...
viper-template:Golang Viper库的扩展，以在viper配置中使用Go模板
04-08
Golang Viper库的扩展，以在viper配置中使用Go模板 用法 package main import ( "fmt" "text/template" "github.com/spf13/viper" vipertemplate "github.com/sv-tools/viper-template" ) func main () { v := ...
Go 每日一库-viper
Anubis
02-08 642
安装go get github.com/spf13/viper功能viper 是用来管理 go 程序的项目配置，是一个完整的解决方案它支持：设置默认值从 JSON、TOML、YAML、HCL、envfile 和 Java 属性配置文件中读取实时观看和重新阅读配置文件(可选)从环境变量中读取从远程配置系统(etcd 或 Consul)读取，并观察变化从命令行标志读取从缓冲区读取设置显式值可以将 Vi...
Viper 大中型项目配置解析神器
最新发布
qq_46281780的博客
04-21 402
之后可以直接操作别名，修改会作用到原本的配置项上。
viper读取配置文件
chencong3139的博客
09-27 1684
Viper是适用于Go应用程序的完整配置解决方案。它被设计用于在应用程序中工作，并且可以处理所有类型的配置需求和格式。设置默认值从JSON、TOML、YAML、HCL、envfile和格式的配置文件读取配置信息实时监控和重新读取配置文件（可选）从环境变量中读取从远程配置系统（etcd或Consul）读取并监控配置变化从命令行参数读取配置从buffer读取配置显式配置值Viper能够帮助我们执行以下操作：查找、加载和反序列化JSON、TOML、YAML、HCL、INI、
Go项目配置管理神器之viper使用详解
程序猿CKeen的专栏
08-08 7728
viper是go一个强大的流行的配置解决方案的库。viper是spf13的另外一个重量级库。有大量项目都使用该库，比如hugo, docker等。它基本上可以处理所有类型的配置需求和格式, viper支持功能。
在 Android 上使用 VIPER 架构
站在巨人的肩膀上
04-30 2638
英文原文：Using the VIPER architecture on Android  我先是一个Android开发者，后来也做了iOS开发，接触过几种不同的架构 - 有好有坏。 在Android中我一直觉得MVP架构用着不错，直到在一个iOS的项目中遇到了VIPER架构，这个架构用了8个月。当我回到Android时，我决定采用这种设计，虽然有人建议说在Android上使用iOS的
VIPER模式介绍
有志者事竟成-程序员也可以创造优雅的艺术
02-10 3637
一、概述 VIPER模式的理念不属于MV(X)系类，其理念来自于建筑设计。 建筑领域流行这样一句话，“我们虽然在营造建筑，但建筑也会重新塑造我们”。正如所有开发者最终领悟到的，这句话同样适用于构建软件。 编写代码中至关重要的是，需要使每一部分容易被识别，赋有一个特定而明显的目的，并与其他部分在逻辑关系中完美契合。这就是我们所说的软件架构。好的架构不仅让一个产品成功投入使用，还可以让产品具有可维护性
从零开发区块链应用(一)--golang配置文件管理工具viper
J_Lee
01-18 8348
[toc] 参考文件：https://github.com/spf13/viper 一、viper简介 viper是一个go 开发工具，主要是用于处理各种格式的配置文件，简化程序配置的读取问题，所以viper 是一个配置解决方案，它拥有丰富的特性： 支持 JSON/TOML/YAML/HCL/envfile/Java properties 等多种格式的配置文件； 可以设置监听配置文件的修改，修改时自动加载新的配置； 从环境变量、命令行选项和io.Reader中读取配置； 从远程配置系统中读取和监听修改，如
golang模块viper读取配置文件
weixin_30823683的博客
07-14 1319
一、介绍 Viper是一个方便Go语言应用程序处理配置信息的库。它可以处理多种格式的配置。它支持的特性： 设置默认值 从JSON、TOML、YAML、HCL和Java properties文件中读取配置数据 可以监视配置文件的变动、重新读取配置文件 从环境变量中读取配置数据 从远端配置系统中读取数据，并监视它们（比如etcd、Consul） 从命令参数中读物配置 从buffer...
github.com/spf13/viper go viper包介绍
热门推荐
好记性不如烂笔头
07-26 2万+
  什么是Viper？ Viper是Go应用程序的完整配置解决方案，包括12-Factor应用程序。它旨在在应用程序中工作，并可以处理所有类型的配置需求和格式。它支持： 设置默认值 从JSON，TOML，YAML，HCL和Java属性配置文件中读取 实时观看和重新读取配置文件（可选） 从环境变量中读取 从远程配置系统（etcd或Consul）读取，并观察变化 从命令行标志读取 从...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值