volume server get notified of leader change

2017-01-18 09:34:27 -08:00
parent 59022b6fe0
commit 5b0f5f456e
5 changed files with 66 additions and 34 deletions
--- a/weed/server/volume_grpc_client.go
+++ b/weed/server/volume_grpc_client.go
@@ -65,6 +65,11 @@ func (vs *VolumeServer) doHeartbeat(sleepInterval time.Duration) error {
 			}
 			vs.store.VolumeSizeLimit = in.GetVolumeSizeLimit()
 			vs.guard.SecretKey = security.Secret(in.GetSecretKey())
+			if in.GetLeader() != "" && masterNode != in.GetLeader() {
+				vs.masterNodes.SetPossibleLeader(in.GetLeader())
+				doneChan <- nil
+				return
+			}
 		}
 	}()