|
@ -3,7 +3,6 @@ package filer |
|
|
import ( |
|
|
import ( |
|
|
"bytes" |
|
|
"bytes" |
|
|
"fmt" |
|
|
"fmt" |
|
|
"golang.org/x/sync/errgroup" |
|
|
|
|
|
"io" |
|
|
"io" |
|
|
"math" |
|
|
"math" |
|
|
"strings" |
|
|
"strings" |
|
@ -14,7 +13,7 @@ import ( |
|
|
"github.com/chrislusf/seaweedfs/weed/wdclient" |
|
|
"github.com/chrislusf/seaweedfs/weed/wdclient" |
|
|
) |
|
|
) |
|
|
|
|
|
|
|
|
func StreamContent(masterClient wdclient.HasLookupFileIdFunction, w io.Writer, chunks []*filer_pb.FileChunk, offset int64, size int64, isCheck bool) error { |
|
|
|
|
|
|
|
|
func StreamContent(masterClient wdclient.HasLookupFileIdFunction, w io.Writer, chunks []*filer_pb.FileChunk, offset int64, size int64) error { |
|
|
|
|
|
|
|
|
glog.V(9).Infof("start to stream content for chunks: %+v\n", chunks) |
|
|
glog.V(9).Infof("start to stream content for chunks: %+v\n", chunks) |
|
|
chunkViews := ViewFromChunks(masterClient.GetLookupFileIdFunction(), chunks, offset, size) |
|
|
chunkViews := ViewFromChunks(masterClient.GetLookupFileIdFunction(), chunks, offset, size) |
|
@ -34,16 +33,6 @@ func StreamContent(masterClient wdclient.HasLookupFileIdFunction, w io.Writer, c |
|
|
fileId2Url[chunkView.FileId] = urlStrings |
|
|
fileId2Url[chunkView.FileId] = urlStrings |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
if isCheck { |
|
|
|
|
|
// Pre-check all chunkViews urls
|
|
|
|
|
|
gErr := new(errgroup.Group) |
|
|
|
|
|
CheckAllChunkViews(chunkViews, &fileId2Url, gErr) |
|
|
|
|
|
if err := gErr.Wait(); err != nil { |
|
|
|
|
|
glog.Errorf("check all chunks: %v", err) |
|
|
|
|
|
return fmt.Errorf("check all chunks: %v", err) |
|
|
|
|
|
} |
|
|
|
|
|
return nil |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
for _, chunkView := range chunkViews { |
|
|
for _, chunkView := range chunkViews { |
|
|
|
|
|
|
|
@ -53,7 +42,6 @@ func StreamContent(masterClient wdclient.HasLookupFileIdFunction, w io.Writer, c |
|
|
glog.Errorf("read chunk: %v", err) |
|
|
glog.Errorf("read chunk: %v", err) |
|
|
return fmt.Errorf("read chunk: %v", err) |
|
|
return fmt.Errorf("read chunk: %v", err) |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
_, err = w.Write(data) |
|
|
_, err = w.Write(data) |
|
|
if err != nil { |
|
|
if err != nil { |
|
|
glog.Errorf("write chunk: %v", err) |
|
|
glog.Errorf("write chunk: %v", err) |
|
@ -65,17 +53,6 @@ func StreamContent(masterClient wdclient.HasLookupFileIdFunction, w io.Writer, c |
|
|
|
|
|
|
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
func CheckAllChunkViews(chunkViews []*ChunkView, fileId2Url *map[string][]string, gErr *errgroup.Group) { |
|
|
|
|
|
for _, chunkView := range chunkViews { |
|
|
|
|
|
urlStrings := (*fileId2Url)[chunkView.FileId] |
|
|
|
|
|
glog.V(9).Infof("Check chunk: %+v\n url: %v", chunkView, urlStrings) |
|
|
|
|
|
gErr.Go(func() error { |
|
|
|
|
|
_, err := retriedFetchChunkData(urlStrings, chunkView.CipherKey, chunkView.IsGzipped, chunkView.IsFullChunk(), chunkView.Offset, int(chunkView.Size)) |
|
|
|
|
|
return err |
|
|
|
|
|
}) |
|
|
|
|
|
} |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
// ---------------- ReadAllReader ----------------------------------
|
|
|
// ---------------- ReadAllReader ----------------------------------
|
|
|
|
|
|
|
|
|
func ReadAll(masterClient *wdclient.MasterClient, chunks []*filer_pb.FileChunk) ([]byte, error) { |
|
|
func ReadAll(masterClient *wdclient.MasterClient, chunks []*filer_pb.FileChunk) ([]byte, error) { |
|
|