Files issue writing zip files

h2oai · Dec 12, 2016 · 995b98a · 995b98a
1 parent a5e7461
commit 995b98a
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/master/upload.go b/master/upload.go
@@ -134,14 +134,17 @@ func (s *UploadHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	dst, err := os.OpenFile(dstPath, os.O_CREATE|os.O_TRUNC, fs.FilePerm)
+	dst, err := os.OpenFile(dstPath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, fs.FilePerm)
 	if err != nil {
 		log.Println("Upload file open operation failed:", err)
 		http.Error(w, fmt.Sprintf("Error writing uploaded file to disk: %s", err), http.StatusInternalServerError)
 		return
 	}
 	defer dst.Close()
-	io.Copy(dst, src)
+	if _, err := io.Copy(dst, src); err != nil {
+		log.Println("Error writing to destination:", err)
+		http.Error(w, fmt.Sprintf("Error writing zip file: %s", err), http.StatusInternalServerError)
+	}
 
 	switch typ {
 	case fs.KindEngine: