seaweedfs/weed/server/filer_server_handlers_write_upload.go

175 lines
5.2 KiB
Go
Raw Normal View History

2021-03-31 04:07:34 +00:00
package weed_server
import (
2021-04-28 17:28:05 +00:00
"bytes"
2021-03-31 04:07:34 +00:00
"crypto/md5"
"hash"
"io"
"io/ioutil"
"net/http"
2021-06-07 03:23:36 +00:00
"sort"
2021-03-31 04:07:34 +00:00
"strings"
"sync"
2021-06-07 06:05:17 +00:00
"sync/atomic"
2021-03-31 04:07:34 +00:00
"time"
"github.com/chrislusf/seaweedfs/weed/filer"
"github.com/chrislusf/seaweedfs/weed/glog"
"github.com/chrislusf/seaweedfs/weed/operation"
"github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
"github.com/chrislusf/seaweedfs/weed/security"
"github.com/chrislusf/seaweedfs/weed/stats"
"github.com/chrislusf/seaweedfs/weed/util"
)
var bufPool = sync.Pool{
New: func() interface{} {
return new(bytes.Buffer)
},
}
func (fs *FilerServer) uploadReaderToChunks(w http.ResponseWriter, r *http.Request, reader io.Reader, chunkSize int32, fileName, contentType string, contentLength int64, so *operation.StorageOption) (fileChunks []*filer_pb.FileChunk, md5Hash hash.Hash, chunkOffset int64, uploadErr error, smallContent []byte) {
2021-03-31 04:07:34 +00:00
md5Hash = md5.New()
2021-03-31 04:07:34 +00:00
var partReader = ioutil.NopCloser(io.TeeReader(reader, md5Hash))
2021-06-07 03:23:36 +00:00
var wg sync.WaitGroup
2021-06-07 06:05:17 +00:00
var bytesBufferCounter int64
bytesBufferLimitCond := sync.NewCond(new(sync.Mutex))
2021-06-07 19:04:50 +00:00
var fileChunksLock sync.Mutex
for {
2021-06-07 03:23:36 +00:00
2021-06-07 06:05:17 +00:00
// need to throttle used byte buffer
bytesBufferLimitCond.L.Lock()
for atomic.LoadInt64(&bytesBufferCounter) >= 4 {
glog.V(4).Infof("waiting for byte buffer %d", bytesBufferCounter)
bytesBufferLimitCond.Wait()
}
atomic.AddInt64(&bytesBufferCounter, 1)
bytesBufferLimitCond.L.Unlock()
2021-06-07 03:23:36 +00:00
bytesBuffer := bufPool.Get().(*bytes.Buffer)
2021-06-07 06:05:17 +00:00
glog.V(4).Infof("received byte buffer %d", bytesBufferCounter)
2021-06-07 03:23:36 +00:00
limitedReader := io.LimitReader(partReader, int64(chunkSize))
2021-03-31 04:07:34 +00:00
bytesBuffer.Reset()
dataSize, err := bytesBuffer.ReadFrom(limitedReader)
// data, err := ioutil.ReadAll(limitedReader)
2021-06-07 03:23:36 +00:00
if err != nil || dataSize == 0 {
2021-06-07 06:05:17 +00:00
bufPool.Put(bytesBuffer)
atomic.AddInt64(&bytesBufferCounter, -1)
bytesBufferLimitCond.Signal()
break
2021-03-31 04:07:34 +00:00
}
if chunkOffset == 0 && !isAppend(r) {
if dataSize < fs.option.SaveToFilerLimit || strings.HasPrefix(r.URL.Path, filer.DirectoryEtcRoot) {
chunkOffset += dataSize
smallContent = make([]byte, dataSize)
2021-06-07 03:57:03 +00:00
bytesBuffer.Read(smallContent)
2021-06-07 06:05:17 +00:00
bufPool.Put(bytesBuffer)
atomic.AddInt64(&bytesBufferCounter, -1)
bytesBufferLimitCond.Signal()
break
}
}
2021-03-31 04:07:34 +00:00
2021-06-07 03:23:36 +00:00
wg.Add(1)
go func(offset int64) {
2021-06-07 06:05:17 +00:00
defer func() {
bufPool.Put(bytesBuffer)
atomic.AddInt64(&bytesBufferCounter, -1)
bytesBufferLimitCond.Signal()
wg.Done()
}()
2021-03-31 04:07:34 +00:00
chunk, toChunkErr := fs.dataToChunk(fileName, contentType, bytesBuffer.Bytes(), offset, so)
2021-06-07 03:23:36 +00:00
if toChunkErr != nil {
uploadErr = toChunkErr
}
if chunk != nil {
2021-06-07 19:04:50 +00:00
fileChunksLock.Lock()
2021-06-07 03:23:36 +00:00
fileChunks = append(fileChunks, chunk)
2021-06-07 19:04:50 +00:00
fileChunksLock.Unlock()
2021-06-07 03:23:36 +00:00
glog.V(4).Infof("uploaded %s chunk %d to %s [%d,%d)", fileName, len(fileChunks), chunk.FileId, offset, offset+int64(chunk.Size))
}
}(chunkOffset)
2021-03-31 04:07:34 +00:00
// reset variables for the next chunk
2021-06-07 03:23:36 +00:00
chunkOffset = chunkOffset + dataSize
2021-03-31 04:07:34 +00:00
// if last chunk was not at full chunk size, but already exhausted the reader
2021-06-07 03:23:36 +00:00
if dataSize < int64(chunkSize) {
break
}
}
2021-06-07 03:23:36 +00:00
wg.Wait()
if uploadErr != nil {
return nil, md5Hash, 0, uploadErr, nil
}
sort.Slice(fileChunks, func(i, j int) bool {
return fileChunks[i].Offset < fileChunks[j].Offset
})
return fileChunks, md5Hash, chunkOffset, nil, smallContent
2021-03-31 04:07:34 +00:00
}
2021-06-07 01:43:04 +00:00
func (fs *FilerServer) doUpload(urlLocation string, limitedReader io.Reader, fileName string, contentType string, pairMap map[string]string, auth security.EncodedJwt) (*operation.UploadResult, error, []byte) {
2021-03-31 04:07:34 +00:00
stats.FilerRequestCounter.WithLabelValues("chunkUpload").Inc()
start := time.Now()
defer func() {
stats.FilerRequestHistogram.WithLabelValues("chunkUpload").Observe(time.Since(start).Seconds())
}()
uploadResult, err, data := operation.Upload(urlLocation, fileName, fs.option.Cipher, limitedReader, false, contentType, pairMap, auth)
if uploadResult != nil && uploadResult.RetryCount > 0 {
stats.FilerRequestCounter.WithLabelValues("chunkUploadRetry").Add(float64(uploadResult.RetryCount))
}
return uploadResult, err, data
}
2021-06-07 01:43:04 +00:00
func (fs *FilerServer) dataToChunk(fileName, contentType string, data []byte, chunkOffset int64, so *operation.StorageOption) (*filer_pb.FileChunk, error) {
2021-06-07 01:43:04 +00:00
dataReader := util.NewBytesReader(data)
// retry to assign a different file id
var fileId, urlLocation string
var auth security.EncodedJwt
var uploadErr error
var uploadResult *operation.UploadResult
for i := 0; i < 3; i++ {
// assign one file id for one chunk
fileId, urlLocation, auth, uploadErr = fs.assignNewFileInfo(so)
if uploadErr != nil {
glog.V(4).Infof("retry later due to assign error: %v", uploadErr)
time.Sleep(time.Duration(i+1) * 251 * time.Millisecond)
continue
}
// upload the chunk to the volume server
uploadResult, uploadErr, _ = fs.doUpload(urlLocation, dataReader, fileName, contentType, nil, auth)
if uploadErr != nil {
glog.V(4).Infof("retry later due to upload error: %v", uploadErr)
time.Sleep(time.Duration(i+1) * 251 * time.Millisecond)
continue
}
break
}
if uploadErr != nil {
glog.Errorf("upload error: %v", uploadErr)
return nil, uploadErr
}
// if last chunk exhausted the reader exactly at the border
if uploadResult.Size == 0 {
return nil, nil
}
return uploadResult.ToPbFileChunk(fileId, chunkOffset), nil
}