go服务运行框架go-svc
go-svc:https://github.com/judwhite/go-svc/svc
go-svc支持linux和windows,应用只需实现Service接口即可。
官方例子
package main
import (
"log"
"os"
"path/filepath"
"github.com/judwhite/go-svc/svc"
)
// implements svc.Service
type program struct {
LogFile *os.File
svr *server
}
func main() {
prg := program{
svr: &server{},
}
defer func() {
if prg.LogFile != nil {
prg.LogFile.Close()
}
}()
// call svc.Run to start your program/service
// svc.Run will call Init, Start, and Stop
if err := svc.Run(prg); err != nil {
log.Fatal(err)
}
}
func (p program) Init(env svc.Environment) error {
log.Printf("is win service? %v\n", env.IsWindowsService())
// write to "example.log" when running as a Windows Service
if env.IsWindowsService() {
dir, err := filepath.Abs(filepath.Dir(os.Args[0]))
if err != nil {
return err
}
logPath := filepath.Join(dir, "example.log")
f, err := os.OpenFile(logPath, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666)
if err != nil {
return err
}
p.LogFile = f
log.SetOutput(f)
}
return nil
}
func (p program) Start() error {
log.Printf("Starting...\n")
go p.svr.start()
return nil
}
func (p program) Stop() error {
log.Printf("Stopping...\n")
if err := p.svr.stop(); err != nil {
return err
}
log.Printf("Stopped.\n")
return nil
}
package main
import (
"log"
"sync"
"time"
)
type server struct {
data chan int
exit chan struct{}
wg sync.WaitGroup
}
func (s *server) start() {
s.data = make(chan int)
s.exit = make(chan struct{})
s.wg.Add(2)
go s.startSender()
go s.startReceiver()
}
func (s *server) stop() error {
close(s.exit)
s.wg.Wait()
return nil
}
func (s *server) startSender() {
ticker := time.NewTicker(time.Second)
count := 1
for {
select {
case <-ticker.C:
s.data <- count
count++
case <-s.exit:
s.wg.Done()
return
}
}
}
func (s *server) startReceiver() {
for {
select {
case n := <-s.data:
log.Printf("%d\n", n)
case <-s.exit:
s.wg.Done()
return
}
}
}
posted on 2017-12-21 10:32 angry-baby 阅读(1360) 评论(0) 编辑 收藏 举报