From 8ba0a125eb7302d8c51d3c85cf8ea8a4de5413db Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期六, 25 五月 2024 17:56:27 +0800
Subject: [PATCH] 修复ape类型判断

---
 main.go |   16 ++++++++++++----
 1 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/main.go b/main.go
index 1c8a2cf..d055edf 100644
--- a/main.go
+++ b/main.go
@@ -3,14 +3,14 @@
 import (
 	"context"
 	"fmt"
-	"gat1400Exchange/cron"
 	"net/http"
 	"os"
 	"os/signal"
 	"syscall"
-	"time"
 
+	"gat1400Exchange/client"
 	"gat1400Exchange/config"
+	"gat1400Exchange/cron"
 	"gat1400Exchange/models"
 	"gat1400Exchange/pkg/logger"
 	"gat1400Exchange/routes"
@@ -37,8 +37,14 @@
 		Handler: r,
 	}
 
+	// 鍚姩1400瀹㈡埛绔�
+	ctx, cancel := context.WithCancel(context.Background())
+	go client.Init1400Client(ctx)
+
 	// 鍚姩缃戠粶瑙嗛瀛楃鍙犲姞鍣ㄦ湇鍔�
-	go service.NVCSServer()
+	go service.StartNVCSServer()
+
+	go service.InitSubscribeTask()
 
 	// 鍚姩瀹氭椂浠诲姟
 	cron.Init()
@@ -65,11 +71,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)
+	//ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
 	defer cancel()
 	if err := srv.Shutdown(ctx); err != nil {
 		logger.Error("Server forced to shutdown:", err)
 	}
 
+	service.StopSubscribeTask()
+
 	logger.Info("Server exiting!")
 }

--
Gitblit v1.8.0