add Destroy methods for cleanup (#16)

lm2.go

@@ -150,6 +150,11 @@ func (rc *recordCache) close() {
 	rc.f.Close()
 }
 
+func (rc *recordCache) destroy() error {
+	rc.close()
+	return os.Remove(rc.f.Name())
+}
+
 func (rc *recordCache) findLastLessThan(key string) int64 {
 	rc.lock.RLock()
 	defer rc.lock.RUnlock()
@@ -827,3 +832,22 @@ func (c *Collection) Version() int64 {
 func (c *Collection) Stats() Stats {
 	return c.stats.clone()
 }
+
+// Destroy closes the collection and removes its associated data files.
+func (c *Collection) Destroy() error {
+	c.Close()
+	var err error
+	err = os.Remove(c.f.Name())
+	if err != nil {
+		return err
+	}
+	err = c.wal.Destroy()
+	if err != nil {
+		return err
+	}
+	err = c.cache.destroy()
+	if err != nil {
+		return err
+	}
+	return nil
+}

lm2_test.go

@@ -29,7 +29,7 @@ func TestCopy(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
-	defer c.Close()
+	defer c.Destroy()
 
 	const N = 1000
 	firstWriteStart := time.Now()
@@ -51,7 +51,7 @@ func TestCopy(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
-	defer c2.Close()
+	defer c2.Destroy()
 
 	cur, err := c.NewCursor()
 	if err != nil {
@@ -121,7 +121,7 @@ func TestWriteBatch(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
-	defer c.Close()
+	defer c.Destroy()
 
 	wb := NewWriteBatch()
 	wb.Set("key1", "1")
@@ -200,7 +200,7 @@ func TestWriteBatch1(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
-	defer c.Close()
+	defer c.Destroy()
 
 	const N = 500
 	for i := 0; i < N; i++ {
@@ -220,7 +220,7 @@ func TestWriteBatch1Concurrent(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
-	defer c.Close()
+	defer c.Destroy()
 
 	const N = 50
 	const NumGoroutines = 8
@@ -271,7 +271,7 @@ func TestWriteBatch2(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
-	defer c.Close()
+	defer c.Destroy()
 
 	wb := NewWriteBatch()
 
@@ -405,6 +405,11 @@ func TestWriteCloseOpen(t *testing.T) {
 		i++
 	}
 	t.Logf("%+v", c.Stats())
+
+	err = c.Destroy()
+	if err != nil {
+		t.Fatal(err)
+	}
 }
 
 func TestReadLastEntry(t *testing.T) {
@@ -436,14 +441,16 @@ func TestReadLastEntry(t *testing.T) {
 		t.Errorf("expected %d records, got %d", 2, entry.NumRecords)
 	}
 	t.Logf("%+v", c.Stats())
+
+	c.Destroy()
 }
 
 func TestSeekToFirstKey(t *testing.T) {
 	c, err := NewCollection("/tmp/test_seektofirstkey.lm2", 100)
 	if err != nil {
 		t.Fatal(err)
 	}
-	defer c.Close()
+	defer c.Destroy()
 
 	wb := NewWriteBatch()
 	wb.Set("a", "1")
@@ -479,7 +486,7 @@ func TestOverwriteFirstKey(t *testing.T) {
 	if err != nil {
 		t.Fatal(err)
 	}
-	defer c.Close()
+	defer c.Destroy()
 
 	wb := NewWriteBatch()
 	wb.Set("a", "1")

wal.go

@@ -260,3 +260,8 @@ func (w *wal) Truncate() error {
 func (w *wal) Close() {
 	w.f.Close()
 }
+
+func (w *wal) Destroy() error {
+	w.Close()
+	return os.Remove(w.f.Name())
+}