yanic/models/nodes.go

91 lines
1.6 KiB
Go
Raw Normal View History

package models
2015-12-29 14:05:47 +01:00
import (
"encoding/json"
"io/ioutil"
"log"
"os"
"sync"
"time"
)
2016-02-25 21:24:54 +01:00
//Node struct
2015-12-29 14:05:47 +01:00
type Node struct {
Firstseen time.Time `json:"firstseen"`
Lastseen time.Time `json:"lastseen"`
Statistics interface{} `json:"statistics"`
Nodeinfo interface{} `json:"nodeinfo"`
Neighbours interface{} `json:"neighbours"`
}
2016-02-25 21:24:54 +01:00
//Nodes struct: cache DB of Node's structs
2015-12-29 14:05:47 +01:00
type Nodes struct {
Version int `json:"version"`
Timestamp time.Time `json:"timestamp"`
List map[string]*Node `json:"nodes"` // the current nodemap
sync.Mutex
}
2016-02-25 21:24:54 +01:00
//NewNodes create Nodes structs (cache DB)
2015-12-29 14:05:47 +01:00
func NewNodes() *Nodes {
nodes := &Nodes{
Version: 1,
List: make(map[string]*Node),
}
return nodes
}
2016-02-25 21:24:54 +01:00
//Get a Node by nodeid
func (nodes *Nodes) Get(nodeID string) *Node {
2015-12-29 14:05:47 +01:00
now := time.Now()
nodes.Lock()
2016-02-25 21:24:54 +01:00
node, _ := nodes.List[nodeID]
2015-12-29 14:05:47 +01:00
if node == nil {
node = &Node{
Firstseen: now,
}
2016-02-25 21:24:54 +01:00
nodes.List[nodeID] = node
2015-12-29 14:05:47 +01:00
}
nodes.Unlock()
node.Lastseen = now
return node
}
2016-02-25 21:24:54 +01:00
//Saver to save the cached DB to json file
func (nodes *Nodes) Saver(outputFile string, saveInterval time.Duration) {
2015-12-29 14:05:47 +01:00
c := time.Tick(saveInterval)
for range c {
nodes.save(outputFile)
2015-12-29 14:05:47 +01:00
}
}
func (nodes *Nodes) save(outputFile string) {
2015-12-29 14:05:47 +01:00
nodes.Timestamp = time.Now()
nodes.Lock()
data, err := json.Marshal(nodes)
nodes.Unlock()
if err != nil {
2016-02-19 11:30:42 +01:00
log.Panic(err)
2016-02-19 11:13:30 +01:00
}
2015-12-29 14:05:47 +01:00
log.Println("saving", len(nodes.List), "nodes")
tmpFile := outputFile + ".tmp"
2016-02-19 11:30:42 +01:00
err = ioutil.WriteFile(tmpFile, data, 0644)
if err != nil {
2016-02-19 11:30:42 +01:00
log.Panic(err)
}
err = os.Rename(tmpFile, outputFile)
if err != nil {
2016-02-19 11:30:42 +01:00
log.Panic(err)
}
2015-12-29 14:05:47 +01:00
}