refactoring

weed/pb/filer_pb/filer_client.go

@@ -0,0 +1,122 @@
+package filer_pb
+
+import (
+	"context"
+	"fmt"
+	"io"
+	"math"
+
+	"github.com/chrislusf/seaweedfs/weed/glog"
+	"github.com/chrislusf/seaweedfs/weed/util"
+)
+
+type FilerClient interface {
+	WithFilerClient(fn func(SeaweedFilerClient) error) error
+	AdjustedUrl(hostAndPort string) string
+}
+
+func GetEntry(filerClient FilerClient, fullFilePath util.FullPath) (entry *Entry, err error) {
+
+	dir, name := fullFilePath.DirAndName()
+
+	err = filerClient.WithFilerClient(func(client SeaweedFilerClient) error {
+
+		request := &LookupDirectoryEntryRequest{
+			Directory: dir,
+			Name:      name,
+		}
+
+		// glog.V(3).Infof("read %s request: %v", fullFilePath, request)
+		resp, err := LookupEntry(client, request)
+		if err != nil {
+			if err == ErrNotFound {
+				return nil
+			}
+			glog.V(3).Infof("read %s %v: %v", fullFilePath, resp, err)
+			return err
+		}
+
+		if resp.Entry == nil {
+			// glog.V(3).Infof("read %s entry: %v", fullFilePath, entry)
+			return nil
+		}
+
+		entry = resp.Entry
+		return nil
+	})
+
+	return
+}
+
+func ReadDirAllEntries(filerClient FilerClient, fullDirPath util.FullPath, prefix string, fn func(entry *Entry, isLast bool)) (err error) {
+
+	err = filerClient.WithFilerClient(func(client SeaweedFilerClient) error {
+
+		lastEntryName := ""
+
+		request := &ListEntriesRequest{
+			Directory:         string(fullDirPath),
+			Prefix:            prefix,
+			StartFromFileName: lastEntryName,
+			Limit:             math.MaxUint32,
+		}
+
+		glog.V(3).Infof("read directory: %v", request)
+		stream, err := client.ListEntries(context.Background(), request)
+		if err != nil {
+			return fmt.Errorf("list %s: %v", fullDirPath, err)
+		}
+
+		var prevEntry *Entry
+		for {
+			resp, recvErr := stream.Recv()
+			if recvErr != nil {
+				if recvErr == io.EOF {
+					if prevEntry != nil {
+						fn(prevEntry, true)
+					}
+					break
+				} else {
+					return recvErr
+				}
+			}
+			if prevEntry != nil {
+				fn(prevEntry, false)
+			}
+			prevEntry = resp.Entry
+		}
+
+		return nil
+
+	})
+
+	return
+}
+
+func Exists(filerClient FilerClient, parentDirectoryPath string, entryName string, isDirectory bool) (exists bool, err error) {
+
+	err = filerClient.WithFilerClient(func(client SeaweedFilerClient) error {
+
+		request := &LookupDirectoryEntryRequest{
+			Directory: parentDirectoryPath,
+			Name:      entryName,
+		}
+
+		glog.V(4).Infof("exists entry %v/%v: %v", parentDirectoryPath, entryName, request)
+		resp, err := LookupEntry(client, request)
+		if err != nil {
+			if err == ErrNotFound {
+				exists = false
+				return nil
+			}
+			glog.V(0).Infof("exists entry %v: %v", request, err)
+			return fmt.Errorf("exists entry %s/%s: %v", parentDirectoryPath, entryName, err)
+		}
+
+		exists = resp.Entry.IsDirectory == isDirectory
+
+		return nil
+	})
+
+	return
+}