podman

1
package client
2

3
import (
4
	"fmt"
5
	"net/http"
6

7
	"github.com/go-openapi/strfmt"
8
	"github.com/opentracing/opentracing-go"
9
	"github.com/opentracing/opentracing-go/ext"
10
	"github.com/opentracing/opentracing-go/log"
11

12
	"github.com/go-openapi/runtime"
13
)
14

15
type tracingTransport struct {
16
	transport runtime.ClientTransport
17
	host      string
18
	opts      []opentracing.StartSpanOption
19
}
20

21
func newOpenTracingTransport(transport runtime.ClientTransport, host string, opts []opentracing.StartSpanOption,
22
) runtime.ClientTransport {
23
	return &tracingTransport{
24
		transport: transport,
25
		host:      host,
26
		opts:      opts,
27
	}
28
}
29

30
func (t *tracingTransport) Submit(op *runtime.ClientOperation) (interface{}, error) {
31
	if op.Context == nil {
32
		return t.transport.Submit(op)
33
	}
34

35
	params := op.Params
36
	reader := op.Reader
37

38
	var span opentracing.Span
39
	defer func() {
40
		if span != nil {
41
			span.Finish()
42
		}
43
	}()
44

45
	op.Params = runtime.ClientRequestWriterFunc(func(req runtime.ClientRequest, reg strfmt.Registry) error {
46
		span = createClientSpan(op, req.GetHeaderParams(), t.host, t.opts)
47
		return params.WriteToRequest(req, reg)
48
	})
49

50
	op.Reader = runtime.ClientResponseReaderFunc(func(response runtime.ClientResponse, consumer runtime.Consumer) (interface{}, error) {
51
		if span != nil {
52
			code := response.Code()
53
			ext.HTTPStatusCode.Set(span, uint16(code))
54
			if code >= 400 {
55
				ext.Error.Set(span, true)
56
			}
57
		}
58
		return reader.ReadResponse(response, consumer)
59
	})
60

61
	submit, err := t.transport.Submit(op)
62
	if err != nil && span != nil {
63
		ext.Error.Set(span, true)
64
		span.LogFields(log.Error(err))
65
	}
66
	return submit, err
67
}
68

69
func createClientSpan(op *runtime.ClientOperation, header http.Header, host string,
70
	opts []opentracing.StartSpanOption) opentracing.Span {
71
	ctx := op.Context
72
	span := opentracing.SpanFromContext(ctx)
73

74
	if span != nil {
75
		opts = append(opts, ext.SpanKindRPCClient)
76
		span, _ = opentracing.StartSpanFromContextWithTracer(
77
			ctx, span.Tracer(), operationName(op), opts...)
78

79
		ext.Component.Set(span, "go-openapi")
80
		ext.PeerHostname.Set(span, host)
81
		span.SetTag("http.path", op.PathPattern)
82
		ext.HTTPMethod.Set(span, op.Method)
83

84
		_ = span.Tracer().Inject(
85
			span.Context(),
86
			opentracing.HTTPHeaders,
87
			opentracing.HTTPHeadersCarrier(header))
88

89
		return span
90
	}
91
	return nil
92
}
93

94
func operationName(op *runtime.ClientOperation) string {
95
	if op.ID != "" {
96
		return op.ID
97
	}
98
	return fmt.Sprintf("%s_%s", op.Method, op.PathPattern)
99
}
100