From 539a78196da60eb97cf7057c5c85dfaa9b240741 Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期五, 11 十月 2024 16:21:32 +0800
Subject: [PATCH] 添加A3平层传感器数据获取方式

---
 main.go |   25 +++++++++++++++++++------
 1 files changed, 19 insertions(+), 6 deletions(-)

diff --git a/main.go b/main.go
index 1e8a151..5d71a57 100644
--- a/main.go
+++ b/main.go
@@ -7,9 +7,10 @@
 	"os"
 	"os/signal"
 	"syscall"
-	"time"
 
+	"gat1400Exchange/client"
 	"gat1400Exchange/config"
+	"gat1400Exchange/cron"
 	"gat1400Exchange/models"
 	"gat1400Exchange/pkg/logger"
 	"gat1400Exchange/routes"
@@ -36,9 +37,7 @@
 		Handler: r,
 	}
 
-	// 鍚姩缃戠粶瑙嗛瀛楃鍙犲姞鍣ㄦ湇鍔�
-	go service.NVCSServer()
-	go service.DeviceInfoReportTask()
+	srv.SetKeepAlivesEnabled(config.ServeConf.Keepalive)
 
 	// Initializing the server in a goroutine so that
 	// it won't block the graceful shutdown handling below
@@ -49,6 +48,18 @@
 	}()
 
 	logger.Info("Server is running at %s:%s", host, port)
+
+	// 鍚姩1400瀹㈡埛绔�
+	ctx, cancel := context.WithCancel(context.Background())
+	go client.Init1400Client(ctx)
+
+	// 鍚姩缃戠粶瑙嗛瀛楃鍙犲姞鍣ㄦ湇鍔�
+	go service.StartNVCSServer()
+
+	go service.InitSubscribeNotificationTasks()
+
+	// 鍚姩瀹氭椂浠诲姟
+	cron.Init()
 
 	// Wait for interrupt signal to gracefully shutdown the server with
 	// a timeout of 5 seconds.
@@ -62,11 +73,13 @@
 
 	// The context is used to inform the server it has 5 seconds to finish
 	// the request it is currently handling
-	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
-	defer cancel()
+	//ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	cancel()
 	if err := srv.Shutdown(ctx); err != nil {
 		logger.Error("Server forced to shutdown:", err)
 	}
 
+	service.StopNotificationTasks()
+
 	logger.Info("Server exiting!")
 }

--
Gitblit v1.8.0