mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-01-19 02:48:24 +00:00
more check in http_util.Delete
add status code in `DeleteResult` struct operation.DeleteFiles maybe unsafe, so `ChunkManifest.DeleteChunks` manually delete each chunks
This commit is contained in:
parent
a9a336fdff
commit
daac5de1ba
|
@ -70,19 +70,11 @@ func (cm *ChunkManifest) GetData() ([]byte, error) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (cm *ChunkManifest) DeleteChunks(master string) error {
|
func (cm *ChunkManifest) DeleteChunks(master string) error {
|
||||||
fileIds := make([]string, 0, len(cm.Chunks))
|
|
||||||
for _, ci := range cm.Chunks {
|
|
||||||
fileIds = append(fileIds, ci.Fid)
|
|
||||||
}
|
|
||||||
results, e := DeleteFiles(master, fileIds)
|
|
||||||
if e != nil {
|
|
||||||
return e
|
|
||||||
}
|
|
||||||
deleteError := 0
|
deleteError := 0
|
||||||
for _, ret := range results.Results {
|
for _, ci := range cm.Chunks {
|
||||||
if ret.Error != "" {
|
if e := DeleteFile(master, ci.Fid, ""); e != nil {
|
||||||
deleteError++
|
deleteError++
|
||||||
glog.V(0).Infoln("delete error:", ret.Error, ret.Fid)
|
glog.V(0).Infoln("delete error:", e, ci.Fid)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if deleteError > 0 {
|
if deleteError > 0 {
|
||||||
|
|
|
@ -7,14 +7,17 @@ import (
|
||||||
"strings"
|
"strings"
|
||||||
"sync"
|
"sync"
|
||||||
|
|
||||||
|
"net/http"
|
||||||
|
|
||||||
"github.com/chrislusf/seaweedfs/go/security"
|
"github.com/chrislusf/seaweedfs/go/security"
|
||||||
"github.com/chrislusf/seaweedfs/go/util"
|
"github.com/chrislusf/seaweedfs/go/util"
|
||||||
)
|
)
|
||||||
|
|
||||||
type DeleteResult struct {
|
type DeleteResult struct {
|
||||||
Fid string `json:"fid"`
|
Fid string `json:"fid"`
|
||||||
Size int `json:"size"`
|
Size int `json:"size"`
|
||||||
Error string `json:"error,omitempty"`
|
Status int `json:"status"`
|
||||||
|
Error string `json:"error,omitempty"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func DeleteFile(master string, fileId string, jwt security.EncodedJwt) error {
|
func DeleteFile(master string, fileId string, jwt security.EncodedJwt) error {
|
||||||
|
@ -45,7 +48,11 @@ func DeleteFiles(master string, fileIds []string) (*DeleteFilesResult, error) {
|
||||||
for _, fileId := range fileIds {
|
for _, fileId := range fileIds {
|
||||||
vid, _, err := ParseFileId(fileId)
|
vid, _, err := ParseFileId(fileId)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ret.Results = append(ret.Results, DeleteResult{Fid: vid, Error: err.Error()})
|
ret.Results = append(ret.Results, DeleteResult{
|
||||||
|
Fid: vid,
|
||||||
|
Status: http.StatusBadRequest,
|
||||||
|
Error: err.Error()},
|
||||||
|
)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
if _, ok := vid_to_fileIds[vid]; !ok {
|
if _, ok := vid_to_fileIds[vid]; !ok {
|
||||||
|
@ -76,6 +83,7 @@ func DeleteFiles(master string, fileIds []string) (*DeleteFilesResult, error) {
|
||||||
}
|
}
|
||||||
|
|
||||||
var wg sync.WaitGroup
|
var wg sync.WaitGroup
|
||||||
|
|
||||||
for server, fidList := range server_to_fileIds {
|
for server, fidList := range server_to_fileIds {
|
||||||
wg.Add(1)
|
wg.Add(1)
|
||||||
go func(server string, fidList []string) {
|
go func(server string, fidList []string) {
|
||||||
|
|
|
@ -9,7 +9,10 @@ import (
|
||||||
"net/url"
|
"net/url"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
|
"encoding/json"
|
||||||
|
|
||||||
"github.com/chrislusf/seaweedfs/go/security"
|
"github.com/chrislusf/seaweedfs/go/security"
|
||||||
|
"github.com/syndtr/goleveldb/leveldb/errors"
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
|
@ -79,10 +82,21 @@ func Delete(url string, jwt security.EncodedJwt) error {
|
||||||
return e
|
return e
|
||||||
}
|
}
|
||||||
defer resp.Body.Close()
|
defer resp.Body.Close()
|
||||||
if _, err := ioutil.ReadAll(resp.Body); err != nil {
|
body, err := ioutil.ReadAll(resp.Body)
|
||||||
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
return nil
|
switch resp.StatusCode {
|
||||||
|
case http.StatusNotFound, http.StatusAccepted, http.StatusOK:
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
m := make(map[string]interface{})
|
||||||
|
if e := json.Unmarshal(body, m); e == nil {
|
||||||
|
if s, ok := m["error"].(string); ok {
|
||||||
|
return errors.New(s)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return errors.New(string(body))
|
||||||
}
|
}
|
||||||
|
|
||||||
func GetBufferStream(url string, values url.Values, allocatedBytes []byte, eachBuffer func([]byte)) error {
|
func GetBufferStream(url string, values url.Values, allocatedBytes []byte, eachBuffer func([]byte)) error {
|
||||||
|
|
|
@ -63,19 +63,20 @@ func (vs *VolumeServer) DeleteHandler(w http.ResponseWriter, r *http.Request) {
|
||||||
|
|
||||||
if n.Cookie != cookie {
|
if n.Cookie != cookie {
|
||||||
glog.V(0).Infoln("delete", r.URL.Path, "with unmaching cookie from ", r.RemoteAddr, "agent", r.UserAgent())
|
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
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
count := int64(n.Size)
|
count := int64(n.Size)
|
||||||
|
|
||||||
if n.IsChunkedManifest(){
|
if n.IsChunkedManifest() {
|
||||||
chunkManifest, e := operation.LoadChunkManifest(n.Data, n.IsGzipped())
|
chunkManifest, e := operation.LoadChunkManifest(n.Data, n.IsGzipped())
|
||||||
if e != nil {
|
if e != nil {
|
||||||
writeJsonError(w, r, http.StatusInternalServerError, errors.New("Load chunks manifest error: " + e.Error()))
|
writeJsonError(w, r, http.StatusInternalServerError, errors.New("Load chunks manifest error: "+e.Error()))
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if e := chunkManifest.DeleteChunks(vs.GetMasterNode()); e != nil {
|
if e := chunkManifest.DeleteChunks(vs.GetMasterNode()); e != nil {
|
||||||
writeJsonError(w, r, http.StatusInternalServerError, errors.New("Delete chunks error: " + e.Error()))
|
writeJsonError(w, r, http.StatusInternalServerError, errors.New("Delete chunks error: "+e.Error()))
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
count = chunkManifest.Size
|
count = chunkManifest.Size
|
||||||
|
@ -100,7 +101,10 @@ func (vs *VolumeServer) batchDeleteHandler(w http.ResponseWriter, r *http.Reques
|
||||||
for _, fid := range r.Form["fid"] {
|
for _, fid := range r.Form["fid"] {
|
||||||
vid, id_cookie, err := operation.ParseFileId(fid)
|
vid, id_cookie, err := operation.ParseFileId(fid)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
ret = append(ret, operation.DeleteResult{Fid: fid, Error: err.Error()})
|
ret = append(ret, operation.DeleteResult{
|
||||||
|
Fid: fid,
|
||||||
|
Status: http.StatusBadRequest,
|
||||||
|
Error: err.Error()})
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
n := new(storage.Needle)
|
n := new(storage.Needle)
|
||||||
|
@ -109,18 +113,45 @@ func (vs *VolumeServer) batchDeleteHandler(w http.ResponseWriter, r *http.Reques
|
||||||
glog.V(4).Infoln("batch deleting", n)
|
glog.V(4).Infoln("batch deleting", n)
|
||||||
cookie := n.Cookie
|
cookie := n.Cookie
|
||||||
if _, err := vs.store.ReadVolumeNeedle(volumeId, n); err != nil {
|
if _, err := vs.store.ReadVolumeNeedle(volumeId, n); err != nil {
|
||||||
ret = append(ret, operation.DeleteResult{Fid: fid, Error: err.Error()})
|
ret = append(ret, operation.DeleteResult{
|
||||||
|
Fid: fid,
|
||||||
|
Status: http.StatusNotFound,
|
||||||
|
Error: err.Error(),
|
||||||
|
})
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if n.IsChunkedManifest() {
|
||||||
|
//Don't allow delete manifest in batch delete mode
|
||||||
|
ret = append(ret, operation.DeleteResult{
|
||||||
|
Fid: fid,
|
||||||
|
Status: http.StatusNotAcceptable,
|
||||||
|
Error: "ChunkManifest: not allow.",
|
||||||
|
})
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
if n.Cookie != cookie {
|
if n.Cookie != cookie {
|
||||||
ret = append(ret, operation.DeleteResult{Fid: fid, Error: "File Random Cookie does not match."})
|
ret = append(ret, operation.DeleteResult{
|
||||||
|
Fid: fid,
|
||||||
|
Status: http.StatusBadRequest,
|
||||||
|
Error: "File Random Cookie does not match.",
|
||||||
|
})
|
||||||
glog.V(0).Infoln("deleting", fid, "with unmaching cookie from ", r.RemoteAddr, "agent", r.UserAgent())
|
glog.V(0).Infoln("deleting", fid, "with unmaching cookie from ", r.RemoteAddr, "agent", r.UserAgent())
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if size, err := vs.store.Delete(volumeId, n); err != nil {
|
if size, err := vs.store.Delete(volumeId, n); err != nil {
|
||||||
ret = append(ret, operation.DeleteResult{Fid: fid, Error: err.Error()})
|
ret = append(ret, operation.DeleteResult{
|
||||||
|
Fid: fid,
|
||||||
|
Status: http.StatusInternalServerError,
|
||||||
|
Error: err.Error()},
|
||||||
|
)
|
||||||
} else {
|
} else {
|
||||||
ret = append(ret, operation.DeleteResult{Fid: fid, Size: int(size)})
|
ret = append(ret, operation.DeleteResult{
|
||||||
|
Fid: fid,
|
||||||
|
Status: http.StatusAccepted,
|
||||||
|
Size: int(size)},
|
||||||
|
)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue