Added delete flag for etcd-import

ake-persson · Sep 21, 2015 · 4a22692 · 4a22692
1 parent 29faa21
commit 4a22692
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/github.com/mickep76/etcd-import/etcd-import.go b/src/github.com/mickep76/etcd-import/etcd-import.go
@@ -25,6 +25,7 @@ func main() {
 
 	// Options.
 	version := flag.Bool("version", false, "Version")
+	delete := flag.Bool("delete", false, "Delete entry before import")
 	node := flag.String("node", "", "Etcd node")
 	port := flag.String("port", "2379", "Etcd port")
 	dir := flag.String("dir", "/", "Etcd directory")
@@ -74,6 +75,10 @@ func main() {
 		log.Fatal("No input provided")
 	}
 
+	if *delete {
+		client.Delete(strings.TrimRight(*dir, "/"), true)
+	}
+
 	if err = etcdmap.Create(client, strings.TrimRight(*dir, "/"), reflect.ValueOf(m)); err != nil {
 		log.Fatal(err.Error())
 	}