package manager
import (
"net/http"
"regexp"
"strconv"
"io"
"fmt"
"github.com/030io/whalefs/manager/volume"
"gopkg.in/redis.v2"
"os"
)
var (
postVolumeUrl *regexp.Regexp //提交卷文件正则表达式
postFileUrl *regexp.Regexp //提交文件正则表达式
deleteFileUrl *regexp.Regexp //删除文件url 正则表达式
)
//初始化文件表达式
func init() { var err error
postVolumeUrl, err = regexp.Compile("/([0-9]*)/") if err != nil { panic(err)
}
postFileUrl, err = regexp.Compile("/([0-9]*)/([0-9]*)/(.*)") if err != nil { panic(err)
}
deleteFileUrl = postFileUrl
}
type Size interface { Size() int64
}
//文件管理处理器
func (vm *VolumeManager)adminEntry(w http.ResponseWriter, r *http.Request) { switch r.Method { case http.MethodGet, http.MethodHead:
vm.publicEntry(w, r)
case http.MethodPost:
if postFileUrl.MatchString(r.URL.Path) { vm.adminPostFile(w, r)
} else if postVolumeUrl.MatchString(r.URL.Path) { vm.adminPostVolume(w, r)
} else { http.Error(w, r.URL.String() + " can't match", http.StatusNotFound)
}
case http.MethodDelete:
if deleteFileUrl.MatchString(r.URL.Path) { vm.adminDeleteFile(w, r)
} else { http.Error(w, r.URL.String() + " can't match", http.StatusNotFound)
}
default:
w.WriteHeader(http.StatusMethodNotAllowed)
}
}
//卷文件管理处理器
func (vm *VolumeManager)adminPostVolume(w http.ResponseWriter, r *http.Request) { match := postVolumeUrl.FindStringSubmatch(r.URL.Path)
vid, _ := strconv.ParseUint(match[1], 10, 64)
volume, err := volume.NewVolume(vm.DataDir, vid)
if err != nil { http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
vm.Volumes[vid] = volume
w.WriteHeader(http.StatusCreated)
}
//上传文件处理器
func (vm *VolumeManager)adminPostFile(w http.ResponseWriter, r *http.Request) { match := postFileUrl.FindStringSubmatch(r.URL.Path)
vid, _ := strconv.ParseUint(match[1], 10, 64)
volume, ok := vm.Volumes[vid]
if !ok { http.Error(w, "can't find volume", http.StatusNotFound)
return
}
file, header, err := r.FormFile("file") if err != nil { http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
defer file.Close()
var fileSize int64
switch file.(type) { case *os.File:
s, _ := file.(*os.File).Stat()
fileSize = s.Size()
case Size:
fileSize = file.(Size).Size()
}
fid, _ := strconv.ParseUint(match[2], 10, 64)
file_, err := volume.NewFile(fid, header.Filename, uint64(fileSize))
if err != nil { http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
defer func() { if err != nil { if err := volume.Delete(file_.Info.Fid, file_.Info.FileName); err != nil { panic(err)
}
}
}()
n, err := io.Copy(file_, file)
if err != nil { http.Error(w, err.Error(), http.StatusInternalServerError)
return
} else if n != fileSize { err = fmt.Errorf("%d != %d", n, fileSize) http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
w.WriteHeader(http.StatusCreated)
}
//文件删除
func (vm *VolumeManager)adminDeleteFile(w http.ResponseWriter, r *http.Request) { match := deleteFileUrl.FindStringSubmatch(r.URL.Path)
vid, _ := strconv.ParseUint(match[1], 10, 64)
volume := vm.Volumes[vid]
if volume == nil { http.Error(w, "can't find volume", http.StatusNotFound)
return
}
fid, _ := strconv.ParseUint(match[2], 10, 64)
err := volume.Delete(fid, match[3])
if err == redis.Nil { http.NotFound(w, r)
} else if err != nil { http.Error(w, err.Error(), http.StatusInternalServerError)
} else { w.WriteHeader(http.StatusAccepted)
}
}