mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-01-19 02:48:24 +00:00
102 lines
2.8 KiB
Go
102 lines
2.8 KiB
Go
package weed_server
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/chrislusf/seaweedfs/weed/glog"
|
|
"github.com/chrislusf/seaweedfs/weed/pb/volume_server_pb"
|
|
"github.com/chrislusf/seaweedfs/weed/storage"
|
|
"github.com/chrislusf/seaweedfs/weed/storage/types"
|
|
)
|
|
|
|
func (vs *VolumeServer) VolumeSyncStatus(ctx context.Context, req *volume_server_pb.VolumeSyncStatusRequest) (*volume_server_pb.VolumeSyncStatusResponse, error) {
|
|
|
|
v := vs.store.GetVolume(storage.VolumeId(req.VolumeId))
|
|
if v == nil {
|
|
return nil, fmt.Errorf("not found volume id %d", req.VolumeId)
|
|
}
|
|
|
|
resp := v.GetVolumeSyncStatus()
|
|
|
|
glog.V(2).Infof("volume sync status %d", req.VolumeId)
|
|
|
|
return resp, nil
|
|
|
|
}
|
|
|
|
func (vs *VolumeServer) VolumeSyncIndex(req *volume_server_pb.VolumeSyncIndexRequest, stream volume_server_pb.VolumeServer_VolumeSyncIndexServer) error {
|
|
|
|
v := vs.store.GetVolume(storage.VolumeId(req.VolumeId))
|
|
if v == nil {
|
|
return fmt.Errorf("not found volume id %d", req.VolumeId)
|
|
}
|
|
|
|
content, err := v.IndexFileContent()
|
|
|
|
if err != nil {
|
|
glog.Errorf("sync volume %d index: %v", req.VolumeId, err)
|
|
} else {
|
|
glog.V(2).Infof("sync volume %d index", req.VolumeId)
|
|
}
|
|
|
|
const blockSizeLimit = 1024 * 1024 * 2
|
|
for i := 0; i < len(content); i += blockSizeLimit {
|
|
blockSize := len(content) - i
|
|
if blockSize > blockSizeLimit {
|
|
blockSize = blockSizeLimit
|
|
}
|
|
resp := &volume_server_pb.VolumeSyncIndexResponse{}
|
|
resp.IndexFileContent = content[i : i+blockSize]
|
|
stream.Send(resp)
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
func (vs *VolumeServer) VolumeSyncData(req *volume_server_pb.VolumeSyncDataRequest, stream volume_server_pb.VolumeServer_VolumeSyncDataServer) error {
|
|
|
|
v := vs.store.GetVolume(storage.VolumeId(req.VolumeId))
|
|
if v == nil {
|
|
return fmt.Errorf("not found volume id %d", req.VolumeId)
|
|
}
|
|
|
|
if uint32(v.SuperBlock.CompactRevision) != req.Revision {
|
|
return fmt.Errorf("requested volume revision is %d, but current revision is %d", req.Revision, v.SuperBlock.CompactRevision)
|
|
}
|
|
|
|
content, err := storage.ReadNeedleBlob(v.DataFile(), int64(req.Offset)*types.NeedlePaddingSize, req.Size, v.Version())
|
|
if err != nil {
|
|
return fmt.Errorf("read offset:%d size:%d", req.Offset, req.Size)
|
|
}
|
|
|
|
id, err := types.ParseNeedleId(req.NeedleId)
|
|
if err != nil {
|
|
return fmt.Errorf("parsing needle id %s: %v", req.NeedleId, err)
|
|
}
|
|
n := new(storage.Needle)
|
|
n.ParseNeedleHeader(content)
|
|
if id != n.Id {
|
|
return fmt.Errorf("expected file entry id %d, but found %d", id, n.Id)
|
|
}
|
|
|
|
if err != nil {
|
|
glog.Errorf("sync volume %d data: %v", req.VolumeId, err)
|
|
}
|
|
|
|
const blockSizeLimit = 1024 * 1024 * 2
|
|
for i := 0; i < len(content); i += blockSizeLimit {
|
|
blockSize := len(content) - i
|
|
if blockSize > blockSizeLimit {
|
|
blockSize = blockSizeLimit
|
|
}
|
|
resp := &volume_server_pb.VolumeSyncDataResponse{}
|
|
resp.FileContent = content[i : i+blockSize]
|
|
stream.Send(resp)
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|