func (s *server) Rename(ctx context.Context, req *pb.RenameRequest) (*pb.RenameReply, error) {
if !enforce.HasQuota(req.Header.ClientID) {
log.Infof("server: out of quota for client %d", req.Header.ClientID)
return &pb.RenameReply{}, nil
}
dn0, ofn, err := splitDiskAndFile(req.Oldname)
if err != nil {
log.Infof("server: rename error (%v)", err)
return &pb.RenameReply{}, nil
}
dn1, nfn, err := splitDiskAndFile(req.Newname)
if err != nil {
log.Infof("server: rename error (%v)", err)
return &pb.RenameReply{}, nil
}
if dn0 != dn1 {
log.Infof("server: rename error (%v)", "not same disk")
return &pb.RenameReply{}, nil
}
d := s.Disk(dn0)
if d == nil {
log.Infof("server: read error (cannot find disk %s)", dn0)
return &pb.RenameReply{}, nil
}
stats.Counter(dn0, "rename").Client(req.Header.ClientID).Add()
err = d.Rename(ofn, nfn)
if err != nil {
log.Infof("server: rename error (%v)", err)
return &pb.RenameReply{}, nil
}
reply := &pb.RenameReply{}
return reply, nil
}
func (s *server) Read(ctx context.Context, req *pb.ReadRequest) (*pb.ReadReply, error) {
if !enforce.HasQuota(req.Header.ClientID) {
log.Infof("server: out of quota for client %d", req.Header.ClientID)
return &pb.ReadReply{}, nil
}
dn, fn, err := splitDiskAndFile(req.Name)
if err != nil {
log.Infof("server: read error (%v)", err)
return &pb.ReadReply{}, nil
}
d := s.Disk(dn)
if d == nil {
log.Infof("server: read error (cannot find disk %s)", dn)
return &pb.ReadReply{}, nil
}
stats.Counter(dn, "read").Client(req.Header.ClientID).Add()
// TODO: reuse buffer
data := make([]byte, req.Length)
n, err := d.ReadAt(fn, data, req.Offset)
// TODO: add error
if err == io.EOF {
log.Infof("server: read %d bytes until EOF", n)
return &pb.ReadReply{BytesRead: int64(n), Data: data[:n]}, nil
}
if err != nil {
log.Infof("server: read error (%v)", err)
return &pb.ReadReply{}, nil
}
reply := &pb.ReadReply{BytesRead: int64(n), Data: data}
return reply, nil
}
func (s *server) Mkdir(ctx context.Context, req *pb.MkdirRequest) (*pb.MkdirReply, error) {
reply := &pb.MkdirReply{}
if !enforce.HasQuota(req.Header.ClientID) {
log.Infof("server: out of quota for client %d", req.Header.ClientID)
return reply, nil
}
dn, fn, err := splitDiskAndFile(req.Name)
if err != nil {
log.Infof("server: mkdir error (%v)", err)
return reply, nil
}
d := s.Disk(dn)
if d == nil {
log.Infof("server: mkdir error (cannot find disk %s)", dn)
return reply, nil
}
stats.Counter(dn, "mkdir").Client(req.Header.ClientID).Add()
err = d.Mkdir(fn, req.All)
if err != nil {
log.Infof("server: mkdir error (%v)", err)
return reply, nil
}
return reply, nil
}
func (s *server) Write(ctx context.Context, req *pb.WriteRequest) (*pb.WriteReply, error) {
if !enforce.HasQuota(req.Header.ClientID) {
log.Infof("server: out of quota for client %d", req.Header.ClientID)
return &pb.WriteReply{}, nil
}
dn, fn, err := splitDiskAndFile(req.Name)
if err != nil {
log.Infof("server: write error (%v)", err)
return &pb.WriteReply{}, nil
}
d := s.Disk(dn)
if d == nil {
log.Infof("server: write error (cannot find disk %s)", dn)
return &pb.WriteReply{}, nil
}
stats.Counter(dn, "write").Client(req.Header.ClientID).Add()
n, err := d.WriteAt(fn, req.Data, req.Offset)
// TODO: add error
if err != nil {
log.Infof("server: write error (%v)", err)
return &pb.WriteReply{}, nil
}
reply := &pb.WriteReply{BytesWritten: int64(n)}
return reply, nil
}
func (s *server) ReadDir(ctx context.Context, req *pb.ReadDirRequest) (*pb.ReadDirReply, error) {
reply := &pb.ReadDirReply{}
if !enforce.HasQuota(req.Header.ClientID) {
log.Infof("server: out of quota for client %d", req.Header.ClientID)
return reply, nil
}
dn, fn, err := splitDiskAndFile(req.Name)
if err != nil {
log.Infof("server: readDir error (%v)", err)
return reply, nil
}
d := s.Disk(dn)
if d == nil {
log.Infof("server: readDir error (cannot find disk %s)", dn)
return reply, nil
}
stats.Counter(dn, "readdir").Client(req.Header.ClientID).Add()
stats, err := d.ReadDir(fn)
if err != nil {
log.Infof("server: readDir error (%v)", err)
return reply, nil
}
reply.FileInfos = make([]*pb.FileInfo, len(stats))
for i, stat := range stats {
reply.FileInfos[i] = &pb.FileInfo{
Name: stat.Name(),
// TODO: Add size
TotalSize: stat.Size(),
IsDir: stat.IsDir(),
}
}
return reply, nil
}
