zoukankan      html  css  js  c++  java
  • 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
    ", 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...
    ")
    	go p.svr.start()
    	return nil
    }
    
    func (p program) Stop() error {
    	log.Printf("Stopping...
    ")
    	if err := p.svr.stop(); err != nil {
    		return err
    	}
    	log.Printf("Stopped.
    ")
    	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)
    		case <-s.exit:
    			s.wg.Done()
    			return
    		}
    	}
    }
    
    
  • 相关阅读:
    千万不要死于无知—几条健康忠告(一)
    2006中国大学生最佳雇主TOP50排行榜(转)
    chromedriver与chrome版本映射表(最新)
    selenium自动化测试学习(一)
    selenium自动化测试——常见的八种元素定位方法
    作业1
    快速搭建属于自己的数据库——mongodb
    通过express搭建自己的服务器
    我的前端之路
    分享ES6中比较常用又强大的新特性
  • 原文地址:https://www.cnblogs.com/cqvoip/p/8078927.html
Copyright © 2011-2022 走看看