fix: 增加读取OSS文件的重试机制

2026-07-08 13:36:34 +08:00
parent 55e4612404
commit 9131b06c8d
1 changed files with 10 additions and 4 deletions
@@ -143,10 +143,16 @@ func Callback(ctx context.Context, req *dto.CallbackReq) error {
 	// 2) 读取 OSS 文件内容
 	var ossContent []byte
 	if req.OssFile != "" {
-		ossContent, err = gateway.GetFileBytesFromURL(ctx, req.OssFile)
-		if err != nil {
-			g.Log().Infof(ctx, "[回调处理] 读取OSS文件 taskId=%s,state=%v,ossFile=%v", req.TaskId, req.State, req.OssFile)
-			g.Log().Warningf(ctx, "[回调处理] 读取OSS失败 taskId=%s err=%v", req.TaskId, err)
+		for attempt := 0; attempt <= 5; attempt++ {
+			ossContent, err = gateway.GetFileBytesFromURL(ctx, req.OssFile)
+			if err != nil {
+				g.Log().Infof(ctx, "[回调处理] 读取OSS文件 taskId=%s,state=%v,ossFile=%v", req.TaskId, req.State, req.OssFile)
+				g.Log().Warningf(ctx, "[回调处理] 读取OSS失败 taskId=%s err=%v", req.TaskId, err)
+			}
+			if err == nil {
+				g.Log().Infof(ctx, "[回调处理] 读取OSS成功 taskId=%s,state=%v,ossFile=%v", req.TaskId, req.State, req.OssFile)
+				break
+			}
 		}
 	}