This commit is contained in:
Tiger Wang 2022-09-08 23:11:15 -04:00
parent 84dfa7f5bf
commit 6ba5420056

29
main.go
View File

@ -28,9 +28,11 @@ const LOCALHOST = "127.0.0.1"
var sqliteDB *gorm.DB var sqliteDB *gorm.DB
var configFlag = flag.String("c", "", "config address") var (
var dbFlag = flag.String("db", "", "db path") configFlag = flag.String("c", "", "config address")
var versionFlag = flag.Bool("v", false, "version") dbFlag = flag.String("db", "", "db path")
versionFlag = flag.Bool("v", false, "version")
)
func init() { func init() {
flag.Parse() flag.Parse()
@ -47,7 +49,7 @@ func init() {
} }
sqliteDB = sqlite.GetDb(*dbFlag) sqliteDB = sqlite.GetDb(*dbFlag)
//gredis.GetRedisConn(config.RedisInfo), // gredis.GetRedisConn(config.RedisInfo),
service.MyService = service.NewService(sqliteDB, config.CommonInfo.RuntimePath) service.MyService = service.NewService(sqliteDB, config.CommonInfo.RuntimePath)
@ -60,7 +62,6 @@ func init() {
// go service.LoopFriend() // go service.LoopFriend()
// go service.MyService.App().CheckNewImage() // go service.MyService.App().CheckNewImage()
} }
// @title casaOS API // @title casaOS API
@ -80,21 +81,21 @@ func main() {
return return
} }
go route.SocketInit(service.NotifyMsg) go route.SocketInit(service.NotifyMsg)
//model.Setup() // model.Setup()
//gredis.Setup() // gredis.Setup()
r := route.InitRouter() r := route.InitRouter()
//service.SyncTask(sqliteDB) // service.SyncTask(sqliteDB)
cron2 := cron.New() cron2 := cron.New()
//every day execution // every day execution
err := cron2.AddFunc("0/5 * * * * *", func() { err := cron2.AddFunc("0/5 * * * * *", func() {
if service.ClientCount > 0 { if service.ClientCount > 0 {
//route.SendNetINfoBySocket() // route.SendNetINfoBySocket()
//route.SendCPUBySocket() // route.SendCPUBySocket()
//route.SendMemBySocket() // route.SendMemBySocket()
// route.SendDiskBySocket() // route.SendDiskBySocket()
//route.SendUSBBySocket() // route.SendUSBBySocket()
route.SendAllHardwareStatusBySocket() route.SendAllHardwareStatusBySocket()
} }
}) })
@ -123,7 +124,7 @@ func main() {
} }
go func() { go func() {
time.Sleep(time.Second * 2) time.Sleep(time.Second * 2)
//v0.3.6 // v0.3.6
if config.ServerInfo.HttpPort != "" { if config.ServerInfo.HttpPort != "" {
changePort := common.ChangePortRequest{} changePort := common.ChangePortRequest{}
changePort.Port = config.ServerInfo.HttpPort changePort.Port = config.ServerInfo.HttpPort