Go教程

从零开发区块链应用(一)--golang配置文件管理工具viper

本文主要是介绍从零开发区块链应用(一)--golang配置文件管理工具viper,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

目录
  • 一、viper简介
  • 二、viper 的基本方法
    • 2.1 安装
    • 2.2 读取配置文件
    • 2.3 viper取值的函数
  • 三、 viper使用举例
    • 3.1 yaml配置文件
    • 3.2 本地配置文件读取方式
    • 3.3 将配置信息序列化为结构体对象
    • 3.4 配置读取

参考文件:https://github.com/spf13/viper

一、viper简介

viper是一个go 开发工具,主要是用于处理各种格式的配置文件,简化程序配置的读取问题,所以viper 是一个配置解决方案,它拥有丰富的特性:

  • 支持 JSON/TOML/YAML/HCL/envfile/Java properties 等多种格式的配置文件;
  • 可以设置监听配置文件的修改,修改时自动加载新的配置;
  • 从环境变量、命令行选项和io.Reader中读取配置;
  • 从远程配置系统中读取和监听修改,如 etcd/Consul;
  • 代码逻辑中显示设置键值。

二、viper 的基本方法

2.1 安装

go get github.com/spf13/viper

2.2 读取配置文件

Viper需要最少知道在哪里查找配置文件的配置。Viper支持JSONTOMLYAMLHCLenvfileJava properties格式的配置文件。Viper可以搜索多个路径,但目前单个Viper实例只支持单个配置文件。Viper不默认任何配置搜索路径,将默认决策留给应用程序。

下面是一个如何使用Viper搜索和读取配置文件的示例。不需要任何特定的路径,但是至少应该提供一个配置文件预期出现的路径。

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

在加载配置文件出错时,你可以像下面这样处理找不到配置文件的特定情况:

if err := viper.ReadInConfig(); err != nil {
    if _, ok := err.(viper.ConfigFileNotFoundError); ok {
        // 配置文件未找到错误;如果需要可以忽略
    } else {
        // 配置文件被找到,但产生了另外的错误
    }
}

// 配置文件找到并成功解析

2.3 viper取值的函数

  • Get(key string) : interface{} //自动判断类型
  • GetBool(key string) : bool //获取bool类型
  • GetFloat64(key string) : float64 //获取浮点
  • GetInt(key string) : int //获取int类型
  • GetString(key string) : string //获取string类型
  • GetIntSlice(key string) : []int //获取slice类型(slice类型为int)
  • GetStringSlice(key string) : []string //获取slice类型(slice类型为string)
  • GetStringMap(key string) : map[string]interface{} // 获取map类型(map自动判断类型)
  • GetStringMapString(key string) : map[string]string //获取map类型(map类型为string类型)
  • GetTime(key string) : time.Time // 获取时间类型
  • GetDuration(key string) : time.Duration //获取持续时间
  • IsSet(key string) : bool //判断某个键值是否存在
  • AllSettings() : map[string]interface{} // 获取所有配置

三、 viper使用举例

3.1 yaml配置文件

# 程序配置
console:
  name: "token"
  version: "v1.0"
  port: ":9100"

# 日志配置
log: "yaml/log.json"

node:
  bsc:
    url: "https://data-seed-prebsc-1-s1.binance.org:8545/"
    chainID: 97
  heco:
    url: "https://http-testnet.huobichain.com"
    chainID: 256

mysql:
  user: root
  password: "123456"
  host: 127.0.0.1
  port: 3306
  database: token

3.2 本地配置文件读取方式

  • 将上述两个配置文件和下面的 main.go 放在统一目录之下,即可实现读取配置文件
package config

import (
	"github.com/token/pkg/go-logger"
	"github.com/spf13/viper"
)

var Conf *Config

// Initialize
func Initialize() {

	// 加载配置文件
	viper.SetConfigType("yaml")
	viper.SetConfigName("config")
	viper.AddConfigPath("./yaml/")
	if err := viper.ReadInConfig(); err != nil {
		logger.Fatal("ReadInConfig error:", err)
	}

	Conf = &Config{
		LogCfg: viper.GetString("log"),
		Console: &Console{
			Name:    viper.GetString("console.name"),
			Version: viper.GetString("console.version"),
			Port:    viper.GetString("console.port"),
		},
		Mysql: &Mysql{
			User:     viper.GetString("mysql.user"),
			Password: viper.GetString("mysql.password"),
			Host:     viper.GetString("mysql.host"),
			Port:     viper.GetString("mysql.port"),
			Database: viper.GetString("mysql.database"),
		},
		Node: &Node{
			BSC: &NodeUrl{
				Url:     viper.GetString("node.bsc.url"),
				ChainID: viper.GetInt64("node.bsc.chainID"),
			},
			HECO: &NodeUrl{
				Url:     viper.GetString("node.heco.url"),
				ChainID: viper.GetInt64("node.heco.chainID"),
			},
		},
	}

	// 初始化日志配置
	logger.SetLogger("./yaml/log.json")
	logger.Info("Successful configuration load")
}

3.3 将配置信息序列化为结构体对象

package config

// yaml配置结构体
type Config struct {
	LogCfg  string   `json:"log"`
	Console *Console `json:"console"`
	Mysql   *Mysql   `json:"mysql"`
	Node    *Node    `json:"node"`
}

// yaml配置结构体
type Console struct {
	Name    string `json:"name"`
	Version string `json:"version"`
	Port    string `json:"port"`
}

type Mysql struct {
	User     string `json:"user"`
	Password string `json:"password"`
	Host     string `json:"host"`
	Port     string `json:"port"`
	Database string `json:"database"`
}

type Node struct {
	BSC  *NodeUrl `json:"bsc"`
	HECO *NodeUrl `json:"heco"`
}

type NodeUrl struct {
	Url     string `json:"url"`
	ChainID int64  `json:"chainID"`
}

3.4 配置读取

  • 读取服务端口

    config.Conf.Console.Port
    
  • 读取BSC节点url

    config.Conf.Node.BSC.Url
    
  • 读取mysql主机IP

    config.Conf.Mysql.Host
    
这篇关于从零开发区块链应用(一)--golang配置文件管理工具viper的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!