73 lines
2.6 KiB
Go
73 lines
2.6 KiB
Go
// Copyright The OpenTelemetry Authors
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package restful
|
|
|
|
import (
|
|
"github.com/emicklei/go-restful/v3"
|
|
|
|
otelglobal "go.opentelemetry.io/otel/api/global"
|
|
otelpropagation "go.opentelemetry.io/otel/api/propagation"
|
|
"go.opentelemetry.io/otel/api/standard"
|
|
oteltrace "go.opentelemetry.io/otel/api/trace"
|
|
)
|
|
|
|
const (
|
|
tracerName = "go.opentelemetry.io/contrib/instrumentation/emicklei/go-restful"
|
|
tracerVersion = "1.0"
|
|
)
|
|
|
|
// OTelFilter returns a restful.FilterFunction which will trace an incoming request.
|
|
//
|
|
// The service parameter should describe the name of the (virtual) server handling
|
|
// the request. Options can be applied to configure the tracer and propagators
|
|
// used for this filter.
|
|
func OTelFilter(service string, opts ...Option) restful.FilterFunction {
|
|
cfg := Config{}
|
|
for _, opt := range opts {
|
|
opt(&cfg)
|
|
}
|
|
if cfg.Tracer == nil {
|
|
cfg.Tracer = otelglobal.TraceProvider().Tracer(tracerName, oteltrace.WithInstrumentationVersion(tracerVersion))
|
|
}
|
|
if cfg.Propagators == nil {
|
|
cfg.Propagators = otelglobal.Propagators()
|
|
}
|
|
return func(req *restful.Request, resp *restful.Response, chain *restful.FilterChain) {
|
|
r := req.Request
|
|
ctx := otelpropagation.ExtractHTTP(r.Context(), cfg.Propagators, r.Header)
|
|
route := req.SelectedRoutePath()
|
|
spanName := route
|
|
|
|
opts := []oteltrace.StartOption{
|
|
oteltrace.WithAttributes(standard.NetAttributesFromHTTPRequest("tcp", r)...),
|
|
oteltrace.WithAttributes(standard.EndUserAttributesFromHTTPRequest(r)...),
|
|
oteltrace.WithAttributes(standard.HTTPServerAttributesFromHTTPRequest(service, route, r)...),
|
|
oteltrace.WithSpanKind(oteltrace.SpanKindServer),
|
|
}
|
|
ctx, span := cfg.Tracer.Start(ctx, spanName, opts...)
|
|
defer span.End()
|
|
|
|
// pass the span through the request context
|
|
req.Request = req.Request.WithContext(ctx)
|
|
|
|
chain.ProcessFilter(req, resp)
|
|
|
|
attrs := standard.HTTPAttributesFromHTTPStatusCode(resp.StatusCode())
|
|
spanStatus, spanMessage := standard.SpanStatusFromHTTPStatusCode(resp.StatusCode())
|
|
span.SetAttributes(attrs...)
|
|
span.SetStatus(spanStatus, spanMessage)
|
|
}
|
|
}
|