seaweedfs/go/operation/upload_content.go

92 lines
2.4 KiB
Go
Raw Normal View History

2012-09-21 08:58:29 +08:00
package operation
import (
2012-09-26 18:27:10 +08:00
"bytes"
"github.com/aszxqw/weed-fs/go/glog"
2012-09-26 18:27:10 +08:00
"encoding/json"
2013-01-17 16:56:56 +08:00
"errors"
"fmt"
2012-09-26 18:27:10 +08:00
"io"
"io/ioutil"
"mime"
2012-09-26 18:27:10 +08:00
"mime/multipart"
"net/http"
"net/textproto"
"path/filepath"
"strings"
2012-09-21 08:58:29 +08:00
)
type UploadResult struct {
2014-04-16 01:01:13 +08:00
Name string `json:"name,omitempty"`
Size uint32 `json:"size,omitempty"`
Error string `json:"error,omitempty"`
2012-09-21 08:58:29 +08:00
}
2014-03-13 05:07:01 +08:00
var (
client *http.Client
)
func init() {
client = &http.Client{Transport: &http.Transport{
MaxIdleConnsPerHost: 1024,
}}
}
var fileNameEscaper = strings.NewReplacer("\\", "\\\\", "\"", "\\\"")
2013-07-30 01:09:36 +08:00
func Upload(uploadUrl string, filename string, reader io.Reader, isGzipped bool, mtype string) (*UploadResult, error) {
return upload_content(uploadUrl, func(w io.Writer) (err error) {
_, err = io.Copy(w, reader)
return
}, filename, isGzipped, mtype)
}
func upload_content(uploadUrl string, fillBufferFunction func(w io.Writer) error, filename string, isGzipped bool, mtype string) (*UploadResult, error) {
2012-09-26 18:27:10 +08: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-30 01:09:36 +08:00
if mtype == "" {
mtype = mime.TypeByExtension(strings.ToLower(filepath.Ext(filename)))
}
if mtype != "" {
h.Set("Content-Type", mtype)
}
if isGzipped {
h.Set("Content-Encoding", "gzip")
}
2014-03-13 06:17:23 +08:00
file_writer, cp_err := body_writer.CreatePart(h)
if cp_err != nil {
glog.V(0).Infoln("error creating form file", cp_err.Error())
return nil, cp_err
2013-02-27 14:54:22 +08:00
}
2014-03-13 06:17:23 +08:00
if err := fillBufferFunction(file_writer); err != nil {
glog.V(0).Infoln("error copying data", err)
2013-02-27 14:54:22 +08:00
return nil, err
}
2013-07-10 15:27:01 +08:00
content_type := body_writer.FormDataContentType()
2014-03-13 06:17:23 +08:00
if err := body_writer.Close(); err != nil {
glog.V(0).Infoln("error closing body", err)
2013-02-27 14:54:22 +08:00
return nil, err
}
2014-03-13 06:17:23 +08:00
resp, post_err := client.Post(uploadUrl, content_type, body_buf)
if post_err != nil {
glog.V(0).Infoln("failing to upload to", uploadUrl, post_err.Error())
return nil, post_err
2012-09-26 18:27:10 +08:00
}
defer resp.Body.Close()
2014-03-13 06:17:23 +08:00
resp_body, ra_err := ioutil.ReadAll(resp.Body)
if ra_err != nil {
return nil, ra_err
2012-09-26 18:27:10 +08:00
}
var ret UploadResult
2014-03-13 06:17:23 +08:00
unmarshal_err := json.Unmarshal(resp_body, &ret)
if unmarshal_err != nil {
2013-08-14 09:21:54 +08:00
glog.V(0).Infoln("failing to read upload resonse", uploadUrl, string(resp_body))
2014-03-13 06:17:23 +08:00
return nil, unmarshal_err
2012-09-26 18:27:10 +08:00
}
2013-01-17 16:56:56 +08:00
if ret.Error != "" {
return nil, errors.New(ret.Error)
2012-09-27 05:28:46 +08:00
}
2012-09-26 18:27:10 +08:00
return &ret, nil
2012-09-21 08:58:29 +08:00
}