mirror of
https://github.com/mainnika/nodesync.git
synced 2026-05-22 15:53:34 +00:00
Make nodesync structure instance
This commit is contained in:
@@ -0,0 +1,5 @@
|
||||
module code.tokarch.uk/mainnika/nodesync
|
||||
|
||||
go 1.17
|
||||
|
||||
require github.com/go-zookeeper/zk v1.0.2
|
||||
@@ -0,0 +1,2 @@
|
||||
github.com/go-zookeeper/zk v1.0.2 h1:4mx0EYENAdX/B/rbunjlt5+4RTA/a9SMHBRuSKdGxPM=
|
||||
github.com/go-zookeeper/zk v1.0.2/go.mod h1:nOB03cncLtlp4t+UAkGSV+9beXP/akpekBwL+UX1Qcw=
|
||||
+48
@@ -0,0 +1,48 @@
|
||||
package sync
|
||||
|
||||
import (
|
||||
"github.com/go-zookeeper/zk"
|
||||
|
||||
"fmt"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// createRecursively creates zookeeper path recursively
|
||||
func (s *NodeSync) createRecursively(path string, acl []zk.ACL) error {
|
||||
|
||||
parts := strings.Split(path, "/")
|
||||
|
||||
if len(parts) < 1 {
|
||||
return fmt.Errorf("path does not contain valid path")
|
||||
}
|
||||
|
||||
for i := range parts {
|
||||
Ensuring:
|
||||
for {
|
||||
child := fmt.Sprintf("%s", strings.Join(parts[:i+1], "/"))
|
||||
|
||||
if len(child) == 0 {
|
||||
break
|
||||
}
|
||||
|
||||
exists, _, err := s.Zk.Exists(child)
|
||||
switch true {
|
||||
case err == zk.ErrSessionExpired:
|
||||
return err
|
||||
case err != nil:
|
||||
continue
|
||||
case exists:
|
||||
break Ensuring
|
||||
default:
|
||||
}
|
||||
|
||||
if _, err = s.Zk.Create(child, nil, 0, acl); err != nil {
|
||||
continue
|
||||
}
|
||||
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
+28
@@ -0,0 +1,28 @@
|
||||
package sync
|
||||
|
||||
import (
|
||||
"github.com/go-zookeeper/zk"
|
||||
)
|
||||
|
||||
// NodeSync structure
|
||||
type NodeSync struct {
|
||||
Zk *zk.Conn
|
||||
|
||||
rootPath string
|
||||
}
|
||||
|
||||
// Inside private environment keep everything open
|
||||
var defaultAcl = zk.WorldACL(zk.PermAll)
|
||||
|
||||
// New creates a new nodesync instance
|
||||
func New(zkconn *zk.Conn, rootPath string) (nodeSync *NodeSync, err error) {
|
||||
|
||||
nodeSync = &NodeSync{Zk: zkconn, rootPath: rootPath}
|
||||
|
||||
err = nodeSync.createRecursively(rootPath, defaultAcl)
|
||||
if err != nil {
|
||||
nodeSync = nil
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
Reference in New Issue
Block a user