|
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758 |
- // Copyright 2016 Google LLC
- //
- // 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 trace is OBSOLETE. See https://cloud.google.com/trace/docs/setup/go.
- package trace // import "cloud.google.com/go/trace"
-
- import (
- "context"
- "crypto/rand"
- "encoding/binary"
- "encoding/json"
- "fmt"
- "log"
- "net/http"
- "runtime"
- "strconv"
- "strings"
- "sync"
- "sync/atomic"
- "time"
-
- api "google.golang.org/api/cloudtrace/v1"
- "google.golang.org/api/gensupport"
- "google.golang.org/api/option"
- "google.golang.org/api/support/bundler"
- htransport "google.golang.org/api/transport/http"
- )
-
- const (
- httpHeader = `X-Cloud-Trace-Context`
- userAgent = `gcloud-golang-trace/20160501`
- cloudPlatformScope = `https://www.googleapis.com/auth/cloud-platform`
- spanKindClient = `RPC_CLIENT`
- spanKindServer = `RPC_SERVER`
- spanKindUnspecified = `SPAN_KIND_UNSPECIFIED`
- maxStackFrames = 20
- labelAgent = `trace.cloud.google.com/agent`
- )
-
- // Stackdriver Trace API predefined labels.
- //
- // Deprecated: see https://cloud.google.com/trace/docs/setup/go.
- const (
- LabelComponent = `trace.cloud.google.com/component`
- LabelErrorMessage = `trace.cloud.google.com/error/message`
- LabelErrorName = `trace.cloud.google.com/error/name`
- LabelHTTPClientCity = `trace.cloud.google.com/http/client_city`
- LabelHTTPClientCountry = `trace.cloud.google.com/http/client_country`
- LabelHTTPClientProtocol = `trace.cloud.google.com/http/client_protocol`
- LabelHTTPClientRegion = `trace.cloud.google.com/http/client_region`
- LabelHTTPHost = `trace.cloud.google.com/http/host`
- LabelHTTPMethod = `trace.cloud.google.com/http/method`
- LabelHTTPRedirectedURL = `trace.cloud.google.com/http/redirected_url`
- LabelHTTPRequestSize = `trace.cloud.google.com/http/request/size`
- LabelHTTPResponseSize = `trace.cloud.google.com/http/response/size`
- LabelHTTPStatusCode = `trace.cloud.google.com/http/status_code`
- LabelHTTPURL = `trace.cloud.google.com/http/url`
- LabelHTTPUserAgent = `trace.cloud.google.com/http/user_agent`
- LabelPID = `trace.cloud.google.com/pid`
- LabelSamplingPolicy = `trace.cloud.google.com/sampling_policy`
- LabelSamplingWeight = `trace.cloud.google.com/sampling_weight`
- LabelStackTrace = `trace.cloud.google.com/stacktrace`
- LabelTID = `trace.cloud.google.com/tid`
- )
-
- const (
- // ScopeTraceAppend grants permissions to write trace data for a project.
- //
- // Deprecated: see https://cloud.google.com/trace/docs/setup/go.
- ScopeTraceAppend = "https://www.googleapis.com/auth/trace.append"
-
- // ScopeCloudPlatform grants permissions to view and manage your data
- // across Google Cloud Platform services.
- //
- // Deprecated: see https://cloud.google.com/trace/docs/setup/go.
- ScopeCloudPlatform = "https://www.googleapis.com/auth/cloud-platform"
- )
-
- type contextKey struct{}
-
- type stackLabelValue struct {
- Frames []stackFrame `json:"stack_frame"`
- }
-
- type stackFrame struct {
- Class string `json:"class_name,omitempty"`
- Method string `json:"method_name"`
- Filename string `json:"file_name"`
- Line int64 `json:"line_number"`
- }
-
- var (
- spanIDCounter uint64
- spanIDIncrement uint64
- )
-
- func init() {
- // Set spanIDCounter and spanIDIncrement to random values. nextSpanID will
- // return an arithmetic progression using these values, skipping zero. We set
- // the LSB of spanIDIncrement to 1, so that the cycle length is 2^64.
- binary.Read(rand.Reader, binary.LittleEndian, &spanIDCounter)
- binary.Read(rand.Reader, binary.LittleEndian, &spanIDIncrement)
- spanIDIncrement |= 1
- // Attach hook for autogenerated Google API calls. This will automatically
- // create trace spans for API calls if there is a trace in the context.
- gensupport.RegisterHook(requestHook)
- }
-
- func requestHook(ctx context.Context, req *http.Request) func(resp *http.Response) {
- span := FromContext(ctx)
- if span == nil || req == nil {
- return nil
- }
- span = span.NewRemoteChild(req)
- return func(resp *http.Response) {
- if resp != nil {
- span.Finish(WithResponse(resp))
- } else {
- span.Finish()
- }
- }
- }
-
- // nextSpanID returns a new span ID. It will never return zero.
- func nextSpanID() uint64 {
- var id uint64
- for id == 0 {
- id = atomic.AddUint64(&spanIDCounter, spanIDIncrement)
- }
- return id
- }
-
- // nextTraceID returns a new trace ID.
- func nextTraceID() string {
- id1 := nextSpanID()
- id2 := nextSpanID()
- return fmt.Sprintf("%016x%016x", id1, id2)
- }
-
- // Client is a client for uploading traces to the Google Stackdriver Trace service.
- // A nil Client will no-op for all of its methods.
- //
- // Deprecated: see https://cloud.google.com/trace/docs/setup/go.
- type Client struct {
- service *api.Service
- projectID string
- policy SamplingPolicy
- bundler *bundler.Bundler
- }
-
- // NewClient creates a new Google Stackdriver Trace client.
- //
- // Deprecated: see https://cloud.google.com/trace/docs/setup/go.
- func NewClient(ctx context.Context, projectID string, opts ...option.ClientOption) (*Client, error) {
- o := []option.ClientOption{
- option.WithScopes(cloudPlatformScope),
- option.WithUserAgent(userAgent),
- }
- o = append(o, opts...)
- hc, basePath, err := htransport.NewClient(ctx, o...)
- if err != nil {
- return nil, fmt.Errorf("creating HTTP client for Google Stackdriver Trace API: %v", err)
- }
- apiService, err := api.New(hc)
- if err != nil {
- return nil, fmt.Errorf("creating Google Stackdriver Trace API client: %v", err)
- }
- if basePath != "" {
- // An option set a basepath, so override api.New's default.
- apiService.BasePath = basePath
- }
- c := &Client{
- service: apiService,
- projectID: projectID,
- }
- bundler := bundler.NewBundler((*api.Trace)(nil), func(bundle interface{}) {
- traces := bundle.([]*api.Trace)
- err := c.upload(traces)
- if err != nil {
- log.Printf("failed to upload %d traces to the Cloud Trace server: %v", len(traces), err)
- }
- })
- bundler.DelayThreshold = 2 * time.Second
- bundler.BundleCountThreshold = 100
- // We're not measuring bytes here, we're counting traces and spans as one "byte" each.
- bundler.BundleByteThreshold = 1000
- bundler.BundleByteLimit = 1000
- bundler.BufferedByteLimit = 10000
- c.bundler = bundler
- return c, nil
- }
-
- // SetSamplingPolicy sets the SamplingPolicy that determines how often traces
- // are initiated by this client.
- //
- // Deprecated: see https://cloud.google.com/trace/docs/setup/go.
- func (c *Client) SetSamplingPolicy(p SamplingPolicy) {
- if c != nil {
- c.policy = p
- }
- }
-
- // SpanFromHeader returns a new trace span based on a provided request header
- // value or nil iff the client is nil.
- //
- // The trace information and identifiers will be read from the header value.
- // Otherwise, a new trace ID is made and the parent span ID is zero.
- // For the exact format of the header value, see
- // https://cloud.google.com/trace/docs/troubleshooting#force-trace
- //
- // The name of the new span is provided as an argument.
- //
- // If a non-nil sampling policy has been set in the client, it can override
- // the options set in the header and choose whether to trace the request.
- //
- // If the header doesn't have existing tracing information, then a *Span is
- // returned anyway, but it will not be uploaded to the server, just as when
- // calling SpanFromRequest on an untraced request.
- //
- // Most users using HTTP should use SpanFromRequest, rather than
- // SpanFromHeader, since it provides additional functionality for HTTP
- // requests. In particular, it will set various pieces of request information
- // as labels on the *Span, which is not available from the header alone.
- //
- // Deprecated: see https://cloud.google.com/trace/docs/setup/go.
- func (c *Client) SpanFromHeader(name string, header string) *Span {
- if c == nil {
- return nil
- }
- traceID, parentSpanID, options, _, ok := traceInfoFromHeader(header)
- if !ok {
- traceID = nextTraceID()
- }
- t := &trace{
- traceID: traceID,
- client: c,
- globalOptions: options,
- localOptions: options,
- }
- span := startNewChild(name, t, parentSpanID)
- span.span.Kind = spanKindServer
- span.rootSpan = true
- configureSpanFromPolicy(span, c.policy, ok)
- return span
- }
-
- // SpanFromRequest returns a new trace span for an HTTP request or nil
- // iff the client is nil.
- //
- // If the incoming HTTP request contains a trace context header, the trace ID,
- // parent span ID, and tracing options will be read from that header.
- // Otherwise, a new trace ID is made and the parent span ID is zero.
- //
- // If a non-nil sampling policy has been set in the client, it can override the
- // options set in the header and choose whether to trace the request.
- //
- // If the request is not being traced, then a *Span is returned anyway, but it
- // will not be uploaded to the server -- it is only useful for propagating
- // trace context to child requests and for getting the TraceID. All its
- // methods can still be called -- the Finish, FinishWait, and SetLabel methods
- // do nothing. NewChild does nothing, and returns the same *Span. TraceID
- // works as usual.
- //
- // Deprecated: see https://cloud.google.com/trace/docs/setup/go.
- func (c *Client) SpanFromRequest(r *http.Request) *Span {
- if c == nil {
- return nil
- }
- traceID, parentSpanID, options, _, ok := traceInfoFromHeader(r.Header.Get(httpHeader))
- if !ok {
- traceID = nextTraceID()
- }
- t := &trace{
- traceID: traceID,
- client: c,
- globalOptions: options,
- localOptions: options,
- }
- span := startNewChildWithRequest(r, t, parentSpanID)
- span.span.Kind = spanKindServer
- span.rootSpan = true
- configureSpanFromPolicy(span, c.policy, ok)
- return span
- }
-
- // NewSpan returns a new trace span with the given name or nil iff the
- // client is nil.
- //
- // A new trace and span ID is generated to trace the span.
- // Returned span need to be finished by calling Finish or FinishWait.
- //
- // Deprecated: see https://cloud.google.com/trace/docs/setup/go.
- func (c *Client) NewSpan(name string) *Span {
- if c == nil {
- return nil
- }
- t := &trace{
- traceID: nextTraceID(),
- client: c,
- localOptions: optionTrace,
- globalOptions: optionTrace,
- }
- span := startNewChild(name, t, 0)
- span.span.Kind = spanKindUnspecified
- span.rootSpan = true
- configureSpanFromPolicy(span, c.policy, false)
- return span
- }
-
- func configureSpanFromPolicy(s *Span, p SamplingPolicy, ok bool) {
- if p == nil {
- return
- }
- d := p.Sample(Parameters{HasTraceHeader: ok})
- if d.Trace {
- // Turn on tracing locally, and in child requests.
- s.trace.localOptions |= optionTrace
- s.trace.globalOptions |= optionTrace
- } else {
- // Turn off tracing locally.
- s.trace.localOptions = 0
- return
- }
- if d.Sample {
- // This trace is in the random sample, so set the labels.
- s.SetLabel(LabelSamplingPolicy, d.Policy)
- s.SetLabel(LabelSamplingWeight, fmt.Sprint(d.Weight))
- }
- }
-
- // NewContext returns a derived context containing the span.
- //
- // Deprecated: see https://cloud.google.com/trace/docs/setup/go.
- func NewContext(ctx context.Context, s *Span) context.Context {
- if s == nil {
- return ctx
- }
- return context.WithValue(ctx, contextKey{}, s)
- }
-
- // FromContext returns the span contained in the context, or nil.
- //
- // Deprecated: see https://cloud.google.com/trace/docs/setup/go.
- func FromContext(ctx context.Context) *Span {
- s, _ := ctx.Value(contextKey{}).(*Span)
- return s
- }
-
- func traceInfoFromHeader(h string) (traceID string, spanID uint64, options optionFlags, optionsOk bool, ok bool) {
- // See https://cloud.google.com/trace/docs/faq for the header format.
- // Return if the header is empty or missing, or if the header is unreasonably
- // large, to avoid making unnecessary copies of a large string.
- if h == "" || len(h) > 200 {
- return "", 0, 0, false, false
-
- }
-
- // Parse the trace id field.
- slash := strings.Index(h, `/`)
- if slash == -1 {
- return "", 0, 0, false, false
-
- }
- traceID, h = h[:slash], h[slash+1:]
-
- // Parse the span id field.
- spanstr := h
- semicolon := strings.Index(h, `;`)
- if semicolon != -1 {
- spanstr, h = h[:semicolon], h[semicolon+1:]
- }
- spanID, err := strconv.ParseUint(spanstr, 10, 64)
- if err != nil {
- return "", 0, 0, false, false
-
- }
-
- // Parse the options field, options field is optional.
- if !strings.HasPrefix(h, "o=") {
- return traceID, spanID, 0, false, true
-
- }
- o, err := strconv.ParseUint(h[2:], 10, 64)
- if err != nil {
- return "", 0, 0, false, false
-
- }
- options = optionFlags(o)
- return traceID, spanID, options, true, true
- }
-
- type optionFlags uint32
-
- const (
- optionTrace optionFlags = 1 << iota
- optionStack
- )
-
- type trace struct {
- mu sync.Mutex
- client *Client
- traceID string
- globalOptions optionFlags // options that will be passed to any child requests
- localOptions optionFlags // options applied in this server
- spans []*Span // finished spans for this trace.
- }
-
- // finish appends s to t.spans. If s is the root span, uploads the trace to the
- // server.
- func (t *trace) finish(s *Span, wait bool, opts ...FinishOption) error {
- for _, o := range opts {
- o.modifySpan(s)
- }
- s.end = time.Now()
- t.mu.Lock()
- t.spans = append(t.spans, s)
- spans := t.spans
- t.mu.Unlock()
- if s.rootSpan {
- if wait {
- return t.client.upload([]*api.Trace{t.constructTrace(spans)})
- }
- go func() {
- tr := t.constructTrace(spans)
- err := t.client.bundler.Add(tr, 1+len(spans))
- if err == bundler.ErrOversizedItem {
- err = t.client.upload([]*api.Trace{tr})
- }
- if err != nil {
- log.Println("error uploading trace:", err)
- }
- }()
- }
- return nil
- }
-
- func (t *trace) constructTrace(spans []*Span) *api.Trace {
- apiSpans := make([]*api.TraceSpan, len(spans))
- for i, sp := range spans {
- sp.span.StartTime = sp.start.In(time.UTC).Format(time.RFC3339Nano)
- sp.span.EndTime = sp.end.In(time.UTC).Format(time.RFC3339Nano)
- if t.localOptions&optionStack != 0 {
- sp.setStackLabel()
- }
- if sp.host != "" {
- sp.SetLabel(LabelHTTPHost, sp.host)
- }
- if sp.url != "" {
- sp.SetLabel(LabelHTTPURL, sp.url)
- }
- if sp.method != "" {
- sp.SetLabel(LabelHTTPMethod, sp.method)
- }
- if sp.statusCode != 0 {
- sp.SetLabel(LabelHTTPStatusCode, strconv.Itoa(sp.statusCode))
- }
- sp.SetLabel(labelAgent, userAgent)
- apiSpans[i] = &sp.span
- }
-
- return &api.Trace{
- ProjectId: t.client.projectID,
- TraceId: t.traceID,
- Spans: apiSpans,
- }
- }
-
- func (c *Client) upload(traces []*api.Trace) error {
- _, err := c.service.Projects.PatchTraces(c.projectID, &api.Traces{Traces: traces}).Do()
- return err
- }
-
- // Span contains information about one span of a trace.
- //
- // Deprecated: see https://cloud.google.com/trace/docs/setup/go.
- type Span struct {
- trace *trace
-
- spanMu sync.Mutex // guards span.Labels
- span api.TraceSpan
-
- start time.Time
- end time.Time
- rootSpan bool
- stack [maxStackFrames]uintptr
- host string
- method string
- url string
- statusCode int
- }
-
- // Traced reports whether the current span is sampled to be traced.
- //
- // Deprecated: see https://cloud.google.com/trace/docs/setup/go.
- func (s *Span) Traced() bool {
- if s == nil {
- return false
- }
- return s.trace.localOptions&optionTrace != 0
- }
-
- // NewChild creates a new span with the given name as a child of s.
- // If s is nil, does nothing and returns nil.
- //
- // Deprecated: see https://cloud.google.com/trace/docs/setup/go.
- func (s *Span) NewChild(name string) *Span {
- if s == nil {
- return nil
- }
- if !s.Traced() {
- // TODO(jbd): Document this behavior in godoc here and elsewhere.
- return s
- }
- return startNewChild(name, s.trace, s.span.SpanId)
- }
-
- // NewRemoteChild creates a new span as a child of s.
- //
- // Some labels in the span are set from the outgoing *http.Request r.
- //
- // A header is set in r so that the trace context is propagated to the
- // destination. The parent span ID in that header is set as follows:
- // - If the request is being traced, then the ID of s is used.
- // - If the request is not being traced, but there was a trace context header
- // in the incoming request for this trace (the request passed to
- // SpanFromRequest), the parent span ID in that header is used.
- // - Otherwise, the parent span ID is zero.
- // The tracing bit in the options is set if tracing is enabled, or if it was
- // set in the incoming request.
- //
- // If s is nil, does nothing and returns nil.
- //
- // Deprecated: see https://cloud.google.com/trace/docs/setup/go.
- func (s *Span) NewRemoteChild(r *http.Request) *Span {
- if s == nil {
- return nil
- }
- if !s.Traced() {
- r.Header[httpHeader] = []string{spanHeader(s.trace.traceID, s.span.ParentSpanId, s.trace.globalOptions)}
- return s
- }
- newSpan := startNewChildWithRequest(r, s.trace, s.span.SpanId)
- r.Header[httpHeader] = []string{spanHeader(s.trace.traceID, newSpan.span.SpanId, s.trace.globalOptions)}
- return newSpan
- }
-
- // Header returns the value of the X-Cloud-Trace-Context header that
- // should be used to propagate the span. This is the inverse of
- // SpanFromHeader.
- //
- // Most users should use NewRemoteChild unless they have specific
- // propagation needs or want to control the naming of their span.
- // Header() does not create a new span.
- //
- // Deprecated: see https://cloud.google.com/trace/docs/setup/go.
- func (s *Span) Header() string {
- if s == nil {
- return ""
- }
- return spanHeader(s.trace.traceID, s.span.SpanId, s.trace.globalOptions)
- }
-
- func startNewChildWithRequest(r *http.Request, trace *trace, parentSpanID uint64) *Span {
- name := r.URL.Host + r.URL.Path // drop scheme and query params
- newSpan := startNewChild(name, trace, parentSpanID)
- if r.Host == "" {
- newSpan.host = r.URL.Host
- } else {
- newSpan.host = r.Host
- }
- newSpan.method = r.Method
- newSpan.url = r.URL.String()
- return newSpan
- }
-
- func startNewChild(name string, trace *trace, parentSpanID uint64) *Span {
- spanID := nextSpanID()
- for spanID == parentSpanID {
- spanID = nextSpanID()
- }
- newSpan := &Span{
- trace: trace,
- span: api.TraceSpan{
- Kind: spanKindClient,
- Name: name,
- ParentSpanId: parentSpanID,
- SpanId: spanID,
- },
- start: time.Now(),
- }
- if trace.localOptions&optionStack != 0 {
- _ = runtime.Callers(1, newSpan.stack[:])
- }
- return newSpan
- }
-
- // TraceID returns the ID of the trace to which s belongs.
- //
- // Deprecated: see https://cloud.google.com/trace/docs/setup/go.
- func (s *Span) TraceID() string {
- if s == nil {
- return ""
- }
- return s.trace.traceID
- }
-
- // SetLabel sets the label for the given key to the given value.
- // If the value is empty, the label for that key is deleted.
- // If a label is given a value automatically and by SetLabel, the
- // automatically-set value is used.
- // If s is nil, does nothing.
- //
- // SetLabel shouldn't be called after Finish or FinishWait.
- //
- // Deprecated: see https://cloud.google.com/trace/docs/setup/go.
- func (s *Span) SetLabel(key, value string) {
- if s == nil {
- return
- }
- if !s.Traced() {
- return
- }
- s.spanMu.Lock()
- defer s.spanMu.Unlock()
-
- if value == "" {
- if s.span.Labels != nil {
- delete(s.span.Labels, key)
- }
- return
- }
- if s.span.Labels == nil {
- s.span.Labels = make(map[string]string)
- }
- s.span.Labels[key] = value
- }
-
- // FinishOption allows users to specify span finalizers.
- // Deprecated: see https://cloud.google.com/trace/docs/setup/go.
- type FinishOption interface {
- modifySpan(s *Span)
- }
-
- type withResponse struct {
- *http.Response
- }
-
- // WithResponse returns an option that can be passed to Finish that indicates
- // that some labels for the span should be set using the given *http.Response.
- //
- // Deprecated: see https://cloud.google.com/trace/docs/setup/go.
- func WithResponse(resp *http.Response) FinishOption {
- return withResponse{resp}
- }
-
- func (u withResponse) modifySpan(s *Span) {
- if u.Response != nil {
- s.statusCode = u.StatusCode
- }
- }
-
- // Finish declares that the span has finished.
- //
- // If s is nil, Finish does nothing and returns nil.
- //
- // If the option trace.WithResponse(resp) is passed, then some labels are set
- // for s using information in the given *http.Response. This is useful when the
- // span is for an outgoing http request; s will typically have been created by
- // NewRemoteChild in this case.
- //
- // If s is a root span (one created by SpanFromRequest) then s, and all its
- // descendant spans that have finished, are uploaded to the Google Stackdriver
- // Trace server asynchronously.
- //
- // Deprecated: see https://cloud.google.com/trace/docs/setup/go.
- func (s *Span) Finish(opts ...FinishOption) {
- if s == nil {
- return
- }
- if !s.Traced() {
- return
- }
- s.trace.finish(s, false, opts...)
- }
-
- // FinishWait is like Finish, but if s is a root span, it waits until uploading
- // is finished, then returns an error if one occurred.
- //
- // Deprecated: see https://cloud.google.com/trace/docs/setup/go.
- func (s *Span) FinishWait(opts ...FinishOption) error {
- if s == nil {
- return nil
- }
- if !s.Traced() {
- return nil
- }
- return s.trace.finish(s, true, opts...)
- }
-
- func spanHeader(traceID string, spanID uint64, options optionFlags) string {
- // See https://cloud.google.com/trace/docs/faq for the header format.
- return fmt.Sprintf("%s/%d;o=%d", traceID, spanID, options)
- }
-
- func (s *Span) setStackLabel() {
- var stack stackLabelValue
- lastSigPanic, inTraceLibrary := false, true
- for _, pc := range s.stack {
- if pc == 0 {
- break
- }
- if !lastSigPanic {
- pc--
- }
- fn := runtime.FuncForPC(pc)
- file, line := fn.FileLine(pc)
- // Name has one of the following forms:
- // path/to/package.Foo
- // path/to/package.(Type).Foo
- // For the first form, we store the whole name in the Method field of the
- // stack frame. For the second form, we set the Method field to "Foo" and
- // the Class field to "path/to/package.(Type)".
- name := fn.Name()
- if inTraceLibrary && !strings.HasPrefix(name, "cloud.google.com/go/trace.") {
- inTraceLibrary = false
- }
- var class string
- if i := strings.Index(name, ")."); i != -1 {
- class, name = name[:i+1], name[i+2:]
- }
- frame := stackFrame{
- Class: class,
- Method: name,
- Filename: file,
- Line: int64(line),
- }
- if inTraceLibrary && len(stack.Frames) == 1 {
- stack.Frames[0] = frame
- } else {
- stack.Frames = append(stack.Frames, frame)
- }
- lastSigPanic = fn.Name() == "runtime.sigpanic"
- }
- if label, err := json.Marshal(stack); err == nil {
- s.SetLabel(LabelStackTrace, string(label))
- }
- }
|