mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-12-23 00:27:55 +08:00
86d92a42b4
* Added global http client * Added Do func for global http client * Changed the code to use the global http client * Fix http client in volume uploader * Fixed pkg name * Fixed http util funcs * Fixed http client for bench_filer_upload * Fixed http client for stress_filer_upload * Fixed http client for filer_server_handlers_proxy * Fixed http client for command_fs_merge_volumes * Fixed http client for command_fs_merge_volumes and command_volume_fsck * Fixed http client for s3api_server * Added init global client for main funcs * Rename global_client to client * Changed: - fixed NewHttpClient; - added CheckIsHttpsClientEnabled func - updated security.toml in scaffold * Reduce the visibility of some functions in the util/http/client pkg * Added the loadSecurityConfig function * Use util.LoadSecurityConfiguration() in NewHttpClient func
44 lines
1.1 KiB
Go
44 lines
1.1 KiB
Go
package repl_util
|
|
|
|
import (
|
|
"github.com/seaweedfs/seaweedfs/weed/filer"
|
|
"github.com/seaweedfs/seaweedfs/weed/glog"
|
|
"github.com/seaweedfs/seaweedfs/weed/replication/source"
|
|
util_http "github.com/seaweedfs/seaweedfs/weed/util/http"
|
|
)
|
|
|
|
func CopyFromChunkViews(chunkViews *filer.IntervalList[*filer.ChunkView], filerSource *source.FilerSource, writeFunc func(data []byte) error) error {
|
|
|
|
for x := chunkViews.Front(); x != nil; x = x.Next {
|
|
chunk := x.Value
|
|
|
|
fileUrls, err := filerSource.LookupFileId(chunk.FileId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var writeErr error
|
|
var shouldRetry bool
|
|
|
|
for _, fileUrl := range fileUrls {
|
|
shouldRetry, err = util_http.ReadUrlAsStream(fileUrl, chunk.CipherKey, chunk.IsGzipped, chunk.IsFullChunk(), chunk.OffsetInChunk, int(chunk.ViewSize), func(data []byte) {
|
|
writeErr = writeFunc(data)
|
|
})
|
|
if err != nil {
|
|
glog.V(1).Infof("read from %s: %v", fileUrl, err)
|
|
} else if writeErr != nil {
|
|
glog.V(1).Infof("copy from %s: %v", fileUrl, writeErr)
|
|
} else {
|
|
break
|
|
}
|
|
}
|
|
if shouldRetry && err != nil {
|
|
return err
|
|
}
|
|
if writeErr != nil {
|
|
return writeErr
|
|
}
|
|
}
|
|
return nil
|
|
}
|