2019-12-05 22:36:58 +08:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
2019-12-08 16:59:58 +08:00
|
|
|
"time"
|
2019-12-05 22:36:58 +08:00
|
|
|
|
2019-12-08 16:59:58 +08:00
|
|
|
"github.com/patrickmn/go-cache"
|
2020-12-19 23:11:16 +08:00
|
|
|
"gorm.io/driver/sqlite"
|
|
|
|
"gorm.io/gorm"
|
2019-12-05 23:42:20 +08:00
|
|
|
|
2020-11-11 10:07:45 +08:00
|
|
|
"github.com/naiba/nezha/cmd/dashboard/controller"
|
|
|
|
"github.com/naiba/nezha/cmd/dashboard/rpc"
|
|
|
|
"github.com/naiba/nezha/model"
|
2020-12-20 12:18:27 +08:00
|
|
|
"github.com/naiba/nezha/service/alertmanager"
|
2020-11-11 10:07:45 +08:00
|
|
|
"github.com/naiba/nezha/service/dao"
|
2019-12-05 22:36:58 +08:00
|
|
|
)
|
|
|
|
|
2019-12-08 16:59:58 +08:00
|
|
|
func init() {
|
|
|
|
var err error
|
2021-01-08 21:04:50 +08:00
|
|
|
dao.ServerList = make(map[uint64]*model.Server)
|
2019-12-20 23:58:09 +08:00
|
|
|
dao.Conf = &model.Config{}
|
|
|
|
err = dao.Conf.Read("data/config.yaml")
|
2019-12-08 16:59:58 +08:00
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
2020-12-19 23:11:16 +08:00
|
|
|
dao.DB, err = gorm.Open(sqlite.Open("data/sqlite.db"), &gorm.Config{})
|
2019-12-05 22:36:58 +08:00
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
2019-12-08 23:18:29 +08:00
|
|
|
if dao.Conf.Debug {
|
|
|
|
dao.DB = dao.DB.Debug()
|
|
|
|
}
|
2019-12-08 16:59:58 +08:00
|
|
|
dao.Cache = cache.New(5*time.Minute, 10*time.Minute)
|
2019-12-08 23:18:29 +08:00
|
|
|
initDB()
|
|
|
|
}
|
|
|
|
|
|
|
|
func initDB() {
|
2021-01-16 00:45:49 +08:00
|
|
|
dao.DB.AutoMigrate(model.Server{}, model.User{},
|
|
|
|
model.Notification{}, model.AlertRule{}, model.Monitor{},
|
|
|
|
model.MonitorHistory{})
|
2019-12-09 16:02:49 +08:00
|
|
|
// load cache
|
|
|
|
var servers []model.Server
|
|
|
|
dao.DB.Find(&servers)
|
|
|
|
for _, s := range servers {
|
2019-12-09 18:14:31 +08:00
|
|
|
innerS := s
|
2020-12-20 12:18:27 +08:00
|
|
|
innerS.Host = &model.Host{}
|
2021-01-16 00:45:49 +08:00
|
|
|
innerS.State = &model.HostState{}
|
2021-01-08 21:04:50 +08:00
|
|
|
dao.ServerList[innerS.ID] = &innerS
|
2019-12-09 16:02:49 +08:00
|
|
|
}
|
2021-01-08 21:04:50 +08:00
|
|
|
dao.ReSortServer()
|
2019-12-08 16:59:58 +08:00
|
|
|
}
|
2019-12-05 22:36:58 +08:00
|
|
|
|
2019-12-08 16:59:58 +08:00
|
|
|
func main() {
|
2020-10-24 21:44:03 +08:00
|
|
|
go controller.ServeWeb(dao.Conf.HTTPPort)
|
2019-12-10 18:05:02 +08:00
|
|
|
go rpc.ServeRPC(5555)
|
2021-01-16 00:45:49 +08:00
|
|
|
go rpc.DispatchTask(time.Minute * 10)
|
2021-01-06 09:35:04 +08:00
|
|
|
alertmanager.Start()
|
2019-12-05 22:36:58 +08:00
|
|
|
}
|