chore: adjusting on error return

aziontech · May 21, 2024 · 33d033a · 33d033a
1 parent a95d7f2
commit 33d033a
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 7 deletions.
diff --git a/pkg/api/storage/storage.go b/pkg/api/storage/storage.go
@@ -105,9 +105,17 @@ func (c *Client) Upload(ctx context.Context, fileOps *contracts.FileOps, conf *c
 	req := c.apiClient.StorageAPI.StorageApiBucketsObjectsCreate(ctx, conf.Bucket, file).Body(fileOps.FileContent).ContentType(fileOps.MimeType)
 	_, httpResp, err := req.Execute()
 	if err != nil {
-		logger.Debug("Error while uploading file <"+fileOps.Path+"> to storage api", zap.Error(err))
+		if httpResp != nil {
+			logger.Debug("Error while uploading file <"+fileOps.Path+"> to storage api", zap.Error(err))
+			err := utils.LogAndRewindBody(httpResp)
+			if err != nil {
+				return err
+			}
+			return utils.ErrorPerStatusCode(httpResp, err)
+		}
 		return utils.ErrorPerStatusCode(httpResp, err)
 	}
+
 	return nil
 }
 

diff --git a/pkg/cmd/deploy/worker.go b/pkg/cmd/deploy/worker.go
@@ -38,14 +38,14 @@ func worker(jobs <-chan contracts.FileOps, results chan<- error, currentFile *in
 				if err != nil {
 					continue
 				}
-				results <- nil
-				atomic.AddInt64(currentFile, 1)
-				return
+				break
 			}
 
-			logger.Debug("There have been 5 retries already, quitting upload")
-			results <- err
-			return
+			if Retries >= 5 {
+				logger.Debug("There have been 5 retries already, quitting upload")
+				results <- err
+				return
+			}
 		}
 
 		atomic.AddInt64(currentFile, 1)