Transfer - Fix "close of closed channel" panic (#1003)

jfrog · Oct 19, 2023 · cc76175 · cc76175
1 parent b1e98b9
commit cc76175
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/artifactory/commands/transferfiles/delayedartifactshandler.go b/artifactory/commands/transferfiles/delayedartifactshandler.go
@@ -369,6 +369,11 @@ func (w *SplitContentWriter) closeCurrentFile() error {
 		if err := w.writer.Close(); err != nil {
 			return err
 		}
+		defer func() {
+			// Reset writer and counter.
+			w.recordCount = 0
+			w.writer = nil
+		}()
 		if w.writer.GetFilePath() != "" {
 			fullPath, err := getUniqueErrorOrDelayFilePath(w.dirPath, func() string {
 				return w.filePrefix
@@ -384,9 +389,6 @@ func (w *SplitContentWriter) closeCurrentFile() error {
 			w.fileIndex++
 		}
 	}
-	// Reset writer and counter.
-	w.recordCount = 0
-	w.writer = nil
 	return nil
 }