/
axiom.go
50 lines (42 loc) 路 1.34 KB
/
axiom.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
package config
import (
odigosv1 "github.com/keyval-dev/odigos/api/odigos/v1alpha1"
commonconf "github.com/keyval-dev/odigos/autoscaler/controllers/common"
"github.com/keyval-dev/odigos/common"
ctrl "sigs.k8s.io/controller-runtime"
)
const (
axiomDatasetKey = "AXIOM_DATASET"
)
type Axiom struct{}
func (a *Axiom) DestType() common.DestinationType {
return common.AxiomDestinationType
}
func (a *Axiom) ModifyConfig(dest *odigosv1.Destination, currentConfig *commonconf.Config) {
dataset, exists := dest.Spec.Data[axiomDatasetKey]
if !exists {
dataset = "default"
ctrl.Log.V(0).Info("Axiom dataset not specified, using default")
}
axiomExporterName := "otlphttp/axiom-" + dest.Name
currentConfig.Exporters[axiomExporterName] = commonconf.GenericMap{
"compression": "gzip",
"endpoint": "https://api.axiom.co",
"headers": commonconf.GenericMap{
"authorization": "Bearer ${AXIOM_API_TOKEN}",
"x-axiom-dataset": dataset,
},
}
if isTracingEnabled(dest) {
tracesPipelineName := "traces/axiom-" + dest.Name
currentConfig.Service.Pipelines[tracesPipelineName] = commonconf.Pipeline{
Exporters: []string{axiomExporterName},
}
}
if isLoggingEnabled(dest) {
logsPipelineName := "logs/axiom-" + dest.Name
currentConfig.Service.Pipelines[logsPipelineName] = commonconf.Pipeline{
Exporters: []string{axiomExporterName},
}
}
}