1 Star 0 Fork 0

jeffops/API-DEMO

加入 Gitee
与超过 1400万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
start.go 2.78 KB
一键复制 编辑 原始数据 按行查看 历史
package cmd
import (
"fmt"
"os"
"os/signal"
"syscall"
"gitee.com/wx_68f311c3da/api-demo/apps"
_ "gitee.com/wx_68f311c3da/api-demo/apps/all"
"gitee.com/wx_68f311c3da/api-demo/conf"
"gitee.com/wx_68f311c3da/api-demo/protocol"
"github.com/infraboard/mcube/logger"
"github.com/infraboard/mcube/logger/zap"
"github.com/spf13/cobra"
)
var (
// pusher service config option
confFile string
)
//start cmd启动时组装
var StartCmd = &cobra.Command{
Use: "start",
Short: "启动 demo-api 后端API",
Long: "启动 demo-api 后端API",
RunE: func(cmd *cobra.Command, args []string) error {
//加载程序配置
err := conf.LoadConfigFromToml(confFile)
if err != nil {
panic(err)
}
loadGlobalLogger()
//加载 hosts service的实体类
//service := impl.NewHostServiceImpl()
//注册HostService的实例到ioc
//apps.HostService = impl.NewHostServiceImpl()
// 通过Host api handeler 提供http接口.
apps.InitImpl()
//注册gin
//e := gin.Default()
//apps.InitGin(e)
//return e.Run(conf.C().App.HttpAddr())
svc := NewManager()
ch := make(chan os.Signal, 1)
defer close(ch)
signal.Notify(ch, syscall.SIGTERM, syscall.SIGHUP, syscall.SIGQUIT, syscall.SIGINT)
go svc.WaitStop(ch)
return svc.Start()
},
}
func NewManager() *manager {
return &manager{
http: *protocol.NewHttpService(),
l: zap.L().Named("CLI"),
}
}
//管理需要启动的服务
type manager struct {
http protocol.HttpService
l logger.Logger
}
func (m *manager) Start() error {
return m.http.Start()
}
//接收信号处理中断
func (m *manager) WaitStop(ch chan os.Signal) {
for v := range ch {
switch v {
default:
m.l.Infof("received signal %s", v)
m.http.Stop()
}
}
}
func (m *manager) Stop() {
m.http.Stop()
}
//日志模块初始化
// log 为全局变量, 只需要load 即可全局可用户, 依赖全局配置先初始化
func loadGlobalLogger() error {
var (
logInitMsg string
level zap.Level
)
lc := conf.C().Log
lv, err := zap.NewLevel(lc.Level)
if err != nil {
logInitMsg = fmt.Sprintf("%s, use default level INFO", err)
level = zap.InfoLevel
} else {
level = lv
logInitMsg = fmt.Sprintf("log level: %s", lv)
}
zapConfig := zap.DefaultConfig()
zapConfig.Level = level
zapConfig.Files.RotateOnStartup = false
switch lc.To {
case conf.ToStdout:
zapConfig.ToStderr = true
zapConfig.ToFiles = false
case conf.ToFile:
zapConfig.Files.Name = "api.log"
zapConfig.Files.Path = lc.PathDir
}
switch lc.Format {
case conf.JSONFormat:
zapConfig.JSON = true
}
if err := zap.Configure(zapConfig); err != nil {
return err
}
zap.L().Named("INIT").Info(logInitMsg)
return nil
}
func init() {
StartCmd.PersistentFlags().StringVarP(&confFile, "config", "f", "etc/demo.toml", "demo-api 配置文件路径")
RootCmd.AddCommand(StartCmd)
}
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Go
1
https://gitee.com/wx_68f311c3da/api-demo.git
git@gitee.com:wx_68f311c3da/api-demo.git
wx_68f311c3da
api-demo
API-DEMO
553a5871bd68

搜索帮助