7 log "github.com/sirupsen/logrus"
10 "github.com/vapor/toolbar/precognitive/database/orm"
13 func (m *monitor) connectionRoutine() {
14 ticker := time.NewTicker(time.Duration(m.cfg.CheckFreqMinutes) * time.Minute)
15 for ; true; <-ticker.C {
16 if err := m.dialNodes(); err != nil {
17 log.WithFields(log.Fields{"err": err}).Error("dialNodes")
22 func (m *monitor) dialNodes() error {
23 log.Info("Start to reconnect to nodes...")
25 if err := m.db.Model(&orm.Node{}).Find(&nodes).Error; err != nil {
29 addresses := make([]*p2p.NetAddress, 0)
30 for i := 0; i < len(nodes); i++ {
31 address := p2p.NewNetAddressIPPort(net.ParseIP(nodes[i].IP), nodes[i].Port)
32 addresses = append(addresses, address)
35 // connected peers will be skipped in switch.DialPeers()
36 m.sw.DialPeers(addresses)
37 log.Info("DialPeers done.")
38 peerList := m.sw.GetPeers().List()
39 m.processDialResults(peerList)
40 m.checkStatus(peerList)
44 func (m *monitor) checkStatus(peerList []*p2p.Peer) {
45 for _, peer := range peerList {
49 log.WithFields(log.Fields{"num": len(m.sw.GetPeers().List()), "peers": m.sw.GetPeers().List()}).Info("connected peers")
51 for _, peerInfo := range m.peers.GetPeerInfos() {
52 if peerInfo.Height > m.bestHeightSeen {
53 m.bestHeightSeen = peerInfo.Height
56 log.WithFields(log.Fields{"bestHeight": m.bestHeightSeen}).Info("peersInfo")
57 m.processPeerInfos(m.peers.GetPeerInfos())
59 for _, peer := range peerList {
60 p := m.peers.GetPeer(peer.ID())
65 m.peers.RemovePeer(p.ID())
67 log.Info("Disonnect all peers.")