Files
MirageServer/server.go
Chenyang Gao 7e14829a0e 清理日志输出
Signed-off-by: Chenyang Gao <gps949@outlook.com>
2023-05-16 11:34:34 +08:00

99 lines
2.2 KiB
Go

package main
import (
"os"
"MirageNetwork/MirageServer/controller"
"github.com/rs/zerolog"
"github.com/rs/zerolog/log"
)
const (
MirageDateTimeFormat = "2006-01-02 15:04:05"
SocketWritePermissions = 0o666
)
func main() {
sysAddr, set := os.LookupEnv("MIRAGE_SYS_ADDR")
if !set {
sysAddr = ":8081"
}
zerolog.TimeFieldFormat = MirageDateTimeFormat
logger := zerolog.New(zerolog.MultiLevelWriter(os.Stdout)).With().Timestamp().Logger()
logger = logger.Level(zerolog.DebugLevel)
log.Logger = logger
datapool := controller.DataPool{}
err := datapool.OpenDB()
if err != nil {
log.Fatal().Caller().Err(err).Msg("Error opening database")
}
ctrlChn := make(chan controller.CtrlMsg)
msgChn := make(chan controller.CtrlMsg)
err = datapool.InitCockpitDB()
if err != nil {
log.Fatal().Caller().Err(err).Msg("Error initializing cockpit database")
}
cockpit, err := controller.NewCockpit(sysAddr, ctrlChn, msgChn, datapool.DB())
if err != nil {
log.Fatal().Caller().Err(err).Msg("Error initializing cockpit")
}
go cockpit.Run()
log.Info().Msg("Cockpit is ready on " + sysAddr + "")
/*
err = controller.LoadConfig()
if err != nil {
log.Fatal().Caller().Err(err).Msgf("Error loading config")
}
cfg, err := controller.GetMirageConfig(srvAddr, serverURL)
if err != nil {
log.Fatal().Caller().Err(err).Msgf("Error loading config")
}
*/
for {
select {
case cockpitMsg := <-ctrlChn:
log.Trace().Msgf("Cockpit message received: %s", cockpitMsg.Msg)
switch cockpitMsg.Msg {
case "start":
err = datapool.InitMirageDB()
if err != nil {
log.Error().Caller().Err(err).Msgf("Error initializing Mirage DB")
msgChn <- controller.CtrlMsg{
Msg: "error",
Err: err,
}
break
}
cockpit.App, err = controller.NewMirage(cockpitMsg.SysCfg, datapool.DB())
if err != nil {
log.Error().Caller().Err(err).Msg("Error initializing Mirage")
msgChn <- controller.CtrlMsg{
Msg: "error",
Err: err,
}
break
}
err = cockpit.App.Serve(ctrlChn)
if err != nil {
log.Error().Caller().Err(err).Msg("Error starting server")
msgChn <- controller.CtrlMsg{
Msg: "error",
Err: err,
}
break
}
}
}
}
}