mirror of
https://github.com/nezhahq/nezha.git
synced 2025-01-22 20:58:14 -05:00
54 lines
1.0 KiB
Go
54 lines
1.0 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/jinzhu/gorm"
|
|
_ "github.com/jinzhu/gorm/dialects/sqlite"
|
|
"github.com/patrickmn/go-cache"
|
|
|
|
"github.com/p14yground/nezha/cmd/dashboard/controller"
|
|
"github.com/p14yground/nezha/cmd/dashboard/rpc"
|
|
"github.com/p14yground/nezha/model"
|
|
"github.com/p14yground/nezha/service/dao"
|
|
)
|
|
|
|
func init() {
|
|
var err error
|
|
dao.ServerList = make(map[string]*model.Server)
|
|
dao.Conf, err = model.ReadInConfig("data/config.yaml")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
dao.Admin = &model.User{
|
|
Login: dao.Conf.GitHub.Admin,
|
|
}
|
|
dao.DB, err = gorm.Open("sqlite3", "data/sqlite.db")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
if dao.Conf.Debug {
|
|
dao.DB = dao.DB.Debug()
|
|
}
|
|
dao.Cache = cache.New(5*time.Minute, 10*time.Minute)
|
|
initDB()
|
|
}
|
|
|
|
func initDB() {
|
|
dao.DB.AutoMigrate(model.Server{})
|
|
// load cache
|
|
var servers []model.Server
|
|
dao.DB.Find(&servers)
|
|
for _, s := range servers {
|
|
innerS := s
|
|
dao.ServerList[fmt.Sprintf("%d", innerS.ID)] = &innerS
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
go controller.ServeWeb()
|
|
go rpc.ServeRPC()
|
|
select {}
|
|
}
|