mirror of
https://github.com/open-telemetry/opentelemetry-go.git
synced 2024-12-20 19:52:56 +02:00
9ed97980e2
* Add Stackdriver Trace exporter for trace. TODOs for future work is: * to replace bundler.Bundler * to add proper tests for the exporter * to move the exporter to proper repository once it will be created. * Change to use functions for the exporter initialization instead of passing option struct directly. This fix is aliging the same fix for Jaeger (#146, #161) * Change Option struct to be function type * Change the original Option struct to be private * Add line comments to maxMessageEventsPerSpan to leave it for future implementation * Fix unnessesary expressions specified by `make precommit` Left errors by `make precommit` in experimental/bridge/opentracing. * Ran make precommit * Add new line at EOF * WIP: Start implementing BatchSpanExporter interfaces * Change to use RegisterSpanProcessor to register bsp * Change function names to fit current implementation of sdk * Removed google.golang.org/api/support/bundler and implement ssp and bsp * Change spanProcessor as a member of Exporter. * Fix option names used for BatchSpanProcessor initialization. * Change Exporter.Shutdown just to unregister spanProcessor. * Removed copyright statements of OpenCensus. * Fix small typo and EOF new line * Fix interfaces of ExportSpan/ExportSpans to meet SpanSyncer/SpanBatcher * Change to follow context.Context passed in ExportSpan/ExportSpans * Fix Stackdriver Exporter to hold sync.Once to lock when it is registered and unregistered.
103 lines
3.0 KiB
Go
103 lines
3.0 KiB
Go
// Copyright 2019, 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 stackdriver
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
traceclient "cloud.google.com/go/trace/apiv2"
|
|
tracepb "google.golang.org/genproto/googleapis/devtools/cloudtrace/v2"
|
|
|
|
"go.opentelemetry.io/sdk/export"
|
|
)
|
|
|
|
// traceExporter is an imeplementation of trace.Exporter and trace.BatchExporter
|
|
// that uploads spans to Stackdriver Trace in batch.
|
|
type traceExporter struct {
|
|
o *options
|
|
projectID string
|
|
// uploadFn defaults in uploadSpans; it can be replaced for tests.
|
|
uploadFn func(ctx context.Context, spans []*tracepb.Span)
|
|
client *traceclient.Client
|
|
}
|
|
|
|
func newTraceExporter(o *options) (*traceExporter, error) {
|
|
ctx := o.Context
|
|
if ctx == nil {
|
|
ctx = context.Background()
|
|
}
|
|
client, err := traceclient.NewClient(ctx, o.TraceClientOptions...)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("Stackdriver: couldn't initiate trace client: %v", err)
|
|
}
|
|
e := &traceExporter{
|
|
projectID: o.ProjectID,
|
|
client: client,
|
|
o: o,
|
|
}
|
|
e.uploadFn = e.uploadSpans
|
|
return e, nil
|
|
}
|
|
|
|
// ExportSpan exports a SpanData to Stackdriver Trace.
|
|
func (e *traceExporter) ExportSpan(ctx context.Context, sd *export.SpanData) {
|
|
protoSpan := protoFromSpanData(sd, e.projectID)
|
|
if ctx == nil {
|
|
ctx = context.Background()
|
|
}
|
|
e.uploadFn(ctx, []*tracepb.Span{protoSpan})
|
|
}
|
|
|
|
// ExportSpans exports a slice of SpanData to Stackdriver Trace in batch
|
|
func (e *traceExporter) ExportSpans(ctx context.Context, sds []*export.SpanData) {
|
|
pbSpans := make([]*tracepb.Span, len(sds))
|
|
for i, sd := range sds {
|
|
pbSpans[i] = protoFromSpanData(sd, e.projectID)
|
|
}
|
|
var cancel func()
|
|
if ctx == nil {
|
|
ctx, cancel = newContextWithTimeout(e.o.Context, e.o.Timeout)
|
|
}
|
|
defer cancel()
|
|
e.uploadFn(ctx, pbSpans)
|
|
}
|
|
|
|
// uploadSpans sends a set of spans to Stackdriver.
|
|
func (e *traceExporter) uploadSpans(ctx context.Context, spans []*tracepb.Span) {
|
|
req := tracepb.BatchWriteSpansRequest{
|
|
Name: "projects/" + e.projectID,
|
|
Spans: spans,
|
|
}
|
|
|
|
// TODO(ymotongpoo): add this part after OTel support NeverSampler
|
|
// for tracer.Start() initialization.
|
|
//
|
|
// tracer := apitrace.Register()
|
|
// ctx, span := tracer.Start(
|
|
// ctx,
|
|
// "go.opentelemetry.io/exporter/stackdriver.uploadSpans",
|
|
// )
|
|
// defer span.End()
|
|
// span.SetAttribute(key.New("num_spans").Int64(int64(len(spans))))
|
|
|
|
err := e.client.BatchWriteSpans(ctx, &req)
|
|
if err != nil {
|
|
// TODO(ymotongpoo): handle detailed error categories
|
|
// span.SetStatus(codes.Unknown)
|
|
e.o.handleError(err)
|
|
}
|
|
}
|