Skip to content

Commit

Permalink
Align telemetry related variables and keys with the new schema (#172)
Browse files Browse the repository at this point in the history
  • Loading branch information
omris94 committed May 1, 2023
1 parent 44730c4 commit a0fdb5c
Show file tree
Hide file tree
Showing 6 changed files with 42 additions and 42 deletions.
2 changes: 1 addition & 1 deletion src/operator/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -148,7 +148,7 @@ func main() {
} else {
kubeSystemUID = string(kubeSystemNs.UID)
}
telemetrysender.SetGlobalPlatformIdentifier(telemetrysender.Anonymize(kubeSystemUID))
telemetrysender.SetGlobalContextId(telemetrysender.Anonymize(kubeSystemUID))

kafkaServersStore := kafkaacls.NewServersStore(tlsSource, enforcementConfig.EnableKafkaACL, kafkaacls.NewKafkaIntentsAdmin, enforcementConfig.EnforcementEnabledGlobally)

Expand Down
20 changes: 10 additions & 10 deletions src/shared/telemetries/telemetriesgql/generated.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

4 changes: 2 additions & 2 deletions src/shared/telemetries/telemetriesgql/schema.graphql

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

30 changes: 15 additions & 15 deletions src/shared/telemetries/telemetrysender/config.go
Original file line number Diff line number Diff line change
Expand Up @@ -6,25 +6,25 @@ import (
)

const (
TelemetriesAPIAddressKey = "telemetries-address"
TimeoutKey = "telemetry-client-timeout"
CloudClientTimeoutDefault = "30s"
TelemetriesEnabledKey = "telemetries-enabled"
TelemetriesEnabledDefault = true
TelemetriesMaxBatchSizeKey = "telemetries-max-batch-size"
TelemetriesMaxBatchSizeDefault = 100
TelemetriesIntervalKey = "telemetries-interval-seconds"
TelemetriesIntervalDefault = 5
TelemetriesAddressDefault = "https://app.otterize.com/api/telemetry/query"
EnvPrefix = "OTTERIZE"
TelemetryAPIAddressKey = "telemetry-address"
TimeoutKey = "telemetry-client-timeout"
CloudClientTimeoutDefault = "30s"
TelemetryEnabledKey = "telemetry-enabled"
TelemetryEnabledDefault = true
TelemetryMaxBatchSizeKey = "telemetry-max-batch-size"
TelemetryMaxBatchSizeDefault = 100
TelemetryIntervalKey = "telemetry-interval-seconds"
TelemetryIntervalDefault = 5
TelemetryAddressDefault = "https://app.otterize.com/api/telemetry/query"
EnvPrefix = "OTTERIZE"
)

func init() {
viper.SetDefault(TelemetriesAPIAddressKey, TelemetriesAddressDefault)
viper.SetDefault(TelemetryAPIAddressKey, TelemetryAddressDefault)
viper.SetDefault(TimeoutKey, CloudClientTimeoutDefault)
viper.SetDefault(TelemetriesIntervalKey, TelemetriesIntervalDefault)
viper.SetDefault(TelemetriesMaxBatchSizeKey, TelemetriesMaxBatchSizeDefault)
viper.SetDefault(TelemetriesEnabledKey, TelemetriesEnabledDefault)
viper.SetDefault(TelemetryIntervalKey, TelemetryIntervalDefault)
viper.SetDefault(TelemetryMaxBatchSizeKey, TelemetryMaxBatchSizeDefault)
viper.SetDefault(TelemetryEnabledKey, TelemetryEnabledDefault)
viper.SetEnvPrefix(EnvPrefix)
viper.SetEnvKeyReplacer(strings.NewReplacer("-", "_"))
viper.AutomaticEnv()
Expand Down
20 changes: 10 additions & 10 deletions src/shared/telemetries/telemetrysender/global_sender.go
Original file line number Diff line number Diff line change
Expand Up @@ -9,30 +9,30 @@ import (
)

var (
senderInitOnce = sync.Once{}
sender *TelemetrySender
globalPlatformIdentifier string
globalRuntimeIdentifier string
senderInitOnce = sync.Once{}
sender *TelemetrySender
globalContextId string
globalComponentInstanceId string
)

func SetGlobalPlatformIdentifier(platformIdentifier string) {
globalPlatformIdentifier = platformIdentifier
func SetGlobalContextId(contextId string) {
globalContextId = contextId
}

func send(componentType telemetriesgql.ComponentType, eventType telemetriesgql.EventType, count int) {
senderInitOnce.Do(func() {
sender = New()
globalRuntimeIdentifier = uuid.NewString()
globalComponentInstanceId = uuid.NewString()
if flag.Lookup("test.v") != nil {
logrus.Infof("Disabling telemetry sender because this is a test")
sender.enabled = false
}
})
if err := sender.Send(
telemetriesgql.Component{
ComponentType: componentType,
Identifier: globalRuntimeIdentifier,
PlatformIdentifier: globalPlatformIdentifier,
ComponentType: componentType,
ComponentInstanceId: globalComponentInstanceId,
ContextId: globalContextId,
}, eventType, count); err != nil {
logrus.Warningf("failed sending telemetry. %s", err)
}
Expand Down
8 changes: 4 additions & 4 deletions src/shared/telemetries/telemetrysender/sender.go
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ type TelemetrySender struct {
}

func newGqlClient() graphql.Client {
apiAddress := viper.GetString(TelemetriesAPIAddressKey)
apiAddress := viper.GetString(TelemetryAPIAddressKey)
clientTimeout := viper.GetDuration(TimeoutKey)
transport := &http.Transport{}
clientWithTimeout := &http.Client{Timeout: clientTimeout, Transport: transport}
Expand All @@ -34,9 +34,9 @@ func batchSendTelemetries(ctx context.Context, telemetriesClient graphql.Client,
}

func New() *TelemetrySender {
enabled := viper.GetBool(TelemetriesEnabledKey)
maxBatchSize := viper.GetInt(TelemetriesMaxBatchSizeKey)
interval := viper.GetInt(TelemetriesIntervalKey)
enabled := viper.GetBool(TelemetryEnabledKey)
maxBatchSize := viper.GetInt(TelemetryMaxBatchSizeKey)
interval := viper.GetInt(TelemetryIntervalKey)
telemetriesClient := newGqlClient()
sender := &TelemetrySender{telemetriesClient: telemetriesClient, enabled: enabled}
batchSendFunc := func(telemetries []telemetriesgql.TelemetryInput) error {
Expand Down

0 comments on commit a0fdb5c

Please sign in to comment.