aboutsummaryrefslogtreecommitdiff
path: root/weed/operation/delete_content.go
diff options
context:
space:
mode:
authorChris Lu <chris.lu@gmail.com>2016-06-02 18:09:14 -0700
committerChris Lu <chris.lu@gmail.com>2016-06-02 18:09:14 -0700
commit5ce6bbf07672bf3f3c8d26cd2ce0e3e853a47c44 (patch)
tree2e4dd2ad0a618ab2b7cdebcdb9c503526c31e2e8 /weed/operation/delete_content.go
parentcaeffa3998adc060fa66c4cd77af971ff2d26c57 (diff)
downloadseaweedfs-5ce6bbf07672bf3f3c8d26cd2ce0e3e853a47c44.tar.xz
seaweedfs-5ce6bbf07672bf3f3c8d26cd2ce0e3e853a47c44.zip
directory structure change to work with glide
glide has its own requirements. My previous workaround caused me some code checkin errors. Need to fix this.
Diffstat (limited to 'weed/operation/delete_content.go')
-rw-r--r--weed/operation/delete_content.go117
1 files changed, 117 insertions, 0 deletions
diff --git a/weed/operation/delete_content.go b/weed/operation/delete_content.go
new file mode 100644
index 000000000..b78221da1
--- /dev/null
+++ b/weed/operation/delete_content.go
@@ -0,0 +1,117 @@
+package operation
+
+import (
+ "encoding/json"
+ "errors"
+ "fmt"
+ "net/url"
+ "strings"
+ "sync"
+
+ "net/http"
+
+ "github.com/chrislusf/seaweedfs/weed/security"
+ "github.com/chrislusf/seaweedfs/weed/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
+}