volume server get notified of leader change

weed/server/master_grpc_server.go

@@ -65,5 +65,17 @@ func (ms MasterServer) SendHeartbeat(stream pb.Seaweed_SendHeartbeatServer) erro
 			}
 			return err
 		}
+
+		if !t.IsLeader() {
+			newLeader, err := t.Leader()
+			if err == nil {
+				if err := stream.Send(&pb.HeartbeatResponse{
+					Leader: newLeader,
+				}); err != nil {
+					return err
+				}
+			}
+		}
+
 	}
 }

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
+			}
 		}
 	}()