Browse Source

Merge pull request #6826 from feisan/dev-kvkv

clientv3/naming: support OpOption when adding an endpoint
Xiang Li 9 years ago
parent
commit
3ed63af51a
1 changed files with 3 additions and 3 deletions
  1. 3 3
      clientv3/naming/grpc.go

+ 3 - 3
clientv3/naming/grpc.go

@@ -31,16 +31,16 @@ type GRPCResolver struct {
 	Client *etcd.Client
 	Client *etcd.Client
 }
 }
 
 
-func (gr *GRPCResolver) Update(ctx context.Context, target string, nm naming.Update) (err error) {
+func (gr *GRPCResolver) Update(ctx context.Context, target string, nm naming.Update, opts ...etcd.OpOption) (err error) {
 	switch nm.Op {
 	switch nm.Op {
 	case naming.Add:
 	case naming.Add:
 		var v []byte
 		var v []byte
 		if v, err = json.Marshal(nm); err != nil {
 		if v, err = json.Marshal(nm); err != nil {
 			return grpc.Errorf(codes.InvalidArgument, err.Error())
 			return grpc.Errorf(codes.InvalidArgument, err.Error())
 		}
 		}
-		_, err = gr.Client.KV.Put(ctx, target+"/"+nm.Addr, string(v))
+		_, err = gr.Client.KV.Put(ctx, target+"/"+nm.Addr, string(v), opts...)
 	case naming.Delete:
 	case naming.Delete:
-		_, err = gr.Client.Delete(ctx, target+"/"+nm.Addr)
+		_, err = gr.Client.Delete(ctx, target+"/"+nm.Addr, opts...)
 	default:
 	default:
 		return grpc.Errorf(codes.InvalidArgument, "naming: bad naming op")
 		return grpc.Errorf(codes.InvalidArgument, "naming: bad naming op")
 	}
 	}