kenneth/gohelpers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

etcd add timeout

Browse Source
This commit is contained in:
kenneth 2022-06-28 14:16:44 +08:00
parent cb501fb3cf
commit 812bd24035
2 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pkg/etcd/discovery.go
View File

@ -49,6 +49,7 @@ func (r *Resolver) Discovery(serviceName string) (*RemoteService, error) {
ctx, cancel := context.WithTimeout(context.Background(), time.Second) ctx, cancel := context.WithTimeout(context.Background(), time.Second)
defer cancel() defer cancel()
resp, err := kv.Get(ctx, serviceName, clientV3.WithPrefix()) resp, err := kv.Get(ctx, serviceName, clientV3.WithPrefix())
if err != nil { if err != nil {
return nil, err return nil, err

10
pkg/etcd/register.go
View File

@ -36,7 +36,10 @@ func (e *Client) Close() error {
// Register 注册服务 // Register 注册服务
func (e *Client) Register(serviceName string, serviceAddress string) error { func (e *Client) Register(serviceName string, serviceAddress string) error {
var err error var err error
ctx := context.Background()
ctx, cancel := context.WithTimeout(context.Background(), time.Second*3)
defer cancel()
e.lease, err = e.v3.Grant(ctx, int64(e.interval+1)) e.lease, err = e.v3.Grant(ctx, int64(e.interval+1))
if err != nil { if err != nil {
return err return err
@ -54,7 +57,10 @@ func (e *Client) Register(serviceName string, serviceAddress string) error {
// Deregister 注销服务 // Deregister 注销服务
func (e *Client) Deregister() error { func (e *Client) Deregister() error {
_, err := e.v3.Delete(context.Background(), e.serviceName) ctx, cancel := context.WithTimeout(context.Background(), time.Second*3)
defer cancel()
_, err := e.v3.Delete(ctx, e.serviceName)
if err != nil { if err != nil {
return err return err
} }