cmd/httpserver/http.go

package main
import (
	"fmt"
	"os"
	"os/signal"
	"pro2d/common"
	"pro2d/common/components"
	"pro2d/common/db/mongoproxy"
	"pro2d/common/db/redisproxy"
	"pro2d/common/etcd"
	"pro2d/common/logger"
	"pro2d/models"
	"syscall"
	"time"
)
type AccountServer struct {
	components.IHttp
	EtcdClient *etcd.EtcdClient
	Sconf      *common.SConf
}
func NewAccountServer(version string, port ...string) *AccountServer {
	return &AccountServer{IHttp: components.NewHttpServer(version, port...)}
}
func (s *AccountServer) Init(sconf *common.SConf) error {
	s.Sconf = sconf
	//mgo init
	err := mongoproxy.ConnectMongo(sconf.MongoConf)
	//redis init
	if err = redisproxy.ConnectRedis(sconf.RedisConf.DB, sconf.RedisConf.Auth, sconf.RedisConf.Address); err != nil {
		return err
	}
	//Etcd 初始化
	s.EtcdClient, err = etcd.NewEtcdClient(common.GlobalConf.Etcd)
	if err != nil {
		return err
	}
	models.NewDBSeed(sconf.ID).InitServerDatabase(models.AccountModels())
	models.DBSeedS().InitAutoIncreUidTable(models.AccountModels())
	s.EtcdClient.PutWithLeasePrefix(sconf.Name, fmt.Sprintf("%d", sconf.ID), fmt.Sprintf("%s:%d", sconf.IP, sconf.Port), 5)
	return nil
}
func (s *AccountServer) TimeOut() {
	models.DBSeedS().SaveAutoincrementTimer(models.AccountModels())
	components.TimeOut(1*time.Second, s.TimeOut)
}
func (s *AccountServer) Start() error {
	if err := s.Init(common.GlobalConf.AccountConf); err != nil {
		return err
	}
	//开始定时器
	s.TimeOut()
	return s.IHttp.Start()
}
func main() {
	err := make(chan error)
	stopChan := make(chan os.Signal)
	signal.Notify(stopChan, syscall.SIGTERM, syscall.SIGINT, syscall.SIGKILL)
	web := NewAccountServer("v1", fmt.Sprintf(":%d", common.GlobalConf.AccountConf.Port))
	web.BindHandler(&AccountAction{HttpServer: web})
	go func() {
		err <- web.Start()
	}()
	select {
	case e := <-err:
		logger.Error("http server error: %v", e)
	case <-stopChan:
		logger.Debug("http stop")
		web.Stop()
	}
}