go 获取命令参数

package main
import (
    "bufio"
    "fmt"
    "os"
)
func main() {
    reader := bufio.NewReader(os.Stdin)
    fmt.Println("请输入你想要输入的内容，按回车结束")
    fmt.Print("-> ")
    text, err := reader.ReadString('\n')
    if err != nil {
        panic(fmt.Errorf("发生致命错误: %w \n", err))
    }
    //TODO 下面我们就可以根据text输入的内容进行调用不同的函数了。
    fmt.Println("刚接收到的内容是：",text)
}

第一种方法：os库

os可以通过变量Args来获取命令参数，os.Args返回一个字符串数组，其中第一个参数就是执行文件本身。

package main
import (
    "fmt"
    "os"
)
func main() {
    fmt.Println("命令行的参数有", len(os.Args))
    // 遍历 os.Args 切片，就可以得到所有的命令行输入参数值
    for i, v := range os.Args {
        fmt.Printf("args[%v]=%v\n", i, v)
    }
}

结果：

> go run  ./main.go   laofan  123  456  789
命令行的参数有 5
args[0]=Temp\go-build\b001\exe\main.exe
args[1]=laofan
args[2]=123
args[3]=456
args[4]=789
>

第二种方法：flag包

如像 mysql 那样：mysql -u root -p 123456，可以指定输入内容，输入顺序可变。 flag包实现了命令行参数的解析。

使用flag来操作命令行参数，支持的格式如下：像flag.Int、flag.Bool、flag.String这样的函数格式都是一样的，第一个参数表示参数名称，第二个参数表示默认值，第三个参数表示使用说明和描述。flag.StringVar这样的函数第一个参数换成了变量地址，后面的参数和flag.String是一样的。

package main
import (
    "flag"
    "fmt"
)
func main() {
    // 定义几个变量，用于接收命令行的参数值
    var user        string
    var password    string
    var host        string
    var port        int
    // &user 就是接收命令行中输入 -u 后面的参数值，其他同理
    flag.StringVar(&user, "u", "root", "账号，默认为root")
    flag.StringVar(&password, "p", "", "密码，默认为空")
    flag.StringVar(&host, "h", "localhost", "主机名，默认为localhost")
    flag.IntVar(&port, "P", 3306, "端口号，默认为3306")
    // 解析命令行参数写入注册的flag里
    flag.Parse()
    // 输出结果
    fmt.Printf("user：%v\npassword：%v\nhost：%v\nport：%v\n",
        user, password, host, port)
}

结果：

> go run  ./main.go   -p 123 -u laofan -P 3306 -h 127.0.0.1
user：laofan
password：123
host：127.0.0.1
port：3306
>

第三种方法：flag包另一种写法

package main
import (
    "flag"
    "fmt"
)
func main () {
   //获取命令参数
  Act := flag.String("act", "", "is ok")
  date := flag.String("date", "", "id")
  // 解析命令行参数写入注册的flag里
  flag.Parse()
  fmt.Println("act :", *Act)
  fmt.Println("date :", *date)
}

结果：

> go run  ./main.go   -act laofan  -date today
act : laofan
date : today
>

参考：https://www.fujieace.com/golang/os-flag.html