80 lines
2.5 KiB
Go
80 lines
2.5 KiB
Go
|
package influxdb
|
||
|
|
|
||
|
|
import (
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/FreifunkBremen/yanic/runtime"
|
||
|
|
|
||
|
|
"github.com/bdlm/log"
|
||
|
|
influxdb "github.com/influxdata/influxdb-client-go/v2"
|
||
|
|
)
|
||
|
|
|
||
|
|
// InsertGlobals implementation of database
|
||
|
|
func (conn *Connection) InsertGlobals(stats *runtime.GlobalStats, time time.Time, site string, domain string) {
|
||
|
|
measurementGlobal := MeasurementGlobal
|
||
|
|
counterMeasurementModel := CounterMeasurementModel
|
||
|
|
counterMeasurementFirmware := CounterMeasurementFirmware
|
||
|
|
counterMeasurementAutoupdater := CounterMeasurementAutoupdater
|
||
|
|
|
||
|
|
if site != runtime.GLOBAL_SITE {
|
||
|
|
measurementGlobal += "_site"
|
||
|
|
counterMeasurementModel += "_site"
|
||
|
|
counterMeasurementFirmware += "_site"
|
||
|
|
counterMeasurementAutoupdater += "_site"
|
||
|
|
}
|
||
|
|
if domain != runtime.GLOBAL_DOMAIN {
|
||
|
|
measurementGlobal += "_domain"
|
||
|
|
counterMeasurementModel += "_domain"
|
||
|
|
counterMeasurementFirmware += "_domain"
|
||
|
|
counterMeasurementAutoupdater += "_domain"
|
||
|
|
}
|
||
|
|
p := influxdb.NewPoint(measurementGlobal,
|
||
|
|
conn.config.Tags(),
|
||
|
|
map[string]interface{}{
|
||
|
|
"nodes": stats.Nodes,
|
||
|
|
"gateways": stats.Gateways,
|
||
|
|
"clients.total": stats.Clients,
|
||
|
|
"clients.wifi": stats.ClientsWifi,
|
||
|
|
"clients.wifi24": stats.ClientsWifi24,
|
||
|
|
"clients.wifi5": stats.ClientsWifi5,
|
||
|
|
"clients.owe": stats.ClientsOWE,
|
||
|
|
"clients.owe24": stats.ClientsOWE24,
|
||
|
|
"clients.owe5": stats.ClientsOWE5,
|
||
|
|
},
|
||
|
|
time)
|
||
|
|
|
||
|
|
if site != runtime.GLOBAL_SITE {
|
||
|
|
p = p.AddTag("site", site)
|
||
|
|
}
|
||
|
|
if domain != runtime.GLOBAL_DOMAIN {
|
||
|
|
p = p.AddTag("domain", domain)
|
||
|
|
}
|
||
|
|
conn.writeAPI[MeasurementGlobal].WritePoint(p)
|
||
|
|
|
||
|
|
conn.addCounterMap(CounterMeasurementModel, counterMeasurementModel, stats.Models, time, site, domain)
|
||
|
|
conn.addCounterMap(CounterMeasurementFirmware, counterMeasurementFirmware, stats.Firmwares, time, site, domain)
|
||
|
|
conn.addCounterMap(CounterMeasurementAutoupdater, counterMeasurementAutoupdater, stats.Autoupdater, time, site, domain)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Saves the values of a CounterMap in the database.
|
||
|
|
// The key are used as 'value' tag.
|
||
|
|
// The value is used as 'counter' field.
|
||
|
|
func (conn *Connection) addCounterMap(writeName, name string, m runtime.CounterMap, t time.Time, site string, domain string) {
|
||
|
|
writeAPI, ok := conn.writeAPI[writeName]
|
||
|
|
if !ok {
|
||
|
|
log.WithField("writeName", writeName).Panic("no writeAPI found for countermap")
|
||
|
|
}
|
||
|
|
for key, count := range m {
|
||
|
|
p := influxdb.NewPoint("stat",
|
||
|
|
conn.config.Tags(),
|
||
|
|
map[string]interface{}{
|
||
|
|
"count": count,
|
||
|
|
},
|
||
|
|
t).
|
||
|
|
AddTag("site", site).
|
||
|
|
AddTag("domain", domain).
|
||
|
|
AddTag("value", key)
|
||
|
|
writeAPI.WritePoint(p)
|
||
|
|
}
|
||
|
|
}
|