auto create path

cmd/zk-syncer/main.go

@@ -18,6 +18,7 @@ func init() {
  flag.StringVar(&src, "src", "127.0.0.1:2181", "src zk addr")
  flag.StringVar(&dst, "dst", "127.0.0.1:2181", "dst zk addr")
  flag.StringVar(&path, "path", "/", "zkpath")
+ flag.Parse()
 }
 
 func main() {
@@ -41,6 +42,7 @@ func main() {
 func iterChild(src, dst *zk.Conn, path string) {
  childs, _, _ := src.Children(path)
  if len(childs) > 0 {
+ dstCheckPath(dst, path)
  for _, child := range childs {
  if path == "/" {
  path = ""
@@ -49,11 +51,33 @@ func iterChild(src, dst *zk.Conn, path string) {
  }
  } else {
  data, stat, _ := src.Get(path)
- _, err := dst.Set(path, data, stat.Version)
+ exist, _, _ := dst.Exists(path)
+ if exist {
+ _, err := dst.Set(path, data, stat.Version)
+ if err != nil {
+ fmt.Printf("path %s err: %s \n", path, err)
+ } else {
+ fmt.Printf("path %s replace ok\n", path)
+ }
+ } else {
+ _, err := dst.Create(path, data, 0, zk.WorldACL(zk.PermAll))
+ if err != nil {
+ fmt.Printf("path %s err: %s \n", path, err)
+ } else {
+ fmt.Printf("path %s create ok\n", path)
+ }
+ }
+ }
+}
+
+func dstCheckPath(dst *zk.Conn, path string) {
+ exist, _, _ := dst.Exists(path)
+ if !exist {
+ _, err := dst.Create(path, []byte{}, 0, zk.WorldACL(zk.PermAll))
  if err != nil {
  fmt.Printf("path %s err: %s \n", path, err)
  } else {
- fmt.Printf("path %s ok\n", path)
+ fmt.Printf("path %s create ok\n", path)
  }
  }
 }