/
span.go
94 lines (82 loc) · 2.33 KB
/
span.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
// Copyright 2021 ZUP IT SERVICOS EM TECNOLOGIA E INOVACAO SA
//
// 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 tracing
import (
"errors"
"fmt"
"net/http"
"runtime/debug"
"github.com/opentracing/opentracing-go"
"github.com/opentracing/opentracing-go/ext"
"github.com/opentracing/opentracing-go/log"
"github.com/uber/jaeger-client-go"
kerrors "k8s.io/apimachinery/pkg/api/errors"
)
type (
Span struct {
opentracing.Span
}
Info struct {
TraceID string
SpanID string
ParentID string
IsSampled bool
}
)
func (s *Span) String() string {
return fmt.Sprintf("%+v", s.Span)
}
func (s *Span) Info() *Info {
if sc, ok := s.Context().(jaeger.SpanContext); ok {
return &Info{
TraceID: sc.TraceID().String(),
SpanID: sc.SpanID().String(),
ParentID: sc.ParentID().String(),
IsSampled: sc.IsSampled(),
}
}
return nil
}
func (s *Span) SetError(err error) {
ext.Error.Set(s, true)
fields := make([]log.Field, 0)
fields = append(fields, log.String("event", "error"), log.String("message", err.Error()))
var serr *kerrors.StatusError
if errors.As(err, &serr) {
status := serr.Status()
fields = append(fields, log.Int32("code", status.Code), log.String("reason", string(status.Reason)))
}
s.LogFields(fields...)
}
func (s *Span) SetHTTPResponseStatus(status int) {
ext.HTTPStatusCode.Set(s, uint16(status))
if status >= 500 && status < 600 {
ext.Error.Set(s, true)
s.SetTag("error.type", fmt.Sprintf("%d: %s", status, http.StatusText(status)))
s.LogKV(
"event", "error",
"message", fmt.Sprintf("%d: %s", status, http.StatusText(status)),
)
}
}
func (s *Span) Panic(err interface{}) {
ext.HTTPStatusCode.Set(s, uint16(500))
ext.Error.Set(s, true)
s.SetTag("error.type", "panic")
s.LogKV("event", "error",
"error.kind", "panic",
"message", err,
"stack", string(debug.Stack()))
panic(err)
}