Logo
12 Oct 2017 | 2 min. (261 words)

golang进阶(三)——后台进程的启动和停止

不想再像java或者其他语言那样为了操作后台程序写各种脚本的话,这篇文章值得你一看

启动命令

和上一篇文章一样,我们先来个非后台运行的启动命令

func init() {
	startCmd := &cobra.Command{
		Use:   "start",
		Short: "Start Gonne",
		Run: func(cmd *cobra.Command, args []string) {
			startHttp()
		},
	}
	startCmd.Flags().BoolVarP(&daemon, "deamon", "d", false, "is daemon?")
	RootCmd.AddCommand(startCmd)

}

startHttp方法启动一个http的web服务

func startHttp() {
	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprintf(w, "Hello cmd!")
	})
	if err := http.ListenAndServe(":9090", nil); err != nil {
		log.Fatal("ListenAndServe: ", err)
	}
}

现在通过gonne start便可以启动一个web服务了,但是程序停留在命令行,如果ctrl+C程序也会终止了

命令行参数

如果想要后台启动,那么得让start命令知道是要后台运行的,参照docker命令行的方式就是加上-d,给一个命令添加参数的判断只需很少的代码

改造一下代码

func init() {
	var daemon bool
	startCmd := &cobra.Command{
		Use:   "start",
		Short: "Start Gonne",
		Run: func(cmd *cobra.Command, args []string) {
			if daemon {
        fmt.Println("gonne start",daemon)        
			}
			startHttp()
		},
	}
	startCmd.Flags().BoolVarP(&daemon, "deamon", "d", false, "is daemon?")
	RootCmd.AddCommand(startCmd)

}

命令行输入

gonne start -d

这样就可以接收到-d参数了,这里要说明一下,第一个参数取值,第二个参数代码--deamon,第三个参数代表-d ,第四个参数代码不加-d时候的默认值,第五参数是描述

后台运行

后台运行其实这里使用的是一个巧妙的方法,就是使用系统的command命令行启动自己的命令行输入,是不是有点绕,再看看看改造后的代码

Run: func(cmd *cobra.Command, args []string) {
  if daemon {
    command := exec.Command("gonne", "start")
    command.Start()
    fmt.Printf("gonne start, [PID] %d running...\n", command.Process.Pid)
    ioutil.WriteFile("gonne.lock", []byte(fmt.Sprintf("%d", command.Process.Pid)), 0666)
    daemon = false
    os.Exit(0)
  } else {
    fmt.Println("gonne start")
  }
  startHttp()
},

用exec的Command启动刚输入的gonne start -d,就会拦截到这条请求然后通过gonne start,但是程序就不会停留在命令行了,然后发现http服务还在,还可以访问。

还有一点就是把pid输出到gonne.lock文件,给停止的程序调用

终止后台程序

有了之前的操作后,停止就简单多了

func init() {
	RootCmd.AddCommand(stopCmd)
}

var stopCmd = &cobra.Command{
	Use:   "stop",
	Short: "Stop Gonne",
	Run: func(cmd *cobra.Command, args []string) {
		strb, _ := ioutil.ReadFile("gonne.lock")
		command := exec.Command("kill", string(strb))
		command.Start()
		println("gonne stop")
	},
}

执行 gonne stop 即可终止之前启动的http服务

help命令

好了,关于命令的操作讲完了,再看看cobra给的福利,自动生成的help命令

这个不需要你做什么操作,只需要输入gonne help,相关信息已经帮你生产好了。

appletekiMacBook-Pro:andev apple$ gonne help
Usage:
  gonne [flags]
  gonne [command]

Available Commands:
  help        Help about any command
  start       Start Gonne
  stop        Stop Gonne
  version     Print the version number of Gonne

Flags:
  -h, --help   help for gonne

Use "gonne [command] --help" for more information about a command.

当然,子命令也有

appletekiMacBook-Pro:andev apple$ gonne start -h
Start Gonne

Usage:
  gonne start [flags]

Flags:
  -d, --deamon   is daemon?
  -h, --help     help for start

自此告别各种脚本

golang cobra

makefile最全文档

或许很多Winodws的程序员都不知道这个东西,因为那些Windows的IDE都为你做了这个工作,但我觉得要作一个好的和professional的程序员,makefile还是要懂…

golang进阶(二)——cobra更实用的命令行

golang自身带的命令行包,各种case,有代码洁癖的人看着就令人头大…

© 2009-2017 lastsweetop.com 版权所有 粤ICP备15102984号
Menu
  • 启动命令
  • 命令行参数
  • 后台运行
  • 终止后台程序
  • help命令