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         // conn "github.com/vapor/p2p/connection"
  15         "github.com/vapor/p2p/signlib"
  16         "github.com/vapor/toolbar/precog/config"
  17         "github.com/vapor/toolbar/precog/database/orm"
  18 )
  19
  20 type monitor struct {
  21         cfg *config.Config
  22         db  *gorm.DB
  23 }
  24
  25 func NewMonitor(cfg *config.Config, db *gorm.DB) *monitor {
  26         return &monitor{
  27                 cfg: cfg,
  28                 db:  db,
  29         }
  30 }
  31
  32 func (m *monitor) Run() {
  33         m.updateBootstrapNodes()
  34         go m.discovery()
  35         ticker := time.NewTicker(time.Duration(m.cfg.CheckFreqSeconds) * time.Second)
  36         for ; true; <-ticker.C {
  37                 // TODO: lock?
  38                 m.monitorRountine()
  39         }
  40 }
  41
  42 // create or update: https://github.com/jinzhu/gorm/issues/1307
  43 func (m *monitor) updateBootstrapNodes() {
  44         for _, node := range m.cfg.Nodes {
  45                 ormNode := &orm.Node{
  46                         PublicKey: node.PublicKey.String(),
  47                         Alias:     node.Alias,
  48                         Host:      node.Host,
  49                         Port:      node.Port,
  50                 }
  51
  52                 if err := m.db.Where(&orm.Node{PublicKey: ormNode.PublicKey}).
  53                         Assign(&orm.Node{
  54                                 Alias: node.Alias,
  55                                 Host:  node.Host,
  56                                 Port:  node.Port,
  57                         }).FirstOrCreate(ormNode).Error; err != nil {
  58                         log.Error(err)
  59                         continue
  60                 }
  61         }
  62 }
  63
  64 // TODO:
  65 // implement logic first, and then refactor
  66 // /home/gavin/work/go/src/github.com/vapor/
  67 // p2p/test_util.go
  68 // p2p/switch_test.go
  69 func (m *monitor) discovery() {
  70         // TODO: clean up
  71         // TODO: 包一下？  common cfg 之类的？
  72         mCfg := cfg.DefaultConfig()
  73         // TODO: fix
  74         dirPath, err := ioutil.TempDir(".", "")
  75         if err != nil {
  76                 log.Fatal(err)
  77         }
  78         mCfg.DBPath = dirPath
  79         defer os.RemoveAll(dirPath)
  80
  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 }