diff --git a/.github/workflows/gf.yml b/.github/workflows/gf.yml index 0bfb9e296..2a6efe3fb 100644 --- a/.github/workflows/gf.yml +++ b/.github/workflows/gf.yml @@ -127,6 +127,11 @@ jobs: ports: - 5236:5236 + zookeeper: + image: loads/zookeeper:3.8 + ports: + - 2181:2181 + strategy: matrix: go-version: [ "1.15", "1.16", "1.17", "1.18" ] diff --git a/contrib/registry/zookeeper/README.MD b/contrib/registry/zookeeper/README.MD new file mode 100644 index 000000000..2cda11ed2 --- /dev/null +++ b/contrib/registry/zookeeper/README.MD @@ -0,0 +1,71 @@ +# GoFrame Etcd Registry + +Use `zookeeper` as service registration and discovery management. + +## Installation +``` +go get -u -v github.com/gogf/gf/contrib/registry/zookeeper/v2 +``` +suggested using `go.mod`: +``` +require github.com/gogf/gf/contrib/registry/zookeeper/v2 latest +``` + +```go +package main + +import ( + "github.com/gogf/gf/contrib/registry/zookeeper/v2" + "github.com/gogf/gf/v2/frame/g" + "github.com/gogf/gf/v2/net/ghttp" + "github.com/gogf/gf/v2/net/gsvc" +) + +func main() { + r := zookeeper.New([]string{"127.0.0.1:2181"}, zookeeper.WithRootPath("/gogf")) + gsvc.SetRegistry(r) + + s := g.Server(`hello.svc`) + s.BindHandler("/", func(r *ghttp.Request) { + g.Log().Info(r.Context(), `request received`) + r.Response.Write(`Hello world`) + }) + s.Run() +} +``` + +```go +package main + +import ( + "fmt" + "time" + + "github.com/gogf/gf/contrib/registry/zookeeper/v2" + "github.com/gogf/gf/v2/frame/g" + "github.com/gogf/gf/v2/net/gsel" + "github.com/gogf/gf/v2/net/gsvc" + "github.com/gogf/gf/v2/os/gctx" +) + +func main() { + gsvc.SetRegistry(zookeeper.New([]string{"127.0.0.1:2181"},zookeeper.WithRootPath("/gogf"))) + gsel.SetBuilder(gsel.NewBuilderRoundRobin()) + + client := g.Client() + for i := 0; i < 100; i++ { + res, err := client.Get(gctx.New(), `http://hello.svc/`) + if err != nil { + panic(err) + } + fmt.Println(res.ReadAllString()) + res.Close() + time.Sleep(time.Second) + } +} +``` + +## License + +`GoFrame zookeeper` is licensed under the [MIT License](../../../LICENSE), 100% free and open-source, forever. + diff --git a/contrib/registry/zookeeper/go.mod b/contrib/registry/zookeeper/go.mod new file mode 100644 index 000000000..d75bb252b --- /dev/null +++ b/contrib/registry/zookeeper/go.mod @@ -0,0 +1,11 @@ +module github.com/gogf/gf/contrib/registry/zookeeper/v2 + +go 1.15 + +require ( + github.com/go-zookeeper/zk v1.0.3 + github.com/gogf/gf/v2 v2.2.2 + golang.org/x/sync v0.1.0 +) + +replace github.com/gogf/gf/v2 => ../../../ \ No newline at end of file diff --git a/contrib/registry/zookeeper/go.sum b/contrib/registry/zookeeper/go.sum new file mode 100644 index 000000000..262e8dfcd --- /dev/null +++ b/contrib/registry/zookeeper/go.sum @@ -0,0 +1,172 @@ +github.com/BurntSushi/toml v1.1.0 h1:ksErzDEI1khOiGPgpwuI7x2ebx/uXQNw7xJpn9Eq1+I= +github.com/BurntSushi/toml v1.1.0/go.mod h1:CxXYINrC8qIiEnFrOxCa7Jy5BFHlXnUU2pbicEuybxQ= +github.com/cespare/xxhash/v2 v2.1.2 h1:YRXhKfTDauu4ajMg1TPgFO5jnlC2HCbmLXMcTG5cbYE= +github.com/cespare/xxhash/v2 v2.1.2/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/clbanning/mxj/v2 v2.5.5 h1:oT81vUeEiQQ/DcHbzSytRngP6Ky9O+L+0Bw0zSJag9E= +github.com/clbanning/mxj/v2 v2.5.5/go.mod h1:hNiWqW14h+kc+MdF9C6/YoRfjEJoR3ou6tn/Qo+ve2s= +github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= +github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f h1:lO4WD4F/rVNCu3HqELle0jiPLLBs70cWOduZpkS1E78= +github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f/go.mod h1:cuUVRXasLTGF7a8hSLbxyZXjz+1KgoB3wDUb6vlszIc= +github.com/fatih/color v1.13.0 h1:8LOYc1KYPPmyKMuN8QV2DNRWNbLo6LZ0iLs8+mlH53w= +github.com/fatih/color v1.13.0/go.mod h1:kLAiJbzzSOZDVNGyDpeOxJ47H46qBXwg5ILebYFFOfk= +github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo= +github.com/fsnotify/fsnotify v1.4.9/go.mod h1:znqG4EE+3YCdAaPaxE2ZRY/06pZUdp0tY4IgpuI1SZQ= +github.com/fsnotify/fsnotify v1.5.4 h1:jRbGcIw6P2Meqdwuo0H1p6JVLbL5DHKAKlYndzMwVZI= +github.com/fsnotify/fsnotify v1.5.4/go.mod h1:OVB6XrOHzAwXMpEM7uPOzcehqUV2UqJxmVXmkdnm1bU= +github.com/go-logr/logr v1.2.2/go.mod h1:jdQByPbusPIv2/zmleS9BjJVeZ6kBagPoEUsqbVz/1A= +github.com/go-logr/logr v1.2.3 h1:2DntVwHkVopvECVRSlL5PSo9eG+cAkDCuckLubN+rq0= +github.com/go-logr/logr v1.2.3/go.mod h1:jdQByPbusPIv2/zmleS9BjJVeZ6kBagPoEUsqbVz/1A= +github.com/go-logr/stdr v1.2.2 h1:hSWxHoqTgW2S2qGc0LTAI563KZ5YKYRhT3MFKZMbjag= +github.com/go-logr/stdr v1.2.2/go.mod h1:mMo/vtBO5dYbehREoey6XUKy/eSumjCCveDpRre4VKE= +github.com/go-redis/redis/v8 v8.11.5 h1:AcZZR7igkdvfVmQTPnu9WE37LRrO/YrBH5zWyjDC0oI= +github.com/go-redis/redis/v8 v8.11.5/go.mod h1:gREzHqY1hg6oD9ngVRbLStwAWKhA0FEgq8Jd4h5lpwo= +github.com/go-task/slim-sprig v0.0.0-20210107165309-348f09dbbbc0/go.mod h1:fyg7847qk6SyHyPtNmDHnmrv/HOrqktSC+C9fM+CJOE= +github.com/go-zookeeper/zk v1.0.3 h1:7M2kwOsc//9VeeFiPtf+uSJlVpU66x9Ba5+8XK7/TDg= +github.com/go-zookeeper/zk v1.0.3/go.mod h1:nOB03cncLtlp4t+UAkGSV+9beXP/akpekBwL+UX1Qcw= +github.com/gogf/gf/v2 v2.2.2 h1:ew4k/VSr/gcPdMZcI8/HZYBYPjB0KOk6bQqA61M8bYE= +github.com/gogf/gf/v2 v2.2.2/go.mod h1:thvkyb43RWUu/m05sRm4CbH9r7t7/FrW2M56L9Ystwk= +github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U= +github.com/golang/protobuf v1.4.0-rc.1/go.mod h1:ceaxUfeHdC40wWswd/P6IGgMaK3YpKi5j83Wpe3EHw8= +github.com/golang/protobuf v1.4.0-rc.1.0.20200221234624-67d41d38c208/go.mod h1:xKAWHe0F5eneWXFV3EuXVDTCmh+JuBKY0li0aMyXATA= +github.com/golang/protobuf v1.4.0-rc.2/go.mod h1:LlEzMj4AhA7rCAGe4KMBDvJI+AwstrUpVNzEA03Pprs= +github.com/golang/protobuf v1.4.0-rc.4.0.20200313231945-b860323f09d0/go.mod h1:WU3c8KckQ9AFe+yFwt9sWVRKCVIyN9cPHBJSNnbL67w= +github.com/golang/protobuf v1.4.0/go.mod h1:jodUvKwWbYaEsadDk5Fwe5c77LiNKVO9IDvqG2KuDX0= +github.com/golang/protobuf v1.4.2/go.mod h1:oDoupMAO8OvCJWAcko0GGGIgR6R6ocIYbsSw735rRwI= +github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk= +github.com/golang/protobuf v1.5.2 h1:ROPKBNFfQgOUMifHyP+KYbvpjbdoFNs+aK7DXlji0Tw= +github.com/golang/protobuf v1.5.2/go.mod h1:XVQd3VNwM+JqD3oG2Ue2ip4fOMUkwXdXDdiuN0vRsmY= +github.com/google/go-cmp v0.3.0/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU= +github.com/google/go-cmp v0.3.1/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU= +github.com/google/go-cmp v0.4.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= +github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= +github.com/google/go-cmp v0.5.7 h1:81/ik6ipDQS2aGcBfIN5dHDB36BwrStyeAQquSYCV4o= +github.com/google/go-cmp v0.5.7/go.mod h1:n+brtR0CgQNWTVd5ZUFpTBC8YFBDLK/h/bpaJ8/DtOE= +github.com/google/pprof v0.0.0-20210407192527-94a9f03dee38/go.mod h1:kpwsk12EmLew5upagYY7GY0pfYCcupk39gWOCRROcvE= +github.com/gorilla/websocket v1.5.0 h1:PPwGk2jz7EePpoHN/+ClbZu8SPxiqlu12wZP/3sWmnc= +github.com/gorilla/websocket v1.5.0/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE= +github.com/grokify/html-strip-tags-go v0.0.1 h1:0fThFwLbW7P/kOiTBs03FsJSV9RM2M/Q/MOnCQxKMo0= +github.com/grokify/html-strip-tags-go v0.0.1/go.mod h1:2Su6romC5/1VXOQMaWL2yb618ARB8iVo6/DR99A6d78= +github.com/hpcloud/tail v1.0.0/go.mod h1:ab1qPbhIpdTxEkNHXyeSf5vhxWSCs/tWer42PpOxQnU= +github.com/ianlancetaylor/demangle v0.0.0-20200824232613-28f6c0f3b639/go.mod h1:aSSvb/t6k1mPoxDqO4vJh6VOCGPwU4O0C2/Eqndh1Sc= +github.com/magiconair/properties v1.8.6 h1:5ibWZ6iY0NctNGWo87LalDlEZ6R41TqbbDamhfG/Qzo= +github.com/magiconair/properties v1.8.6/go.mod h1:y3VJvCyxH9uVvJTWEGAELF3aiYNyPKd5NZ3oSwXrF60= +github.com/mattn/go-colorable v0.1.9 h1:sqDoxXbdeALODt0DAeJCVp38ps9ZogZEAXjus69YV3U= +github.com/mattn/go-colorable v0.1.9/go.mod h1:u6P/XSegPjTcexA+o6vUJrdnUu04hMope9wVRipJSqc= +github.com/mattn/go-isatty v0.0.12/go.mod h1:cbi8OIDigv2wuxKPP5vlRcQ1OAZbq2CE4Kysco4FUpU= +github.com/mattn/go-isatty v0.0.14 h1:yVuAays6BHfxijgZPzw+3Zlu5yQgKGP2/hcQbHb7S9Y= +github.com/mattn/go-isatty v0.0.14/go.mod h1:7GGIvUiUoEMVVmxf/4nioHXj79iQHKdU27kJ6hsGG94= +github.com/mattn/go-runewidth v0.0.9 h1:Lm995f3rfxdpd6TSmuVCHVb/QhupuXlYr8sCI/QdE+0= +github.com/mattn/go-runewidth v0.0.9/go.mod h1:H031xJmbD/WCDINGzjvQ9THkh0rPKHF+m2gUSrubnMI= +github.com/nxadm/tail v1.4.4/go.mod h1:kenIhsEOeOJmVchQTgglprH7qJGnHDVpk1VPCcaMI8A= +github.com/nxadm/tail v1.4.8 h1:nPr65rt6Y5JFSKQO7qToXr7pePgD6Gwiw05lkbyAQTE= +github.com/nxadm/tail v1.4.8/go.mod h1:+ncqLTQzXmGhMZNUePPaPqPvBxHAIsmXswZKocGu+AU= +github.com/olekukonko/tablewriter v0.0.5 h1:P2Ga83D34wi1o9J6Wh1mRuqd4mF/x/lgBS7N7AbDhec= +github.com/olekukonko/tablewriter v0.0.5/go.mod h1:hPp6KlRPjbx+hW8ykQs1w3UBbZlj6HuIJcUGPhkA7kY= +github.com/onsi/ginkgo v1.6.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE= +github.com/onsi/ginkgo v1.12.1/go.mod h1:zj2OWP4+oCPe1qIXoGWkgMRwljMUYCdkwsT2108oapk= +github.com/onsi/ginkgo v1.16.4/go.mod h1:dX+/inL/fNMqNlz0e9LfyB9TswhZpCVdJM/Z6Vvnwo0= +github.com/onsi/ginkgo v1.16.5 h1:8xi0RTUf59SOSfEtZMvwTvXYMzG4gV23XVHOZiXNtnE= +github.com/onsi/ginkgo v1.16.5/go.mod h1:+E8gABHa3K6zRBolWtd+ROzc/U5bkGt0FwiG042wbpU= +github.com/onsi/ginkgo/v2 v2.0.0 h1:CcuG/HvWNkkaqCUpJifQY8z7qEMBJya6aLPx6ftGyjQ= +github.com/onsi/ginkgo/v2 v2.0.0/go.mod h1:vw5CSIxN1JObi/U8gcbwft7ZxR2dgaR70JSE3/PpL4c= +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.17.0/go.mod h1:HnhC7FXeEQY45zxNK3PPoIUhzk/80Xly9PcubAlGdZY= +github.com/onsi/gomega v1.18.1 h1:M1GfJqGRrBrrGGsbxzV5dqM2U2ApXefZCQpkukxYRLE= +github.com/onsi/gomega v1.18.1/go.mod h1:0q+aL8jAiMXy9hbwj2mr5GziHiwhAIQpFmmtT5hitRs= +github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= +github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= +github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA= +github.com/stretchr/testify v1.7.1 h1:5TQK59W5E3v0r2duFAb7P95B6hEeOyEnHRa8MjYSMTY= +github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/yuin/goldmark v1.2.1/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74= +github.com/yuin/goldmark v1.4.0/go.mod h1:mwnBkeHKe2W/ZEtQ+71ViKU8L12m81fl3OWwC1Zlc8k= +go.opentelemetry.io/otel v1.7.0 h1:Z2lA3Tdch0iDcrhJXDIlC94XE+bxok1F9B+4Lz/lGsM= +go.opentelemetry.io/otel v1.7.0/go.mod h1:5BdUoMIz5WEs0vt0CUEMtSSaTSHBBVwrhnz7+nrD5xk= +go.opentelemetry.io/otel/sdk v1.7.0 h1:4OmStpcKVOfvDOgCt7UriAPtKolwIhxpnSNI/yK+1B0= +go.opentelemetry.io/otel/sdk v1.7.0/go.mod h1:uTEOTwaqIVuTGiJN7ii13Ibp75wJmYUDe374q6cZwUU= +go.opentelemetry.io/otel/trace v1.7.0 h1:O37Iogk1lEkMRXewVtZ1BBTVn5JEp8GrJvP92bJqC6o= +go.opentelemetry.io/otel/trace v1.7.0/go.mod h1:fzLSB9nqR2eXzxPXb2JW9IKE+ScyXA48yyE4TNvoHqU= +golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= +golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI= +golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto= +golang.org/x/mod v0.3.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA= +golang.org/x/mod v0.4.2/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA= +golang.org/x/net v0.0.0-20180906233101-161cd47e91fd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= +golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= +golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= +golang.org/x/net v0.0.0-20200520004742-59133d7f0dd7/go.mod h1:qpuaurCH72eLCgpAm/N6yyVIVM9cpaDIP3A8BGJEC5A= +golang.org/x/net v0.0.0-20201021035429-f5854403a974/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU= +golang.org/x/net v0.0.0-20210428140749-89ef3d95e781/go.mod h1:OJAsFXCWl8Ukc7SiCT/9KSuxbyM7479/AVlXFRxuMCk= +golang.org/x/net v0.0.0-20210805182204-aaa1db679c0d/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y= +golang.org/x/net v0.0.0-20211112202133-69e39bad7dc2 h1:CIJ76btIcR3eFI5EgSo6k1qKw9KJexJuRLI9G7Hp5wE= +golang.org/x/net v0.0.0-20211112202133-69e39bad7dc2/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y= +golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20201020160332-67f06af15bc9/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20210220032951-036812b2e83c/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.1.0 h1:wsuoTGHzEhffawBOhz5CYhcrV4IdKZbEyZjBMuTp12o= +golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sys v0.0.0-20180909124046-d0be0721c37e/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-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20190904154756-749cb33beabd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20191005200804-aed5e4c7ecf9/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20191120155948-bd437916bb0e/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20191204072324-ce4227a45e2e/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20200116001909-b77594299b42/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20200223170610-d5e6a3e2c0ae/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20200323222414-85ca7c5b95cd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20200930185726-fdedc70b468f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20210112080510-489259a85091/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20210423185535-09eb48e85fd7/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20210630005230-0f9fa26af87c/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20210809222454-d867a43fc93e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20211216021012-1d35b9e2eb4e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20220412211240-33da011f77ad h1:ntjMns5wyP/fN65tdBD4g8J5w8n015+iIIs9rtjXkY0= +golang.org/x/sys v0.0.0-20220412211240-33da011f77ad/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo= +golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= +golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= +golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= +golang.org/x/text v0.3.8-0.20211105212822-18b340fc7af2 h1:GLw7MR8AfAG2GmGcmVgObFOHXYypgGjnGno25RDwn3Y= +golang.org/x/text v0.3.8-0.20211105212822-18b340fc7af2/go.mod h1:EFNZuWvGYxIRUEX+K8UmCFwYmZjqcrnq15ZuVldZkZ0= +golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo= +golang.org/x/tools v0.0.0-20201224043029-2b0845dc783e/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA= +golang.org/x/tools v0.1.7/go.mod h1:LGqMHiF4EqQNHR1JncWGqT5BVaXmza+X+BDGol+dOxo= +golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1 h1:go1bK/D/BFZV2I8cIQd1NKEZ+0owSTG1fDTci4IqFcE= +golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +google.golang.org/protobuf v0.0.0-20200109180630-ec00e32a8dfd/go.mod h1:DFci5gLYBciE7Vtevhsrf46CRTquxDuWsQurQQe4oz8= +google.golang.org/protobuf v0.0.0-20200221191635-4d8936d0db64/go.mod h1:kwYJMbMJ01Woi6D6+Kah6886xMZcty6N08ah7+eCXa0= +google.golang.org/protobuf v0.0.0-20200228230310-ab0ca4ff8a60/go.mod h1:cfTl7dwQJ+fmap5saPgwCLgHXTUD7jkjRqWcaiX5VyM= +google.golang.org/protobuf v1.20.1-0.20200309200217-e05f789c0967/go.mod h1:A+miEFZTKqfCUM6K7xSMQL9OKL/b6hQv+e19PK+JZNE= +google.golang.org/protobuf v1.21.0/go.mod h1:47Nbq4nVaFHyn7ilMalzfO3qCViNmqZ2kzikPIcrTAo= +google.golang.org/protobuf v1.23.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU= +google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw= +google.golang.org/protobuf v1.26.0 h1:bxAC2xTBsZGibn2RTntX0oH50xLsqy1OxA9tTL3p/lk= +google.golang.org/protobuf v1.26.0/go.mod h1:9q0QmTI4eRPtz6boOQmLYwt+qCgq0jsYwAQnmE0givc= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/fsnotify.v1 v1.4.7/go.mod h1:Tz8NjZHkW78fSQdbUxIjBTcgA1z1m8ZHf0WmKUhAMys= +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.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= +gopkg.in/yaml.v2 v2.2.4/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= +gopkg.in/yaml.v2 v2.3.0/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= +gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY= +gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ= +gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b h1:h8qDotaEPuJATrMmW04NCwg7v22aHH28wwpauUhK9Oo= +gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= diff --git a/contrib/registry/zookeeper/zookeeper.go b/contrib/registry/zookeeper/zookeeper.go new file mode 100644 index 000000000..53a7aa87d --- /dev/null +++ b/contrib/registry/zookeeper/zookeeper.go @@ -0,0 +1,71 @@ +// Copyright GoFrame Author(https://goframe.org). All Rights Reserved. +// +// This Source Code Form is subject to the terms of the MIT License. +// If a copy of the MIT was not distributed with this file, +// You can obtain one at https://github.com/gogf/gf. + +// Package zookeeper implements service Registry and Discovery using zookeeper. +package zookeeper + +import ( + "github.com/go-zookeeper/zk" + "github.com/gogf/gf/v2/errors/gerror" + "github.com/gogf/gf/v2/net/gsvc" + "golang.org/x/sync/singleflight" + "time" +) + +var _ gsvc.Registry = &Registry{} + +type Content struct { + Key string + Value string +} + +// Option is etcd registry option. +type Option func(o *options) + +type options struct { + namespace string + user string + password string +} + +// WithRootPath with registry root path. +func WithRootPath(path string) Option { + return func(o *options) { o.namespace = path } +} + +// WithDigestACL with registry password. +func WithDigestACL(user string, password string) Option { + return func(o *options) { + o.user = user + o.password = password + } +} + +// Registry is consul registry +type Registry struct { + opts *options + conn *zk.Conn + group singleflight.Group +} + +func New(address []string, opts ...Option) *Registry { + conn, _, err := zk.Connect(address, time.Second*120) + if err != nil { + panic(gerror.Wrapf(err, + "Error with connect to zookeeper"), + ) + } + options := &options{ + namespace: "/microservices", + } + for _, o := range opts { + o(options) + } + return &Registry{ + opts: options, + conn: conn, + } +} diff --git a/contrib/registry/zookeeper/zookeeper_discovery.go b/contrib/registry/zookeeper/zookeeper_discovery.go new file mode 100644 index 000000000..35329185e --- /dev/null +++ b/contrib/registry/zookeeper/zookeeper_discovery.go @@ -0,0 +1,71 @@ +// Copyright GoFrame Author(https://goframe.org). All Rights Reserved. +// +// This Source Code Form is subject to the terms of the MIT License. +// If a copy of the MIT was not distributed with this file, +// You can obtain one at https://github.com/gogf/gf. + +package zookeeper + +import ( + "context" + "github.com/gogf/gf/v2/errors/gerror" + "github.com/gogf/gf/v2/net/gsvc" + "path" + "strings" +) + +// Search is the etcd discovery search function. +func (r *Registry) Search(_ context.Context, in gsvc.SearchInput) ([]gsvc.Service, error) { + prefix := strings.TrimPrefix(strings.ReplaceAll(in.Prefix, "/", "-"), "-") + instances, err, _ := r.group.Do(prefix, func() (interface{}, error) { + serviceNamePath := path.Join(r.opts.namespace, prefix) + servicesID, _, err := r.conn.Children(serviceNamePath) + if err != nil { + return nil, gerror.Wrapf( + err, + "Error with search the children node under %s", + serviceNamePath, + ) + } + items := make([]gsvc.Service, 0, len(servicesID)) + for _, service := range servicesID { + servicePath := path.Join(serviceNamePath, service) + byteData, _, err := r.conn.Get(servicePath) + if err != nil { + return nil, gerror.Wrapf( + err, + "Error with node data which name is %s", + servicePath, + ) + } + item, err := unmarshal(byteData) + if err != nil { + return nil, gerror.Wrapf( + err, + "Error with unmarshal node data to Content", + ) + } + svc, err := gsvc.NewServiceWithKV(item.Key, item.Value) + if err != nil { + return nil, gerror.Wrapf( + err, + "Error with new service with KV in Content", + ) + } + items = append(items, svc) + } + return items, nil + }) + if err != nil { + return nil, gerror.Wrapf( + err, + "Error with group do", + ) + } + return instances.([]gsvc.Service), nil +} + +// Watch is the etcd discovery watch function. +func (r *Registry) Watch(ctx context.Context, key string) (gsvc.Watcher, error) { + return newWatcher(ctx, r.opts.namespace, key, r.conn) +} diff --git a/contrib/registry/zookeeper/zookeeper_func.go b/contrib/registry/zookeeper/zookeeper_func.go new file mode 100644 index 000000000..0f3f72d04 --- /dev/null +++ b/contrib/registry/zookeeper/zookeeper_func.go @@ -0,0 +1,20 @@ +// Copyright GoFrame Author(https://goframe.org). All Rights Reserved. +// +// This Source Code Form is subject to the terms of the MIT License. +// If a copy of the MIT was not distributed with this file, +// You can obtain one at https://github.com/gogf/gf. + +package zookeeper + +import ( + "encoding/json" +) + +func unmarshal(data []byte) (c *Content, err error) { + err = json.Unmarshal(data, &c) + return +} + +func marshal(c *Content) ([]byte, error) { + return json.Marshal(c) +} diff --git a/contrib/registry/zookeeper/zookeeper_registrar.go b/contrib/registry/zookeeper/zookeeper_registrar.go new file mode 100644 index 000000000..a69060666 --- /dev/null +++ b/contrib/registry/zookeeper/zookeeper_registrar.go @@ -0,0 +1,139 @@ +// Copyright GoFrame Author(https://goframe.org). All Rights Reserved. +// +// This Source Code Form is subject to the terms of the MIT License. +// If a copy of the MIT was not distributed with this file, +// You can obtain one at https://github.com/gogf/gf. + +package zookeeper + +import ( + "context" + "github.com/go-zookeeper/zk" + "github.com/gogf/gf/v2/errors/gerror" + "path" + "strings" + "time" + + "github.com/gogf/gf/v2/net/gsvc" +) + +// Register registers `service` to Registry. +// Note that it returns a new Service if it changes the input Service with custom one. +func (r *Registry) Register(_ context.Context, service gsvc.Service) (gsvc.Service, error) { + var ( + data []byte + err error + ) + if err = r.ensureName(r.opts.namespace, []byte(""), 0); err != nil { + return service, gerror.Wrapf( + err, + "Error Creat node which name is %s", + r.opts.namespace, + ) + } + prefix := strings.TrimPrefix(strings.ReplaceAll(service.GetPrefix(), "/", "-"), "-") + servicePrefixPath := path.Join(r.opts.namespace, prefix) + if err = r.ensureName(servicePrefixPath, []byte(""), 0); err != nil { + return service, gerror.Wrapf( + err, + "Error Creat node which name is %s", + servicePrefixPath, + ) + } + + if data, err = marshal(&Content{ + Key: service.GetKey(), + Value: service.GetValue(), + }); err != nil { + return service, gerror.Wrapf( + err, + "Error with marshal Content to Json string", + ) + } + servicePath := path.Join(servicePrefixPath, service.GetName()) + if err = r.ensureName(servicePath, data, zk.FlagEphemeral); err != nil { + return service, gerror.Wrapf( + err, + "Error Creat node which name is %s", + servicePath, + ) + } + go r.reRegister(servicePath, data) + return service, nil +} + +// Deregister off-lines and removes `service` from the Registry. +func (r *Registry) Deregister(ctx context.Context, service gsvc.Service) error { + ch := make(chan error, 1) + prefix := strings.TrimPrefix(strings.ReplaceAll(service.GetPrefix(), "/", "-"), "-") + servicePath := path.Join(r.opts.namespace, prefix, service.GetName()) + go func() { + err := r.conn.Delete(servicePath, -1) + ch <- err + }() + var err error + select { + case <-ctx.Done(): + err = ctx.Err() + case err = <-ch: + } + return gerror.Wrapf(err, + "Error with deregister service:%s", + service.GetName(), + ) +} + +// ensureName ensure node exists, if not exist, create and set data +func (r *Registry) ensureName(path string, data []byte, flags int32) error { + exists, stat, err := r.conn.Exists(path) + if err != nil { + return gerror.Wrapf(err, + "Error with check node exist which name is %s", + path, + ) + } + // ephemeral nodes handling after restart + // fixes a race condition if the server crashes without using CreateProtectedEphemeralSequential() + if flags&zk.FlagEphemeral == zk.FlagEphemeral { + err = r.conn.Delete(path, stat.Version) + if err != nil && err != zk.ErrNoNode { + return gerror.Wrapf(err, + "Error with delete node which name is %s", + path, + ) + } + exists = false + } + if !exists { + if len(r.opts.user) > 0 && len(r.opts.password) > 0 { + _, err = r.conn.Create(path, data, flags, zk.DigestACL(zk.PermAll, r.opts.user, r.opts.password)) + } else { + _, err = r.conn.Create(path, data, flags, zk.WorldACL(zk.PermAll)) + } + if err != nil { + return gerror.Wrapf(err, + "Error with create node which name is %s", + path, + ) + } + } + return nil +} + +// reRegister re-register data node info when bad connection recovered +func (r *Registry) reRegister(path string, data []byte) { + sessionID := r.conn.SessionID() + ticker := time.NewTicker(time.Second) + defer ticker.Stop() + for range ticker.C { + cur := r.conn.SessionID() + // sessionID changed + if cur > 0 && sessionID != cur { + // re-ensureName + if err := r.ensureName(path, data, zk.FlagEphemeral); err != nil { + return + } + sessionID = cur + } + } +} diff --git a/contrib/registry/zookeeper/zookeeper_watcher.go b/contrib/registry/zookeeper/zookeeper_watcher.go new file mode 100644 index 000000000..3b03b5f7a --- /dev/null +++ b/contrib/registry/zookeeper/zookeeper_watcher.go @@ -0,0 +1,138 @@ +// Copyright GoFrame Author(https://goframe.org). All Rights Reserved. +// +// This Source Code Form is subject to the terms of the MIT License. +// If a copy of the MIT was not distributed with this file, +// You can obtain one at https://github.com/gogf/gf. + +package zookeeper + +import ( + "context" + "errors" + "github.com/go-zookeeper/zk" + "github.com/gogf/gf/v2/errors/gerror" + "github.com/gogf/gf/v2/net/gsvc" + "golang.org/x/sync/singleflight" + "path" + "strings" +) + +var _ gsvc.Watcher = (*watcher)(nil) + +var ErrWatcherStopped = errors.New("watcher stopped") + +type watcher struct { + ctx context.Context + event chan zk.Event + conn *zk.Conn + cancel context.CancelFunc + prefix string + nameSpace string + group singleflight.Group +} + +func newWatcher(ctx context.Context, nameSpace, prefix string, conn *zk.Conn) (*watcher, error) { + w := &watcher{ + conn: conn, + event: make(chan zk.Event, 1), + nameSpace: nameSpace, + prefix: prefix, + } + w.ctx, w.cancel = context.WithCancel(ctx) + go w.watch(w.ctx) + return w, nil +} + +func (w *watcher) Proceed() ([]gsvc.Service, error) { + select { + case <-w.ctx.Done(): + return nil, w.ctx.Err() + case e := <-w.event: + if e.State == zk.StateDisconnected { + return nil, gerror.Wrapf( + ErrWatcherStopped, + "watcher stopped", + ) + } + if e.Err != nil { + return nil, e.Err + } + return w.getServicesByPrefix() + } +} + +func (w *watcher) getServicesByPrefix() ([]gsvc.Service, error) { + prefix := strings.TrimPrefix(strings.ReplaceAll(w.prefix, "/", "-"), "-") + serviceNamePath := path.Join(w.nameSpace, prefix) + instances, err, _ := w.group.Do(serviceNamePath, func() (interface{}, error) { + servicesID, _, err := w.conn.Children(serviceNamePath) + if err != nil { + return nil, gerror.Wrapf( + err, + "Error with search the children node under %s", + serviceNamePath, + ) + } + items := make([]gsvc.Service, 0, len(servicesID)) + for _, service := range servicesID { + servicePath := path.Join(serviceNamePath, service) + byteData, _, err := w.conn.Get(servicePath) + if err != nil { + return nil, gerror.Wrapf( + err, + "Error with node data which name is %s", + servicePath, + ) + } + item, err := unmarshal(byteData) + if err != nil { + return nil, gerror.Wrapf( + err, + "Error with unmarshal node data to Content", + ) + } + svc, err := gsvc.NewServiceWithKV(item.Key, item.Value) + if err != nil { + return nil, gerror.Wrapf( + err, + "Error with new service with KV in Content", + ) + } + items = append(items, svc) + } + return items, nil + }) + if err != nil { + return nil, gerror.Wrapf( + err, + "Error with group do", + ) + } + return instances.([]gsvc.Service), nil +} + +func (w *watcher) Close() error { + w.cancel() + return nil +} + +func (w *watcher) watch(ctx context.Context) { + prefix := strings.TrimPrefix(strings.ReplaceAll(w.prefix, "/", "-"), "-") + serviceNamePath := path.Join(w.nameSpace, prefix) + for { + + if w.conn.State() == zk.StateConnected || w.conn.State() == zk.StateHasSession { + // each watch action is only valid once + _, _, ch, err := w.conn.ChildrenW(serviceNamePath) + if err != nil { + w.event <- zk.Event{Err: err} + } + select { + case <-ctx.Done(): + return + default: + w.event <- <-ch + } + } + } +} diff --git a/contrib/registry/zookeeper/zookeeper_z_test.go b/contrib/registry/zookeeper/zookeeper_z_test.go new file mode 100644 index 000000000..207e82376 --- /dev/null +++ b/contrib/registry/zookeeper/zookeeper_z_test.go @@ -0,0 +1,189 @@ +// Copyright GoFrame Author(https://goframe.org). All Rights Reserved. +// +// This Source Code Form is subject to the terms of the MIT License. +// If a copy of the MIT was not distributed with this file, +// You can obtain one at https://github.com/gogf/gf. + +package zookeeper + +import ( + "context" + "testing" + "time" + + "github.com/gogf/gf/v2/frame/g" + "github.com/gogf/gf/v2/net/gsvc" + "github.com/gogf/gf/v2/os/gctx" +) + +// TestRegistry TestRegistryManyService +func TestRegistry(t *testing.T) { + r := New([]string{"127.0.0.1:2181"}, WithRootPath("/gogf")) + ctx := context.Background() + + svc := &gsvc.LocalService{ + Name: "goframe-provider-0-tcp", + Version: "test", + Metadata: map[string]interface{}{"app": "goframe", gsvc.MDProtocol: "tcp"}, + Endpoints: gsvc.NewEndpoints("127.0.0.1:9000"), + } + + s, err := r.Register(ctx, svc) + if err != nil { + t.Fatal(err) + } + + err = r.Deregister(ctx, s) + if err != nil { + t.Fatal(err) + } +} + +// TestRegistryMany TestRegistryManyService +func TestRegistryMany(t *testing.T) { + r := New([]string{"127.0.0.1:2181"}, WithRootPath("/gogf")) + + svc := &gsvc.LocalService{ + Name: "goframe-provider-1-tcp", + Version: "test", + Metadata: map[string]interface{}{"app": "goframe", gsvc.MDProtocol: "tcp"}, + Endpoints: gsvc.NewEndpoints("127.0.0.1:9000"), + } + svc1 := &gsvc.LocalService{ + Name: "goframe-provider-2-tcp", + Version: "test", + Metadata: map[string]interface{}{"app": "goframe", gsvc.MDProtocol: "tcp"}, + Endpoints: gsvc.NewEndpoints("127.0.0.1:9001"), + } + svc2 := &gsvc.LocalService{ + Name: "goframe-provider-3-tcp", + Version: "test", + Metadata: map[string]interface{}{"app": "goframe", gsvc.MDProtocol: "tcp"}, + Endpoints: gsvc.NewEndpoints("127.0.0.1:9002"), + } + + s0, err := r.Register(context.Background(), svc) + if err != nil { + t.Fatal(err) + } + + s1, err := r.Register(context.Background(), svc1) + if err != nil { + t.Fatal(err) + } + + s2, err := r.Register(context.Background(), svc2) + if err != nil { + t.Fatal(err) + } + + err = r.Deregister(context.Background(), s0) + if err != nil { + t.Fatal(err) + } + + err = r.Deregister(context.Background(), s1) + if err != nil { + t.Fatal(err) + } + + err = r.Deregister(context.Background(), s2) + if err != nil { + t.Fatal(err) + } +} + +// TestGetService Test GetService +func TestGetService(t *testing.T) { + r := New([]string{"127.0.0.1:2181"}, WithRootPath("/gogf")) + ctx := context.Background() + + svc := &gsvc.LocalService{ + Name: "goframe-provider-4-tcp", + Version: "test", + Metadata: map[string]interface{}{"app": "goframe", gsvc.MDProtocol: "tcp"}, + Endpoints: gsvc.NewEndpoints("127.0.0.1:9000"), + } + + s, err := r.Register(ctx, svc) + if err != nil { + t.Fatal(err) + } + time.Sleep(time.Second * 1) + serviceInstances, err := r.Search(ctx, gsvc.SearchInput{ + Prefix: s.GetPrefix(), + Name: svc.Name, + Version: svc.Version, + Metadata: svc.Metadata, + }) + if err != nil { + t.Fatal(err) + } + for _, instance := range serviceInstances { + g.Log().Info(ctx, instance) + } + + err = r.Deregister(ctx, s) + if err != nil { + t.Fatal(err) + } +} + +// TestWatch Test Watch +func TestWatch(t *testing.T) { + r := New([]string{"127.0.0.1:2181"}, WithRootPath("/gogf")) + + ctx := gctx.New() + + svc := &gsvc.LocalService{ + Name: "goframe-provider-4-tcp", + Version: "test", + Metadata: map[string]interface{}{"app": "goframe", gsvc.MDProtocol: "tcp"}, + Endpoints: gsvc.NewEndpoints("127.0.0.1:9000"), + } + t.Log("watch") + watch, err := r.Watch(context.Background(), svc.GetPrefix()) + if err != nil { + t.Fatal(err) + } + + s1, err := r.Register(context.Background(), svc) + if err != nil { + t.Fatal(err) + } + // watch svc + // svc register, AddEvent + next, err := watch.Proceed() + if err != nil { + t.Fatal(err) + } + for _, instance := range next { + // it will output one instance + g.Log().Info(ctx, "Register Proceed service: ", instance) + } + + err = r.Deregister(context.Background(), s1) + if err != nil { + t.Fatal(err) + } + + // svc deregister, DeleteEvent + next, err = watch.Proceed() + if err != nil { + t.Fatal(err) + } + for _, instance := range next { + // it will output nothing + g.Log().Info(ctx, "Deregister Proceed service: ", instance) + } + + err = watch.Close() + if err != nil { + t.Fatal(err) + } + _, err = watch.Proceed() + if err == nil { + // if nil, stop failed + t.Fatal() + } +}