seaweedfs/weed/server/filer_grpc_server.go

456 lines
12 KiB
Go
Raw Normal View History

2018-05-08 16:59:43 +08:00
package weed_server
import (
"context"
2018-05-28 02:52:26 +08:00
"fmt"
2018-05-27 14:59:56 +08:00
"os"
"path/filepath"
"strconv"
"strings"
2018-05-28 02:52:26 +08:00
"time"
2018-05-27 14:59:56 +08:00
2018-05-14 14:56:16 +08:00
"github.com/chrislusf/seaweedfs/weed/filer2"
"github.com/chrislusf/seaweedfs/weed/glog"
2018-05-28 02:52:26 +08:00
"github.com/chrislusf/seaweedfs/weed/operation"
"github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
"github.com/chrislusf/seaweedfs/weed/pb/master_pb"
2020-03-23 15:01:34 +08:00
"github.com/chrislusf/seaweedfs/weed/util"
2018-05-08 16:59:43 +08:00
)
2018-05-10 14:18:02 +08:00
func (fs *FilerServer) LookupDirectoryEntry(ctx context.Context, req *filer_pb.LookupDirectoryEntryRequest) (*filer_pb.LookupDirectoryEntryResponse, error) {
2018-05-08 16:59:43 +08:00
glog.V(4).Infof("LookupDirectoryEntry %s", filepath.Join(req.Directory, req.Name))
2020-04-06 04:11:43 +08:00
entry, err := fs.filer.FindEntry(ctx, util.JoinPath(req.Directory, req.Name))
2020-03-08 09:01:39 +08:00
if err == filer_pb.ErrNotFound {
return &filer_pb.LookupDirectoryEntryResponse{}, err
2020-01-25 10:07:34 +08:00
}
2018-05-08 16:59:43 +08:00
if err != nil {
2020-01-25 09:26:03 +08:00
glog.V(3).Infof("LookupDirectoryEntry %s: %+v, ", filepath.Join(req.Directory, req.Name), err)
return nil, err
2018-05-08 16:59:43 +08:00
}
2018-05-10 14:18:02 +08:00
return &filer_pb.LookupDirectoryEntryResponse{
Entry: &filer_pb.Entry{
2018-05-08 16:59:43 +08:00
Name: req.Name,
2018-05-14 14:56:16 +08:00
IsDirectory: entry.IsDirectory(),
Attributes: filer2.EntryAttributeToPb(entry),
Chunks: entry.Chunks,
2019-12-18 13:39:48 +08:00
Extended: entry.Extended,
2018-05-08 16:59:43 +08:00
},
}, nil
}
2019-12-13 16:22:37 +08:00
func (fs *FilerServer) ListEntries(req *filer_pb.ListEntriesRequest, stream filer_pb.SeaweedFiler_ListEntriesServer) error {
2018-05-08 16:59:43 +08:00
glog.V(4).Infof("ListEntries %v", req)
2018-07-22 16:14:36 +08:00
limit := int(req.Limit)
if limit == 0 {
limit = fs.option.DirListingLimit
2018-05-08 16:59:43 +08:00
}
paginationLimit := filer2.PaginationSize
if limit < paginationLimit {
paginationLimit = limit
}
2018-07-22 16:14:36 +08:00
lastFileName := req.StartFromFileName
includeLastFile := req.InclusiveStartFrom
for limit > 0 {
2020-03-23 15:01:34 +08:00
entries, err := fs.filer.ListDirectoryEntries(stream.Context(), util.FullPath(req.Directory), lastFileName, includeLastFile, paginationLimit)
2018-07-22 16:14:36 +08:00
if err != nil {
2019-12-13 16:22:37 +08:00
return err
2018-07-22 16:14:36 +08:00
}
if len(entries) == 0 {
2019-12-13 16:22:37 +08:00
return nil
2018-07-22 16:14:36 +08:00
}
includeLastFile = false
for _, entry := range entries {
lastFileName = entry.Name()
if req.Prefix != "" {
if !strings.HasPrefix(entry.Name(), req.Prefix) {
continue
}
}
2019-12-13 16:22:37 +08:00
if err := stream.Send(&filer_pb.ListEntriesResponse{
Entry: &filer_pb.Entry{
Name: entry.Name(),
IsDirectory: entry.IsDirectory(),
Chunks: entry.Chunks,
Attributes: filer2.EntryAttributeToPb(entry),
2019-12-18 13:39:48 +08:00
Extended: entry.Extended,
2019-12-13 16:22:37 +08:00
},
}); err != nil {
return err
}
2018-07-22 16:14:36 +08:00
limit--
if limit == 0 {
2019-12-13 16:22:37 +08:00
return nil
}
2018-07-22 16:14:36 +08:00
}
if len(entries) < paginationLimit {
break
}
2018-05-08 16:59:43 +08:00
}
2019-12-13 16:22:37 +08:00
return nil
2018-05-08 16:59:43 +08:00
}
func (fs *FilerServer) LookupVolume(ctx context.Context, req *filer_pb.LookupVolumeRequest) (*filer_pb.LookupVolumeResponse, error) {
2018-05-08 16:59:43 +08:00
resp := &filer_pb.LookupVolumeResponse{
LocationsMap: make(map[string]*filer_pb.Locations),
2018-05-08 16:59:43 +08:00
}
for _, vidString := range req.VolumeIds {
vid, err := strconv.Atoi(vidString)
if err != nil {
2018-07-29 17:25:03 +08:00
glog.V(1).Infof("Unknown volume id %d", vid)
return nil, err
}
var locs []*filer_pb.Location
locations, found := fs.filer.MasterClient.GetLocations(uint32(vid))
if !found {
continue
}
for _, loc := range locations {
locs = append(locs, &filer_pb.Location{
Url: loc.Url,
PublicUrl: loc.PublicUrl,
})
}
resp.LocationsMap[vidString] = &filer_pb.Locations{
Locations: locs,
}
}
return resp, nil
2018-05-08 16:59:43 +08:00
}
func (fs *FilerServer) CreateEntry(ctx context.Context, req *filer_pb.CreateEntryRequest) (resp *filer_pb.CreateEntryResponse, err error) {
2018-09-23 13:11:37 +08:00
glog.V(4).Infof("CreateEntry %v", req)
resp = &filer_pb.CreateEntryResponse{}
2018-09-23 13:11:37 +08:00
chunks, garbages := filer2.CompactFileChunks(req.Entry.Chunks)
2018-12-03 14:57:59 +08:00
if req.Entry.Attributes == nil {
2020-01-25 09:55:39 +08:00
glog.V(3).Infof("CreateEntry %s: nil attributes", filepath.Join(req.Directory, req.Entry.Name))
resp.Error = fmt.Sprintf("can not create entry with empty attributes")
return
2018-12-03 14:57:59 +08:00
}
createErr := fs.filer.CreateEntry(ctx, &filer2.Entry{
2020-04-06 04:11:43 +08:00
FullPath: util.JoinPath(req.Directory, req.Entry.Name),
Attr: filer2.PbToEntryAttribute(req.Entry.Attributes),
2018-09-23 13:11:37 +08:00
Chunks: chunks,
}, req.OExcl)
if createErr == nil {
fs.filer.DeleteChunks(garbages)
2020-01-25 09:55:39 +08:00
} else {
glog.V(3).Infof("CreateEntry %s: %v", filepath.Join(req.Directory, req.Entry.Name), createErr)
resp.Error = createErr.Error()
}
return
}
func (fs *FilerServer) UpdateEntry(ctx context.Context, req *filer_pb.UpdateEntryRequest) (*filer_pb.UpdateEntryResponse, error) {
glog.V(4).Infof("UpdateEntry %v", req)
2020-04-06 04:11:43 +08:00
fullpath := util.Join(req.Directory, req.Entry.Name)
2020-03-23 15:01:34 +08:00
entry, err := fs.filer.FindEntry(ctx, util.FullPath(fullpath))
if err != nil {
return &filer_pb.UpdateEntryResponse{}, fmt.Errorf("not found %s: %v", fullpath, err)
}
// remove old chunks if not included in the new ones
2019-06-23 11:04:56 +08:00
unusedChunks := filer2.MinusChunks(entry.Chunks, req.Entry.Chunks)
chunks, garbages := filer2.CompactFileChunks(req.Entry.Chunks)
2018-05-23 18:08:46 +08:00
newEntry := &filer2.Entry{
2020-04-06 04:11:43 +08:00
FullPath: util.JoinPath(req.Directory, req.Entry.Name),
2018-05-23 18:08:46 +08:00
Attr: entry.Attr,
2019-12-18 13:39:48 +08:00
Extended: req.Entry.Extended,
2018-05-23 18:08:46 +08:00
Chunks: chunks,
}
2019-12-18 13:39:48 +08:00
glog.V(3).Infof("updating %s: %+v, chunks %d: %v => %+v, chunks %d: %v, extended: %v => %v",
2018-05-23 18:08:46 +08:00
fullpath, entry.Attr, len(entry.Chunks), entry.Chunks,
2019-12-18 13:39:48 +08:00
req.Entry.Attributes, len(req.Entry.Chunks), req.Entry.Chunks,
entry.Extended, req.Entry.Extended)
2018-05-23 18:08:46 +08:00
if req.Entry.Attributes != nil {
if req.Entry.Attributes.Mtime != 0 {
newEntry.Attr.Mtime = time.Unix(req.Entry.Attributes.Mtime, 0)
}
if req.Entry.Attributes.FileMode != 0 {
newEntry.Attr.Mode = os.FileMode(req.Entry.Attributes.FileMode)
}
newEntry.Attr.Uid = req.Entry.Attributes.Uid
newEntry.Attr.Gid = req.Entry.Attributes.Gid
newEntry.Attr.Mime = req.Entry.Attributes.Mime
newEntry.Attr.UserName = req.Entry.Attributes.UserName
newEntry.Attr.GroupNames = req.Entry.Attributes.GroupName
2018-05-23 18:08:46 +08:00
}
2018-08-13 16:20:49 +08:00
if filer2.EqualEntry(entry, newEntry) {
return &filer_pb.UpdateEntryResponse{}, err
}
2019-03-16 06:55:34 +08:00
if err = fs.filer.UpdateEntry(ctx, entry, newEntry); err == nil {
fs.filer.DeleteChunks(unusedChunks)
fs.filer.DeleteChunks(garbages)
2020-01-25 09:55:39 +08:00
} else {
glog.V(3).Infof("UpdateEntry %s: %v", filepath.Join(req.Directory, req.Entry.Name), err)
}
2018-09-17 15:27:56 +08:00
fs.filer.NotifyUpdateEvent(entry, newEntry, true)
2018-08-13 16:20:49 +08:00
return &filer_pb.UpdateEntryResponse{}, err
}
2020-04-17 17:28:09 +08:00
func (fs *FilerServer) AppendToEntry(ctx context.Context, req *filer_pb.AppendToEntryRequest) (*filer_pb.AppendToEntryResponse, error) {
glog.V(4).Infof("AppendToEntry %v", req)
2020-04-17 17:28:09 +08:00
fullpath := util.NewFullPath(req.Directory, req.EntryName)
var offset int64 = 0
entry, err := fs.filer.FindEntry(ctx, util.FullPath(fullpath))
if err == filer_pb.ErrNotFound {
entry = &filer2.Entry{
FullPath: fullpath,
Attr: filer2.Attr{
Crtime: time.Now(),
Mtime: time.Now(),
Mode: os.FileMode(0644),
Uid: OS_UID,
Gid: OS_GID,
},
}
} else {
offset = int64(filer2.TotalSize(entry.Chunks))
}
for _, chunk := range req.Chunks {
chunk.Offset = offset
offset += int64(chunk.Size)
}
entry.Chunks = append(entry.Chunks, req.Chunks...)
err = fs.filer.CreateEntry(context.Background(), entry, false)
return &filer_pb.AppendToEntryResponse{}, err
}
2018-05-10 14:18:02 +08:00
func (fs *FilerServer) DeleteEntry(ctx context.Context, req *filer_pb.DeleteEntryRequest) (resp *filer_pb.DeleteEntryResponse, err error) {
glog.V(4).Infof("DeleteEntry %v", req)
2020-04-06 04:11:43 +08:00
err = fs.filer.DeleteEntryMetaAndData(ctx, util.JoinPath(req.Directory, req.Name), req.IsRecursive, req.IgnoreRecursiveError, req.IsDeleteData)
2020-02-26 06:38:36 +08:00
resp = &filer_pb.DeleteEntryResponse{}
if err != nil {
resp.Error = err.Error()
}
return resp, nil
}
func (fs *FilerServer) AssignVolume(ctx context.Context, req *filer_pb.AssignVolumeRequest) (resp *filer_pb.AssignVolumeResponse, err error) {
2018-06-12 14:13:33 +08:00
ttlStr := ""
if req.TtlSec > 0 {
ttlStr = strconv.Itoa(int(req.TtlSec))
}
2020-04-12 14:37:10 +08:00
collection, replication, _ := fs.detectCollection(req.ParentPath, req.Collection, req.Replication)
2018-06-12 14:13:33 +08:00
2018-07-10 15:20:50 +08:00
var altRequest *operation.VolumeAssignRequest
dataCenter := req.DataCenter
2018-07-15 05:09:45 +08:00
if dataCenter == "" {
dataCenter = fs.option.DataCenter
}
assignRequest := &operation.VolumeAssignRequest{
Count: uint64(req.Count),
Replication: replication,
Collection: collection,
2018-06-12 14:13:33 +08:00
Ttl: ttlStr,
DataCenter: dataCenter,
}
if dataCenter != "" {
2018-07-10 15:20:50 +08:00
altRequest = &operation.VolumeAssignRequest{
Count: uint64(req.Count),
Replication: replication,
Collection: collection,
Ttl: ttlStr,
DataCenter: "",
}
}
2019-02-19 04:11:52 +08:00
assignResult, err := operation.Assign(fs.filer.GetMaster(), fs.grpcDialOption, assignRequest, altRequest)
if err != nil {
2020-01-25 10:07:34 +08:00
glog.V(3).Infof("AssignVolume: %v", err)
return &filer_pb.AssignVolumeResponse{Error: fmt.Sprintf("assign volume: %v", err)}, nil
}
if assignResult.Error != "" {
2020-01-25 10:07:34 +08:00
glog.V(3).Infof("AssignVolume error: %v", assignResult.Error)
return &filer_pb.AssignVolumeResponse{Error: fmt.Sprintf("assign volume result: %v", assignResult.Error)}, nil
}
return &filer_pb.AssignVolumeResponse{
FileId: assignResult.Fid,
Count: int32(assignResult.Count),
Url: assignResult.Url,
PublicUrl: assignResult.PublicUrl,
Auth: string(assignResult.Auth),
Collection: collection,
Replication: replication,
}, nil
}
func (fs *FilerServer) DeleteCollection(ctx context.Context, req *filer_pb.DeleteCollectionRequest) (resp *filer_pb.DeleteCollectionResponse, err error) {
glog.V(4).Infof("DeleteCollection %v", req)
err = fs.filer.MasterClient.WithClient(func(client master_pb.SeaweedClient) error {
_, err := client.CollectionDelete(context.Background(), &master_pb.CollectionDeleteRequest{
2019-03-19 20:19:37 +08:00
Name: req.GetCollection(),
})
return err
})
return &filer_pb.DeleteCollectionResponse{}, err
}
func (fs *FilerServer) Statistics(ctx context.Context, req *filer_pb.StatisticsRequest) (resp *filer_pb.StatisticsResponse, err error) {
2020-03-27 13:46:49 +08:00
var output *master_pb.StatisticsResponse
err = fs.filer.MasterClient.WithClient(func(masterClient master_pb.SeaweedClient) error {
grpcResponse, grpcErr := masterClient.Statistics(context.Background(), &master_pb.StatisticsRequest{
Replication: req.Replication,
Collection: req.Collection,
Ttl: req.Ttl,
})
if grpcErr != nil {
return grpcErr
}
output = grpcResponse
return nil
})
if err != nil {
return nil, err
}
return &filer_pb.StatisticsResponse{
TotalSize: output.TotalSize,
UsedSize: output.UsedSize,
FileCount: output.FileCount,
}, nil
}
func (fs *FilerServer) GetFilerConfiguration(ctx context.Context, req *filer_pb.GetFilerConfigurationRequest) (resp *filer_pb.GetFilerConfigurationResponse, err error) {
t := &filer_pb.GetFilerConfigurationResponse{
Masters: fs.option.Masters,
Collection: fs.option.Collection,
Replication: fs.option.DefaultReplication,
MaxMb: uint32(fs.option.MaxMB),
2020-02-27 16:07:13 +08:00
DirBuckets: fs.filer.DirBucketsPath,
Cipher: fs.filer.Cipher,
}
glog.V(4).Infof("GetFilerConfiguration: %v", t)
return t, nil
}
2020-05-05 17:05:28 +08:00
func (fs *FilerServer) KeepConnected(stream filer_pb.SeaweedFiler_KeepConnectedServer) error {
req, err := stream.Recv()
if err != nil {
return err
}
clientName := fmt.Sprintf("%s:%d", req.Name, req.GrpcPort)
2020-05-08 17:47:22 +08:00
m := make(map[string]bool)
for _, tp := range req.Resources {
m[tp] = true
}
2020-05-05 17:05:28 +08:00
fs.brokersLock.Lock()
2020-05-08 17:47:22 +08:00
fs.brokers[clientName] = m
2020-05-05 17:05:28 +08:00
glog.V(0).Infof("+ broker %v", clientName)
fs.brokersLock.Unlock()
defer func() {
fs.brokersLock.Lock()
delete(fs.brokers, clientName)
glog.V(0).Infof("- broker %v: %v", clientName, err)
fs.brokersLock.Unlock()
}()
for {
if err := stream.Send(&filer_pb.KeepConnectedResponse{}); err != nil {
glog.V(0).Infof("send broker %v: %+v", clientName, err)
return err
}
// println("replied")
if _, err := stream.Recv(); err != nil {
glog.V(0).Infof("recv broker %v: %v", clientName, err)
return err
}
// println("received")
}
}
2020-05-08 17:47:22 +08:00
func (fs *FilerServer) LocateBroker(ctx context.Context, req *filer_pb.LocateBrokerRequest) (resp *filer_pb.LocateBrokerResponse, err error) {
resp = &filer_pb.LocateBrokerResponse{}
fs.brokersLock.Lock()
defer fs.brokersLock.Unlock()
var localBrokers []*filer_pb.LocateBrokerResponse_Resource
for b, m := range fs.brokers {
if _, found := m[req.Resource]; found {
resp.Found = true
resp.Resources = []*filer_pb.LocateBrokerResponse_Resource{
{
GrpcAddresses: b,
ResourceCount: int32(len(m)),
},
}
return
}
localBrokers = append(localBrokers, &filer_pb.LocateBrokerResponse_Resource{
GrpcAddresses: b,
ResourceCount: int32(len(m)),
})
}
resp.Resources = localBrokers
return resp, nil
}