118 lines
3.4 KiB
Go
118 lines
3.4 KiB
Go
package telemetry_test
|
|
|
|
import (
|
|
"context"
|
|
omni "git.yuntuo.hk/k2safe/go-omni"
|
|
"git.yuntuo.hk/k2safe/go-omni/httpx"
|
|
"git.yuntuo.hk/k2safe/go-omni/logx"
|
|
"git.yuntuo.hk/k2safe/go-omni/telemetry"
|
|
"github.com/redis/go-redis/v9"
|
|
sdkmetric "go.opentelemetry.io/otel/sdk/metric"
|
|
"go.opentelemetry.io/otel/sdk/metric/metricdata"
|
|
"go.opentelemetry.io/otel/sdk/trace/tracetest"
|
|
"go.opentelemetry.io/otel/trace"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestHTTPTracePropagationMetricsAndRedisNil(t *testing.T) {
|
|
exporter := tracetest.NewInMemoryExporter()
|
|
reader := sdkmetric.NewManualReader()
|
|
runtime, err := telemetry.New(telemetry.Config{Service: "test", SampleRatio: 1, Exporter: exporter, Reader: reader})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer runtime.Close()
|
|
ctx := context.Background()
|
|
upstream := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if !strings.HasPrefix(r.Header.Get("Traceparent"), "00-11111111111111111111111111111111-") {
|
|
t.Error("missing propagated trace")
|
|
}
|
|
io.WriteString(w, "ok")
|
|
}))
|
|
defer upstream.Close()
|
|
app := omni.New(omni.WithLogger(false))
|
|
app.Use(func(next httpx.Handler) httpx.Handler {
|
|
return func(c *httpx.Context) error {
|
|
err := next(c)
|
|
if logx.TraceID(c.Context()) != "11111111111111111111111111111111" {
|
|
t.Error("outer access log lost trace correlation")
|
|
}
|
|
return err
|
|
}
|
|
})
|
|
app.Use(runtime.Middleware())
|
|
app.GET("/users/:id", func(c *httpx.Context) error {
|
|
if logx.TraceID(c.Context()) != "11111111111111111111111111111111" {
|
|
t.Error("log correlation")
|
|
}
|
|
req, _ := http.NewRequestWithContext(c.Context(), "GET", upstream.URL+"?secret=hidden", nil)
|
|
resp, err := (telemetry.Transport{Runtime: runtime}).RoundTrip(req)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
io.ReadAll(resp.Body)
|
|
resp.Body.Close()
|
|
if req.Header.Get("Traceparent") != "" {
|
|
t.Error("caller headers mutated")
|
|
}
|
|
return c.OK("ok")
|
|
})
|
|
req := httptest.NewRequest("GET", "/users/privateUser?secret=hidden", nil)
|
|
req.Header.Set("Traceparent", "00-11111111111111111111111111111111-2222222222222222-01")
|
|
rec := httptest.NewRecorder()
|
|
app.ServeHTTP(rec, req)
|
|
if rec.Code != 200 {
|
|
t.Fatal(rec.Code)
|
|
}
|
|
hook := telemetry.RedisHook{Runtime: runtime}
|
|
if err = hook.ProcessHook(func(context.Context, redis.Cmder) error { return redis.Nil })(ctx, redis.NewStringCmd(ctx, "GET", "secret")); err != redis.Nil {
|
|
t.Fatal("cache miss was swallowed", err)
|
|
}
|
|
if err = runtime.Flush(ctx); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
spans := exporter.GetSpans()
|
|
if len(spans) != 3 {
|
|
t.Fatalf("got %d spans", len(spans))
|
|
}
|
|
var client, server bool
|
|
for _, s := range spans {
|
|
if strings.Contains(s.Name, "privateUser") || strings.Contains(s.Name, "secret") {
|
|
t.Fatal(s.Name)
|
|
}
|
|
if s.SpanKind == trace.SpanKindServer {
|
|
server = true
|
|
if s.Name != "GET /users/:id" || s.Parent.SpanID().String() != "2222222222222222" {
|
|
t.Fatal(s.Name, s.Parent)
|
|
}
|
|
}
|
|
if s.SpanKind == trace.SpanKindClient {
|
|
client = true
|
|
}
|
|
}
|
|
if !client || !server {
|
|
t.Fatal("missing client/server span")
|
|
}
|
|
var metrics metricdata.ResourceMetrics
|
|
if err = reader.Collect(ctx, &metrics); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
var count int64
|
|
for _, scope := range metrics.ScopeMetrics {
|
|
for _, m := range scope.Metrics {
|
|
if sum, ok := m.Data.(metricdata.Sum[int64]); ok {
|
|
for _, p := range sum.DataPoints {
|
|
count += p.Value
|
|
}
|
|
}
|
|
}
|
|
}
|
|
if count != 1 {
|
|
t.Fatal("request metric", count)
|
|
}
|
|
}
|