From ac37ca0d9825297bf28644caab4206a32563cf84 Mon Sep 17 00:00:00 2001
From: Shreyas Goenka <shreyas.goenka@databricks.com>
Date: Thu, 2 Jan 2025 18:54:35 +0530
Subject: [PATCH] -

---
 libs/filer/dbfs_client.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/libs/filer/dbfs_client.go b/libs/filer/dbfs_client.go
index 6a7f90600..0a799e7ad 100644
--- a/libs/filer/dbfs_client.go
+++ b/libs/filer/dbfs_client.go
@@ -168,6 +168,10 @@ func (w *DbfsClient) putFile(ctx context.Context, path string, overwrite bool, f
 			return
 		}
 		_, err = io.Copy(contents, file)
+		if err != nil {
+			pw.CloseWithError(fmt.Errorf("error while streaming file to dbfs: %w", err))
+			return
+		}
 		err = writer.Close()
 		if err != nil {
 			pw.CloseWithError(fmt.Errorf("failed to close multipart form writer: %w", err))