admin-handlers.go

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)

    }

}