incus/file/pull: Read files in chunks

Signed-off-by: Stéphane Graber <[email protected]>
lxc · Feb 2, 2025 · 04eab98 · 04eab98
1 parent c06318b
commit 04eab98
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/cmd/incus/file.go b/cmd/incus/file.go
@@ -686,10 +686,17 @@ func (c *cmdFilePull) Run(cmd *cobra.Command, args []string) error {
 				return err
 			}
 		} else {
-			_, err = io.Copy(writer, src)
-			if err != nil {
-				progress.Done("")
-				return err
+			for {
+				// Read 1MB at a time.
+				_, err = io.CopyN(writer, src, 1024 * 1024 * 1024)
+				if err != nil {
+					if err == io.EOF {
+						break
+					}
+
+					progress.Done("")
+					return err
+				}
 			}
 		}