59 lines
1.9 KiB
Go
59 lines
1.9 KiB
Go
|
package exporters
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"git.ma-al.com/gora_filip/pkg/exporters/console_exporter"
|
||
|
|
gelf_exporter "git.ma-al.com/gora_filip/pkg/exporters/gelf_exporter"
|
||
|
|
otlphttp_exporter "go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracehttp"
|
||
|
|
sdktrace "go.opentelemetry.io/otel/sdk/trace"
|
||
|
|
)
|
||
|
|
|
||
|
|
func NewWithConfig(exporter sdktrace.SpanExporter) ExporterWithConfig {
|
||
|
|
return ExporterWithConfig{
|
||
|
|
exporter: exporter,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Combined exporter with batch processor config
|
||
|
|
type ExporterWithConfig struct {
|
||
|
|
exporter sdktrace.SpanExporter
|
||
|
|
config []sdktrace.BatchSpanProcessorOption
|
||
|
|
}
|
||
|
|
|
||
|
|
func (ecfg ExporterWithConfig) Add(opt sdktrace.BatchSpanProcessorOption) ExporterWithConfig {
|
||
|
|
ecfg.config = append(ecfg.config, opt)
|
||
|
|
return ecfg
|
||
|
|
}
|
||
|
|
|
||
|
|
func (ecfg ExporterWithConfig) IntoTraceProviderOption() sdktrace.TracerProviderOption {
|
||
|
|
return sdktrace.WithBatcher(ecfg.exporter, ecfg.config...)
|
||
|
|
}
|
||
|
|
|
||
|
|
// An exporter printing to console with very small delay
|
||
|
|
func DevConsoleExporter(opts ...console_exporter.ExporterOptions) ExporterWithConfig {
|
||
|
|
batchTimeout := (time.Millisecond * 250)
|
||
|
|
exportTimeout := (time.Millisecond * 250)
|
||
|
|
var exporter ExporterWithConfig
|
||
|
|
if len(opts) > 0 {
|
||
|
|
exporter = NewWithConfig(console_exporter.NewExporter(opts[0]))
|
||
|
|
} else {
|
||
|
|
exporter = NewWithConfig(console_exporter.DefaultConsoleExporter())
|
||
|
|
}
|
||
|
|
return exporter.Add(sdktrace.WithBatchTimeout(batchTimeout)).Add(sdktrace.WithExportTimeout(exportTimeout))
|
||
|
|
}
|
||
|
|
|
||
|
|
// Default exporter to Graylog.
|
||
|
|
func GelfExporter(opts ...gelf_exporter.Option) (ExporterWithConfig, error) {
|
||
|
|
gelfExp, err := gelf_exporter.New(opts...)
|
||
|
|
return NewWithConfig(gelfExp), err
|
||
|
|
}
|
||
|
|
|
||
|
|
// Exporter for traces over HTTP. Can be used with Jaeger.
|
||
|
|
// See documentation of [otlhttp_exporter] for details.
|
||
|
|
func OtlpHTTPExporter(opts ...otlphttp_exporter.Option) (ExporterWithConfig, error) {
|
||
|
|
otlpExp, err := otlphttp_exporter.New(context.Background(), opts...)
|
||
|
|
return NewWithConfig(otlpExp), err
|
||
|
|
}
|