mirror of
https://github.com/kubeshark/kubeshark.git
synced 2025-10-22 07:51:30 +00:00
115 lines
3.0 KiB
Go
115 lines
3.0 KiB
Go
package telemetry
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"github.com/denisbrodbeck/machineid"
|
|
"github.com/up9inc/mizu/cli/config"
|
|
"github.com/up9inc/mizu/cli/kubernetes"
|
|
"github.com/up9inc/mizu/cli/logger"
|
|
"github.com/up9inc/mizu/cli/mizu"
|
|
"io/ioutil"
|
|
"net/http"
|
|
)
|
|
|
|
const telemetryUrl = "https://us-east4-up9-prod.cloudfunctions.net/mizu-telemetry"
|
|
|
|
func ReportRun(cmd string, args interface{}) {
|
|
if !shouldRunTelemetry() {
|
|
logger.Log.Debugf("not reporting telemetry")
|
|
return
|
|
}
|
|
|
|
argsBytes, _ := json.Marshal(args)
|
|
argsMap := map[string]interface{}{
|
|
"cmd": cmd,
|
|
"args": string(argsBytes),
|
|
}
|
|
|
|
if err := sendTelemetry("Execution", argsMap); err != nil {
|
|
logger.Log.Debug(err)
|
|
return
|
|
}
|
|
|
|
logger.Log.Debugf("successfully reported telemetry for cmd %v", cmd)
|
|
}
|
|
|
|
func ReportAPICalls(mizuPort uint16) {
|
|
if !shouldRunTelemetry() {
|
|
logger.Log.Debugf("not reporting telemetry")
|
|
return
|
|
}
|
|
|
|
mizuProxiedUrl := kubernetes.GetMizuApiServerProxiedHostAndPath(mizuPort)
|
|
generalStatsUrl := fmt.Sprintf("http://%s/api/generalStats", mizuProxiedUrl)
|
|
|
|
response, requestErr := http.Get(generalStatsUrl)
|
|
if requestErr != nil {
|
|
logger.Log.Debugf("ERROR: failed to get general stats for telemetry, err: %v", requestErr)
|
|
return
|
|
} else if response.StatusCode != 200 {
|
|
logger.Log.Debugf("ERROR: failed to get general stats for telemetry, status code: %v", response.StatusCode)
|
|
return
|
|
}
|
|
|
|
defer func() { _ = response.Body.Close() }()
|
|
|
|
data, readErr := ioutil.ReadAll(response.Body)
|
|
if readErr != nil {
|
|
logger.Log.Debugf("ERROR: failed to read general stats for telemetry, err: %v", readErr)
|
|
return
|
|
}
|
|
|
|
var generalStats map[string]interface{}
|
|
if parseErr := json.Unmarshal(data, &generalStats); parseErr != nil {
|
|
logger.Log.Debugf("ERROR: failed to parse general stats for telemetry, err: %v", parseErr)
|
|
return
|
|
}
|
|
|
|
argsMap := map[string]interface{}{
|
|
"apiCallsCount": generalStats["EntriesCount"],
|
|
"firstAPICallTimestamp": generalStats["FirstEntryTimestamp"],
|
|
"lastAPICallTimestamp": generalStats["LastEntryTimestamp"],
|
|
}
|
|
|
|
if err := sendTelemetry("APICalls", argsMap); err != nil {
|
|
logger.Log.Debug(err)
|
|
return
|
|
}
|
|
|
|
logger.Log.Debugf("successfully reported telemetry of api calls")
|
|
}
|
|
|
|
func shouldRunTelemetry() bool {
|
|
if !config.Config.Telemetry {
|
|
return false
|
|
}
|
|
|
|
if mizu.Branch != "main" && mizu.Branch != "develop" {
|
|
return false
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
func sendTelemetry(telemetryType string, argsMap map[string]interface{}) error {
|
|
argsMap["telemetryType"] = telemetryType
|
|
argsMap["component"] = "mizu_cli"
|
|
argsMap["buildTimestamp"] = mizu.BuildTimestamp
|
|
argsMap["branch"] = mizu.Branch
|
|
argsMap["version"] = mizu.SemVer
|
|
|
|
if machineId, err := machineid.ProtectedID("mizu"); err == nil {
|
|
argsMap["machineId"] = machineId
|
|
}
|
|
|
|
jsonValue, _ := json.Marshal(argsMap)
|
|
|
|
if resp, err := http.Post(telemetryUrl, "application/json", bytes.NewBuffer(jsonValue)); err != nil {
|
|
return fmt.Errorf("ERROR: failed sending telemetry, err: %v, response %v", err, resp)
|
|
}
|
|
|
|
return nil
|
|
}
|