diff --git a/logic/telemetry.go b/logic/telemetry.go index 5ecbe4cd..4b9e8635 100644 --- a/logic/telemetry.go +++ b/logic/telemetry.go @@ -5,6 +5,7 @@ import ( "time" "github.com/gravitl/netmaker/database" + "github.com/gravitl/netmaker/logger" "github.com/gravitl/netmaker/models" "github.com/gravitl/netmaker/servercfg" "github.com/posthog/posthog-go" @@ -84,7 +85,7 @@ func fetchTelemetryData() (telemetryData, error) { data.Users = getDBLength(database.USERS_TABLE_NAME) data.Networks = getDBLength(database.NETWORKS_TABLE_NAME) data.Version = servercfg.GetVersion() - //data.Servers = GetServerCount() + data.Servers = getServerCount() nodes, err := GetAllNodes() if err == nil { data.Nodes = len(nodes) @@ -93,6 +94,15 @@ func fetchTelemetryData() (telemetryData, error) { 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 func setTelemetryTimestamp(telRecord *models.Telemetry) error { lastsend := time.Now().Unix()