Go获取命令行参数及信号量处理

kenkao · · 7072 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

部署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的处理

 

本文来自:CSDN博客

感谢作者:kenkao

查看原文:Go获取命令行参数及信号量处理

入群交流(该群和以上内容无关):Go中文网 QQ交流群:798786647 或 加微信入微信群:274768166 备注:入群; 公众号:Go语言中文网

7072 次点击  
加入收藏 微博
1 回复  |  直到 2000-01-01 00:00:00
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传