add a command to force compaction of a volume, removing deleted files

2 files changed, 45 insertions, 0 deletions

diff --git a/go/weed/compact.go b/go/weed/compact.go
new file mode 100644
index 000000000..30ae6abd2
--- /dev/null
+++ b/go/weed/compact.go
@@ -0,0 +1,44 @@
+package main
+
+import (
+	"code.google.com/p/weed-fs/go/glog"
+	"code.google.com/p/weed-fs/go/storage"
+)
+
+func init() {
+	cmdCompact.Run = runCompact // break init cycle
+	cmdCompact.IsDebug = cmdCompact.Flag.Bool("debug", false, "enable debug mode")
+}
+
+var cmdCompact = &Command{
+	UsageLine: "compact -dir=/tmp -volumeId=234",
+	Short:     "run weed tool compact on volume file if corrupted",
+	Long: `Force an compaction to remove deleted files from volume files.
+  The compacted .dat file is stored as .cpd file.
+  The compacted .idx file is stored as .cpx file.
+
+  `,
+}
+
+var (
+	compactVolumePath = cmdCompact.Flag.String("dir", "/tmp", "data directory to store files")
+	compactVolumeId   = cmdCompact.Flag.Int("volumeId", -1, "a volume id. The volume should already exist in the dir.")
+)
+
+func runCompact(cmd *Command, args []string) bool {
+
+	if *compactVolumeId == -1 {
+		return false
+	}
+
+	vid := storage.VolumeId(*compactVolumeId)
+	v, err := storage.NewVolume(*compactVolumePath, vid, storage.CopyNil)
+	if err != nil {
+		glog.Fatalf("Load Volume [ERROR] %s\n", err)
+	}
+	if err = v.Compact(); err != nil {
+		glog.Fatalf("Compact Volume [ERROR] %s\n", err)
+	}
+
+	return true
+}
diff --git a/go/weed/weed.go b/go/weed/weed.go
index e9bc4e37d..0b1512636 100644
--- a/go/weed/weed.go
+++ b/go/weed/weed.go
@@ -22,6 +22,7 @@ var IsDebug *bool
 var server *string
 
 var commands = []*Command{
+  cmdCompact,
 	cmdFix,
 	cmdMaster,
 	cmdUpload,