@@ -2,22 +2,30 @@ package observability
22
33import (
44 "context"
5+ "fmt"
56 "os"
67 "strings"
78
89 "github.com/MrAlias/flow"
910 "github.com/pkg/errors"
11+ "github.com/prometheus/client_golang/prometheus"
1012 "go.opentelemetry.io/otel"
1113 "go.opentelemetry.io/otel/exporters/otlp/otlptrace"
1214 "go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracehttp"
15+ otelProm "go.opentelemetry.io/otel/exporters/prometheus"
16+ otelMetric "go.opentelemetry.io/otel/metric"
1317 "go.opentelemetry.io/otel/propagation"
18+ "go.opentelemetry.io/otel/sdk/metric"
1419 "go.opentelemetry.io/otel/sdk/resource"
1520 sdkTrace "go.opentelemetry.io/otel/sdk/trace"
1621 semconv "go.opentelemetry.io/otel/semconv/v1.4.0"
1722 "go.opentelemetry.io/otel/trace"
1823)
1924
25+ var Meter otelMetric.Meter
26+
2027func InitTracer (serviceName , serviceVersion string ) (* sdkTrace.TracerProvider , trace.Tracer , error ) {
28+ ctx := context .Background ()
2129
2230 otlpEndpoint , ok := os .LookupEnv ("OTLP_ENDPOINT" )
2331 otlpInsecure := os .Getenv ("OTLP_INSECURE" )
@@ -37,7 +45,7 @@ func InitTracer(serviceName, serviceVersion string) (*sdkTrace.TracerProvider, t
3745
3846 client := otlptracehttp .NewClient (otlpOptions ... )
3947
40- otlptracehttpExporter , err := otlptrace .New (context . Background () , client )
48+ otlptracehttpExporter , err := otlptrace .New (ctx , client )
4149 if err != nil {
4250 return nil , nil , errors .Wrap (err , "failed creating OTLP trace exporter" )
4351 }
@@ -48,7 +56,7 @@ func InitTracer(serviceName, serviceVersion string) (*sdkTrace.TracerProvider, t
4856 }
4957
5058 resources , err := resource .New (
51- context . Background () ,
59+ ctx ,
5260 resource .WithFromEnv (), // pull attributes from OTEL_RESOURCE_ATTRIBUTES and OTEL_SERVICE_NAME environment variables
5361 resource .WithOS (), // This option configures a set of Detectors that discover OS information
5462 resource .WithContainer (), // This option configures a set of Detectors that discover container information
@@ -80,3 +88,68 @@ func InitTracer(serviceName, serviceVersion string) (*sdkTrace.TracerProvider, t
8088
8189 return traceProvider , trace , nil
8290}
91+
92+ func InitMetrics (serviceName , serviceVersion string ) (* metric.MeterProvider , error ) {
93+ ctx := context .Background ()
94+
95+ otlpEndpoint , ok := os .LookupEnv ("OTLP_ENDPOINT" )
96+ otlpInsecure := os .Getenv ("OTLP_INSECURE" )
97+
98+ otlpOptions := make ([]otlptracehttp.Option , 0 )
99+
100+ if ok {
101+ otlpOptions = append (otlpOptions , otlptracehttp .WithEndpoint (otlpEndpoint ))
102+
103+ if strings .ToLower (otlpInsecure ) == "true" {
104+ otlpOptions = append (otlpOptions , otlptracehttp .WithInsecure ())
105+ }
106+ } else {
107+ otlpOptions = append (otlpOptions , otlptracehttp .WithEndpoint ("localhost:4318" ))
108+ otlpOptions = append (otlpOptions , otlptracehttp .WithInsecure ())
109+ }
110+
111+ registry := prometheus .NewRegistry ()
112+ exporter , err := otelProm .New (
113+ otelProm .WithoutUnits (),
114+ otelProm .WithRegisterer (registry ),
115+ )
116+
117+ if err != nil {
118+ return nil , err
119+ }
120+
121+ hostname , err := os .Hostname ()
122+ if err != nil {
123+ return nil , err
124+ }
125+
126+ resources , err := resource .New (
127+ ctx ,
128+ resource .WithFromEnv (), // pull attributes from OTEL_RESOURCE_ATTRIBUTES and OTEL_SERVICE_NAME environment variables
129+ resource .WithOS (), // This option configures a set of Detectors that discover OS information
130+ resource .WithContainer (), // This option configures a set of Detectors that discover container information
131+ resource .WithHost (), // This option configures a set of Detectors that discover host information
132+ resource .WithAttributes (
133+ semconv .ServiceNameKey .String (serviceName ),
134+ semconv .ServiceVersionKey .String (serviceVersion ),
135+ semconv .ServiceInstanceIDKey .String (hostname ),
136+ ),
137+ )
138+ if err != nil {
139+ return nil , err
140+ }
141+
142+ resources , err = resource .Merge (resource .Default (), resources )
143+ if err != nil {
144+ return nil , err
145+ }
146+
147+ provider := metric .NewMeterProvider (
148+ metric .WithResource (resources ),
149+ metric .WithReader (exporter ),
150+ )
151+
152+ Meter = provider .Meter (fmt .Sprintf ("%sMeter" , serviceName ))
153+
154+ return provider , nil
155+ }
0 commit comments