mirror of
https://github.com/nezhahq/nezha.git
synced 2025-01-23 05:08:13 -05:00
57 lines
1.5 KiB
Go
57 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
|
|
"github.com/naiba/nezha/cmd/dashboard/controller"
|
|
"github.com/naiba/nezha/cmd/dashboard/rpc"
|
|
"github.com/naiba/nezha/model"
|
|
"github.com/naiba/nezha/service/singleton"
|
|
"github.com/ory/graceful"
|
|
)
|
|
|
|
func init() {
|
|
// 初始化 dao 包
|
|
singleton.Init()
|
|
singleton.InitConfigFromPath("data/config.yaml")
|
|
singleton.InitDBFromPath("data/sqlite.db")
|
|
singleton.InitLocalizer()
|
|
initSystem()
|
|
}
|
|
|
|
func initSystem() {
|
|
// 启动 singleton 包下的所有服务
|
|
singleton.LoadSingleton()
|
|
|
|
// 每天的3:30 对 监控记录 和 流量记录 进行清理
|
|
if _, err := singleton.Cron.AddFunc("0 30 3 * * *", singleton.CleanMonitorHistory); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
// 每小时对流量记录进行打点
|
|
if _, err := singleton.Cron.AddFunc("0 0 * * * *", singleton.RecordTransferHourlyUsage); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
singleton.CleanMonitorHistory()
|
|
go rpc.ServeRPC(singleton.Conf.GRPCPort)
|
|
serviceSentinelDispatchBus := make(chan model.Monitor) // 用于传递服务监控任务信息的channel
|
|
go rpc.DispatchTask(serviceSentinelDispatchBus)
|
|
go rpc.DispatchKeepalive()
|
|
go singleton.AlertSentinelStart()
|
|
singleton.NewServiceSentinel(serviceSentinelDispatchBus)
|
|
srv := controller.ServeWeb(singleton.Conf.HTTPPort)
|
|
graceful.Graceful(func() error {
|
|
return srv.ListenAndServe()
|
|
}, func(c context.Context) error {
|
|
log.Println("NEZHA>> Graceful::START")
|
|
singleton.RecordTransferHourlyUsage()
|
|
log.Println("NEZHA>> Graceful::END")
|
|
srv.Shutdown(c)
|
|
return nil
|
|
})
|
|
}
|