2012-09-21 00:58:29 +00:00
|
|
|
package operation
|
|
|
|
|
|
|
|
import (
|
2012-09-26 10:27:10 +00:00
|
|
|
"bytes"
|
2019-04-07 18:01:28 +00:00
|
|
|
"compress/flate"
|
2019-04-06 21:14:28 +00:00
|
|
|
"compress/gzip"
|
2012-09-26 10:27:10 +00:00
|
|
|
"encoding/json"
|
2013-01-17 08:56:56 +00:00
|
|
|
"errors"
|
2013-07-12 02:14:55 +00:00
|
|
|
"fmt"
|
2012-09-26 10:27:10 +00:00
|
|
|
"io"
|
|
|
|
"io/ioutil"
|
2013-07-12 02:14:55 +00:00
|
|
|
"mime"
|
2012-09-26 10:27:10 +00:00
|
|
|
"mime/multipart"
|
|
|
|
"net/http"
|
2013-07-12 02:14:55 +00:00
|
|
|
"net/textproto"
|
|
|
|
"path/filepath"
|
2013-07-12 04:16:54 +00:00
|
|
|
"strings"
|
2014-10-26 18:34:55 +00:00
|
|
|
|
2016-06-03 01:09:14 +00:00
|
|
|
"github.com/chrislusf/seaweedfs/weed/glog"
|
|
|
|
"github.com/chrislusf/seaweedfs/weed/security"
|
2012-09-21 00:58:29 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
type UploadResult struct {
|
2014-04-15 17:01:13 +00:00
|
|
|
Name string `json:"name,omitempty"`
|
|
|
|
Size uint32 `json:"size,omitempty"`
|
|
|
|
Error string `json:"error,omitempty"`
|
2019-01-02 20:57:54 +00:00
|
|
|
ETag string `json:"eTag,omitempty"`
|
2012-09-21 00:58:29 +00:00
|
|
|
}
|
|
|
|
|
2014-03-12 21:07:01 +00:00
|
|
|
var (
|
|
|
|
client *http.Client
|
|
|
|
)
|
|
|
|
|
|
|
|
func init() {
|
2019-01-18 22:14:47 +00:00
|
|
|
client = &http.Client{Transport: &http.Transport{
|
|
|
|
MaxIdleConnsPerHost: 1024,
|
|
|
|
}}
|
2014-03-12 21:07:01 +00:00
|
|
|
}
|
|
|
|
|
2013-07-16 00:26:00 +00:00
|
|
|
var fileNameEscaper = strings.NewReplacer("\\", "\\\\", "\"", "\\\"")
|
|
|
|
|
2018-11-04 05:32:21 +00:00
|
|
|
// Upload sends a POST request to a volume server to upload the content
|
2017-01-08 14:34:26 +00:00
|
|
|
func Upload(uploadUrl string, filename string, reader io.Reader, isGzipped bool, mtype string, pairMap map[string]string, jwt security.EncodedJwt) (*UploadResult, error) {
|
2019-04-06 21:14:28 +00:00
|
|
|
contentIsGzipped := isGzipped
|
|
|
|
shouldGzipNow := false
|
|
|
|
if !isGzipped {
|
|
|
|
if shouldBeZipped, iAmSure := IsGzippableFileType(filepath.Base(filename), mtype); iAmSure && shouldBeZipped {
|
|
|
|
shouldGzipNow = true
|
|
|
|
contentIsGzipped = true
|
|
|
|
}
|
|
|
|
}
|
2013-08-14 06:26:51 +00:00
|
|
|
return upload_content(uploadUrl, func(w io.Writer) (err error) {
|
2019-04-06 21:14:28 +00:00
|
|
|
if shouldGzipNow {
|
2019-04-07 18:01:28 +00:00
|
|
|
gzWriter, _ := gzip.NewWriterLevel(w, flate.BestSpeed)
|
2019-04-06 21:14:28 +00:00
|
|
|
_, err = io.Copy(gzWriter, reader)
|
|
|
|
gzWriter.Close()
|
|
|
|
} else {
|
|
|
|
_, err = io.Copy(w, reader)
|
|
|
|
}
|
2013-08-14 06:26:51 +00:00
|
|
|
return
|
2019-04-06 21:14:28 +00:00
|
|
|
}, filename, contentIsGzipped, mtype, pairMap, jwt)
|
2013-08-14 06:26:51 +00:00
|
|
|
}
|
2017-01-08 14:34:26 +00:00
|
|
|
func upload_content(uploadUrl string, fillBufferFunction func(w io.Writer) error, filename string, isGzipped bool, mtype string, pairMap map[string]string, jwt security.EncodedJwt) (*UploadResult, error) {
|
2012-09-26 10:27:10 +00:00
|
|
|
body_buf := bytes.NewBufferString("")
|
|
|
|
body_writer := multipart.NewWriter(body_buf)
|
2013-07-12 02:14:55 +00:00
|
|
|
h := make(textproto.MIMEHeader)
|
2013-07-16 00:26:00 +00:00
|
|
|
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)))
|
|
|
|
}
|
2013-08-14 06:26:51 +00:00
|
|
|
if mtype != "" {
|
|
|
|
h.Set("Content-Type", mtype)
|
|
|
|
}
|
2013-07-15 18:04:43 +00:00
|
|
|
if isGzipped {
|
|
|
|
h.Set("Content-Encoding", "gzip")
|
|
|
|
}
|
2017-01-08 01:16:29 +00:00
|
|
|
|
2014-03-12 22:17:23 +00: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 06:54:22 +00:00
|
|
|
}
|
2014-03-12 22:17:23 +00:00
|
|
|
if err := fillBufferFunction(file_writer); err != nil {
|
2013-08-09 06:57:22 +00:00
|
|
|
glog.V(0).Infoln("error copying data", err)
|
2013-02-27 06:54:22 +00:00
|
|
|
return nil, err
|
|
|
|
}
|
2013-07-10 07:27:01 +00:00
|
|
|
content_type := body_writer.FormDataContentType()
|
2014-03-12 22:17:23 +00:00
|
|
|
if err := body_writer.Close(); err != nil {
|
2013-08-09 06:57:22 +00:00
|
|
|
glog.V(0).Infoln("error closing body", err)
|
2013-02-27 06:54:22 +00:00
|
|
|
return nil, err
|
|
|
|
}
|
2017-01-08 01:16:29 +00:00
|
|
|
|
|
|
|
req, postErr := http.NewRequest("POST", uploadUrl, body_buf)
|
|
|
|
if postErr != nil {
|
|
|
|
glog.V(0).Infoln("failing to upload to", uploadUrl, postErr.Error())
|
|
|
|
return nil, postErr
|
|
|
|
}
|
|
|
|
req.Header.Set("Content-Type", content_type)
|
|
|
|
for k, v := range pairMap {
|
|
|
|
req.Header.Set(k, v)
|
|
|
|
}
|
2019-02-15 08:09:19 +00:00
|
|
|
if jwt != "" {
|
|
|
|
req.Header.Set("Authorization", "BEARER "+string(jwt))
|
|
|
|
}
|
2017-01-08 01:16:29 +00:00
|
|
|
resp, post_err := client.Do(req)
|
2014-03-12 22:17:23 +00:00
|
|
|
if post_err != nil {
|
|
|
|
glog.V(0).Infoln("failing to upload to", uploadUrl, post_err.Error())
|
|
|
|
return nil, post_err
|
2012-09-26 10:27:10 +00:00
|
|
|
}
|
|
|
|
defer resp.Body.Close()
|
2018-09-23 05:12:21 +00:00
|
|
|
etag := getEtag(resp)
|
2014-03-12 22:17:23 +00:00
|
|
|
resp_body, ra_err := ioutil.ReadAll(resp.Body)
|
|
|
|
if ra_err != nil {
|
|
|
|
return nil, ra_err
|
2012-09-26 10:27:10 +00:00
|
|
|
}
|
|
|
|
var ret UploadResult
|
2014-03-12 22:17:23 +00:00
|
|
|
unmarshal_err := json.Unmarshal(resp_body, &ret)
|
|
|
|
if unmarshal_err != nil {
|
2017-01-08 01:16:29 +00:00
|
|
|
glog.V(0).Infoln("failing to read upload response", uploadUrl, string(resp_body))
|
2014-03-12 22:17:23 +00:00
|
|
|
return nil, unmarshal_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
|
|
|
}
|
2018-09-23 05:12:21 +00:00
|
|
|
ret.ETag = etag
|
2012-09-26 10:27:10 +00:00
|
|
|
return &ret, nil
|
2012-09-21 00:58:29 +00:00
|
|
|
}
|
2018-09-23 05:12:21 +00:00
|
|
|
|
|
|
|
func getEtag(r *http.Response) (etag string) {
|
|
|
|
etag = r.Header.Get("ETag")
|
|
|
|
if strings.HasPrefix(etag, "\"") && strings.HasSuffix(etag, "\"") {
|
|
|
|
etag = etag[1 : len(etag)-1]
|
|
|
|
}
|
|
|
|
return
|
|
|
|
}
|