From 293a982eee0f3d8747450518ceb9e915d74885db Mon Sep 17 00:00:00 2001 From: Anthony Romano Date: Thu, 30 Mar 2017 15:43:07 -0700 Subject: [PATCH] embed: add Election service --- embed/serve.go | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/embed/serve.go b/embed/serve.go index fb05df730ea..c7b0d80304c 100644 --- a/embed/serve.go +++ b/embed/serve.go @@ -26,6 +26,8 @@ import ( "github.com/coreos/etcd/etcdserver" "github.com/coreos/etcd/etcdserver/api/v3client" + "github.com/coreos/etcd/etcdserver/api/v3election" + "github.com/coreos/etcd/etcdserver/api/v3election/v3electionpb" "github.com/coreos/etcd/etcdserver/api/v3lock" "github.com/coreos/etcd/etcdserver/api/v3lock/v3lockpb" "github.com/coreos/etcd/etcdserver/api/v3rpc" @@ -71,6 +73,7 @@ func (sctx *serveCtx) serve(s *etcdserver.EtcdServer, tlscfg *tls.Config, handle if sctx.insecure { gs := v3rpc.Server(s, nil) + v3electionpb.RegisterElectionServer(gs, v3election.NewElectionServer(v3client.New(s))) v3lockpb.RegisterLockServer(gs, v3lock.NewLockServer(v3client.New(s))) if sctx.serviceRegister != nil { sctx.serviceRegister(gs) @@ -99,6 +102,7 @@ func (sctx *serveCtx) serve(s *etcdserver.EtcdServer, tlscfg *tls.Config, handle if sctx.secure { gs := v3rpc.Server(s, tlscfg) + v3electionpb.RegisterElectionServer(gs, v3election.NewElectionServer(v3client.New(s))) v3lockpb.RegisterLockServer(gs, v3lock.NewLockServer(v3client.New(s))) if sctx.serviceRegister != nil { sctx.serviceRegister(gs)