toolbar/precog/monitor/monitor.go

   1 package monitor
   2
   3 import (
   4         "io/ioutil"
   5         "os"
   6         "time"
   7
   8         "github.com/jinzhu/gorm"
   9         log "github.com/sirupsen/logrus"
  10         dbm "github.com/vapor/database/leveldb"
  11
  12         cfg "github.com/vapor/config"
  13         "github.com/vapor/p2p"
  14         "github.com/vapor/p2p/signlib"
  15         "github.com/vapor/toolbar/precog/config"
  16         "github.com/vapor/toolbar/precog/database/orm"
  17 )
  18
  19 type monitor struct {
  20         cfg *config.Config
  21         db  *gorm.DB
  22 }
  23
  24 func NewMonitor(cfg *config.Config, db *gorm.DB) *monitor {
  25         return &monitor{
  26                 cfg: cfg,
  27                 db:  db,
  28         }
  29 }
  30
  31 func (m *monitor) Run() {
  32         m.updateBootstrapNodes()
  33         go m.discovery()
  34         ticker := time.NewTicker(time.Duration(m.cfg.CheckFreqSeconds) * time.Second)
  35         for ; true; <-ticker.C {
  36                 // TODO: lock?
  37                 m.monitorRountine()
  38         }
  39 }
  40
  41 // create or update: https://github.com/jinzhu/gorm/issues/1307
  42 func (m *monitor) updateBootstrapNodes() {
  43         for _, node := range m.cfg.Nodes {
  44                 ormNode := &orm.Node{
  45                         PublicKey: node.PublicKey.String(),
  46                         Alias:     node.Alias,
  47                         Host:      node.Host,
  48                         Port:      node.Port,
  49                 }
  50
  51                 if err := m.db.Where(&orm.Node{PublicKey: ormNode.PublicKey}).
  52                         Assign(&orm.Node{
  53                                 Alias: node.Alias,
  54                                 Host:  node.Host,
  55                                 Port:  node.Port,
  56                         }).FirstOrCreate(ormNode).Error; err != nil {
  57                         log.Error(err)
  58                         continue
  59                 }
  60         }
  61 }
  62
  63 // TODO:
  64 // implement logic first, and then refactor
  65 // /home/gavin/work/go/src/github.com/vapor/
  66 // p2p/test_util.go
  67 // p2p/switch_test.go
  68 func (m *monitor) discovery() {
  69         mCfg := &cfg.Config{}
  70         // TODO: fix
  71         dirPath, err := ioutil.TempDir(".", "")
  72         if err != nil {
  73                 log.Fatal(err)
  74         }
  75         defer os.RemoveAll(dirPath)
  76
  77         // TODO: fix
  78         mCfg.DBPath = dirPath
  79         // TODO: fix
  80         mCfg.P2P.ListenAddress = "127.0.1.1:0"
  81         swPrivKey, err := signlib.NewPrivKey()
  82         if err != nil {
  83                 log.Fatal(err)
  84         }
  85
  86         testDB := dbm.NewDB("testdb", "leveldb", dirPath)
  87         // initSwitchFunc
  88         sw := p2p.MakeSwitch(mCfg, testDB, swPrivKey, initSwitchFunc)
  89         sw.Start()
  90         defer sw.Stop()
  91 }
  92
  93 func (m *monitor) monitorRountine() error {
  94         // TODO: dail nodes, get lantency & best_height
  95         // TODO: decide check_height("best best_height" - "confirmations")
  96         // TODO: get blockhash by check_height, get latency
  97         // TODO: update lantency, active_time and status
  98         return nil
  99 }