1

config/serverinit/usersubscribeinit.go

@@ -0,0 +1,85 @@
+package serverinit
+
+import (
+	"context"
+	"go-admin/app/admin/models"
+	"go-admin/app/admin/service"
+	"go-admin/common/const/rediskey"
+	"go-admin/common/helper"
+	"go-admin/pkg/utility"
+	"go-admin/services/excservice"
+	"os"
+	"time"
+
+	"github.com/bytedance/sonic"
+	log "github.com/go-admin-team/go-admin-core/logger"
+	"gorm.io/gorm"
+)
+
+// 初始化用户订阅ws连接
+func UserSubscribeInit(orm *gorm.DB, ctx context.Context) {
+	var list []models.LineApiUser
+
+	err := orm.Model(&models.LineApiUser{}).Where("open_status=1 ").Find(&list).Error
+
+	if err != nil {
+		log.Error("获取用户api失败", err)
+		os.Exit(1)
+	}
+
+	for _, item := range list {
+		service.OpenUserBinanceWebsocket(item)
+	}
+
+	utility.SafeGo(func() {
+		ticker := time.NewTicker(time.Second * 5)
+		defer ticker.Stop()
+
+		for {
+			select {
+			case <-ctx.Done():
+				log.Info("ctx.Done() 被触发，原因:", ctx.Err()) // 打印上下文错误信息
+				return
+			case <-ticker.C:
+				deleteKeys, _ := helper.DefaultRedis.GetAllList(rediskey.ApiUserDeleteList)
+				activeApis, _ := helper.DefaultRedis.GetAllList(rediskey.ApiUserActiveList)
+
+				//移除连接
+				for _, item := range deleteKeys {
+					if wm, ok := excservice.SpotSockets[item]; ok {
+						wm.Stop()
+
+						delete(excservice.FutureSockets, item)
+					}
+
+					if wm, ok := excservice.SpotSockets[item]; ok {
+						wm.Stop()
+
+						delete(excservice.SpotSockets, item)
+					}
+
+					if _, err := helper.DefaultRedis.LRem(rediskey.ApiUserDeleteList, item); err != nil {
+						log.Error("移除 待关闭websocket 失败:", err)
+					}
+				}
+
+				//连接websocket
+				var apiUser models.LineApiUser
+				for _, item := range activeApis {
+					if item == "" {
+						continue
+					}
+
+					sonic.Unmarshal([]byte(item), &apiUser)
+
+					if apiUser.Id > 0 {
+						service.OpenUserBinanceWebsocket(apiUser)
+						if _, err := helper.DefaultRedis.LRem(rediskey.ApiUserActiveList, item); err != nil {
+							log.Error("删除 待触发的apiUser失败", err)
+						}
+					}
+				}
+			}
+		}
+	})
+}