Zookeeper lock

components/go.mod

@@ -6,8 +6,9 @@ require (
 	github.com/alicebob/miniredis/v2 v2.13.3
 	github.com/fsnotify/fsnotify v1.4.9 // indirect
 	github.com/go-redis/redis/v8 v8.8.0
+	github.com/go-zookeeper/zk v1.0.2
 	github.com/golang/protobuf v1.4.2 // indirect
-	github.com/google/go-cmp v0.5.5 // indirect
+    github.com/google/go-cmp v0.5.5 // indirect
 	github.com/google/uuid v1.1.1
 	github.com/stretchr/testify v1.7.0
 	github.com/valyala/fasthttp v1.26.0

components/go.sum

@@ -24,6 +24,10 @@ github.com/TarsCloud/TarsGo v1.1.4/go.mod h1:jkslu43B0kZitq9ztCH6ZM6l4rCKGkMjdqE
 github.com/alecthomas/template v0.0.0-20160405071501-a0175ee3bccc/go.mod h1:LOuyumcjzFXgccqObfd/Ljyb9UuFJ6TxHnclSeseNhc=
 github.com/alecthomas/units v0.0.0-20151022065526-2efee857e7cf/go.mod h1:ybxpYRFXyAe+OPACYpWeL0wqObRcbAqCMya13uyzqw0=
 github.com/alibaba/sentinel-golang v1.0.2-0.20210112133552-db6063eb263e/go.mod h1:uHDiUhfQjc9ACs9AdwP9CxmSze/B7tJ74QCdAbWl7x0=
+github.com/alicebob/gopher-json v0.0.0-20200520072559-a9ecdc9d1d3a h1:HbKu58rmZpUGpz5+4FfNmIU+FmZg2P3Xaj2v2bfNWmk=
+github.com/alicebob/gopher-json v0.0.0-20200520072559-a9ecdc9d1d3a/go.mod h1:SGnFV6hVsYE877CKEZ6tDNTjaSXYUk6QqoIK6PrAtcc=
+github.com/alicebob/miniredis/v2 v2.13.3 h1:kohgdtN58KW/r9ZDVmMJE3MrfbumwsDQStd0LPAGmmw=
+github.com/alicebob/miniredis/v2 v2.13.3/go.mod h1:uS970Sw5Gs9/iK3yBg0l9Uj9s25wXxSpQUE9EaJ/Blg=
 github.com/andybalholm/brotli v1.0.0/go.mod h1:loMXtMfwqflxFJPmdbJO0a3KNoPuLBgiu3qAvBg8x/Y=
 github.com/andybalholm/brotli v1.0.2 h1:JKnhI/XQ75uFBTiuzXpzFrUriDPiZjlOSzh6wXogP0E=
 github.com/andybalholm/brotli v1.0.2/go.mod h1:loMXtMfwqflxFJPmdbJO0a3KNoPuLBgiu3qAvBg8x/Y=
@@ -51,6 +55,9 @@ github.com/cespare/xxhash v1.1.0 h1:a6HrQnmkObjyL+Gs60czilIUGqrzKutQD6XZog3p+ko=
 github.com/cespare/xxhash v1.1.0/go.mod h1:XrSqR1VqqWfGrhpAt58auRo0WTKS1nRRg3ghfAqPWnc=
 github.com/cespare/xxhash/v2 v2.1.1 h1:6MnRN8NT7+YBpUIWxHtefFZOKTAPgGjpQSxqLNn0+qY=
 github.com/cespare/xxhash/v2 v2.1.1/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
+github.com/chzyer/logex v1.1.10/go.mod h1:+Ywpsq7O8HXn0nuIou7OrIPyXbp3wmkHB+jjWRnGsAI=
+github.com/chzyer/readline v0.0.0-20180603132655-2972be24d48e/go.mod h1:nSuG5e5PlCu98SY8svDHJxuZscDgtXS6KTTbou5AhLI=
+github.com/chzyer/test v0.0.0-20180213035817-a1ea475d72b1/go.mod h1:Q3SI9o4m/ZMnBNeIyt5eFwwo7qiLfzFZmjNmxjkiQlU=
 github.com/client9/misspell v0.3.4/go.mod h1:qj6jICC3Q7zFZvVWo7KLAzC3yx5G7kyvSDkc90ppPyw=
 github.com/cncf/udpa/go v0.0.0-20191209042840-269d4d468f6f h1:WBZRG4aNOuI15bLRrCgN8fCq8E5Xuty6jGbmSNEvSsU=
 github.com/cncf/udpa/go v0.0.0-20191209042840-269d4d468f6f/go.mod h1:M8M6+tZqaGXZJjfX53e64911xZQV5JYwmTeXPW+k8Sc=
@@ -117,6 +124,8 @@ github.com/go-playground/validator/v10 v10.2.0/go.mod h1:uOYAAleCW8F/7oMFd6aG0GO
 github.com/go-redis/redis/v8 v8.8.0 h1:fDZP58UN/1RD3DjtTXP/fFZ04TFohSYhjZDkcDe2dnw=
 github.com/go-redis/redis/v8 v8.8.0/go.mod h1:F7resOH5Kdug49Otu24RjHWwgK7u9AmtqWMnCV1iP5Y=
 github.com/go-stack/stack v1.8.0/go.mod h1:v0f6uXyyMGvRgIKkXu+yp6POWl0qKG85gN/melR3HDY=
+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=
 github.com/gogo/protobuf v1.1.1/go.mod h1:r8qH/GZQm5c6nD/R0oafs1akxWv10x8SbQlK7atdtwQ=
 github.com/gogo/protobuf v1.2.1 h1:/s5zKNz0uPFCZ5hddgPdo2TK2TVrUNMn0OOX8/aZMTE=
 github.com/gogo/protobuf v1.2.1/go.mod h1:hp+jE20tsWTFYpLwKvXlhS1hjn+gTNwPg2I6zVXpSg4=
@@ -171,6 +180,7 @@ github.com/google/martian v2.1.0+incompatible/go.mod h1:9I4somxYTbIHy5NJKHRl3wXi
 github.com/google/pprof v0.0.0-20181206194817-3ea8567a2e57/go.mod h1:zfwlbNMJ+OItoe0UupaVj+oy1omPYYDuagoSzA8v9mc=
 github.com/google/pprof v0.0.0-20190515194954-54271f7e092f/go.mod h1:zfwlbNMJ+OItoe0UupaVj+oy1omPYYDuagoSzA8v9mc=
 github.com/google/renameio v0.1.0/go.mod h1:KWCgfxg9yswjAJkECMjeO8J8rahYeXnNhOm40UhjYkI=
+github.com/google/uuid v1.1.1 h1:Gkbcsh/GbpXz7lPftLA3P6TYMwjCLYm83jiFQZF/3gY=
 github.com/google/uuid v1.1.1/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
 github.com/googleapis/gax-go/v2 v2.0.4/go.mod h1:0Wqv26UfaUD9n4G6kQubkQ+KchISgw+vpHVxEJEs9eg=
 github.com/googleapis/gax-go/v2 v2.0.5/go.mod h1:DWXyrwAJ9X0FpwwEdw+IPEYBICEFu5mhpdKc/us6bOk=
@@ -282,18 +292,21 @@ github.com/mosn/registry v0.0.0-20210108061200-d7b63bc1904b/go.mod h1:sO2+AtmZqI
 github.com/munnerz/goautoneg v0.0.0-20120707110453-a547fc61f48d/go.mod h1:+n7T8mK8HuQTcFwEeznm/DIxMOiR9yIdICNftLE1DvQ=
 github.com/mwitkow/go-conntrack v0.0.0-20161129095857-cc309e4a2223/go.mod h1:qRWi+5nqEBWmkhHvq77mSJWrCKwh8bxhgT7d/eI7P4U=
 github.com/mxk/go-flowrate v0.0.0-20140419014527-cca7078d478f/go.mod h1:ZdcZmHo+o7JKHSa8/e818NopupXU1YMK5fe1lsApnBw=
+github.com/nxadm/tail v1.4.4 h1:DQuhQpB1tVlglWS2hLQ5OV6B5r8aGxSrPc5Qo6uTN78=
 github.com/nxadm/tail v1.4.4/go.mod h1:kenIhsEOeOJmVchQTgglprH7qJGnHDVpk1VPCcaMI8A=
 github.com/oklog/run v1.0.0/go.mod h1:dlhp/R75TPv97u0XWUtDeV/lRKWPKSdTuV0TZvrmrQA=
 github.com/oklog/ulid v1.3.1/go.mod h1:CirwcVhetQ6Lv90oh/F+FBtV6XMibvdAFo93nm5qn4U=
 github.com/onsi/ginkgo v0.0.0-20170829012221-11459a886d9c/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE=
 github.com/onsi/ginkgo v1.6.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE=
 github.com/onsi/ginkgo v1.10.1/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE=
 github.com/onsi/ginkgo v1.12.1/go.mod h1:zj2OWP4+oCPe1qIXoGWkgMRwljMUYCdkwsT2108oapk=
+github.com/onsi/ginkgo v1.15.0 h1:1V1NfVQR87RtWAgp1lv9JZJ5Jap+XFGKPi00andXGi4=
 github.com/onsi/ginkgo v1.15.0/go.mod h1:hF8qUzuuC8DJGygJH3726JnCZX4MYbRB8yFfISqnKUg=
 github.com/onsi/gomega v0.0.0-20170829124025-dcabb60a477c/go.mod h1:C1qb7wdrVGGVU+Z6iS04AVkA3Q65CEZX59MT0QO5uiA=
 github.com/onsi/gomega v1.7.0/go.mod h1:ex+gbHU/CVuBBDIJjb2X0qEXbFg53c61hWP/1CpauHY=
 github.com/onsi/gomega v1.7.1/go.mod h1:XdKZgCCFLUoM/7CFJVPcG8C1xQ1AJ0vpAezJrB7JYyY=
 github.com/onsi/gomega v1.10.1/go.mod h1:iN09h71vgCQne3DLsj+A5owkum+a2tYe+TOCB1ybHNo=
+github.com/onsi/gomega v1.10.5 h1:7n6FEkpFmfCoo2t+YYqXH0evK+a9ICQz0xcAy9dYcaQ=
 github.com/onsi/gomega v1.10.5/go.mod h1:gza4q3jKQJijlu05nKWRCW/GavJumGt8aNRxWg7mt48=
 github.com/opentracing/opentracing-go v1.1.0/go.mod h1:UkNAQd3GIcIGf0SeVgPpRdFStlNbqXla1AfSYxPUl2o=
 github.com/openzipkin-contrib/zipkin-go-opentracing v0.3.5/go.mod h1:uVHyebswE1cCXr2A73cRM2frx5ld1RJUCJkFNZ90ZiI=
@@ -393,6 +406,8 @@ github.com/wasmerio/wasmer-go v1.0.3/go.mod h1:0gzVdSfg6pysA6QVp6iVRPTagC6Wq9pOE
 github.com/xiang90/probing v0.0.0-20190116061207-43a291ad63a2 h1:eY9dn8+vbi4tKz5Qo6v2eYzo7kUS51QINcR5jNpbZS8=
 github.com/xiang90/probing v0.0.0-20190116061207-43a291ad63a2/go.mod h1:UETIi67q53MR2AWcXfiuqkDkRtnGDLqkBTpCHuJHxtU=
 github.com/yuin/goldmark v1.2.1/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
+github.com/yuin/gopher-lua v0.0.0-20191220021717-ab39c6098bdb h1:ZkM6LRnq40pR1Ox0hTHlnpkcOTuFIDQpZ1IN8rKKhX0=
+github.com/yuin/gopher-lua v0.0.0-20191220021717-ab39c6098bdb/go.mod h1:gqRgreBUhTSL0GeU64rtZ3Uq3wtjOa/TB2YfrtkCbVQ=
 github.com/zouyx/agollo/v4 v4.0.7 h1:aUwdnCz+1jExZbZvN1nZ+kKjn62Ula5joekZHTccj2E=
 github.com/zouyx/agollo/v4 v4.0.7/go.mod h1:unhojnZiGLlT4gLpWz3Oa7sGcChZWv/1DJBkV6s8uAE=
 go.etcd.io/bbolt v1.3.2 h1:Z/90sZLPOeCy2PwprqkFa25PdkusRzaj9P8zm/KNyvk=
@@ -403,6 +418,7 @@ go.opentelemetry.io/otel v0.19.0 h1:Lenfy7QHRXPZVsw/12CWpxX6d/JkrX8wrx2vO8G80Ng=
 go.opentelemetry.io/otel v0.19.0/go.mod h1:j9bF567N9EfomkSidSfmMwIwIBuP37AMAIzVW85OxSg=
 go.opentelemetry.io/otel/metric v0.19.0 h1:dtZ1Ju44gkJkYvo+3qGqVXmf88tc+a42edOywypengg=
 go.opentelemetry.io/otel/metric v0.19.0/go.mod h1:8f9fglJPRnXuskQmKpnad31lcLJ2VmNNqIsx/uIwBSc=
+go.opentelemetry.io/otel/oteltest v0.19.0 h1:YVfA0ByROYqTwOxqHVZYZExzEpfZor+MU1rU+ip2v9Q=
 go.opentelemetry.io/otel/oteltest v0.19.0/go.mod h1:tI4yxwh8U21v7JD6R3BcA/2+RBoTKFexE/PJ/nSO7IA=
 go.opentelemetry.io/otel/trace v0.19.0 h1:1ucYlenXIDA1OlHVLDZKX0ObXV5RLaq06DtUKz5e5zc=
 go.opentelemetry.io/otel/trace v0.19.0/go.mod h1:4IXiNextNOpPnRlI4ryK69mn5iC84bjBWZQA5DXz/qg=
@@ -501,6 +517,7 @@ golang.org/x/sys v0.0.0-20181026203630-95b1ffbd15a5/go.mod h1:STP8DvDyc/dI5b8T5h
 golang.org/x/sys v0.0.0-20181107165924-66b7b1311ac8/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
 golang.org/x/sys v0.0.0-20181116152217-5ac8a444bdc5/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
 golang.org/x/sys v0.0.0-20190129075346-302c3dd5f1cc/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
+golang.org/x/sys v0.0.0-20190204203706-41f3e6584952/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
 golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
 golang.org/x/sys v0.0.0-20190312061237-fead79001313/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
@@ -625,6 +642,7 @@ gopkg.in/mgo.v2 v2.0.0-20190816093944-a6b53ec6cb22/go.mod h1:yeKp02qBN3iKW1OzL3M
 gopkg.in/natefinch/lumberjack.v2 v2.0.0 h1:1Lc07Kr7qY4U2YPouBjpCLxpiyxIVoxqXgkXLknAOE8=
 gopkg.in/natefinch/lumberjack.v2 v2.0.0/go.mod h1:l0ndWWf7gzL7RNwBG7wST/UCcT4T24xpD6X8LsfU/+k=
 gopkg.in/resty.v1 v1.12.0/go.mod h1:mDo4pnntr5jdWRML875a/NmxYqAlA73dVijT2AXvQQo=
+gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7 h1:uRGJdciOHaEIrze2W8Q3AKkepLTh2hOroT7a+7czfdQ=
 gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7/go.mod h1:dt/ZhP58zS4L8KSrWDmTeBkI65Dw0HsyUHuEVlX15mw=
 gopkg.in/yaml.v2 v2.0.0-20170812160011-eb3733d160e7/go.mod h1:JAlM8MvJe8wmxCU4Bli9HhUf9+ttbYbLASfIpnQbh74=
 gopkg.in/yaml.v2 v2.2.1/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=

components/lock/zookeeper/zookeeper_lock.go

@@ -0,0 +1,144 @@
+package zookeeper
+
+import (
+	"errors"
+	"fmt"
+	"github.com/go-zookeeper/zk"
+	"mosn.io/layotto/components/lock"
+	"mosn.io/pkg/log"
+	"strconv"
+	"strings"
+	"time"
+)
+
+const (
+	host                  = "zookeeperHosts"
+	password              = "zookeeperPassword"
+	sessionTimeout        = "sessionTimeout"
+	defaultSessionTimeout = time.Second * 3
+)
+
+type ZookeeperLock struct {
+	metadata metadata
+	logger   log.ErrorLogger
+}
+
+func NewZookeeperLock(logger log.ErrorLogger) *ZookeeperLock {
+	lock := &ZookeeperLock{
+		logger: logger,
+	}
+	return lock
+}
+
+func (p *ZookeeperLock) newConnection() (*zk.Conn, error) {
+	//make sure a lock and a connection
+	conn, _, err := zk.Connect(p.metadata.hosts, p.metadata.sessionTimeout)
+	if err != nil {
+		return nil, err
+	}
+	return conn, nil
+}
+
+func (p *ZookeeperLock) Init(metadata lock.Metadata) error {
+
+	m, err := parseZookeeperMetadata(metadata)
+	if err != nil {
+		return err
+	}
+	p.metadata = m
+
+	return nil
+}
+
+func (p *ZookeeperLock) Features() []lock.Feature {
+	return nil
+}
+func (p *ZookeeperLock) TryLock(req *lock.TryLockRequest) (*lock.TryLockResponse, error) {
+
+	conn, err := p.newConnection()
+	if err != nil {
+		return &lock.TryLockResponse{}, err
+	}
+	//1.create zk ephemeral node
+	_, err = conn.Create("/"+req.ResourceId, []byte(req.LockOwner), zk.FlagEphemeral, zk.WorldACL(zk.PermAll))
+
+	//2.1 create node fail ,indicates lock fail
+	if err != nil {
+		return &lock.TryLockResponse{
+			Success: false,
+		}, nil
+	}
+
+	//2.2 create node success, asyn  to make sure zkclient alive for need time
+	go func() {
+		// make sure close connecion
+		defer conn.Close()
+		//can also
+		//time.Sleep(time.Second * time.Duration(req.Expire))
+		timeAfterTrigger := time.After(time.Second * time.Duration(req.Expire))
+		<-timeAfterTrigger
+	}()
+
+	return &lock.TryLockResponse{
+		Success: true,
+	}, nil
+
+}
+func (p *ZookeeperLock) Unlock(req *lock.UnlockRequest) (*lock.UnlockResponse, error) {
+
+	conn, err := p.newConnection()
+	defer conn.Close()
+
+	if err != nil {
+		return &lock.UnlockResponse{Status: lock.INTERNAL_ERROR}, err
+	}
+
+	path := "/" + req.ResourceId
+	owner, state, err := conn.Get(path)
+	//1. node does not exist, indicates this lock has expired or wrong unlock
+	if err != nil {
+		return &lock.UnlockResponse{Status: lock.LOCK_UNEXIST}, nil
+	}
+	//2. node exist ,but owner not this, indicates this lock has expired or wrong unlock
+	if string(owner) != req.LockOwner {
+		return &lock.UnlockResponse{Status: lock.LOCK_BELONG_TO_OTHERS}, nil
+	}
+	err = conn.Delete(path, state.Version)
+	//3.owner is this, but delete fail, indicates this lock has expired
+	//this step contains problem, the lock maybe also UNEXIST
+	if err != nil {
+		return &lock.UnlockResponse{Status: lock.LOCK_BELONG_TO_OTHERS}, nil
+	}
+	//4.delete success, unlock succes
+	return &lock.UnlockResponse{Status: lock.SUCCESS}, nil
+}
+
+type metadata struct {
+	hosts          []string
+	password       string
+	sessionTimeout time.Duration
+}
+
+func parseZookeeperMetadata(meta lock.Metadata) (metadata, error) {
+	m := metadata{}
+	if val, ok := meta.Properties[host]; ok && val != "" {
+		split := strings.Split(val, ";")
+		m.hosts = append(m.hosts, split...)
+	} else {
+		return m, errors.New("zookeeper store error: missing host address")
+	}
+
+	if val, ok := meta.Properties[password]; ok && val != "" {
+		m.password = val
+	}
+
+	m.sessionTimeout = defaultSessionTimeout
+	if val, ok := meta.Properties[sessionTimeout]; ok && val != "" {
+		parsedVal, err := strconv.Atoi(val)
+		if err != nil {
+			return m, fmt.Errorf("zookeeper store error: can't parse sessionTimeout field: %s", err)
+		}
+		m.sessionTimeout = time.Duration(parsedVal)
+	}
+	return m, nil
+}

components/lock/zookeeper/zookeeper_lock_test.go

@@ -0,0 +1,133 @@
+package zookeeper
+
+import (
+	"github.com/stretchr/testify/assert"
+	"mosn.io/layotto/components/lock"
+	"mosn.io/pkg/log"
+	"testing"
+	"time"
+)
+
+const resouseId = "resoure_1"
+const lockOwerA = "p1"
+const lockOwerB = "p2"
+const expireTime = 5
+
+var cfg = lock.Metadata{
+	Properties: make(map[string]string),
+}
+
+func TestMain(m *testing.M) {
+	cfg.Properties["zookeeperHosts"] = "127.0.0.1;127.0.0.1"
+	cfg.Properties["zookeeperPassword"] = ""
+	m.Run()
+}
+
+// A lock ,A unlock
+func TestZookeeperLock_One(t *testing.T) {
+	comp := NewZookeeperLock(log.DefaultLogger)
+
+	comp.Init(cfg)
+
+	tryLock, err := comp.TryLock(&lock.TryLockRequest{
+		ResourceId: resouseId,
+		LockOwner:  lockOwerA,
+		Expire:     expireTime,
+	})
+	assert.NoError(t, err)
+	assert.Equal(t, tryLock.Success, true)
+	unlock, _ := comp.Unlock(&lock.UnlockRequest{
+		ResourceId: resouseId,
+		LockOwner:  lockOwerA,
+	})
+	assert.NoError(t, err)
+	assert.Equal(t, unlock.Status, lock.SUCCESS)
+
+}
+
+// A lock ,B unlock
+func TestZookeeperLock_Two(t *testing.T) {
+	comp := NewZookeeperLock(log.DefaultLogger)
+
+	comp.Init(cfg)
+
+	tryLock, err := comp.TryLock(&lock.TryLockRequest{
+		ResourceId: resouseId,
+		LockOwner:  lockOwerA,
+		Expire:     expireTime,
+	})
+	assert.NoError(t, err)
+	assert.Equal(t, tryLock.Success, true)
+	unlock, err := comp.Unlock(&lock.UnlockRequest{
+		ResourceId: resouseId,
+		LockOwner:  lockOwerB,
+	})
+	assert.NoError(t, err)
+	assert.Equal(t, unlock.Status, lock.LOCK_BELONG_TO_OTHERS)
+
+}
+
+// A lock , B lock ,A unlock ,A lock,B lock,B unlock
+func TestZookeeperLock_Three(t *testing.T) {
+	comp := NewZookeeperLock(log.DefaultLogger)
+
+	comp.Init(cfg)
+	//A lock
+	tryLock, err := comp.TryLock(&lock.TryLockRequest{
+		ResourceId: resouseId,
+		LockOwner:  lockOwerA,
+		Expire:     expireTime,
+	})
+	assert.NoError(t, err)
+	assert.Equal(t, tryLock.Success, true)
+	//B lock
+	tryLock, err = comp.TryLock(&lock.TryLockRequest{
+		ResourceId: resouseId,
+		LockOwner:  lockOwerB,
+		Expire:     expireTime,
+	})
+	assert.NoError(t, err)
+	assert.Equal(t, tryLock.Success, false)
+	//A unlock
+	unlock, _ := comp.Unlock(&lock.UnlockRequest{
+		ResourceId: resouseId,
+		LockOwner:  lockOwerA,
+	})
+	assert.NoError(t, err)
+	assert.Equal(t, unlock.Status, lock.SUCCESS)
+
+	//B lock
+	tryLock, err = comp.TryLock(&lock.TryLockRequest{
+		ResourceId: resouseId,
+		LockOwner:  lockOwerB,
+		Expire:     expireTime,
+	})
+	assert.NoError(t, err)
+	assert.Equal(t, tryLock.Success, true)
+
+	//B unlock
+	unlock, _ = comp.Unlock(&lock.UnlockRequest{
+		ResourceId: resouseId,
+		LockOwner:  lockOwerB,
+	})
+	assert.NoError(t, err)
+	assert.Equal(t, unlock.Status, lock.SUCCESS)
+}
+
+//A lock
+func TestZookeeperLock_Four(t *testing.T) {
+	comp := NewZookeeperLock(log.DefaultLogger)
+	comp.Init(cfg)
+	go func() {
+		tryLock, err := comp.TryLock(&lock.TryLockRequest{
+			ResourceId: resouseId,
+			LockOwner:  lockOwerA,
+			Expire:     expireTime,
+		})
+		assert.NoError(t, err)
+		assert.Equal(t, tryLock.Success, true)
+	}()
+
+	time.Sleep(time.Second * 9999)
+
+}