new func getServerCount

This commit is contained in:
Matthew R Kasun 2023-03-08 13:23:25 -05:00
parent f651e65eff
commit ecba6e4622

View file

@ -5,6 +5,7 @@ import (
"time" "time"
"github.com/gravitl/netmaker/database" "github.com/gravitl/netmaker/database"
"github.com/gravitl/netmaker/logger"
"github.com/gravitl/netmaker/models" "github.com/gravitl/netmaker/models"
"github.com/gravitl/netmaker/servercfg" "github.com/gravitl/netmaker/servercfg"
"github.com/posthog/posthog-go" "github.com/posthog/posthog-go"
@ -84,7 +85,7 @@ func fetchTelemetryData() (telemetryData, error) {
data.Users = getDBLength(database.USERS_TABLE_NAME) data.Users = getDBLength(database.USERS_TABLE_NAME)
data.Networks = getDBLength(database.NETWORKS_TABLE_NAME) data.Networks = getDBLength(database.NETWORKS_TABLE_NAME)
data.Version = servercfg.GetVersion() data.Version = servercfg.GetVersion()
//data.Servers = GetServerCount() data.Servers = getServerCount()
nodes, err := GetAllNodes() nodes, err := GetAllNodes()
if err == nil { if err == nil {
data.Nodes = len(nodes) data.Nodes = len(nodes)
@ -93,6 +94,15 @@ func fetchTelemetryData() (telemetryData, error) {
return data, err return data, err
} }
// getServerCount returns number of servers from database
func getServerCount() int {
data, err := database.FetchRecords(database.SERVER_UUID_TABLE_NAME)
if err != nil {
logger.Log(0, "errror retrieving server data", err.Error())
}
return len(data)
}
// setTelemetryTimestamp - Give the entry in the DB a new timestamp // setTelemetryTimestamp - Give the entry in the DB a new timestamp
func setTelemetryTimestamp(telRecord *models.Telemetry) error { func setTelemetryTimestamp(telRecord *models.Telemetry) error {
lastsend := time.Now().Unix() lastsend := time.Now().Unix()