聊聊kingbus的startRaftPeer

孤者浪人 提交于 2020-08-10 21:07:15

本文主要研究一下kingbus的startRaftPeer

startRaftPeer

kingbus/server/server.go

func (s *KingbusServer) startRaftPeer(peerURLs types.URLs) error {
	err := s.raftNode.NewPeerListener(peerURLs)
	if err != nil {
		return err
	}
	s.raftNode.SetPeerHandler()
	s.raftNode.PeerHandlerServe()
	log.Log.Infof("startRaftPeer success")
	return nil
}
  • startRaftPeer方法根据peerURLs通过s.raftNode.NewPeerListener开启listener,之后执行s.raftNode.SetPeerHandler()、s.raftNode.PeerHandlerServe()

NewPeerListener

kingbus/raft/peer_handler.go

//NewPeerListener create listener of peer
func (r *Node) NewPeerListener(peerURLs types.URLs) (err error) {
	peers := make([]*peerListener, len(peerURLs))
	defer func() {
		if err == nil {
			return
		}
		for i := range peers {
			if peers[i] != nil && peers[i].close != nil {
				log.Log.Info("stopping listening for peers on ", peerURLs.String())
				peers[i].close(context.Background())
			}
		}
	}()

	for i, u := range peerURLs {
		peers[i] = &peerListener{close: func(context.Context) error { return nil }}
		peers[i].Listener, err = rafthttp.NewListener(u, nil)
		if err != nil {
			return err
		}
		peers[i].close = func(context.Context) error {
			return peers[i].Listener.Close()
		}
		log.Log.Info("listening for peers on ", u.String())
	}
	r.PeerListener = peers
	return nil
}
  • NewPeerListener方法遍历peerURLs通过rafthttp.NewListener创建peerListener

PeerHandlerServe

kingbus/raft/peer_handler.go

//PeerHandlerServe serve
func (r *Node) PeerHandlerServe() {
	for _, peer := range r.PeerListener {
		go peer.serve()
	}
}
  • PeerHandlerServe方法则执行peer.serve()

小结

startRaftPeer方法根据peerURLs通过s.raftNode.NewPeerListener开启listener,之后执行s.raftNode.SetPeerHandler()、s.raftNode.PeerHandlerServe()

doc

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!