mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-12-20 22:47:51 +08:00
118 lines
2.8 KiB
Go
118 lines
2.8 KiB
Go
package operation
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"net/url"
|
|
"strings"
|
|
"sync"
|
|
|
|
"net/http"
|
|
|
|
"github.com/chrislusf/seaweedfs/go/security"
|
|
"github.com/chrislusf/seaweedfs/go/util"
|
|
)
|
|
|
|
type DeleteResult struct {
|
|
Fid string `json:"fid"`
|
|
Size int `json:"size"`
|
|
Status int `json:"status"`
|
|
Error string `json:"error,omitempty"`
|
|
}
|
|
|
|
func DeleteFile(master string, fileId string, jwt security.EncodedJwt) error {
|
|
fileUrl, err := LookupFileId(master, fileId)
|
|
if err != nil {
|
|
return fmt.Errorf("Failed to lookup %s:%v", fileId, err)
|
|
}
|
|
err = util.Delete(fileUrl, jwt)
|
|
if err != nil {
|
|
return fmt.Errorf("Failed to delete %s:%v", fileUrl, err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func ParseFileId(fid string) (vid string, key_cookie string, err error) {
|
|
commaIndex := strings.Index(fid, ",")
|
|
if commaIndex <= 0 {
|
|
return "", "", errors.New("Wrong fid format.")
|
|
}
|
|
return fid[:commaIndex], fid[commaIndex+1:], nil
|
|
}
|
|
|
|
type DeleteFilesResult struct {
|
|
Errors []string
|
|
Results []DeleteResult
|
|
}
|
|
|
|
func DeleteFiles(master string, fileIds []string) (*DeleteFilesResult, error) {
|
|
vid_to_fileIds := make(map[string][]string)
|
|
ret := &DeleteFilesResult{}
|
|
var vids []string
|
|
for _, fileId := range fileIds {
|
|
vid, _, err := ParseFileId(fileId)
|
|
if err != nil {
|
|
ret.Results = append(ret.Results, DeleteResult{
|
|
Fid: vid,
|
|
Status: http.StatusBadRequest,
|
|
Error: err.Error()},
|
|
)
|
|
continue
|
|
}
|
|
if _, ok := vid_to_fileIds[vid]; !ok {
|
|
vid_to_fileIds[vid] = make([]string, 0)
|
|
vids = append(vids, vid)
|
|
}
|
|
vid_to_fileIds[vid] = append(vid_to_fileIds[vid], fileId)
|
|
}
|
|
|
|
lookupResults, err := LookupVolumeIds(master, vids)
|
|
if err != nil {
|
|
return ret, err
|
|
}
|
|
|
|
server_to_fileIds := make(map[string][]string)
|
|
for vid, result := range lookupResults {
|
|
if result.Error != "" {
|
|
ret.Errors = append(ret.Errors, result.Error)
|
|
continue
|
|
}
|
|
for _, location := range result.Locations {
|
|
if _, ok := server_to_fileIds[location.Url]; !ok {
|
|
server_to_fileIds[location.Url] = make([]string, 0)
|
|
}
|
|
server_to_fileIds[location.Url] = append(
|
|
server_to_fileIds[location.Url], vid_to_fileIds[vid]...)
|
|
}
|
|
}
|
|
|
|
var wg sync.WaitGroup
|
|
|
|
for server, fidList := range server_to_fileIds {
|
|
wg.Add(1)
|
|
go func(server string, fidList []string) {
|
|
defer wg.Done()
|
|
values := make(url.Values)
|
|
for _, fid := range fidList {
|
|
values.Add("fid", fid)
|
|
}
|
|
jsonBlob, err := util.Post("http://"+server+"/delete", values)
|
|
if err != nil {
|
|
ret.Errors = append(ret.Errors, err.Error()+" "+string(jsonBlob))
|
|
return
|
|
}
|
|
var result []DeleteResult
|
|
err = json.Unmarshal(jsonBlob, &result)
|
|
if err != nil {
|
|
ret.Errors = append(ret.Errors, err.Error()+" "+string(jsonBlob))
|
|
return
|
|
}
|
|
ret.Results = append(ret.Results, result...)
|
|
}(server, fidList)
|
|
}
|
|
wg.Wait()
|
|
|
|
return ret, nil
|
|
}
|