示例#1
0
文件: service.go 项目: ycaihua/bamboo
func All(conn *zk.Conn, zkConf conf.Zookeeper) (map[string]Service, error) {

	err := ensurePathExists(conn, zkConf.Path)
	if err != nil {
		return nil, err
	}

	services := map[string]Service{}
	keys, _, err2 := conn.Children(zkConf.Path)

	if err2 != nil {
		return nil, err2
	}

	for _, childPath := range keys {
		bite, _, e := conn.Get(zkConf.Path + "/" + childPath)
		if e != nil {
			return nil, e
			break
		}
		appId, _ := unescapeSlashes(childPath)
		services[appId] = Service{Id: appId, Acl: string(bite)}
	}
	return services, nil
}
示例#2
0
func pollZooKeeper(conn *zk.Conn, path string, evts chan zk.Event, quit chan bool) {
	logger.Println("####################################Pooling zookeeper")
	logger.Println(path)

	children, _, err := conn.Children(path)
	if err != nil {
		fmt.Printf("%+v", err)
		panic(err)
	}

	watcherControl := make([]chan<- bool, len(children)+2)
	watcherControl[0] = sinkSelfEvents(conn, path, evts)
	watcherControl[1] = sinkChildEvents(conn, path, evts)

	for i, child := range children {
		p := path + "/" + child
		watcherControl[i+2] = sinkSelfEvents(conn, p, evts)
	}

	<-quit

	for _, ch := range watcherControl {
		ch <- true
	}
}
示例#3
0
func loadToZK(conn *zk.Conn, data [][2]string) {
	for _, entry := range data {
		k := entry[0]
		v := entry[1]
		_, err := conn.Create(k, []byte(v), 0, defaultACL())
		orPanic(err)
	}
}
示例#4
0
/*
   Read ZK ACL:
   http://zookeeper.apache.org/doc/trunk/zookeeperProgrammers.html#sc_ACLPermissions
*/
func Create(conn *zk.Conn, zkConf conf.Zookeeper, appId string, domainValue string) (string, error) {
	path := concatPath(zkConf.Path, validateAppId(appId))
	resPath, err := conn.Create(path, []byte(domainValue), 0, defaultACL())
	if err != nil {
		return "", err
	}

	return resPath, nil
}
示例#5
0
func ensurePathExists(conn *zk.Conn, path string) error {
	pathExists, _, _ := conn.Exists(path)
	if pathExists {
		return nil
	}

	_, err := conn.Create(path, []byte{}, 0, defaultACL())
	if err != nil {
		return err
	}

	return nil
}
示例#6
0
func deleteRecursive(conn *zk.Conn, path string) {
	children, _, err := conn.Children(path)
	if err != nil {
		log.Printf("failed to get children %s: %s", path, err)
		return
	}
	for _, child := range children {
		deleteRecursive(conn, path+"/"+child)
	}
	err = conn.Delete(path, -1)
	if err != nil {
		log.Printf("failed to delete %s: %s", path, err)
	}
}
示例#7
0
func ListenToConn(c *zk.Conn, path string, deb bool, repDelay time.Duration) (chan zk.Event, chan bool) {
	exists, _, err := c.Exists(path)

	if err != nil {
		logger.Fatalf("Couldn't determine whether node %v exists in Zookeeper", path)
	}
	if !exists {
		logger.Printf("Node '%v' does not exist in Zookeeper, creating...", path)
		err := zkNodeCreateByPath(path, c)
		if err != nil {
			logger.Fatalf("Unable to create path '%v': %v", path, err)
		}

	}

	quit := make(chan bool)
	evts := make(chan zk.Event)

	go pollZooKeeper(c, path, evts, quit)

	if deb {
		evts = debounce(evts, 100*time.Millisecond)
	}
	if repDelay > 0 {
		evts = delay(evts, repDelay)
	}
	return evts, quit
}
示例#8
0
func Put(conn *zk.Conn, zkConf conf.Zookeeper, appId string, domainValue string) (*zk.Stat, error) {
	path := concatPath(zkConf.Path, validateAppId(appId))
	err := ensurePathExists(conn, path)
	if err != nil {
		return nil, err
	}

	stats, err := conn.Set(path, []byte(domainValue), -1)
	if err != nil {
		return nil, err
	}

	// Force triger an event on parent
	conn.Set(zkConf.Path, []byte{}, -1)

	return stats, nil
}
示例#9
0
func zkCreateNodes(path string, nodes []string, c *zk.Conn) error {
	if len(nodes) > 0 {
		// strings.Split will return empty-strings for leading split chars, lets skip over these.
		if len(nodes[0]) == 0 {
			return zkCreateNodes(path, nodes[1:], c)
		}
		fqPath := path + "/" + nodes[0]
		log.Printf("Creating path: %v", fqPath)
		exists, _, err := c.Exists(fqPath)
		if err != nil {
			return err
		}
		if !exists {
			_, err := c.Create(fqPath, []byte{}, 0, zk.WorldACL(zk.PermAll))
			if err != nil {
				return err
			}
		}
		return zkCreateNodes(fqPath, nodes[1:], c)
	}
	return nil
}
示例#10
0
func sinkSelfEvents(conn *zk.Conn, path string, sink chan<- zk.Event) chan<- bool {
	control := make(chan bool)
	go func() {
		_, _, selfCh, err := conn.GetW(path)
		if err != nil {
			logger.Panicf("failed to set listener on path: %s", err.Error())
		}
		for {
			select {
			case _ = <-control:
				break
			case ev := <-selfCh:
				sink <- ev
				_, _, selfCh, err = conn.GetW(path)
				if err != nil {
					logger.Printf("failed to set listener on path: %s\n", err.Error())
				}
			}
		}
	}()

	return control
}
示例#11
0
func sinkChildEvents(conn *zk.Conn, path string, sink chan<- zk.Event) chan<- bool {
	control := make(chan bool)
	go func() {
		_, _, selfCh, err := conn.ChildrenW(path)
		if err != nil {
			logger.Panic("failed to set listener on path")
		}
		for {
			select {
			case _ = <-control:
				break
			case ev := <-selfCh:
				sink <- ev
				_, _, selfCh, err = conn.ChildrenW(path)
				if err != nil {
					logger.Panic("failed to set listener on path")
				}
			}
		}
	}()

	return control
}
示例#12
0
func Delete(conn *zk.Conn, zkConf conf.Zookeeper, appId string) error {
	path := concatPath(zkConf.Path, validateAppId(appId))
	return conn.Delete(path, -1)
}