nezha/cmd/dashboard/main.go

55 lines
1.1 KiB
Go
Raw Normal View History

2019-12-05 22:36:58 +08:00
package main
import (
2019-12-09 16:02:49 +08:00
"fmt"
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"
"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
2019-12-09 18:14:31 +08:00
dao.ServerList = make(map[string]*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)
}
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() {
dao.DB.AutoMigrate(model.Server{}, model.User{}, model.Notification{}, model.AlertRule{})
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{}
innerS.State = &model.State{}
2019-12-09 18:14:31 +08:00
dao.ServerList[fmt.Sprintf("%d", innerS.ID)] = &innerS
2019-12-09 16:02:49 +08:00
}
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-06 09:35:04 +08:00
alertmanager.Start()
2019-12-05 22:36:58 +08:00
}