|
2 | 2 | package volumes |
3 | 3 |
|
4 | 4 | import ( |
5 | | - "bytes" |
6 | 5 | "encoding/json" |
7 | 6 | "fmt" |
8 | 7 | "io" |
@@ -43,21 +42,28 @@ func GetFile(c *drycc.Client, appID, volumeID, path string) (*http.Response, err |
43 | 42 |
|
44 | 43 | // Put file to an app's volume. |
45 | 44 | func PostFile(c *drycc.Client, appID, volumeID, volumePath, name string, reader io.Reader) (*http.Response, error) { |
46 | | - buffer := new(bytes.Buffer) |
47 | | - writer := multipart.NewWriter(buffer) |
48 | | - if part, err := writer.CreateFormFile("file", name); err != nil { |
49 | | - return nil, err |
50 | | - } else if _, err = io.Copy(part, reader); err != nil { |
51 | | - return nil, err |
52 | | - } |
53 | | - |
54 | | - if err := writer.WriteField("path", volumePath); err != nil { |
55 | | - return nil, err |
56 | | - } |
57 | | - writer.Close() |
| 45 | + pr, pw := io.Pipe() |
| 46 | + writer := multipart.NewWriter(pw) |
| 47 | + go func() { |
| 48 | + if err := writer.WriteField("path", volumePath); err != nil { |
| 49 | + pw.CloseWithError(err) |
| 50 | + return |
| 51 | + } |
| 52 | + part, err := writer.CreateFormFile("file", name) |
| 53 | + if err != nil { |
| 54 | + pw.CloseWithError(err) |
| 55 | + return |
| 56 | + } |
| 57 | + _, err = io.Copy(part, reader) |
| 58 | + if err != nil { |
| 59 | + pw.CloseWithError(err) |
| 60 | + return |
| 61 | + } |
| 62 | + pw.CloseWithError(writer.Close()) |
| 63 | + }() |
58 | 64 |
|
59 | 65 | u := fmt.Sprintf("/v2/apps/%s/volumes/%s/client/", appID, volumeID) |
60 | | - r, err := c.NewRequest("POST", u, buffer) |
| 66 | + r, err := c.NewRequest("POST", u, pr) |
61 | 67 | if err != nil { |
62 | 68 | return nil, err |
63 | 69 | } |
|
0 commit comments