fix: 部分情况缺少返回预扣

(cherry picked from commit 96373455521a38095706bd81c57f9a18557d9c2e)
2024-11-05 19:28:08 +08:00
parent d9c7e98115
commit 14f4320923
5 changed files with 33 additions and 18 deletions
@@ -391,7 +391,7 @@ func OpenaiRealtimeHandler(c *gin.Context, info *relaycommon.RelayInfo) (*dto.Op
 	localUsage := &dto.RealtimeUsage{}
 	sumUsage := &dto.RealtimeUsage{}

-	go func() {
+	gopool.Go(func() {
 		for {
 			select {
 			case <-c.Done():
@@ -444,9 +444,9 @@ func OpenaiRealtimeHandler(c *gin.Context, info *relaycommon.RelayInfo) (*dto.Op
 				}
 			}
 		}
-	}()
+	})

-	go func() {
+	gopool.Go(func() {
 		for {
 			select {
 			case <-c.Done():
@@ -541,7 +541,7 @@ func OpenaiRealtimeHandler(c *gin.Context, info *relaycommon.RelayInfo) (*dto.Op
 				}
 			}
 		}
-	}()
+	})

 	select {
 	case <-clientClosed: