chore(filer): change post file procotol

Author: duanhongyi

volumes/filer.go

@@ -2,14 +2,11 @@
 package volumes
 
 import (
-	"bytes"
 	"encoding/json"
 	"fmt"
 	"io"
-	"mime/multipart"
 	"net/http"
 	"net/url"
-	"strings"
 
 	drycc "github.com/drycc/controller-sdk-go"
 	"github.com/drycc/controller-sdk-go/api"
@@ -45,29 +42,14 @@ func GetFile(c *drycc.Client, appID, volumeID, path string) (*http.Response, err
 
 // Put file to an app's volume.
 func PostFile(c *drycc.Client, appID, volumeID, volumePath, name string, size int64, reader io.Reader) (*http.Response, error) {
-
-	buffer := new(bytes.Buffer)
-	writer := multipart.NewWriter(buffer)
-	if err := writer.WriteField("path", volumePath); err != nil {
-		return nil, err
-	}
-	if _, err := writer.CreateFormFile("file", name); err != nil {
-		return nil, err
-	}
-	size += int64(buffer.Len())
-	head := strings.NewReader(buffer.String())
-	buffer.Reset()
-	writer.Close()
-	bottom := strings.NewReader(buffer.String())
-	size += int64(buffer.Len())
 	u := fmt.Sprintf("/v2/apps/%s/volumes/%s/client/", appID, volumeID)
-
-	r, err := c.NewRequest("POST", u, io.MultiReader(head, reader, bottom))
+	r, err := c.NewRequest("POST", u, reader)
 	if err != nil {
 		return nil, err
 	}
 	r.ContentLength = size
-	r.Header.Add("Content-Type", writer.FormDataContentType())
+	r.Header.Add("Content-Type", "filer/octet-stream")
+	r.Header.Add("Content-Disposition", fmt.Sprintf(`attachment; filename="%s"; filepath="%s"`, name, volumePath))
 	return c.Do(r)
 }
 

volumes/filer_test.go

@@ -38,27 +38,17 @@ func (f *fakeFilerServer) ServeHTTP(res http.ResponseWriter, req *http.Request)
 			res.Write([]byte(`{"results":[], "count": 0}`))
 			return
 		} else if req.Method == "POST" {
-			if err := req.ParseMultipartForm(1024 * 1024); err != nil {
-				http.Error(res, fmt.Sprintf("Parse multipart form error: %v", err), http.StatusBadRequest)
-				return
+			body, err := io.ReadAll(req.Body)
+			if err != nil {
+				fmt.Println(err)
+				res.WriteHeader(http.StatusInternalServerError)
+				res.Write(nil)
 			}
-			for _, tmpFiles := range req.MultipartForm.File {
-				for _, tmpFile := range tmpFiles {
-					srcFile, err := tmpFile.Open()
-					if err != nil {
-						return
-					}
-					body, err := io.ReadAll(srcFile)
-					if err != nil {
-						return
-					}
-					if string(body) != volumeFileContentExpected {
-						fmt.Printf("Expected '%s', Got '%s'\n", volumeFileContentExpected, body)
-						res.WriteHeader(http.StatusInternalServerError)
-						res.Write(nil)
-						return
-					}
-				}
+			if string(body) != volumeFileContentExpected {
+				fmt.Printf("Expected '%s', Got '%s'\n", volumeFileContentExpected, body)
+				res.WriteHeader(http.StatusInternalServerError)
+				res.Write(nil)
+				return
 			}
 			return
 		}