seaweedfs/go/operation/upload_content.go

73 lines
1.7 KiB
Go
Raw Normal View History

2012-09-21 00:58:29 +00:00
package operation
import (
2012-09-26 10:27:10 +00:00
"bytes"
"encoding/json"
2013-01-17 08:56:56 +00:00
"errors"
"fmt"
2012-09-26 10:27:10 +00:00
"io"
"io/ioutil"
2013-01-17 08:56:56 +00:00
"log"
"mime"
2012-09-26 10:27:10 +00:00
"mime/multipart"
"net/http"
"net/textproto"
"path/filepath"
"strings"
2012-09-21 00:58:29 +00:00
)
type UploadResult struct {
2013-01-17 08:56:56 +00:00
Size int
Error string
2012-09-21 00:58:29 +00:00
}
var fileNameEscaper = strings.NewReplacer("\\", "\\\\", "\"", "\\\"")
2013-07-29 17:09:36 +00:00
func Upload(uploadUrl string, filename string, reader io.Reader, isGzipped bool, mtype string) (*UploadResult, error) {
2012-09-26 10:27:10 +00:00
body_buf := bytes.NewBufferString("")
body_writer := multipart.NewWriter(body_buf)
h := make(textproto.MIMEHeader)
h.Set("Content-Disposition", fmt.Sprintf(`form-data; name="file"; filename="%s"`, fileNameEscaper.Replace(filename)))
2013-07-29 17:09:36 +00:00
if mtype == "" {
mtype = mime.TypeByExtension(strings.ToLower(filepath.Ext(filename)))
}
h.Set("Content-Type", mtype)
if isGzipped {
h.Set("Content-Encoding", "gzip")
}
file_writer, err := body_writer.CreatePart(h)
2013-02-27 06:54:22 +00:00
if err != nil {
log.Println("error creating form file", err)
return nil, err
}
if _, err = io.Copy(file_writer, reader); err != nil {
log.Println("error copying data", err)
return nil, err
}
2013-07-10 07:27:01 +00:00
content_type := body_writer.FormDataContentType()
2013-02-27 06:54:22 +00:00
if err = body_writer.Close(); err != nil {
log.Println("error closing body", err)
return nil, err
}
2012-09-26 10:27:10 +00:00
resp, err := http.Post(uploadUrl, content_type, body_buf)
if err != nil {
2013-01-17 08:56:56 +00:00
log.Println("failing to upload to", uploadUrl)
2012-09-26 10:27:10 +00:00
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 {
2013-01-17 08:56:56 +00:00
log.Println("failing to read upload resonse", uploadUrl, resp_body)
return nil, err
2012-09-26 10:27:10 +00:00
}
2013-01-17 08:56:56 +00:00
if ret.Error != "" {
return nil, errors.New(ret.Error)
2012-09-26 21:28:46 +00:00
}
2012-09-26 10:27:10 +00:00
return &ret, nil
2012-09-21 00:58:29 +00:00
}