~~~ package zookeeper import ( "errors" "github.com/samuel/go-zookeeper/zk" "project/library/dao/common" "project/library/dao/config" "time" ) type ZookeeperMgr struct { Conn *zk.Conn /* flags有4種取值: 0:永久,除非手動刪除 1:短暫,session斷開則改節點也被刪除 2:會自動在節點后面添加序號 3:即,短暫且自動添加序號 */ //初始化時默認0 flags int32 //初始化時默認允許所有操作 acl []zk.ACL } func InitZookeeper(cfg *config.ZkCfg) *ZookeeperMgr { conn, _, err := zk.Connect(cfg.Servers, time.Second*5) if err != nil { common.SimplePanic("InitZookeeper失敗", err.Error()) } return &ZookeeperMgr{Conn: conn, acl: zk.WorldACL(zk.PermAll)} } func InitZookeeper2(cfg *config.ZkCfg) (error, *ZookeeperMgr) { conn, _, err := zk.Connect(cfg.Servers, time.Second*5) if err != nil { return errors.New("InitZookeeper失敗"), nil } return nil, &ZookeeperMgr{Conn: conn, acl: zk.WorldACL(zk.PermAll)} } func (mgr *ZookeeperMgr) Create(path string, data []byte) (string, error) { return mgr.Conn.Create(path, data, mgr.flags, mgr.acl) } func (mgr *ZookeeperMgr) Delete(path string) error { _, stat, err := mgr.Conn.Get(path) if err != nil { return err } return mgr.Conn.Delete(path, stat.Version) } func (mgr *ZookeeperMgr) Update(path string, data []byte) error { _, stat, err := mgr.Conn.Get(path) if err != nil { return err } _, err = mgr.Conn.Set(path, data, stat.Version) return err } func (mgr *ZookeeperMgr) Get(path string) ([]byte, *zk.Stat, error) { return mgr.Conn.Get(path) } ~~~