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/consensus"
  17         "github.com/vapor/toolbar/precog/config"
  18         "github.com/vapor/toolbar/precog/database/orm"
  19 )
  20
  21 type monitor struct {
  22         cfg     *config.Config
  23         db      *gorm.DB
  24         dirPath *os.File
  25 }
  26
  27 func NewMonitor(cfg *config.Config, db *gorm.DB) *monitor {
  28         dirPath, err := ioutil.TempDir(".", "")
  29         if err != nil {
  30                 log.Fatal(err)
  31         }
  32
  33         return &monitor{
  34                 cfg:     cfg,
  35                 db:      db,
  36                 dirPath: dirPath,
  37         }
  38 }
  39
  40 func (m *monitor) Run() {
  41         defer os.RemoveAll(m.dirPath)
  42
  43         m.updateBootstrapNodes()
  44         go m.discovery()
  45         ticker := time.NewTicker(time.Duration(m.cfg.CheckFreqSeconds) * time.Second)
  46         for ; true; <-ticker.C {
  47                 // TODO: lock?
  48                 m.monitorRountine()
  49         }
  50 }
  51
  52 // create or update: https://github.com/jinzhu/gorm/issues/1307
  53 func (m *monitor) updateBootstrapNodes() {
  54         for _, node := range m.cfg.Nodes {
  55                 ormNode := &orm.Node{
  56                         PublicKey: node.PublicKey.String(),
  57                         Alias:     node.Alias,
  58                         Host:      node.Host,
  59                         Port:      node.Port,
  60                 }
  61
  62                 if err := m.db.Where(&orm.Node{PublicKey: ormNode.PublicKey}).
  63                         Assign(&orm.Node{
  64                                 Alias: node.Alias,
  65                                 Host:  node.Host,
  66                                 Port:  node.Port,
  67                         }).FirstOrCreate(ormNode).Error; err != nil {
  68                         log.Error(err)
  69                         continue
  70                 }
  71         }
  72 }
  73
  74 // TODO:
  75 // implement logic first, and then refactor
  76 // /home/gavin/work/go/src/github.com/vapor/
  77 // p2p/test_util.go
  78 // p2p/switch_test.go
  79 func (m *monitor) discovery() {
  80         // TODO: 包一下？  common cfg 之类的？
  81         mCfg := &cfg.Config{
  82                 BaseConfig: cfg.DefaultBaseConfig(),
  83                 P2P:        cfg.DefaultP2PConfig(),
  84                 Federation: cfg.DefaultFederationConfig(),
  85         }
  86         mCfg.DBPath = m.dirPath
  87
  88         // swPrivKey, err := signlib.NewPrivKey()
  89         // if err != nil {
  90         //      log.Fatal(err)
  91         // }
  92
  93         // TODO: whatz that for
  94         // testDB := dbm.NewDB("testdb", "leveldb", dirPath)
  95         // TODO: clean up
  96         // log.Println("Federation.Xpubs", mCfg.Federation.Xpubs)
  97         sw, err := p2p.NewSwitch(mCfg)
  98         if err != nil {
  99                 log.Fatal(err)
 100         }
 101
 102         sw.Start()
 103         defer sw.Stop()
 104 }
 105
 106 func (m *monitor) makeSwitch() {
 107
 108 }
 109
 110 func (m *monitor) monitorRountine() error {
 111         // TODO: dail nodes, get lantency & best_height
 112         // TODO: decide check_height("best best_height" - "confirmations")
 113         // TODO: get blockhash by check_height, get latency
 114         // TODO: update lantency, active_time and status
 115         return nil
 116 }