seaweedfs/weed/filesys/file.go

package filesys

import (
	"context"
	"fmt"

	"bazil.org/fuse"
	"bazil.org/fuse/fs"
	"github.com/chrislusf/seaweedfs/weed/glog"
	"github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
	"path/filepath"
	"os"
	"time"
	"bytes"
	"github.com/chrislusf/seaweedfs/weed/operation"
)

var _ = fs.Node(&File{})
// var _ = fs.NodeOpener(&File{})
// var _ = fs.NodeFsyncer(&File{})
var _ = fs.Handle(&File{})
var _ = fs.HandleReadAller(&File{})
// var _ = fs.HandleReader(&File{})
var _ = fs.HandleFlusher(&File{})
var _ = fs.HandleWriter(&File{})

type File struct {
	Chunks []*filer_pb.FileChunk
	Name   string
	dir    *Dir
	wfs    *WFS
}

func (file *File) Attr(context context.Context, attr *fuse.Attr) error {
	fullPath := filepath.Join(file.dir.Path, file.Name)
	item := file.wfs.listDirectoryEntriesCache.Get(fullPath)
	var attributes *filer_pb.FuseAttributes
	if item != nil {
		attributes = item.Value().(*filer_pb.FuseAttributes)
		glog.V(1).Infof("read cached file %v attributes", file.Name)
	} else {
		err := file.wfs.withFilerClient(func(client filer_pb.SeaweedFilerClient) error {

			request := &filer_pb.GetFileAttributesRequest{
				Name:      file.Name,
				ParentDir: file.dir.Path,
			}

			glog.V(1).Infof("read file size: %v", request)
			resp, err := client.GetFileAttributes(context, request)
			if err != nil {
				return err
			}

			attributes = resp.Attributes

			return nil
		})

		if err != nil {
			return err
		}
	}

	attr.Mode = os.FileMode(attributes.FileMode)
	attr.Size = attributes.FileSize
	attr.Mtime = time.Unix(attributes.Mtime, 0)
	attr.Gid = attributes.Gid
	attr.Uid = attributes.Uid
	return nil

}

func (file *File) ReadAll(ctx context.Context) (content []byte, err error) {

	if len(file.Chunks) == 0 {
		return
	}

	err = file.wfs.withFilerClient(func(client filer_pb.SeaweedFilerClient) error {

		// FIXME: need to either use Read() or implement differently
		request := &filer_pb.GetFileContentRequest{
			FileId: file.Chunks[0].FileId,
		}

		glog.V(1).Infof("read file content: %v", request)
		resp, err := client.GetFileContent(ctx, request)
		if err != nil {
			return err
		}

		content = resp.Content

		return nil
	})

	return content, err
}

func (file *File) Flush(ctx context.Context, req *fuse.FlushRequest) error {
	// write the file chunks to the filer
	fmt.Printf("flush file %+v\n", req)

	err := file.wfs.withFilerClient(func(client filer_pb.SeaweedFilerClient) error {

		request := &filer_pb.AppendFileChunksRequest{
			Directory: file.dir.Path,
			Entry: &filer_pb.Entry{
				Name:   file.Name,
				Chunks: file.Chunks,
			},
		}

		glog.V(1).Infof("append chunks: %v", request)
		if _, err := client.AppendFileChunks(ctx, request); err != nil {
			return fmt.Errorf("create file: %v", err)
		}

		return nil
	})

	return err
}

func (file *File) Write(ctx context.Context, req *fuse.WriteRequest, resp *fuse.WriteResponse) error {
	// write the request to volume servers
	fmt.Printf("write file %+v\n", req)

	var fileId, host string

	if err := file.wfs.withFilerClient(func(client filer_pb.SeaweedFilerClient) error {

		request := &filer_pb.AssignVolumeRequest{
			Count:       1,
			Replication: "000",
			Collection:  "",
		}

		glog.V(1).Infof("assign volume: %v", request)
		resp, err := client.AssignVolume(ctx, request)
		if err != nil {
			return err
		}

		fileId, host = resp.FileId, resp.Url

		return nil
	}); err != nil {
		return fmt.Errorf("filer assign volume: %v", err)
	}

	fileUrl := fmt.Sprintf("http://%s/%s", host, fileId)
	bufReader := bytes.NewReader(req.Data)
	uploadResult, err := operation.Upload(fileUrl, file.Name, bufReader, false, "application/octet-stream", nil, "")
	if err != nil {
		return fmt.Errorf("upload data: %v", err)
	}
	if uploadResult.Error != "" {
		return fmt.Errorf("upload result: %v", uploadResult.Error)
	}

	glog.V(1).Infof("uploaded %s/%s to: %v", file.dir.Path, file.Name, fileUrl)

	file.Chunks = append(file.Chunks, &filer_pb.FileChunk{
		FileId: fileId,
		Offset: req.Offset,
		Size:   uint64(uploadResult.Size),
	})

	return nil
}