基于 Zookeeper 实现的分布式锁

在 go-zookeeper Lock 的基础上新增了context.Context，同时能够对给定的key加锁，减小锁的颗粒度。

Example

package main

import (
	"context"
	"github.com/go-zookeeper/zk"
	zklock "github.com/xuqil/zk-lock"
	"log"
	"os"
	"os/signal"
	"time"
)

var servers = []string{"localhost:2181"}
var notify = make(chan struct{})

func main() {
	conn, _, err := zk.Connect(servers, 5*time.Second)
	if err != nil {
		log.Fatalln(err)
	}
	defer conn.Close()

	for i := 0; i < 100; i++ {
		go run(conn, i)
	}
	quit := make(chan os.Signal, 1)
	signal.Notify(quit, os.Interrupt, os.Kill)
	<-quit
	close(notify)

}

func run(conn *zk.Conn, i int) {
	for {
		select {
		case <-notify:
			return
		default:
			lock := zklock.NewLock(conn, "key", zk.WorldACL(zk.PermAll))

			if err := lock.Lock(context.Background()); err != nil {
				log.Fatalf("goroutine[%d] lock failed: %s", i, err)
			}
			log.Printf("goroutine[%d] lock", i)
			
			if err := lock.Unlock(context.Background()); err != nil {
				log.Fatalf("goroutine[%d] unlock failed: %s", i, err)
			}
			log.Printf("goroutine[%d] unlock", i)
		}
	}
}