From 6597287fc412ce914aba41fe571ed05817cf371c Mon Sep 17 00:00:00 2001
From: sunty <1172534965@qq.com>
Date: 星期一, 27 五月 2024 11:34:58 +0800
Subject: [PATCH] fix script
---
main.go | 48 ++++++++++++++++++++++++++++++++++++++++++++++--
1 files changed, 46 insertions(+), 2 deletions(-)
diff --git a/main.go b/main.go
index b03808f..f1d4320 100644
--- a/main.go
+++ b/main.go
@@ -1,7 +1,18 @@
package main
import (
+ "basic.com/CloudAI/protomsg.git/analysis"
+ "fmt"
+ "google.golang.org/grpc"
+ "net"
+ "ruleModelEngine/api/analysisApi"
+ "ruleModelEngine/cache"
"ruleModelEngine/config"
+ "ruleModelEngine/db"
+ "ruleModelEngine/task"
+
+ //"ruleModelEngine/task"
+
//"annotation_service/db"
"basic.com/valib/logger.git"
"flag"
@@ -16,7 +27,33 @@
var logFile = config.LogConf.Path + "annotation_service.log"
logger.InitLogger(logFile, config.LogConf.Level, config.LogConf.MaxSize, config.LogConf.MaxBackups, config.LogConf.MaxAge)
logger.Info("loginit success !")
+ dbErr := db.ConnectDB()
+ if dbErr != nil {
+ logger.Error("鏁版嵁搴撹繛鎺ュけ璐�", dbErr)
+ return
+ }
+ logger.Info("ConnectDB success !")
+ cacheErr := cache.DeviceCacheInit()
+ if cacheErr != nil {
+ logger.Error("璁惧琛ㄧ紦瀛樺け璐�", cacheErr)
+ return
+ }
+ logger.Info("cache device success !")
+}
+func initApiServer() {
+ ln, err := net.Listen("tcp", config.Api.Host+":"+config.Api.Port)
+ if err != nil {
+ logger.Fatal("grpc server init error: %v", err.Error())
+ //panic(fmt.Sprintf("grpc server init error: %v", err.Error()))
+ }
+ s := grpc.NewServer()
+ analysis.RegisterAnalysisServiceServer(s, &analysisApi.Server{})
+ err = s.Serve(ln)
+ if err != nil {
+ logger.Fatal("grpc server init error: %v", err.Error())
+ //panic(fmt.Sprintf("grpc server init error: %v", err.Error()))
+ }
}
func main() {
@@ -28,11 +65,18 @@
logger.Info("Executing immediately...")
rule.ExecuteTask()
}
-
+ r, _ := db.QueryByDays(3)
+ for _, id := range r {
+ task.TaskAnalysisService(id)
+ }
+ fmt.Println("ids: ", len(r))
+ return
+ //task.TaskAnalysisService("3407b20e-6343-4832-88fa-a8e986552532")
+ go initApiServer()
+ //return
now := time.Now()
next := time.Date(now.Year(), now.Month(), now.Day()+1, 1, 0, 0, 0, now.Location())
duration := next.Sub(now)
-
timer := time.NewTimer(duration)
logger.Info("The program has started and will execute at one o'clock in the early morning every night.")
for {
--
Gitblit v1.8.0