toolbar/precog/monitor/discover.go

   1 package monitor
   2
   3 import (
   4         "time"
   5
   6         log "github.com/sirupsen/logrus"
   7
   8         "github.com/vapor/p2p/discover/dht"
   9         "github.com/vapor/toolbar/precog/config"
  10 )
  11
  12 var (
  13         nodesToDiscv = 150
  14         discvFreqSec = 60
  15 )
  16
  17 func (m *monitor) discoveryRoutine() {
  18         // TODO: ip & port?
  19         // discvMap maps a node's public key to the node itself
  20         discvMap := make(map[string]*dht.Node)
  21         ticker := time.NewTicker(time.Duration(discvFreqSec) * time.Second)
  22         for range ticker.C {
  23                 nodes := make([]*dht.Node, nodesToDiscv)
  24                 num := m.sw.GetDiscv().ReadRandomNodes(nodes)
  25                 for _, node := range nodes[:num] {
  26                         if n, ok := discvMap[node.ID.String()]; ok && n.String() == node.String() {
  27                                 continue
  28                         }
  29
  30                         log.WithFields(log.Fields{"new node": node}).Info("discover")
  31
  32                         if err := m.upsertNode(&config.Node{
  33                                 PublicKey: node.ID.String(),
  34                                 IP:        node.IP.String(),
  35                                 Port:      node.TCP,
  36                         }); err != nil {
  37                                 log.WithFields(log.Fields{"node": node, "err": err}).Error("upsertNode")
  38                         } else {
  39                                 discvMap[node.ID.String()] = node
  40                         }
  41                 }
  42         }
  43 }