dashboard: experiment 5

This commit is contained in:
Kurkó Mihály 2018-09-27 12:25:01 +03:00
parent a9cedd619e
commit dd9dd0ddc7
4 changed files with 349 additions and 188 deletions

View file

@ -42,7 +42,7 @@ import (
) )
const ( const (
sampleLimit = 200 // Maximum number of data samples sampleLimit = 5 // Maximum number of data samples
) )
// Dashboard contains the dashboard internals. // Dashboard contains the dashboard internals.
@ -100,9 +100,6 @@ func New(config *Config, commit string, logdir string) *Dashboard {
DiskRead: emptyChartEntries(now, sampleLimit, config.Refresh), DiskRead: emptyChartEntries(now, sampleLimit, config.Refresh),
DiskWrite: emptyChartEntries(now, sampleLimit, config.Refresh), DiskWrite: emptyChartEntries(now, sampleLimit, config.Refresh),
}, },
//Network: &NetworkMessage{
// PeerBundles: make(map[string]*PeerBundle),
//},
}, },
logdir: logdir, logdir: logdir,
} }

View file

@ -42,6 +42,14 @@ type GeoDBInfo struct {
} `maxminddb:"location" json:"location,omitempty"` } `maxminddb:"location" json:"location,omitempty"`
} }
// GeoLocation contains geographical information.
type GeoLocation struct {
Country string `json:"country,omitempty"`
City string `json:"city,omitempty"`
Latitude float64 `json:"latitude,omitempty"`
Longitude float64 `json:"longitude,omitempty"`
}
// GeoDB represents a geoip database that can be queried for IP to geographical // GeoDB represents a geoip database that can be queried for IP to geographical
// information conversions. // information conversions.
type GeoDB struct { type GeoDB struct {

View file

@ -26,7 +26,8 @@ type Message struct {
Home *HomeMessage `json:"home,omitempty"` Home *HomeMessage `json:"home,omitempty"`
Chain *ChainMessage `json:"chain,omitempty"` Chain *ChainMessage `json:"chain,omitempty"`
TxPool *TxPoolMessage `json:"txpool,omitempty"` TxPool *TxPoolMessage `json:"txpool,omitempty"`
Network *PeerContainer `json:"network,omitempty"` Network *NetworkMessage `json:"network,omitempty"`
Peers *PeersMessage `json:"peers,omitempty"`
System *SystemMessage `json:"system,omitempty"` System *SystemMessage `json:"system,omitempty"`
Logs *LogsMessage `json:"logs,omitempty"` Logs *LogsMessage `json:"logs,omitempty"`
} }
@ -57,83 +58,9 @@ type TxPoolMessage struct {
// NetworkMessage contains information about the peers organized based on the IP address. // NetworkMessage contains information about the peers organized based on the IP address.
type NetworkMessage struct { type NetworkMessage struct {
PeerBundles map[string]*PeerBundle `json:"peerBundles,omitempty"` /* TODO (kurkomisi) */
} }
// getOrInitBundle returns the peer bundle belonging to the given IP, or
// initializes the bundle if it doesn't exist.
//func (m *NetworkMessage) getOrInitBundle(ip string) *PeerBundle {
// if _, ok := m.PeerBundles[ip]; !ok {
// m.PeerBundles[ip] = &PeerBundle{
// Peers: make(PeerMap),
// FailedPeers: make(PeerMap),
// }
// }
// return m.PeerBundles[ip]
//}
//
//// getOrInitPeer returns the peer belonging to the given IP and node id, or
//// initializes the peer if it doesn't exist.
//func (m *NetworkMessage) getOrInitPeer(ip, id string) *Peer {
// return m.getOrInitBundle(ip).getOrInitPeer(id)
//}
//type PeerMap map[string]*Peer
//func (pm PeerMap) getOrInitBundle(id string) *Peer {
// if _, ok := pm[id]; !ok {
// pm[id] = new(Peer)
// }
// return pm[id]
//}
//
//func (pm PeerMap) remove(id string) {
// delete(pm, id)
//}
//// PeerBundle contains information about the peers pertaining to an IP address.
//type PeerBundle struct {
// Location *GeoLocation `json:"location,omitempty"` // geographical information based on IP
// Peers PeerMap `json:"peers,omitempty"` // the peers' node id is used as key
// FailedPeers PeerMap `json:"failedPeers,omitempty"`
//}
//
//func (b *PeerBundle) getOrInitPeer(id string) *Peer {
// return b.Peers.getOrInitBundle(id)
//}
//
//func (b *PeerBundle) removePeer(id string) {
// b.Peers.remove(id)
//}
//
//func (b *PeerBundle) getOrInitFailedPeer(id string) *Peer {
// return b.FailedPeers.getOrInitBundle(id)
//}
//
//func (b * PeerBundle) removeFailedPeer(id string) {
// b.FailedPeers.remove(id)
//}
// GeoLocation contains geographical information.
type GeoLocation struct {
Country string `json:"country,omitempty"`
City string `json:"city,omitempty"`
Latitude float64 `json:"latitude,omitempty"`
Longitude float64 `json:"longitude,omitempty"`
}
// Peer contains lifecycle timestamps and traffic information of a given peer.
//type Peer struct {
// Connected []time.Time `json:"connected,omitempty"`
// Disconnected []time.Time `json:"disconnected,omitempty"`
//
// Ingress ChartEntries `json:"ingress,omitempty"`
// Egress ChartEntries `json:"egress,omitempty"`
//
// element *list.Element
// ip, id string
//}
// SystemMessage contains the metered system data samples. // SystemMessage contains the metered system data samples.
type SystemMessage struct { type SystemMessage struct {
ActiveMemory ChartEntries `json:"activeMemory,omitempty"` ActiveMemory ChartEntries `json:"activeMemory,omitempty"`

View file

@ -34,149 +34,286 @@ const (
unknownPeerLimit = p2p.MeteredPeerLimit unknownPeerLimit = p2p.MeteredPeerLimit
) )
type PeerContainer struct { // PeersMessage contains information about the node's peers. This data structure
Bundles map[string]*PeerBundle `json:"peerBundles,omitempty"` // tries to maintain the metered peer data based on the different behaviours of
// the peers.
//
// Every peer has an IP address, and the peers that manage to make the handshake
// have node IDs. There can appear more peers with the same IP, therefore the
// peer maintainer data structure is a tree consisting of a map of maps, where
// the first key groups the peers by IP, while the second one groups them by the
// node ID. The peers failing before the handshake only have IP addresses, so
// they are stored as part of the value of the outer map. A peer can connect
// multiple times, so an array is needed to store the consecutive sessions.
//
// Another criteria is to limit the number of metered peers so that they don't
// fill the memory. The metered peers are selected based on their activity: the
// peers that are inactive for the longest time are thrown first. For the selection a fifo
// list is used which is linked to the bottom of the peer tree, and when a peer
// is removed from the list, it is also removed from the tree. The active peers
// have priority over the disconnected ones, therefore this list is extended by
// a separator, which is a pointer to a list element. The separator separates the
// active peers from the inactive ones, and it is the entry for the list. If the
// peer that is to be inserted is active, it goes before the separator, otherwise
// it goes after. This way the peers that are active for the longest time are at
// the beginning of the list, and the inactive ones move to the end. When a peer
// has some activity, it is removed from and reinserted into the list.
//
// The peers that don't manage to make handshake are not inserted into the list,
// their sessions are not stored, only their connection attempts are appended
// to the array belonging to their IP. In order to keep the fifo principle, a super
// array contains the order of the attempts, and when the peer count reaches the
// limit, the earliest attempt is removed from the beginning of its array.
type PeersMessage struct {
// Bundles is the outer map using the peer's IP address as key.
Bundles map[string]*PeerBundle `json:"bundles,omitempty"`
RemovedKnown []string `json:"removedKnown,omitempty"`
RemovedUnknown []string `json:"removedUnknown,omitempty"`
// activeSeparator is a pointer to the last active peer element, splitting
// the list into an active and inactive part, and forming the entry for the
// peer list.
activeSeparator *list.Element activeSeparator *list.Element
// knownPeers contains the peers that managed to make handshake.
knownPeers *list.List knownPeers *list.List
// unknownPeers contains pointers to the peer bundles that belong to the
// IP addresses, from which the peers attempted to connect then failed.
// Its values are appended in the moment of the attempt, so in chronological
// order, which means that the oldest attempt is at the beginning of the array.
// When the first element is removed, the first element of the linked bundle's
// attempt array is also removed, ensuring that always the latest attempts are stored.
unknownPeers []*PeerBundle unknownPeers []*PeerBundle
// geodb is used to look up the geographical data based on the IP.
geodb *GeoDB geodb *GeoDB
refresh time.Duration
} }
func NewPeerContainer(geodb *GeoDB, refresh time.Duration) *PeerContainer { // NewPeersMessage returns a new instance of the metered peer maintainer data structure.
return &PeerContainer{ func NewPeersMessage(geodb *GeoDB) *PeersMessage {
return &PeersMessage{
Bundles: make(map[string]*PeerBundle), Bundles: make(map[string]*PeerBundle),
knownPeers: list.New(), knownPeers: list.New(),
unknownPeers: make([]*PeerBundle, 0, unknownPeerLimit), unknownPeers: make([]*PeerBundle, 0, unknownPeerLimit),
geodb: geodb, geodb: geodb,
refresh: refresh,
} }
} }
func (pc *PeerContainer) getOrInitBundle(ip string) *PeerBundle { func (m *PeersMessage) hasBundle(ip string) bool {
if _, ok := pc.Bundles[ip]; !ok { _, ok := m.Bundles[ip]
pc.Bundles[ip] = &PeerBundle{ return ok
Location: pc.geodb.Location(ip), }
func (m *PeersMessage) hasPeer(ip, id string) bool {
if !m.hasBundle(ip) {
return false
}
return m.Bundles[ip].has(id)
}
// getOrInitBundle returns the bundle belonging to the given IP.
// Inserts a new bundle into the map if it doesn't already exist.
func (m *PeersMessage) getOrInitBundle(ip string) *PeerBundle {
if _, ok := m.Bundles[ip]; !ok {
m.Bundles[ip] = &PeerBundle{
Location: m.geodb.Location(ip),
KnownPeers: make(map[string]*KnownPeer), KnownPeers: make(map[string]*KnownPeer),
parent: pc, root: m,
ip: ip, ip: ip,
} }
} }
return pc.Bundles[ip] return m.Bundles[ip]
} }
func (pc *PeerContainer) updateKnown(ip, id string, session *PeerSession) (removedIP, removedID string) { func (m *PeersMessage) getOrInitKnownPeer(ip, id string) *KnownPeer {
peer := pc.getOrInitBundle(ip).getOrInitKnown(id) return m.getOrInitBundle(ip).getOrInitKnownPeer(id)
if peer.Sessions == nil {
peer.Sessions = []*PeerSession{{
Ingress: emptyChartEntries(time.Now(), 5, pc.refresh),
Egress: emptyChartEntries(time.Now(), 5, pc.refresh),
}}
} }
// updateKnownPeer updates the last session of the peer belonging to the given IP
// and ID and returns the IP and the ID of the removed peer if there is any,
// (nil, nil) otherwise.
func (m *PeersMessage) updateKnownPeer(ip, id string, session *PeerSession) (removedKey string) {
peer := m.getOrInitKnownPeer(ip, id)
if peer.listElement != nil { if peer.listElement != nil {
if peer.listElement == pc.activeSeparator { // If the peer is already part of the list, remove it first.
pc.activeSeparator = pc.activeSeparator.Prev() if peer.listElement == m.activeSeparator {
m.activeSeparator = m.activeSeparator.Prev()
} }
pc.knownPeers.Remove(peer.listElement) m.knownPeers.Remove(peer.listElement)
} }
if pc.knownPeers.Len() >= knownPeerLimit { if m.knownPeers.Len() >= knownPeerLimit {
removed := pc.knownPeers.Remove(pc.knownPeers.Back()) // If the peer count reached the limit, remove the last element of the
// list, which is the oldest inactive one.
removed := m.knownPeers.Remove(m.knownPeers.Back())
if p, ok := removed.(*KnownPeer); ok { if p, ok := removed.(*KnownPeer); ok {
removedIP, removedID = p.parent.ip, p.id removedKey = fmt.Sprintf("%s/%s", p.bundle.ip, p.id)
p.delete() p.delete() // Remove the peer from the tree.
} else { } else {
log.Warn("Bad value used as peer", "value", removed) log.Warn("Bad value used as peer", "value", removed)
} }
} }
if pc.activeSeparator == nil { if m.activeSeparator == nil {
peer.listElement = pc.knownPeers.PushFront(peer) // If there isn't any active peer in the list, push the new peer to the
// front of the list.
peer.listElement = m.knownPeers.PushFront(peer)
} else { } else {
peer.listElement = pc.knownPeers.InsertAfter(peer, pc.activeSeparator) // If there are active peers in the list, push the new peer after them.
peer.listElement = m.knownPeers.InsertAfter(peer, m.activeSeparator)
} }
peer.update(session) peer.update(session)
if peer.Sessions[len(peer.Sessions)-1].Disconnected == nil { if peer.Sessions[len(peer.Sessions)-1].Disconnected == nil {
pc.activeSeparator = peer.listElement // If the new peer is active, step to it with the separator.
m.activeSeparator = peer.listElement
} }
return removedIP, removedID return removedKey
} }
func (pc *PeerContainer) updateUnknown(ip string, peer *UnknownPeer) { // updateUnknownPeer inserts a peer connection attempt into the peer tree.
if len(pc.unknownPeers) >= unknownPeerLimit { func (m *PeersMessage) updateUnknownPeer(ip string, peer *UnknownPeer) (removedKey string) {
removed := pc.unknownPeers[0] if len(m.unknownPeers) >= unknownPeerLimit {
// If the count of the metered unknown peers reached the limit,
// remove the oldest attempt, which is the first element of the
// array belonging to the peer bundle pointed by the first element
// of the super unknown peer array.
removed := m.unknownPeers[0]
removed.UnknownPeers = removed.UnknownPeers[1:] removed.UnknownPeers = removed.UnknownPeers[1:]
pc.unknownPeers = pc.unknownPeers[1:] m.unknownPeers = m.unknownPeers[1:]
removedKey = removed.ip
} }
bundle := pc.getOrInitBundle(ip) bundle := m.getOrInitBundle(ip)
bundle.UnknownPeers = append(bundle.UnknownPeers, peer) bundle.UnknownPeers = append(bundle.UnknownPeers, peer)
pc.unknownPeers = append(pc.unknownPeers, bundle) m.unknownPeers = append(m.unknownPeers, bundle)
return removedKey
} }
func (pc *PeerContainer) clear() { // clear removes the elements of the metered peer list, and removes the linked
for pc.knownPeers.Front() != nil { // peers from the peer tree along the way.
first := pc.knownPeers.Front() func (m *PeersMessage) clear() {
if p, ok := first.Value.(*KnownPeer); ok { for m.knownPeers.Front() != nil {
first := m.knownPeers.Remove(m.knownPeers.Front())
if p, ok := first.(*KnownPeer); ok {
p.delete() p.delete()
} else { } else {
log.Warn("Bad value used as peer", "value", first) log.Warn("Bad value used as peer", "value", first)
} }
} }
pc.activeSeparator = nil m.activeSeparator = nil
} }
func (pc *PeerContainer) updateTraffic(traffic *peerTraffic) { func (m *PeersMessage) updateTraffic(ip, id string, ingress, egress float64) {
now := time.Now() if !m.hasPeer(ip, id) {
for e := pc.knownPeers.Front(); e != nil; e = e.Next() { m.updateKnownPeer(ip, id, &PeerSession{
if p, ok := e.Value.(*KnownPeer); ok { Ingress: ChartEntries{&ChartEntry{
key := fmt.Sprintf("%s/%s", p.parent.ip, p.id) Value: ingress,
p.updateTraffic( }},
&ChartEntry{Time: now, Value: float64(traffic.ingress[key])}, Egress: ChartEntries{&ChartEntry{
&ChartEntry{Time: now, Value: float64(traffic.egress[key])}, Value: egress,
}},
})
return
}
m.getOrInitKnownPeer(ip, id).updateTraffic(
&ChartEntry{Value: ingress},
&ChartEntry{Value: egress},
) )
} else {
log.Warn("Bad value used as peer", "value", e.Value)
}
}
} }
//func (m *PeersMessage) updateTraffic(ingress, egress *map[string]float64) {
// now := time.Now()
// for k, v := range *ingress {
//
// }
// for e := m.knownPeers.Front(); e != nil; e = e.Next() {
// if p, ok := e.Value.(*KnownPeer); ok {
// key := fmt.Sprintf("%s/%s", p.bundle.ip, p.id)
// p.updateTraffic(
// &ChartEntry{Time: now, Value: (*ingress)[key]},
// &ChartEntry{Time: now, Value: (*egress)[key]},
// )
// } else {
// log.Warn("Bad value used as peer", "value", e.Value)
// }
// }
//}
// append
func (m *PeersMessage) append(n *PeersMessage) (removedKnown, removedUnknown []string) {
for _, bundle := range n.Bundles {
for _, peer := range bundle.KnownPeers {
for _, session := range peer.Sessions {
removedKey := m.updateKnownPeer(bundle.ip, peer.id, session)
if removedKey != "" {
removedKnown = append(removedKnown, removedKey)
}
}
}
for _, peer := range bundle.UnknownPeers {
removedKey := m.updateUnknownPeer(bundle.ip, peer)
if removedKey != "" {
removedUnknown = append(removedUnknown, removedKey)
}
}
}
return removedKnown, removedUnknown
}
// PeerBundle contains the peers belonging to a given IP address
type PeerBundle struct { type PeerBundle struct {
Location *GeoLocation `json:"location,omitempty"` Location *GeoLocation `json:"location,omitempty"` // Geographical location based on IP
// KnownPeers is the inner map of the metered peer maintainer data structure
// using the node ID as key.
KnownPeers map[string]*KnownPeer `json:"knownPeers,omitempty"` KnownPeers map[string]*KnownPeer `json:"knownPeers,omitempty"`
// UnknownPeers contains the failed connection attempts of the peers
// belonging to a given IP address in chronological order.
UnknownPeers []*UnknownPeer `json:"unknownPeers,omitempty"` UnknownPeers []*UnknownPeer `json:"unknownPeers,omitempty"`
parent *PeerContainer root *PeersMessage // Pointer to the outer map.
ip string ip string // Key of the bundle in the outer map.
element list.Element
} }
func (b *PeerBundle) getOrInitKnown(id string) *KnownPeer { func (b *PeerBundle) has(id string) bool {
_, ok := b.KnownPeers[id]
return ok
}
// getOrInitKnownPeer returns the peer belonging to the given ID.
// Initializes it if it doesn't already exist.
func (b *PeerBundle) getOrInitKnownPeer(id string) *KnownPeer {
if _, ok := b.KnownPeers[id]; !ok { if _, ok := b.KnownPeers[id]; !ok {
b.KnownPeers[id] = &KnownPeer{ b.KnownPeers[id] = &KnownPeer{
parent: b, sampleCount: sampleLimit,
bundle: b,
id: id, id: id,
} }
} }
return b.KnownPeers[id] return b.KnownPeers[id]
} }
// KnownPeer contains the metered data of a particular peer.
type KnownPeer struct { type KnownPeer struct {
// Sessions contains the metered data in a session of a peer.
Sessions []*PeerSession `json:"sessions,omitempty"` Sessions []*PeerSession `json:"sessions,omitempty"`
parent *PeerBundle sampleCount int
id string
listElement *list.Element bundle *PeerBundle // Pointer to the inner map.
id string // Key of the peer in the inner map.
listElement *list.Element // Pointer to the peer element in the list.
} }
func (peer *KnownPeer) delete() { func (peer *KnownPeer) delete() {
delete(peer.parent.KnownPeers, peer.id) delete(peer.bundle.KnownPeers, peer.id)
fmt.Println(peer, "to make sure p is not cleared") if len(peer.bundle.KnownPeers) < 1 && len(peer.bundle.UnknownPeers) < 1 {
if len(peer.parent.KnownPeers) < 1 && len(peer.parent.UnknownPeers) < 1 { delete(peer.bundle.root.Bundles, peer.bundle.ip)
delete(peer.parent.parent.Bundles, peer.parent.ip)
} }
peer.listElement = nil peer.listElement = nil
peer.parent = nil peer.bundle = nil
for i := range peer.Sessions { for i := range peer.Sessions {
peer.Sessions[i] = nil peer.Sessions[i] = nil
} }
@ -184,16 +321,35 @@ func (peer *KnownPeer) delete() {
} }
func (peer *KnownPeer) update(session *PeerSession) { func (peer *KnownPeer) update(session *PeerSession) {
if peer.Sessions == nil {
peer.Sessions = []*PeerSession{session}
if session.Connected != nil && session.Disconnected != nil && session.Connected.After(*session.Disconnected) { if session.Connected != nil && session.Disconnected != nil && session.Connected.After(*session.Disconnected) {
peer.Sessions[len(peer.Sessions)-1].Disconnected = session.Disconnected peer.Sessions = append(peer.Sessions, &PeerSession{Connected: session.Connected})
session.Connected = nil
}
return
}
if session.Connected != nil && session.Disconnected != nil && session.Connected.After(*session.Disconnected) {
last := peer.Sessions[len(peer.Sessions)-1]
if last.Connected == nil {
// This may happen at the beginning, if the connection was established
// before the initialization of the peer event feed. Otherwise it should
// not happen to handle a disconnect event before the connect, because
// both of them are coming on the same channel consecutively.
log.Warn("Disconnect event appeared without connect")
}
last.Disconnected = session.Disconnected
last.Ingress = append(last.Ingress, session.Ingress...)
last.Egress = append(last.Egress, session.Egress...)
session.Disconnected = nil session.Disconnected = nil
session.Ingress = nil
session.Egress = nil
} }
if session.Connected != nil { if session.Connected != nil {
peer.Sessions = append(peer.Sessions, session) peer.Sessions = append(peer.Sessions, session)
} }
last := peer.Sessions[len(peer.Sessions)-1]
if session.Disconnected != nil { if session.Disconnected != nil {
last.Disconnected = session.Disconnected peer.Sessions[len(peer.Sessions)-1].Disconnected = session.Disconnected
} }
for i := 0; i < len(session.Ingress) && i < len(session.Egress); i++ { for i := 0; i < len(session.Ingress) && i < len(session.Egress); i++ {
peer.updateTraffic(session.Ingress[i], session.Egress[i]) peer.updateTraffic(session.Ingress[i], session.Egress[i])
@ -201,15 +357,36 @@ func (peer *KnownPeer) update(session *PeerSession) {
} }
func (peer *KnownPeer) updateTraffic(ingress, egress *ChartEntry) { func (peer *KnownPeer) updateTraffic(ingress, egress *ChartEntry) {
first, last := peer.Sessions[0], peer.Sessions[len(peer.Sessions)-1] if ingress == nil {
ingress = new(ChartEntry)
}
if egress == nil {
egress = new(ChartEntry)
}
first := peer.Sessions[0]
if len(first.Ingress) < 1 || len(first.Egress) < 1 {
first.Ingress = append(first.Ingress, ingress)
first.Egress = append(first.Egress, egress)
peer.sampleCount = 1
return
}
if peer.sampleCount >= sampleLimit {
if len(first.Ingress) < 2 || len(first.Egress) < 2 { if len(first.Ingress) < 2 || len(first.Egress) < 2 {
peer.Sessions = peer.Sessions[1:] peer.Sessions = peer.Sessions[1:]
} else { } else {
first.Ingress = first.Ingress[1:] first.Ingress = first.Ingress[1:]
first.Egress = first.Egress[1:] first.Egress = first.Egress[1:]
} }
peer.sampleCount--
}
last := peer.Sessions[len(peer.Sessions)-1]
last.Ingress = append(last.Ingress, ingress) last.Ingress = append(last.Ingress, ingress)
last.Egress = append(last.Egress, egress) last.Egress = append(last.Egress, egress)
peer.sampleCount++
}
func (peer *KnownPeer) len() int {
return peer.sampleCount
} }
type PeerSession struct { type PeerSession struct {
@ -221,13 +398,8 @@ type PeerSession struct {
} }
type UnknownPeer struct { type UnknownPeer struct {
Connected time.Time `json:"connected,omitempty"` Connected time.Time `json:"connected"`
Disconnected time.Time `json:"disconnected,omitempty"` Disconnected time.Time `json:"disconnected"`
}
type peerTraffic struct {
ingress map[string]int64
egress map[string]int64
} }
// collectPeerData gathers data about the peers and sends it to the clients. // collectPeerData gathers data about the peers and sends it to the clients.
@ -250,19 +422,24 @@ func (db *Dashboard) collectPeerData() {
ticker := time.NewTicker(db.config.Refresh) ticker := time.NewTicker(db.config.Refresh)
defer ticker.Stop() defer ticker.Stop()
pc := NewPeerContainer(db.geodb, db.config.Refresh) db.peerLock.Lock()
db.history.Peers = NewPeersMessage(db.geodb)
db.peerLock.Unlock()
diff := NewPeersMessage(db.geodb)
trafficUpdater := func(prefix string) func(*map[string]int64) func(name string, i interface{}) { trafficCollector := func(prefix string) func(*map[string]float64) func(name string, i interface{}) {
return func(entryMap *map[string]int64) func(name string, i interface{}) { return func(traffic *map[string]float64) func(name string, i interface{}) {
return func(name string, i interface{}) { return func(name string, i interface{}) {
if m, ok := i.(metrics.Meter); ok { if m, ok := i.(metrics.Meter); ok {
(*entryMap)[strings.TrimPrefix(name, prefix)] = m.Count() (*traffic)[strings.TrimPrefix(name, prefix)] = float64(m.Count())
} else {
log.Warn("Bad value used as meter", "name", name)
} }
} }
} }
} }
updateIngress := trafficUpdater(p2p.MetricsInboundTraffic + "/") collectIngress := trafficCollector(p2p.MetricsInboundTraffic + "/")
updateEgress := trafficUpdater(p2p.MetricsOutboundTraffic + "/") collectEgress := trafficCollector(p2p.MetricsOutboundTraffic + "/")
for { for {
select { select {
@ -271,11 +448,11 @@ func (db *Dashboard) collectPeerData() {
switch event.Type { switch event.Type {
case p2p.PeerConnected: case p2p.PeerConnected:
connected := now.Add(-event.Elapsed) connected := now.Add(-event.Elapsed)
pc.updateKnown(event.IP.String(), event.ID, &PeerSession{ diff.updateKnownPeer(event.IP.String(), event.ID, &PeerSession{
Connected: &connected, Connected: &connected,
}) })
case p2p.PeerDisconnected: case p2p.PeerDisconnected:
pc.updateKnown(event.IP.String(), event.ID, &PeerSession{ diff.updateKnownPeer(event.IP.String(), event.ID, &PeerSession{
Disconnected: &now, Disconnected: &now,
Ingress: ChartEntries{ Ingress: ChartEntries{
&ChartEntry{ &ChartEntry{
@ -291,7 +468,7 @@ func (db *Dashboard) collectPeerData() {
}, },
}) })
case p2p.PeerHandshakeFailed: case p2p.PeerHandshakeFailed:
pc.updateUnknown(event.IP.String(), &UnknownPeer{ diff.updateUnknownPeer(event.IP.String(), &UnknownPeer{
Connected: now.Add(-event.Elapsed), Connected: now.Add(-event.Elapsed),
Disconnected: now, Disconnected: now,
}) })
@ -299,16 +476,68 @@ func (db *Dashboard) collectPeerData() {
log.Error("Unknown metered peer event type", "type", event.Type) log.Error("Unknown metered peer event type", "type", event.Type)
} }
case <-ticker.C: case <-ticker.C:
traffic := peerTraffic{ ingress, egress := make(map[string]float64), make(map[string]float64)
ingress: make(map[string]int64), db.peerLock.Lock()
egress: make(map[string]int64), for e := db.history.Peers.knownPeers.Front(); e != nil; e = e.Next() {
if p, ok := e.Value.(*KnownPeer); ok {
key := fmt.Sprintf("%s/%s", p.bundle.ip, p.id)
ingress[key] = 0
egress[key] = 0
} }
p2p.PeerIngressRegistry.Each(updateIngress(&traffic.ingress)) }
p2p.PeerEgressRegistry.Each(updateEgress(&traffic.egress)) p2p.PeerIngressRegistry.Each(collectIngress(&ingress))
pc.updateTraffic(&traffic) p2p.PeerEgressRegistry.Each(collectEgress(&egress))
s, _ := json.MarshalIndent(pc, "", " ") //diff.updateTraffic(&ingress, &egress)
for key := range ingress {
if k := strings.Split(key, "/"); len(k) == 2 {
diff.updateTraffic(k[0], k[1], ingress[key], egress[key])
} else {
log.Warn("Bad key", "key", key)
}
}
for key := range egress {
if _, ok := ingress[key]; !ok {
if k := strings.Split(key, "/"); len(k) == 2 {
diff.updateTraffic(k[0], k[1], ingress[key], egress[key])
} else {
log.Warn("Bad key", "key", key)
}
}
}
for _, bundle := range diff.Bundles {
for _, peer := range bundle.KnownPeers {
var ipExists, idExists bool
var b *PeerBundle
if b, ipExists = db.history.Peers.Bundles[bundle.ip]; ipExists {
_, idExists = b.KnownPeers[peer.id]
}
if !idExists || !ipExists {
fmt.Println("doesn't exist ", bundle.ip, peer.id)
//t, n := time.Now(), sampleLimit-peer.len()
//if len(peer.Sessions) > 0 && peer.Sessions[0].Connected != nil {
// t = *peer.Sessions[0].Connected
//}
//peer.Sessions = append([]*PeerSession{{
// Ingress: emptyChartEntries(t, n, db.config.Refresh),
// Egress: emptyChartEntries(t, n, db.config.Refresh),
//}}, peer.Sessions...)
} else {
fmt.Println("does exist ", bundle.ip, peer.id)
s, _ := json.MarshalIndent(bundle, "", " ")
fmt.Println(string(s)) fmt.Println(string(s))
//db.sendToAll(&Message{Network: pc}) fmt.Println(ingress[fmt.Sprintf("%s/%s", bundle.ip, peer.id)])
fmt.Println(egress[fmt.Sprintf("%s/%s", bundle.ip, peer.id)])
}
}
}
diff.RemovedKnown, diff.RemovedUnknown = db.history.Peers.append(diff)
//sh, _ := json.MarshalIndent(db.history.Network, "", " ")
//fmt.Println(string(sh))
db.peerLock.Unlock()
//s, _ := json.MarshalIndent(diff, "", " ")
//fmt.Println(string(s))
//db.sendToAll(&Message{Peers: diff})
diff.clear()
case err := <-subPeer.Err(): case err := <-subPeer.Err():
log.Warn("Peer subscription error", "err", err) log.Warn("Peer subscription error", "err", err)
return return