mirror of
https://github.com/rocky-linux/peridot.git
synced 2024-12-19 09:28:29 +00:00
100 lines
2.4 KiB
Go
100 lines
2.4 KiB
Go
package client
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/go-openapi/strfmt"
|
|
"github.com/opentracing/opentracing-go"
|
|
"github.com/opentracing/opentracing-go/ext"
|
|
"github.com/opentracing/opentracing-go/log"
|
|
|
|
"github.com/go-openapi/runtime"
|
|
)
|
|
|
|
type tracingTransport struct {
|
|
transport runtime.ClientTransport
|
|
host string
|
|
opts []opentracing.StartSpanOption
|
|
}
|
|
|
|
func newOpenTracingTransport(transport runtime.ClientTransport, host string, opts []opentracing.StartSpanOption,
|
|
) runtime.ClientTransport {
|
|
return &tracingTransport{
|
|
transport: transport,
|
|
host: host,
|
|
opts: opts,
|
|
}
|
|
}
|
|
|
|
func (t *tracingTransport) Submit(op *runtime.ClientOperation) (interface{}, error) {
|
|
if op.Context == nil {
|
|
return t.transport.Submit(op)
|
|
}
|
|
|
|
params := op.Params
|
|
reader := op.Reader
|
|
|
|
var span opentracing.Span
|
|
defer func() {
|
|
if span != nil {
|
|
span.Finish()
|
|
}
|
|
}()
|
|
|
|
op.Params = runtime.ClientRequestWriterFunc(func(req runtime.ClientRequest, reg strfmt.Registry) error {
|
|
span = createClientSpan(op, req.GetHeaderParams(), t.host, t.opts)
|
|
return params.WriteToRequest(req, reg)
|
|
})
|
|
|
|
op.Reader = runtime.ClientResponseReaderFunc(func(response runtime.ClientResponse, consumer runtime.Consumer) (interface{}, error) {
|
|
if span != nil {
|
|
code := response.Code()
|
|
ext.HTTPStatusCode.Set(span, uint16(code))
|
|
if code >= 400 {
|
|
ext.Error.Set(span, true)
|
|
}
|
|
}
|
|
return reader.ReadResponse(response, consumer)
|
|
})
|
|
|
|
submit, err := t.transport.Submit(op)
|
|
if err != nil && span != nil {
|
|
ext.Error.Set(span, true)
|
|
span.LogFields(log.Error(err))
|
|
}
|
|
return submit, err
|
|
}
|
|
|
|
func createClientSpan(op *runtime.ClientOperation, header http.Header, host string,
|
|
opts []opentracing.StartSpanOption) opentracing.Span {
|
|
ctx := op.Context
|
|
span := opentracing.SpanFromContext(ctx)
|
|
|
|
if span != nil {
|
|
opts = append(opts, ext.SpanKindRPCClient)
|
|
span, _ = opentracing.StartSpanFromContextWithTracer(
|
|
ctx, span.Tracer(), operationName(op), opts...)
|
|
|
|
ext.Component.Set(span, "go-openapi")
|
|
ext.PeerHostname.Set(span, host)
|
|
span.SetTag("http.path", op.PathPattern)
|
|
ext.HTTPMethod.Set(span, op.Method)
|
|
|
|
_ = span.Tracer().Inject(
|
|
span.Context(),
|
|
opentracing.HTTPHeaders,
|
|
opentracing.HTTPHeadersCarrier(header))
|
|
|
|
return span
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func operationName(op *runtime.ClientOperation) string {
|
|
if op.ID != "" {
|
|
return op.ID
|
|
}
|
|
return fmt.Sprintf("%s_%s", op.Method, op.PathPattern)
|
|
}
|