package operation import ( "context" "errors" "fmt" "net/http" "strings" "sync" "google.golang.org/grpc" "github.com/chrislusf/seaweedfs/weed/pb/volume_server_pb" ) type DeleteResult struct { Fid string `json:"fid"` Size int `json:"size"` Status int `json:"status"` Error string `json:"error,omitempty"` } 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 } // DeleteFiles batch deletes a list of fileIds func DeleteFiles(master string, grpcDialOption grpc.DialOption, fileIds []string) ([]*volume_server_pb.DeleteResult, error) { lookupFunc := func(vids []string) (map[string]LookupResult, error) { return LookupVolumeIds(master, grpcDialOption, vids) } return DeleteFilesWithLookupVolumeId(grpcDialOption, fileIds, lookupFunc) } func DeleteFilesWithLookupVolumeId(grpcDialOption grpc.DialOption, fileIds []string, lookupFunc func(vid []string) (map[string]LookupResult, error)) ([]*volume_server_pb.DeleteResult, error) { var ret []*volume_server_pb.DeleteResult vid_to_fileIds := make(map[string][]string) var vids []string for _, fileId := range fileIds { vid, _, err := ParseFileId(fileId) if err != nil { ret = append(ret, &volume_server_pb.DeleteResult{ FileId: fileId, 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 := lookupFunc(vids) if err != nil { return ret, err } server_to_fileIds := make(map[string][]string) for vid, result := range lookupResults { if result.Error != "" { ret = append(ret, &volume_server_pb.DeleteResult{ FileId: vid, Status: http.StatusBadRequest, Error: err.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]...) } } resultChan := make(chan []*volume_server_pb.DeleteResult, len(server_to_fileIds)) var wg sync.WaitGroup for server, fidList := range server_to_fileIds { wg.Add(1) go func(server string, fidList []string) { defer wg.Done() if deleteResults, deleteErr := DeleteFilesAtOneVolumeServer(server, grpcDialOption, fidList); deleteErr != nil { err = deleteErr } else if deleteResults != nil { resultChan <- deleteResults } }(server, fidList) } wg.Wait() close(resultChan) for result := range resultChan { ret = append(ret, result...) } return ret, err } // DeleteFilesAtOneVolumeServer deletes a list of files that is on one volume server via gRpc func DeleteFilesAtOneVolumeServer(volumeServer string, grpcDialOption grpc.DialOption, fileIds []string) (ret []*volume_server_pb.DeleteResult, err error) { err = WithVolumeServerClient(volumeServer, grpcDialOption, func(volumeServerClient volume_server_pb.VolumeServerClient) error { req := &volume_server_pb.BatchDeleteRequest{ FileIds: fileIds, } resp, err := volumeServerClient.BatchDelete(context.Background(), req) // fmt.Printf("deleted %v %v: %v\n", fileIds, err, resp) if err != nil { return err } ret = append(ret, resp.Results...) return nil }) if err != nil { return } for _, result := range ret { if result.Error != "" && result.Error != "not found" { return nil, fmt.Errorf("delete fileId %s: %v", result.FileId, result.Error) } } return }