diff options
| author | Chris Lu <chris.lu@gmail.com> | 2021-11-06 14:25:06 -0700 |
|---|---|---|
| committer | Chris Lu <chris.lu@gmail.com> | 2021-11-06 14:26:26 -0700 |
| commit | 751a7073e3bf7e45f75a7638da256720c2d902a7 (patch) | |
| tree | 67998d568bc5aa69ab0c6aac5be7f109e447beeb /weed/wdclient/masterclient.go | |
| parent | 04663c3611326a8bac87d47fc4a2c546c6d9acdd (diff) | |
| parent | e0fc2898e9f9ec24ac2a94f7fba4b440058e2d25 (diff) | |
| download | seaweedfs-751a7073e3bf7e45f75a7638da256720c2d902a7.tar.xz seaweedfs-751a7073e3bf7e45f75a7638da256720c2d902a7.zip | |
Merge branch 'flexible_filer_addition'
Diffstat (limited to 'weed/wdclient/masterclient.go')
| -rw-r--r-- | weed/wdclient/masterclient.go | 13 |
1 files changed, 9 insertions, 4 deletions
diff --git a/weed/wdclient/masterclient.go b/weed/wdclient/masterclient.go index a2f6c7ffb..727d9cd34 100644 --- a/weed/wdclient/masterclient.go +++ b/weed/wdclient/masterclient.go @@ -21,6 +21,8 @@ type MasterClient struct { grpcDialOption grpc.DialOption vidMap + + OnPeerUpdate func(update *master_pb.ClusterNodeUpdate) } func NewMasterClient(grpcDialOption grpc.DialOption, clientType string, clientHost pb.ServerAddress, clientDataCenter string, masters []pb.ServerAddress) *MasterClient { @@ -151,10 +153,13 @@ func (mc *MasterClient) tryConnectToMaster(master pb.ServerAddress) (nextHintedL if resp.ClusterNodeUpdate != nil { update := resp.ClusterNodeUpdate - if update.IsAdd { - glog.V(0).Infof("+ %s %s leader:%v\n", update.NodeType, update.Address, update.IsLeader) - } else { - glog.V(0).Infof("- %s %s leader:%v\n", update.NodeType, update.Address, update.IsLeader) + if mc.OnPeerUpdate != nil { + if update.IsAdd { + glog.V(0).Infof("+ %s %s leader:%v\n", update.NodeType, update.Address, update.IsLeader) + } else { + glog.V(0).Infof("- %s %s leader:%v\n", update.NodeType, update.Address, update.IsLeader) + } + mc.OnPeerUpdate(update) } } |