mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-01-19 02:48:24 +00:00
82 lines
2.2 KiB
Go
82 lines
2.2 KiB
Go
package weed_server
|
|
|
|
import (
|
|
"net"
|
|
"strings"
|
|
|
|
"github.com/chrislusf/seaweedfs/weed/glog"
|
|
"github.com/chrislusf/seaweedfs/weed/pb"
|
|
"github.com/chrislusf/seaweedfs/weed/storage"
|
|
"github.com/chrislusf/seaweedfs/weed/topology"
|
|
"google.golang.org/grpc/peer"
|
|
)
|
|
|
|
func (ms MasterServer) SendHeartbeat(stream pb.Seaweed_SendHeartbeatServer) error {
|
|
var dn *topology.DataNode
|
|
t := ms.Topo
|
|
for {
|
|
heartbeat, err := stream.Recv()
|
|
if err == nil {
|
|
if dn == nil {
|
|
t.Sequence.SetMax(heartbeat.MaxFileKey)
|
|
if heartbeat.Ip == "" {
|
|
if pr, ok := peer.FromContext(stream.Context()); ok {
|
|
if pr.Addr != net.Addr(nil) {
|
|
heartbeat.Ip = pr.Addr.String()[0:strings.LastIndex(pr.Addr.String(), ":")]
|
|
glog.V(0).Infof("remote IP address is detected as %v", heartbeat.Ip)
|
|
}
|
|
}
|
|
}
|
|
dcName, rackName := t.Configuration.Locate(heartbeat.Ip, heartbeat.DataCenter, heartbeat.Rack)
|
|
dc := t.GetOrCreateDataCenter(dcName)
|
|
rack := dc.GetOrCreateRack(rackName)
|
|
dn = rack.GetOrCreateDataNode(heartbeat.Ip,
|
|
int(heartbeat.Port), heartbeat.PublicUrl,
|
|
int(heartbeat.MaxVolumeCount))
|
|
glog.V(0).Infof("added volume server %v:%d", heartbeat.GetIp(), heartbeat.GetPort())
|
|
if err := stream.Send(&pb.HeartbeatResponse{
|
|
VolumeSizeLimit: uint64(ms.volumeSizeLimitMB) * 1024 * 1024,
|
|
SecretKey: string(ms.guard.SecretKey),
|
|
}); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
var volumeInfos []storage.VolumeInfo
|
|
for _, v := range heartbeat.Volumes {
|
|
if vi, err := storage.NewVolumeInfo(v); err == nil {
|
|
volumeInfos = append(volumeInfos, vi)
|
|
} else {
|
|
glog.V(0).Infof("Fail to convert joined volume information: %v", err)
|
|
}
|
|
}
|
|
deletedVolumes := dn.UpdateVolumes(volumeInfos)
|
|
for _, v := range volumeInfos {
|
|
t.RegisterVolumeLayout(v, dn)
|
|
}
|
|
for _, v := range deletedVolumes {
|
|
t.UnRegisterVolumeLayout(v, dn)
|
|
}
|
|
|
|
} else {
|
|
glog.V(0).Infof("lost volume server %s:%d", dn.Ip, dn.Port)
|
|
if dn != nil {
|
|
t.UnRegisterDataNode(dn)
|
|
}
|
|
return err
|
|
}
|
|
|
|
if !t.IsLeader() {
|
|
newLeader, err := t.Leader()
|
|
if err == nil {
|
|
if err := stream.Send(&pb.HeartbeatResponse{
|
|
Leader: newLeader,
|
|
}); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|
|
}
|