Fix s3 upload stuck (#54)

open-gpdb · Aug 24, 2024 · 22e6d9a · 22e6d9a
1 parent 8c492ef
commit 22e6d9a
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/pkg/proc/interaction.go b/pkg/proc/interaction.go
@@ -28,14 +28,6 @@ func ProcessPutExtended(
 	var w io.WriteCloser
 	r, w := io.Pipe()
 
-	err := s.PutFileToDest(name, r, settings)
-
-	if err != nil {
-		_ = ycl.ReplyError(err, "failed to upload")
-
-		return err
-	}
-
 	defer r.Close()
 	defer w.Close()
 
@@ -103,6 +95,15 @@ func ProcessPutExtended(
 		}
 	}()
 
+	/* Should go after reader dispatch! */
+	err := s.PutFileToDest(name, r, settings)
+
+	if err != nil {
+		_ = ycl.ReplyError(err, "failed to upload")
+
+		return err
+	}
+
 	wg.Wait()
 
 	_, err = ycl.GetRW().Write(message.NewReadyForQueryMessage().Encode())