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: 包一下？  common cfg 之类的？
  71         mCfg := &cfg.Config{
  72                 BaseConfig: cfg.DefaultBaseConfig(),
  73                 P2P:        cfg.DefaultP2PConfig(),
  74                 Federation: cfg.DefaultFederationConfig(),
  75         }
  76         dirPath, err := ioutil.TempDir(".", "")
  77         if err != nil {
  78                 log.Fatal(err)
  79         }
  80         // TODO: fix
  81         mCfg.DBPath = dirPath
  82         defer os.RemoveAll(dirPath)
  83
  84         // swPrivKey, err := signlib.NewPrivKey()
  85         // if err != nil {
  86         //      log.Fatal(err)
  87         // }
  88
  89         // testDB := dbm.NewDB("testdb", "leveldb", dirPath)
  90         // TODO: clean up
  91         sw, err := p2p.NewSwitch(mCfg)
  92         if err != nil {
  93                 log.Fatal(err)
  94         }
  95
  96         sw.Start()
  97         defer sw.Stop()
  98 }
  99
 100 func (m *monitor) monitorRountine() error {
 101         // TODO: dail nodes, get lantency & best_height
 102         // TODO: decide check_height("best best_height" - "confirmations")
 103         // TODO: get blockhash by check_height, get latency
 104         // TODO: update lantency, active_time and status
 105         return nil
 106 }