yanic/models/nodes.go

104 lines
2.0 KiB
Go

package models
import (
"encoding/json"
"io/ioutil"
"log"
"os"
"sync"
"time"
"github.com/FreifunkBremen/respond-collector/data"
)
// Node struct
type Node struct {
Firstseen time.Time `json:"firstseen"`
Lastseen time.Time `json:"lastseen"`
Statistics *data.Statistics `json:"statistics"`
Nodeinfo *data.NodeInfo `json:"nodeinfo"`
Neighbours *data.Neighbours `json:"-"`
}
type NodeElement struct {
NodeId string
}
// Nodes struct: cache DB of Node's structs
type Nodes struct {
Version int `json:"version"`
Timestamp time.Time `json:"timestamp"`
List map[string]*Node `json:"nodes"` // the current nodemap, indexed by node ID
sync.Mutex
}
// NewNodes create Nodes structs (cache DB)
func NewNodes() *Nodes {
nodes := &Nodes{
Version: 2,
List: make(map[string]*Node),
}
return nodes
}
// Get a Node by nodeid
func (nodes *Nodes) Get(nodeID string) *Node {
now := time.Now()
nodes.Lock()
node, _ := nodes.List[nodeID]
if node == nil {
node = &Node{
Firstseen: now,
Nodeinfo: &data.NodeInfo{},
Statistics: &data.Statistics{},
Neighbours: &data.Neighbours{},
}
nodes.List[nodeID] = node
}
nodes.Unlock()
node.Lastseen = now
return node
}
// Saves the cached DB to json file periodically
func (nodes *Nodes) Saver(config *Config) {
c := time.Tick(time.Second * time.Duration(config.Nodes.SaveInterval))
for range c {
log.Println("saving", len(nodes.List), "nodes")
nodes.Timestamp = time.Now()
nodes.Lock()
if path := config.Nodes.NodesPath; path != "" {
save(nodes, path)
}
if path := config.Nodes.GraphsPath; path != "" {
save(nodes.BuildGraph(config.Nodes.VpnAddresses), path)
}
nodes.Unlock()
}
}
func save(input interface{}, outputFile string) {
data, err := json.Marshal(input)
if err != nil {
log.Panic(err)
}
tmpFile := outputFile + ".tmp"
err = ioutil.WriteFile(tmpFile, data, 0644)
if err != nil {
log.Panic(err)
}
err = os.Rename(tmpFile, outputFile)
if err != nil {
log.Panic(err)
}
}