环节处理改为批处理

et_Info/InfoHandler.go

@@ -3,9 +3,9 @@ package et_Info
 import (
 	"gitea.anxinyun.cn/container/common_models"
 	"gitea.anxinyun.cn/container/common_utils"
-	"gitea.anxinyun.cn/container/common_utils/configLoad"
 	"log"
 	"node/stages"
+	"sync"
 )
 
 type InfoHandler struct {
@@ -13,13 +13,13 @@ type InfoHandler struct {
 	stage        *stages.Stage
 }
 
-func NewInfoHandler() *InfoHandler {
-	redisAddr := configLoad.LoadConfig().GetString("redis.address")
+func NewInfoHandler(configHelper *common_utils.ConfigHelper) *InfoHandler {
+	//redisAddr := configLoad.LoadConfig().GetString("redis.address")
 	the := &InfoHandler{
-		configHelper: common_utils.NewConfigHelper(redisAddr),
+		configHelper: configHelper, //common_utils.NewConfigHelper(redisAddr),
 		stage:        stages.NewStage("测点信息获取"),
 	}
-	the.stage.AddProcess(the.getStationInfo)
+	the.stage.AddProcess(the.getStationInfos)
 	return the
 }
 
@@ -27,8 +27,20 @@ func (the *InfoHandler) GetStage() stages.Stage {
 	return *the.stage
 }
 
-func (the *InfoHandler) getStationInfo(p *common_models.ProcessData) *common_models.ProcessData {
+func (the *InfoHandler) getStationInfos(data []*common_models.ProcessData) []*common_models.ProcessData {
+	var wg sync.WaitGroup // 初始化 WaitGroup
+	for _, processData := range data {
+		wg.Add(1)
+		go func(pd *common_models.ProcessData) {
+			defer wg.Done()
+			the.getStationInfo(pd)
+		}(processData)
+	}
+	wg.Wait()
+	return data
+}
 
+func (the *InfoHandler) getStationInfo(p *common_models.ProcessData) *common_models.ProcessData {
 	s, err := the.configHelper.GetDeviceStationObjs(p.DeviceData.DeviceId)
 	if err == nil && s != nil {
 		p.Stations = s