فهرست منبع

fix:edit agent/data/batchDelete

jimmyyem 5 ماه پیش
والد
کامیت
a2764ce743
1فایلهای تغییر یافته به همراه17 افزوده شده و 10 حذف شده
  1. 17 10
      internal/logic/agent/batch_delete_agent_data_logic.go

+ 17 - 10
internal/logic/agent/batch_delete_agent_data_logic.go

@@ -3,8 +3,8 @@ package agent
 import (
 	"context"
 	"fmt"
+	"sync"
 	"wechat-api/hook/fastgpt"
-
 	"wechat-api/internal/svc"
 	"wechat-api/internal/types"
 
@@ -28,17 +28,24 @@ func (l *BatchDeleteAgentDataLogic) BatchDeleteAgentData(req *types.BatchDeleteI
 	var success, fail int
 
 	if len(req.Ids) > 0 {
+		var wg sync.WaitGroup
 		for _, sid := range req.Ids {
-			response, err := fastgpt.DeleteData(sid)
-			fmt.Printf("id=%s, response=%+v\n", sid, response)
-			if err != nil {
-				fail++
-				fmt.Printf("delete fastgpt data failed:%v\n", err)
-			}
-			if response != nil && response.Code == 200 {
-				success++
-			}
+			wg.Add(1)
+			go func(sid string) {
+				defer wg.Done()
+
+				response, err := fastgpt.DeleteData(sid)
+				fmt.Printf("id=%s, response=%+v\n", sid, response)
+				if err != nil {
+					fail++
+					fmt.Printf("delete fastgpt data failed:%v\n", err)
+				}
+				if response != nil && response.Code == 200 {
+					success++
+				}
+			}(sid)
 		}
+		wg.Wait()
 	}
 
 	resp := types.BaseDataInfo{}