toolbar/precog/monitor/connection.go

   1 package monitor
   2
   3 import (
   4         "net"
   5         "time"
   6
   7         log "github.com/sirupsen/logrus"
   8         "github.com/vapor/p2p"
   9         "github.com/vapor/toolbar/precog/database/orm"
  10 )
  11
  12 func (m *monitor) connectNodesRoutine() {
  13         ticker := time.NewTicker(time.Duration(m.cfg.CheckFreqMinutes) * time.Minute)
  14         for ; true; <-ticker.C {
  15                 <-m.dialCh
  16                 m.Lock()
  17
  18                 if err := m.dialNodes(); err != nil {
  19                         log.Error(err)
  20                 }
  21         }
  22 }
  23
  24 func (m *monitor) dialNodes() error {
  25         log.Info("Start to reconnect to nodes...")
  26         var nodes []*orm.Node
  27         if err := m.db.Model(&orm.Node{}).Find(&nodes).Error; err != nil {
  28                 return err
  29         }
  30
  31         addresses := make([]*p2p.NetAddress, 0)
  32         for i := 0; i < len(nodes); i++ {
  33                 address := p2p.NewNetAddressIPPort(net.ParseIP(nodes[i].IP), nodes[i].Port)
  34                 addresses = append(addresses, address)
  35         }
  36
  37         // connected peers will be skipped in switch.DialPeers()
  38         m.sw.DialPeers(addresses)
  39         log.Info("DialPeers done.")
  40         m.processDialResults()
  41         m.checkStatusCh <- struct{}{}
  42         return nil
  43 }