update raft

1 files changed, 4 insertions, 2 deletions

diff --git a/weed/server/raft_server.go b/weed/server/raft_server.go
index d718ecac7..16736eb5b 100644
--- a/weed/server/raft_server.go
+++ b/weed/server/raft_server.go
@@ -3,6 +3,7 @@ package weed_server
 import (
 	"encoding/json"
 	transport "github.com/Jille/raft-grpc-transport"
+	"github.com/seaweedfs/raft/protobuf"
 	"io"
 	"math/rand"
 	"os"
@@ -33,6 +34,8 @@ type RaftServerOption struct {
 }
 
 type RaftServer struct {
+	protobuf.UnimplementedRaftServer
+	raftGrpcServer   *raft.GrpcServer
 	peers            map[string]pb.ServerAddress // initial peers to join with
 	raftServer       raft.Server
 	RaftHashicorp    *hashicorpRaft.Raft
@@ -40,7 +43,6 @@ type RaftServer struct {
 	dataDir          string
 	serverAddr       pb.ServerAddress
 	topo             *topology.Topology
-	*raft.GrpcServer
 }
 
 type StateMachine struct {
@@ -162,7 +164,7 @@ func NewRaftServer(option *RaftServerOption) (*RaftServer, error) {
 		}
 	}
 
-	s.GrpcServer = raft.NewGrpcServer(s.raftServer)
+	s.raftGrpcServer = raft.NewGrpcServer(s.raftServer)
 
 	glog.V(0).Infof("current cluster leader: %v", s.raftServer.Leader())