embed: add Election service

etcd-io · Mar 31, 2017 · 293a982 · 293a982
1 parent 5aa78e3
commit 293a982
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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)