mirror of
https://github.com/grafana/grafana.git
synced 2024-12-02 05:29:42 -06:00
131 lines
3.4 KiB
Go
131 lines
3.4 KiB
Go
package metrics
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/grafana/grafana/pkg/bus"
|
|
"github.com/grafana/grafana/pkg/log"
|
|
m "github.com/grafana/grafana/pkg/models"
|
|
"github.com/grafana/grafana/pkg/plugins"
|
|
"github.com/grafana/grafana/pkg/setting"
|
|
)
|
|
|
|
var metricsLogger log.Logger = log.New("metrics")
|
|
var metricPublishCounter int64 = 0
|
|
|
|
func Init() {
|
|
settings := readSettings()
|
|
initMetricVars(settings)
|
|
go instrumentationLoop(settings)
|
|
}
|
|
|
|
func instrumentationLoop(settings *MetricSettings) chan struct{} {
|
|
M_Instance_Start.Inc(1)
|
|
|
|
onceEveryDayTick := time.NewTicker(time.Hour * 24)
|
|
secondTicker := time.NewTicker(time.Second * time.Duration(settings.IntervalSeconds))
|
|
|
|
for {
|
|
select {
|
|
case <-onceEveryDayTick.C:
|
|
sendUsageStats()
|
|
case <-secondTicker.C:
|
|
if settings.Enabled {
|
|
sendMetrics(settings)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func sendMetrics(settings *MetricSettings) {
|
|
if len(settings.Publishers) == 0 {
|
|
return
|
|
}
|
|
|
|
updateTotalStats()
|
|
|
|
metrics := MetricStats.GetSnapshots()
|
|
for _, publisher := range settings.Publishers {
|
|
publisher.Publish(metrics)
|
|
}
|
|
}
|
|
|
|
func updateTotalStats() {
|
|
|
|
// every interval also publish totals
|
|
metricPublishCounter++
|
|
if metricPublishCounter%10 == 0 {
|
|
// get stats
|
|
statsQuery := m.GetSystemStatsQuery{}
|
|
if err := bus.Dispatch(&statsQuery); err != nil {
|
|
metricsLogger.Error("Failed to get system stats", "error", err)
|
|
return
|
|
}
|
|
|
|
M_StatTotal_Dashboards.Update(statsQuery.Result.DashboardCount)
|
|
M_StatTotal_Users.Update(statsQuery.Result.UserCount)
|
|
M_StatTotal_Playlists.Update(statsQuery.Result.PlaylistCount)
|
|
M_StatTotal_Orgs.Update(statsQuery.Result.OrgCount)
|
|
}
|
|
}
|
|
|
|
func sendUsageStats() {
|
|
if !setting.ReportingEnabled {
|
|
return
|
|
}
|
|
|
|
metricsLogger.Debug("Sending anonymous usage stats to stats.grafana.org")
|
|
|
|
version := strings.Replace(setting.BuildVersion, ".", "_", -1)
|
|
|
|
metrics := map[string]interface{}{}
|
|
report := map[string]interface{}{
|
|
"version": version,
|
|
"metrics": metrics,
|
|
}
|
|
|
|
statsQuery := m.GetSystemStatsQuery{}
|
|
if err := bus.Dispatch(&statsQuery); err != nil {
|
|
metricsLogger.Error("Failed to get system stats", "error", err)
|
|
return
|
|
}
|
|
|
|
metrics["stats.dashboards.count"] = statsQuery.Result.DashboardCount
|
|
metrics["stats.users.count"] = statsQuery.Result.UserCount
|
|
metrics["stats.orgs.count"] = statsQuery.Result.OrgCount
|
|
metrics["stats.playlist.count"] = statsQuery.Result.PlaylistCount
|
|
metrics["stats.plugins.apps.count"] = len(plugins.Apps)
|
|
metrics["stats.plugins.panels.count"] = len(plugins.Panels)
|
|
metrics["stats.plugins.datasources.count"] = len(plugins.DataSources)
|
|
metrics["stats.alerts.count"] = statsQuery.Result.AlertCount
|
|
|
|
dsStats := m.GetDataSourceStatsQuery{}
|
|
if err := bus.Dispatch(&dsStats); err != nil {
|
|
metricsLogger.Error("Failed to get datasource stats", "error", err)
|
|
return
|
|
}
|
|
|
|
// send counters for each data source
|
|
// but ignore any custom data sources
|
|
// as sending that name could be sensitive information
|
|
dsOtherCount := 0
|
|
for _, dsStat := range dsStats.Result {
|
|
if m.IsKnownDataSourcePlugin(dsStat.Type) {
|
|
metrics["stats.ds."+dsStat.Type+".count"] = dsStat.Count
|
|
} else {
|
|
dsOtherCount += dsStat.Count
|
|
}
|
|
}
|
|
metrics["stats.ds.other.count"] = dsOtherCount
|
|
|
|
out, _ := json.MarshalIndent(report, "", " ")
|
|
data := bytes.NewBuffer(out)
|
|
|
|
client := http.Client{Timeout: time.Duration(5 * time.Second)}
|
|
go client.Post("https://stats.grafana.org/grafana-usage-report", "application/json", data)
|
|
}
|