You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
114 lines
3.2 KiB
114 lines
3.2 KiB
package weed_server
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/seaweedfs/seaweedfs/weed/glog"
|
|
"github.com/seaweedfs/seaweedfs/weed/pb"
|
|
"github.com/seaweedfs/seaweedfs/weed/pb/filer_pb"
|
|
"time"
|
|
)
|
|
|
|
// Lock is a grpc handler to handle FilerServer's LockRequest
|
|
func (fs *FilerServer) Lock(ctx context.Context, req *filer_pb.LockRequest) (resp *filer_pb.LockResponse, err error) {
|
|
|
|
resp = &filer_pb.LockResponse{}
|
|
|
|
var movedTo pb.ServerAddress
|
|
expiredAtNs := time.Now().Add(time.Duration(req.SecondsToLock) * time.Second).UnixNano()
|
|
resp.RenewToken, movedTo, err = fs.filer.Dlm.Lock(fs.option.Host, req.Name, expiredAtNs, req.RenewToken)
|
|
if !req.IsMoved && movedTo != "" {
|
|
err = pb.WithFilerClient(false, 0, movedTo, fs.grpcDialOption, func(client filer_pb.SeaweedFilerClient) error {
|
|
secondResp, err := client.Lock(context.Background(), &filer_pb.LockRequest{
|
|
Name: req.Name,
|
|
SecondsToLock: req.SecondsToLock,
|
|
RenewToken: req.RenewToken,
|
|
IsMoved: true,
|
|
})
|
|
if err == nil {
|
|
resp.RenewToken = secondResp.RenewToken
|
|
} else {
|
|
resp.Error = secondResp.Error
|
|
}
|
|
return err
|
|
})
|
|
}
|
|
|
|
if err != nil {
|
|
resp.Error = fmt.Sprintf("%v", err)
|
|
}
|
|
if movedTo != "" {
|
|
resp.MovedTo = string(movedTo)
|
|
}
|
|
|
|
return resp, nil
|
|
}
|
|
|
|
// Unlock is a grpc handler to handle FilerServer's UnlockRequest
|
|
func (fs *FilerServer) Unlock(ctx context.Context, req *filer_pb.UnlockRequest) (resp *filer_pb.UnlockResponse, err error) {
|
|
|
|
resp = &filer_pb.UnlockResponse{}
|
|
|
|
var movedTo pb.ServerAddress
|
|
movedTo, err = fs.filer.Dlm.Unlock(fs.option.Host, req.Name, req.RenewToken)
|
|
|
|
if !req.IsMoved && movedTo != "" {
|
|
err = pb.WithFilerClient(false, 0, movedTo, fs.grpcDialOption, func(client filer_pb.SeaweedFilerClient) error {
|
|
secondResp, err := client.Unlock(context.Background(), &filer_pb.UnlockRequest{
|
|
Name: req.Name,
|
|
RenewToken: req.RenewToken,
|
|
IsMoved: true,
|
|
})
|
|
resp.Error = secondResp.Error
|
|
return err
|
|
})
|
|
}
|
|
|
|
if err != nil {
|
|
resp.Error = fmt.Sprintf("%v", err)
|
|
}
|
|
if movedTo != "" {
|
|
resp.MovedTo = string(movedTo)
|
|
}
|
|
|
|
return resp, nil
|
|
|
|
}
|
|
|
|
// TransferLocks is a grpc handler to handle FilerServer's TransferLocksRequest
|
|
func (fs *FilerServer) TransferLocks(ctx context.Context, req *filer_pb.TransferLocksRequest) (*filer_pb.TransferLocksResponse, error) {
|
|
|
|
for _, lock := range req.Locks {
|
|
fs.filer.Dlm.InsertLock(lock.Name, lock.ExpiredAtNs, lock.RenewToken)
|
|
}
|
|
|
|
return &filer_pb.TransferLocksResponse{}, nil
|
|
|
|
}
|
|
|
|
func (fs *FilerServer) OnDlmChangeSnapshot(snapshot []pb.ServerAddress) {
|
|
locks := fs.filer.Dlm.SelectNotOwnedLocks(fs.option.Host, snapshot)
|
|
if len(locks) == 0 {
|
|
return
|
|
}
|
|
|
|
for _, lock := range locks {
|
|
server := fs.filer.Dlm.CalculateTargetServer(lock.Key, snapshot)
|
|
if err := pb.WithFilerClient(false, 0, server, fs.grpcDialOption, func(client filer_pb.SeaweedFilerClient) error {
|
|
_, err := client.TransferLocks(context.Background(), &filer_pb.TransferLocksRequest{
|
|
Locks: []*filer_pb.Lock{
|
|
{
|
|
Name: lock.Key,
|
|
RenewToken: lock.Token,
|
|
ExpiredAtNs: lock.ExpiredAtNs,
|
|
},
|
|
},
|
|
})
|
|
return err
|
|
}); err != nil {
|
|
// it may not be worth retrying, since the lock may have expired
|
|
glog.Errorf("transfer lock %v to %v: %v", lock.Key, server, err)
|
|
}
|
|
}
|
|
|
|
}
|