Run a program as a service (daemon)-GO

fpkoko發表於2018-05-03

所用到的庫

github.com/kardianos/service
需要golang.org/x/sys做依賴庫.請注意翻牆
或者從https://github.com/golang/sys先下載下來
複製程式碼

程式碼片段

package main
import (
	"github.com/kardianos/service"
	"os"
	"sync"
)

type program struct {
	log service.Logger
	cfg *service.Config
}

func (p *program) Start(s service.Service) error {
	go p.run()
	return nil
}

func (p *program) run() {
    //這裡寫執行時的程式碼
    wg.Done()
}

func (p *program) Stop(s service.Service) error {
	return nil
}

var wg sync.WaitGroup

func main() {
	wg.Add(1)
	svcConfig := &service.Config{
		Name:        "名稱", //在windows下可用net start 
		DisplayName: "顯示名稱", //顯示的名稱
		Description: "這是一個service程式", //詳情
	}
	prg := &program{}
	s, err := service.New(prg, svcConfig)
	if err != nil {
		fmt.Println(err.Error())
	}
	if len(os.Args) > 1 {
		if os.Args[1] == "install" {
			x := s.Install()
			if x != nil {
				fmt.Println("error:", x.Error())
				return
			}
			fmt.Println("服務安裝成功")
			return
		} else if os.Args[1] == "uninstall" {
			x := s.Uninstall()
			if x != nil {
				fmt.Println("error:", x.Error())
				return
			}
			fmt.Println("服務解除安裝成功")
			return
		}
	}
	err = s.Run()
	if err != nil {
		fmt.Println(err.Error())
	}
	wg.Wait()
}
複製程式碼

在run函式中寫入要執行的程式即可 引數install即為安裝,uninstall即為解除安裝.什麼引數都沒有,即為執行.

windows下請使用管理員模式,其他系統請sudo

相關文章