add collection.delete

4 files changed, 41 insertions, 2 deletions

diff --git a/weed/storage/disk_location.go b/weed/storage/disk_location.go
index 0aedb0f47..234d1a5f4 100644
--- a/weed/storage/disk_location.go
+++ b/weed/storage/disk_location.go
@@ -114,17 +114,31 @@ func (l *DiskLocation) loadExistingVolumes(needleMapKind NeedleMapType) {
 }
 
 func (l *DiskLocation) DeleteCollectionFromDiskLocation(collection string) (e error) {
-	l.Lock()
-	defer l.Unlock()
 
+	l.Lock()
 	for k, v := range l.volumes {
 		if v.Collection == collection {
 			e = l.deleteVolumeById(k)
 			if e != nil {
+				l.Unlock()
+				return
+			}
+		}
+	}
+	l.Unlock()
+
+	l.ecVolumesLock.Lock()
+	for k, v := range l.ecVolumes {
+		if v.Collection == collection {
+			e = l.deleteEcVolumeById(k)
+			if e != nil {
+				l.ecVolumesLock.Unlock()
 				return
 			}
 		}
 	}
+	l.ecVolumesLock.Unlock()
+
 	return
 }
 
diff --git a/weed/storage/disk_location_ec.go b/weed/storage/disk_location_ec.go
index bcc09a1c0..ff72baf33 100644
--- a/weed/storage/disk_location_ec.go
+++ b/weed/storage/disk_location_ec.go
@@ -148,3 +148,13 @@ func (l *DiskLocation) loadAllEcShards() (err error) {
 	}
 	return nil
 }
+
+func (l *DiskLocation) deleteEcVolumeById(vid needle.VolumeId) (e error) {
+	ecVolume, ok := l.ecVolumes[vid]
+	if !ok {
+		return
+	}
+	ecVolume.Destroy()
+	delete(l.ecVolumes, vid)
+	return
+}
diff --git a/weed/storage/erasure_coding/ec_shard.go b/weed/storage/erasure_coding/ec_shard.go
index 037c4ba48..687a582a7 100644
--- a/weed/storage/erasure_coding/ec_shard.go
+++ b/weed/storage/erasure_coding/ec_shard.go
@@ -64,6 +64,10 @@ func (shard *EcVolumeShard) Close() {
 	}
 }
 
+func (shard *EcVolumeShard) Destroy() {
+	os.Remove(shard.FileName() + ToExt(int(shard.ShardId)))
+}
+
 func (shard *EcVolumeShard) ReadAt(buf []byte, offset int64) (int, error) {
 
 	return shard.ecdFile.ReadAt(buf, offset)
diff --git a/weed/storage/erasure_coding/ec_volume.go b/weed/storage/erasure_coding/ec_volume.go
index 32f66260a..28dbd8c7f 100644
--- a/weed/storage/erasure_coding/ec_volume.go
+++ b/weed/storage/erasure_coding/ec_volume.go
@@ -94,6 +94,17 @@ func (ev *EcVolume) Close() {
 	}
 }
 
+func (ev *EcVolume) Destroy() {
+
+	ev.Close()
+
+	baseFileName := EcShardFileName(ev.Collection, ev.dir, int(ev.VolumeId))
+	for _, s := range ev.Shards {
+		s.Destroy()
+	}
+	os.Remove(baseFileName + ".ecx")
+}
+
 func (ev *EcVolume) ToVolumeEcShardInformationMessage() (messages []*master_pb.VolumeEcShardInformationMessage) {
 	prevVolumeId := needle.VolumeId(math.MaxUint32)
 	var m *master_pb.VolumeEcShardInformationMessage