部署golang项目时难免要通过命令行来设置一些参数,那么在golang中如何操作命令行参数呢?可以使用os库和flag库。
golang os库获取命令行参数
os可以通过变量Args来获取命令参数,os.Args返回一个字符串数组,其中第一个参数就是执行文件本身。
1
2
3
4
5
6
7
8
9
10
|
package
main
import
(
"fmt"
"os"
)
func
main() {
fmt.Println(os.Args)
}
|
编译执行后执行
$ ./cmd -user="root" [./cmd -user=root]
这种方式操作起来要自己封装,比较费时费劲。golang提供了flag库,可以很方便的操作命名行参数,下面介绍下flag的用法。
golang flag获取命令行参数
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
package
main
import
(
"flag"
"fmt"
)
func
main() {
ok := flag.Bool(
"ok"
, false,
"is ok"
)
id := flag.Int(
"id"
,
0
,
"id"
)
port := flag.String(
"port"
,
":8080"
,
"http listen port"
)
var
name string
flag.StringVar(&name,
"name"
,
"123"
,
"name"
)
flag.Parse()
fmt.Println(
"ok:"
, *ok)
fmt.Println(
"id:"
, *id)
fmt.Println(
"port:"
, *port)
fmt.Println(
"name:"
, name)
}
|
像flag.Int、flag.Bool、flag.String这样的函数格式都是一样的,第一个参数表示参数名称,第二个参数表示默认值,第三个参数表示使用说明和描述。flag.StringVar这样的函数第一个参数换成了变量地址,后面的参数和flag.String是一样的。
使用flag来操作命令行参数,支持的格式如下:
-id=1 --id=1 -id 1 --id 1
还是非常方便的。
执行一下:
$ go run flag.go -id=2 -name="golang" ok: false id: 2 port: :8080 name: golang
使用-h参数可以查看使用帮助:
$ go run flag.go -h -id=0: id -name="123": name -ok=false: is ok -port=":8080": http listen port
转载请注明:快乐编程 » golang获取命令行参数
在实际项目中我们修改了配置文件后,希望在不重启进程的情况下重新加载配置文件,这时候就需要通过信号传递来进行处理了。golang中对信号的处理主要使用os/signal包中的两个方法:一个是notify方法用来监听收到的信号;一个是 stop方法用来取消监听。下面给一个些示例。
监听信号
notify方法原型
func Notify(c chan<- os.Signal, sig ...os.Signal)
第一个参数表示接收信号的管道
第二个及后面的参数表示设置要监听的信号,如果不设置表示监听所有的信号。
package main import ( "fmt" "os" "os/signal" //"syscall" ) func main() { c := make(chan os.Signal) signal.Notify(c) //监听指定信号 //signal.Notify(c, syscall.SIGHUP, syscall.SIGUSR2) //阻塞直至有信号传入 s := <-c fmt.Println("get signal:", s) }
运行该程序后,搜索到该程序的进程编号,运行kill命令就可以看到信号的输出。
终端1:
$ go run main.go //此时处于阻塞状态,当终端2执行kill时输出如下信息 get signal: user defined signal 2
终端2:
$ ps -ef | grep main 501 839 543 0 11:16PM ttys000 0:00.03 go run main.go 501 842 839 0 11:16PM ttys000 0:00.00 /var/folders/s8/kbqz28xd3wl8q5rw8vsr129c0000gn/T/go-build384577908/command-line-arguments/_obj/exe/main 501 844 709 0 11:16PM ttys001 0:00.00 grep main $ kill -USR2 842
当然这样的程序在实际中并没有什么用处,改进一下就可以用来接收信号并且处理一些内容了。
package main import ( "fmt" "os" "os/signal" "syscall" "time" ) func main() { go signalListen() for { time.Sleep(10 * time.Second) } } func signalListen() { c := make(chan os.Signal) signal.Notify(c, syscall.SIGUSR2) for { s := <-c //收到信号后的处理,这里只是输出信号内容,可以做一些更有意思的事 fmt.Println("get signal:", s) } }
主要就是开启一个gorutine单独处理信号的监听。
停止监听
package main import ( "fmt" "os" "os/signal" "syscall" ) func main() { c := make(chan os.Signal) signal.Notify(c, syscall.SIGUSR2) //当调用了该方法后,下面的for循环内<-c接收到一个信号就退出了。 signal.Stop(c) for { s := <-c fmt.Println("get signal:", s) } }
小结
golang中处理信号非常简单,但是关于信号本身需要了解的还有很多,建议可以参考《Unix高级编程》中的信号篇章。
转载请注明:快乐编程 » golang信号signal的处理