go语言开发中常用库总结

原创已于 2022-08-13 22:19:27 修改 · 322 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#golang #开发语言 #后端

于 2022-08-13 21:20:25 首次发布

本文介绍了使用Go语言进行命令行参数解析、配置文件解析及命令行框架搭建的方法。通过Pflag库实现命令行参数解析；Viper库完成YAML配置文件读取；Cobra库构建命令行应用程序。

一、命令行参数解析

Pflag使用示例:

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

pflag.BoolVarP(&version, "version", "v", true, "show version infomation")

二、配置解析

Viper使用示例：

package main

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

func main() {
    viper.SetConfigFile("dbconfig")  
    viper.SetConfigType("yaml") 
    if err := viper.ReadInConfig(); err != nil { 
       panic(fmt.Errorf("Fatal error config file: %s \n", err))
  }
}

三、命令行框架

Cobra


package cmd

import (
  "fmt"

  "github.com/spf13/cobra"
)

func init() {
  rootCmd.AddCommand(versionCmd)
}

var versionCmd = &cobra.Command{
  Use:   "version",
  Short: "Print the version",
  Long:  `Print the version`,
  Run: func(cmd *cobra.Command, args []string) {
    fmt.Println("v1.0.0")
  },
}