clientv3/naming: support OpOption when adding an endpoint
if we want to add an endpoint with lease, we need this option. for example: resp, err := cli.Grant(context.TODO(), 5) if err != nil { log.Fatal(err) } err = r.Update(context.TODO(), serviceName, naming.Update{Op:naming.Add, Addr: exposedAddr}, clientv3.WithLease(resp.ID)) if err != nil { log.Fatalf(err) }release-3.1
parent
9c7a0a68e5
commit
c20d31adc5
|
@ -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")
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue