config networkID

[bytom/vapor.git] / toolbar / precog / monitor / monitor.go

diff --git a/toolbar/precog/monitor/monitor.go b/toolbar/precog/monitor/monitor.go
index d0bb6cd..e11a657 100644 (file)
--- a/toolbar/precog/monitor/monitor.go
+++ b/toolbar/precog/monitor/monitor.go
@@ -1,26 +1,123 @@
 package monitor
 
 import (
-       "github.com/jinzhu/gorm"
+       // "encoding/binary"
+       // "io/ioutil"
+       "fmt"
+       "os"
+       "strings"
+       "time"
 
+       "github.com/jinzhu/gorm"
        log "github.com/sirupsen/logrus"
+       // dbm "github.com/vapor/database/leveldb"
+
+       vaporCfg "github.com/vapor/config"
+       "github.com/vapor/p2p"
+       // conn "github.com/vapor/p2p/connection"
+       // "github.com/vapor/consensus"
+       // "github.com/vapor/crypto/sha3pool"
+       "github.com/vapor/p2p/discover/dht"
+       "github.com/vapor/p2p/discover/mdns"
+       "github.com/vapor/p2p/signlib"
+       "github.com/vapor/toolbar/precog/config"
+       "github.com/vapor/toolbar/precog/database/orm"
 )
 
 type monitor struct {
-       db *gorm.DB
+       cfg     *config.Config
+       db      *gorm.DB
+       nodeCfg *vaporCfg.Config
 }
 
-func NewMonitor(db *gorm.DB) *monitor {
-       return &monitor{db: db}
+func NewMonitor(cfg *config.Config, db *gorm.DB) *monitor {
+       nodeCfg := &vaporCfg.Config{
+               BaseConfig: vaporCfg.DefaultBaseConfig(),
+               P2P:        vaporCfg.DefaultP2PConfig(),
+               Federation: vaporCfg.DefaultFederationConfig(),
+       }
+       nodeCfg.DBPath = "vapor_precog_data"
+
+       return &monitor{
+               cfg:     cfg,
+               db:      db,
+               nodeCfg: nodeCfg,
+       }
 }
 
-func (s *monitor) Run() {
-       if err := s.updateNodesHostPort(); err != nil {
-               // TODO: redirect output to logfile
+func (m *monitor) Run() {
+       defer os.RemoveAll(m.nodeCfg.DBPath)
+
+       m.updateBootstrapNodes()
+       go m.discovery()
+       ticker := time.NewTicker(time.Duration(m.cfg.CheckFreqSeconds) * time.Second)
+       for ; true; <-ticker.C {
+               // TODO: lock?
+               m.monitorRountine()
+       }
+}
+
+// create or update: https://github.com/jinzhu/gorm/issues/1307
+func (m *monitor) updateBootstrapNodes() {
+       var seeds []string
+       for _, node := range m.cfg.Nodes {
+               ormNode := &orm.Node{
+                       PublicKey: node.PublicKey.String(),
+                       Alias:     node.Alias,
+                       Host:      node.Host,
+                       Port:      node.Port,
+               }
+               seeds = append(seeds, fmt.Sprintf("%s:%d", node.Host, node.Port))
+
+               if err := m.db.Where(&orm.Node{PublicKey: ormNode.PublicKey}).
+                       Assign(&orm.Node{
+                               Alias: node.Alias,
+                               Host:  node.Host,
+                               Port:  node.Port,
+                       }).FirstOrCreate(ormNode).Error; err != nil {
+                       log.Error(err)
+                       continue
+               }
+       }
+       m.nodeCfg.P2P.Seeds = strings.Join(seeds, ",")
+}
+
+// TODO:
+// implement logic first, and then refactor
+// /home/gavin/work/go/src/github.com/vapor/
+// p2p/test_util.go
+// p2p/switch_test.go
+// syncManager
+// notificationMgr
+func (m *monitor) discovery() {
+       sw, err := m.makeSwitch()
+       if err != nil {
                log.Fatal(err)
        }
+
+       sw.Start()
+}
+
+func (m *monitor) makeSwitch() (*p2p.Switch, error) {
+       swPrivKey, err := signlib.NewPrivKey()
+       if err != nil {
+               return nil, err
+       }
+
+       l, listenAddr := p2p.GetListener(m.nodeCfg.P2P)
+       discv, err := dht.NewDiscover(m.nodeCfg, swPrivKey, l.ExternalAddress().Port, m.cfg.NetworkID)
+       if err != nil {
+               return nil, err
+       }
+
+       lanDiscv := mdns.NewLANDiscover(mdns.NewProtocol(), int(l.ExternalAddress().Port))
+       return p2p.NewSwitch(m.nodeCfg, discv, lanDiscv, l, swPrivKey, listenAddr, m.cfg.NetworkID)
 }
 
-func (s *monitor) updateNodesHostPort() error {
+func (m *monitor) monitorRountine() error {
+       // TODO: dail nodes, get lantency & best_height
+       // TODO: decide check_height("best best_height" - "confirmations")
+       // TODO: get blockhash by check_height, get latency
+       // TODO: update lantency, active_time and status
        return nil
 }