mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-01-19 02:48:24 +00:00
41 lines
962 B
Go
41 lines
962 B
Go
|
package operation
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"encoding/json"
|
||
|
|
"mime/multipart"
|
||
|
|
"net/http"
|
||
|
|
_ "fmt"
|
||
|
|
"io"
|
||
|
|
"io/ioutil"
|
||
|
|
)
|
||
|
|
|
||
|
|
type UploadResult struct {
|
||
|
|
Size int
|
||
|
|
}
|
||
|
|
|
||
|
|
func Upload(server string, fid string, filename string, reader io.Reader) (*UploadResult, error) {
|
||
|
|
body_buf := bytes.NewBufferString("")
|
||
|
|
body_writer := multipart.NewWriter(body_buf)
|
||
|
|
file_writer, err := body_writer.CreateFormFile("file", filename)
|
||
|
|
io.Copy(file_writer, reader)
|
||
|
|
content_type := body_writer.FormDataContentType()
|
||
|
|
body_writer.Close()
|
||
|
|
resp, err := http.Post("http://"+server+"/"+fid, content_type, body_buf)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
defer resp.Body.Close()
|
||
|
|
resp_body, err := ioutil.ReadAll(resp.Body)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
var ret UploadResult
|
||
|
|
err = json.Unmarshal(resp_body, &ret)
|
||
|
|
if err != nil {
|
||
|
|
panic(err.Error())
|
||
|
|
}
|
||
|
|
//fmt.Println("Uploaded " + strconv.Itoa(ret.Size) + " Bytes to " + uploadUrl)
|
||
|
|
return &ret, nil
|
||
|
|
}
|