mirror of
https://github.com/chrislusf/seaweedfs
synced 2024-08-25 03:01:15 +02:00
166 lines
4.4 KiB
Go
166 lines
4.4 KiB
Go
package weed_server
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/seaweedfs/seaweedfs/weed/glog"
|
|
"github.com/seaweedfs/seaweedfs/weed/operation"
|
|
"github.com/seaweedfs/seaweedfs/weed/storage/needle"
|
|
"github.com/seaweedfs/seaweedfs/weed/topology"
|
|
)
|
|
|
|
func (vs *VolumeServer) PostHandler(w http.ResponseWriter, r *http.Request) {
|
|
if e := r.ParseForm(); e != nil {
|
|
glog.V(0).Infoln("form parse error:", e)
|
|
writeJsonError(w, r, http.StatusBadRequest, e)
|
|
return
|
|
}
|
|
|
|
vid, fid, _, _, _ := parseURLPath(r.URL.Path)
|
|
volumeId, ve := needle.NewVolumeId(vid)
|
|
if ve != nil {
|
|
glog.V(0).Infoln("NewVolumeId error:", ve)
|
|
writeJsonError(w, r, http.StatusBadRequest, ve)
|
|
return
|
|
}
|
|
|
|
if !vs.maybeCheckJwtAuthorization(r, vid, fid, true) {
|
|
writeJsonError(w, r, http.StatusUnauthorized, errors.New("wrong jwt"))
|
|
return
|
|
}
|
|
|
|
bytesBuffer := bufPool.Get().(*bytes.Buffer)
|
|
defer bufPool.Put(bytesBuffer)
|
|
|
|
reqNeedle, originalSize, contentMd5, ne := needle.CreateNeedleFromRequest(r, vs.FixJpgOrientation, vs.fileSizeLimitBytes, bytesBuffer)
|
|
if ne != nil {
|
|
writeJsonError(w, r, http.StatusBadRequest, ne)
|
|
return
|
|
}
|
|
|
|
ret := operation.UploadResult{}
|
|
isUnchanged, writeError := topology.ReplicatedWrite(vs.GetMaster, vs.grpcDialOption, vs.store, volumeId, reqNeedle, r)
|
|
if writeError != nil {
|
|
writeJsonError(w, r, http.StatusInternalServerError, writeError)
|
|
}
|
|
|
|
// http 204 status code does not allow body
|
|
if writeError == nil && isUnchanged {
|
|
setEtag(w, reqNeedle.Etag())
|
|
w.WriteHeader(http.StatusNoContent)
|
|
return
|
|
}
|
|
|
|
httpStatus := http.StatusCreated
|
|
if reqNeedle.HasName() {
|
|
ret.Name = string(reqNeedle.Name)
|
|
}
|
|
ret.Size = uint32(originalSize)
|
|
ret.ETag = reqNeedle.Etag()
|
|
ret.Mime = string(reqNeedle.Mime)
|
|
setEtag(w, ret.ETag)
|
|
w.Header().Set("Content-MD5", contentMd5)
|
|
writeJsonQuiet(w, r, httpStatus, ret)
|
|
}
|
|
|
|
func (vs *VolumeServer) DeleteHandler(w http.ResponseWriter, r *http.Request) {
|
|
n := new(needle.Needle)
|
|
vid, fid, _, _, _ := parseURLPath(r.URL.Path)
|
|
volumeId, _ := needle.NewVolumeId(vid)
|
|
n.ParsePath(fid)
|
|
|
|
if !vs.maybeCheckJwtAuthorization(r, vid, fid, true) {
|
|
writeJsonError(w, r, http.StatusUnauthorized, errors.New("wrong jwt"))
|
|
return
|
|
}
|
|
|
|
// glog.V(2).Infof("volume %s deleting %s", vid, n)
|
|
|
|
cookie := n.Cookie
|
|
|
|
ecVolume, hasEcVolume := vs.store.FindEcVolume(volumeId)
|
|
|
|
if hasEcVolume {
|
|
count, err := vs.store.DeleteEcShardNeedle(ecVolume, n, cookie)
|
|
writeDeleteResult(err, count, w, r)
|
|
return
|
|
}
|
|
|
|
_, ok := vs.store.ReadVolumeNeedle(volumeId, n, nil, nil)
|
|
if ok != nil {
|
|
m := make(map[string]uint32)
|
|
m["size"] = 0
|
|
writeJsonQuiet(w, r, http.StatusNotFound, m)
|
|
return
|
|
}
|
|
|
|
if n.Cookie != cookie {
|
|
glog.V(0).Infoln("delete", r.URL.Path, "with unmaching cookie from ", r.RemoteAddr, "agent", r.UserAgent())
|
|
writeJsonError(w, r, http.StatusBadRequest, errors.New("File Random Cookie does not match."))
|
|
return
|
|
}
|
|
|
|
count := int64(n.Size)
|
|
|
|
if n.IsChunkedManifest() {
|
|
chunkManifest, e := operation.LoadChunkManifest(n.Data, n.IsCompressed())
|
|
if e != nil {
|
|
writeJsonError(w, r, http.StatusInternalServerError, fmt.Errorf("Load chunks manifest error: %v", e))
|
|
return
|
|
}
|
|
// make sure all chunks had deleted before delete manifest
|
|
if e := chunkManifest.DeleteChunks(vs.GetMaster, false, vs.grpcDialOption); e != nil {
|
|
writeJsonError(w, r, http.StatusInternalServerError, fmt.Errorf("Delete chunks error: %v", e))
|
|
return
|
|
}
|
|
count = chunkManifest.Size
|
|
}
|
|
|
|
n.LastModified = uint64(time.Now().Unix())
|
|
if len(r.FormValue("ts")) > 0 {
|
|
modifiedTime, err := strconv.ParseInt(r.FormValue("ts"), 10, 64)
|
|
if err == nil {
|
|
n.LastModified = uint64(modifiedTime)
|
|
}
|
|
}
|
|
|
|
_, err := topology.ReplicatedDelete(vs.GetMaster, vs.grpcDialOption, vs.store, volumeId, n, r)
|
|
|
|
writeDeleteResult(err, count, w, r)
|
|
|
|
}
|
|
|
|
func writeDeleteResult(err error, count int64, w http.ResponseWriter, r *http.Request) {
|
|
if err == nil {
|
|
m := make(map[string]int64)
|
|
m["size"] = count
|
|
writeJsonQuiet(w, r, http.StatusAccepted, m)
|
|
} else {
|
|
writeJsonError(w, r, http.StatusInternalServerError, fmt.Errorf("Deletion Failed: %v", err))
|
|
}
|
|
}
|
|
|
|
func setEtag(w http.ResponseWriter, etag string) {
|
|
if etag != "" {
|
|
if strings.HasPrefix(etag, "\"") {
|
|
w.Header().Set("ETag", etag)
|
|
} else {
|
|
w.Header().Set("ETag", "\""+etag+"\"")
|
|
}
|
|
}
|
|
}
|
|
|
|
func getEtag(resp *http.Response) (etag string) {
|
|
etag = resp.Header.Get("ETag")
|
|
if strings.HasPrefix(etag, "\"") && strings.HasSuffix(etag, "\"") {
|
|
return etag[1 : len(etag)-1]
|
|
}
|
|
return
|
|
}
|