2020-03-25 23:47:17 +02:00
|
|
|
// Copyright The OpenTelemetry Authors
|
2020-03-11 23:49:02 +02:00
|
|
|
//
|
|
|
|
// 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.
|
|
|
|
|
|
|
|
// Command zipkin is an example program that creates spans
|
|
|
|
// and uploads to openzipkin collector.
|
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
2020-04-23 22:11:24 +02:00
|
|
|
"flag"
|
2020-03-11 23:49:02 +02:00
|
|
|
"log"
|
|
|
|
"os"
|
2022-05-19 22:15:07 +02:00
|
|
|
"os/signal"
|
2020-03-11 23:49:02 +02:00
|
|
|
"time"
|
|
|
|
|
2020-11-16 19:30:54 +02:00
|
|
|
"go.opentelemetry.io/otel"
|
2021-06-12 20:39:22 +02:00
|
|
|
"go.opentelemetry.io/otel/exporters/zipkin"
|
2021-03-16 19:38:48 +02:00
|
|
|
"go.opentelemetry.io/otel/sdk/resource"
|
2020-03-11 23:49:02 +02:00
|
|
|
sdktrace "go.opentelemetry.io/otel/sdk/trace"
|
2022-07-13 15:55:43 +02:00
|
|
|
semconv "go.opentelemetry.io/otel/semconv/v1.12.0"
|
2021-03-08 21:23:15 +02:00
|
|
|
"go.opentelemetry.io/otel/trace"
|
2020-03-11 23:49:02 +02:00
|
|
|
)
|
|
|
|
|
|
|
|
var logger = log.New(os.Stderr, "zipkin-example", log.Ldate|log.Ltime|log.Llongfile)
|
|
|
|
|
|
|
|
// initTracer creates a new trace provider instance and registers it as global trace provider.
|
2022-05-19 22:15:07 +02:00
|
|
|
func initTracer(url string) (func(context.Context) error, error) {
|
2020-07-22 20:57:48 +02:00
|
|
|
// Create Zipkin Exporter and install it as a global tracer.
|
|
|
|
//
|
2020-03-11 23:49:02 +02:00
|
|
|
// For demoing purposes, always sample. In a production application, you should
|
2020-09-13 20:45:46 +02:00
|
|
|
// configure the sampler to a trace.ParentBased(trace.TraceIDRatioBased) set at the desired
|
2020-09-03 16:28:01 +02:00
|
|
|
// ratio.
|
2021-06-10 18:22:47 +02:00
|
|
|
exporter, err := zipkin.New(
|
2020-07-22 20:57:48 +02:00
|
|
|
url,
|
|
|
|
zipkin.WithLogger(logger),
|
2020-03-11 23:49:02 +02:00
|
|
|
)
|
|
|
|
if err != nil {
|
2022-05-19 22:15:07 +02:00
|
|
|
return nil, err
|
2020-03-11 23:49:02 +02:00
|
|
|
}
|
2021-03-08 21:23:15 +02:00
|
|
|
|
|
|
|
batcher := sdktrace.NewBatchSpanProcessor(exporter)
|
|
|
|
|
2021-03-16 19:38:48 +02:00
|
|
|
tp := sdktrace.NewTracerProvider(
|
|
|
|
sdktrace.WithSpanProcessor(batcher),
|
|
|
|
sdktrace.WithResource(resource.NewWithAttributes(
|
2021-06-08 18:46:42 +02:00
|
|
|
semconv.SchemaURL,
|
2021-03-16 19:38:48 +02:00
|
|
|
semconv.ServiceNameKey.String("zipkin-test"),
|
|
|
|
)),
|
|
|
|
)
|
2021-03-08 21:23:15 +02:00
|
|
|
otel.SetTracerProvider(tp)
|
|
|
|
|
2022-05-19 22:15:07 +02:00
|
|
|
return tp.Shutdown, nil
|
2020-03-11 23:49:02 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
func main() {
|
2020-04-23 22:11:24 +02:00
|
|
|
url := flag.String("zipkin", "http://localhost:9411/api/v2/spans", "zipkin url")
|
|
|
|
flag.Parse()
|
|
|
|
|
2022-05-19 22:15:07 +02:00
|
|
|
ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt)
|
|
|
|
defer cancel()
|
2020-03-11 23:49:02 +02:00
|
|
|
|
2022-05-19 22:15:07 +02:00
|
|
|
shutdown, err := initTracer(*url)
|
|
|
|
if err != nil {
|
|
|
|
log.Fatal(err)
|
|
|
|
}
|
|
|
|
defer func() {
|
|
|
|
if err := shutdown(ctx); err != nil {
|
|
|
|
log.Fatal("failed to shutdown TracerProvider: %w", err)
|
|
|
|
}
|
|
|
|
}()
|
2020-03-11 23:49:02 +02:00
|
|
|
|
2020-11-16 19:30:54 +02:00
|
|
|
tr := otel.GetTracerProvider().Tracer("component-main")
|
2021-03-08 21:23:15 +02:00
|
|
|
ctx, span := tr.Start(ctx, "foo", trace.WithSpanKind(trace.SpanKindServer))
|
2020-03-11 23:49:02 +02:00
|
|
|
<-time.After(6 * time.Millisecond)
|
|
|
|
bar(ctx)
|
|
|
|
<-time.After(6 * time.Millisecond)
|
|
|
|
span.End()
|
|
|
|
}
|
|
|
|
|
|
|
|
func bar(ctx context.Context) {
|
2020-11-16 19:30:54 +02:00
|
|
|
tr := otel.GetTracerProvider().Tracer("component-bar")
|
2020-03-11 23:49:02 +02:00
|
|
|
_, span := tr.Start(ctx, "bar")
|
|
|
|
<-time.After(6 * time.Millisecond)
|
|
|
|
span.End()
|
|
|
|
}
|