第一个golang项目增加help指令并调整指令模式

调整指令模式
增加help指令
减少了配置文件的解析读取次数
新指令模式
打包并运行

上一篇文章

调整指令模式

version指令修改为-v和-version
replace指令修改为-r和-replace

dir参数修改为-d和 -directory

package commands

import (
"flag"
"fmt"
"log"
"os"
"strings"

复制代码

  "github.com/spf13/viper"

)

var (
help bool
version bool
replace bool

复制代码

  directory string

)

func Init() {
flag.BoolVar(&help, "h", false, "this help and exit")
flag.BoolVar(&help, "help", false, "this help and exit")
flag.BoolVar(&version, "v", false, "show version and exit")
flag.BoolVar(&version, "version", false, "show version and exit")
flag.BoolVar(&replace, "r", false, "send replace to process and replace content for scan files under directory.")
flag.BoolVar(&replace, "replace", false, "send replace to process and replace content for scan files under directory.")

复制代码

  // 注意 `directory`。默认是 -d string，有了 `directory` 之后，变为 -d directory
  flag.StringVar(&directory, "d", "", "send `directory` to process: standard, standard-fix, saas, saas-fix")
  flag.StringVar(&directory, "directory", "", "send `directory` to process: standard, standard-fix, saas, saas-fix")

  // 改变默认的 Usage
  flag.Usage = usage

  viper.SetConfigName("config") // 配置文件名（不包含扩展名）
  viper.SetConfigType("toml")   // 配置文件格式
  viper.AddConfigPath(".")      // 查找配置文件的路径
  if err := viper.ReadInConfig(); err != nil {
  	log.Fatalf("Error reading config file, %s", err)
  }

  flag.Parse()

}

func usage() {
version := viper.Get("version").(string)
cmd := os.Args[0]
fmt.Fprintf(os.Stderr, `%s version: %s
Usage: %s [-hvq] [-d,directory directory]

Options:
`, cmd, version, cmd)
flag.PrintDefaults()
}

func Run() {
Init()

复制代码

  if help {
  	flag.Usage()
  } else if version {
  	Version()
  } else if replace {
  	dir := "standard"
  	if strings.TrimSpace(directory) != "" {
  		dir = directory
  	}
  	Replace(dir)
  } else {
  	flag.Usage()
  }

}

增加help指令

-h和-help指令，打印程序已知参数列表及参数说明

func usage() {
version := viper.Get("version").(string)
cmd := os.Args[0]
fmt.Fprintf(os.Stderr, `%s version: %s
Usage: %s [-hvq] [-d,directory directory]

Options:
`, cmd, version, cmd)
flag.PrintDefaults()
}

减少了配置文件的解析读取次数

在程序执行初始化时解析一次，后续皆可使用

复制代码

  // 注意 `directory`。默认是 -d string，有了 `directory` 之后，变为 -d directory
  flag.StringVar(&directory, "d", "", "send `directory` to process: standard, standard-fix, saas, saas-fix")
  flag.StringVar(&directory, "directory", "", "send `directory` to process: standard, standard-fix, saas, saas-fix")

  // 改变默认的 Usage
  flag.Usage = usage

  viper.SetConfigName("config") // 配置文件名（不包含扩展名）
  viper.SetConfigType("toml")   // 配置文件格式
  viper.AddConfigPath(".")      // 查找配置文件的路径
  if err := viper.ReadInConfig(); err != nil {
  	log.Fatalf("Error reading config file, %s", err)
  }

  flag.Parse()

}

新指令模式

复制代码

func Run() {
	Init()

	if help {
		flag.Usage()
	} else if version {
		Version()
	} else if replace {
		dir := "standard"
		if strings.TrimSpace(directory) != "" {
			dir = directory
		}
		Replace(dir)
	} else {
		flag.Usage()
	}
}

打包并运行

复制代码

go build -v -ldflags "-linkmode external -extldflags '-static' -w" -o devtools.exe main.go
devtools.exe -h
devtools.exe -help
devtools.exe -v
devtools.exe -version
devtools.exe -r -d saas
devtools.exe -replace -directory saas

https://gitee.com/xqxyxchy/dev-tools

尽情享用吧