mirror of
https://github.com/zap-proto/zip.git
synced 2026-10-19 15:38:57 +00:00
zeekay
f2a543133e
traceOf read the caller's ids as slices of a view over the header buffer, then rewrote that buffer in place: every shipped span named itself as its parent, and once the connection served its next request the span carried that request's trace id. Claude-Session: https://claude.ai/code/session_01HuMDfgFssPUptYUzgZ6Cp8
74 lines
2.6 KiB
Go
74 lines
2.6 KiB
Go
package zip
|
|
|
|
import (
|
|
"context"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/valyala/fasthttp"
|
|
)
|
|
|
|
// sentTrace is the traceparent forwardIdentity writes onto an outbound request.
|
|
func sentTrace(ctx context.Context) string {
|
|
req := fasthttp.AcquireRequest()
|
|
defer fasthttp.ReleaseRequest(req)
|
|
forwardIdentity(ctx, req)
|
|
return string(req.Header.Peek(HeaderTrace))
|
|
}
|
|
|
|
// A host's own trace position rides a call that has no request behind it, wins
|
|
// over the request's when there is one, and an empty answer leaves the request's.
|
|
func TestTraceparentHook(t *testing.T) {
|
|
const own = "00-0af7651916cd43dd8448eb211c80319c-b7ad6b7169203331-01"
|
|
old := Traceparent
|
|
t.Cleanup(func() { Traceparent = old })
|
|
|
|
Traceparent = nil
|
|
if got := sentTrace(context.Background()); got != "" {
|
|
t.Fatalf("no hook, no request: sent %q, want nothing", got)
|
|
}
|
|
|
|
Traceparent = func(context.Context) string { return own }
|
|
if got := sentTrace(context.Background()); got != own {
|
|
t.Fatalf("detached call: sent %q, want %q", got, own)
|
|
}
|
|
|
|
var fromRequest, overridden string
|
|
app := New(Config{AppName: "front", DisableStartupMessage: true})
|
|
app.Raw("GET", "/ask", func(c *Ctx) error {
|
|
Traceparent = func(context.Context) string { return "" }
|
|
fromRequest = sentTrace(c.Forward())
|
|
Traceparent = func(context.Context) string { return own }
|
|
overridden = sentTrace(c.Forward())
|
|
return c.JSON(200, map[string]string{"ok": "1"})
|
|
})
|
|
resp, err := app.Fiber().Test(httptest.NewRequest("GET", "/ask", nil))
|
|
if err != nil {
|
|
t.Fatalf("GET: %v", err)
|
|
}
|
|
_ = resp.Body.Close()
|
|
if fromRequest == "" || fromRequest == own {
|
|
t.Fatalf("empty answer: sent %q, want the request's own traceparent", fromRequest)
|
|
}
|
|
if overridden != own {
|
|
t.Fatalf("hook answer: sent %q, want %q over the request's", overridden, own)
|
|
}
|
|
}
|
|
|
|
// The span a hop ships names its caller as parent and its own id as itself, even
|
|
// after the hop has rewritten the header the caller's ids were read from, and
|
|
// after the connection's next request has overwritten it.
|
|
func TestTraceOfKeepsTheCallersIDs(t *testing.T) {
|
|
const (
|
|
caller = "4bf92f3577b34da6a3ce929d0e0e4736"
|
|
parent = "00f067aa0ba902b7"
|
|
)
|
|
c := New(Config{AppName: "front", DisableStartupMessage: true}).TestCtx("GET", "/ask")
|
|
c.fc.Request().Header.Set(HeaderTrace, "00-"+caller+"-"+parent+"-01")
|
|
tr := traceOf(c)
|
|
c.fc.Request().Header.Set(HeaderTrace, "00-ffffffffffffffffffffffffffffffff-ffffffffffffffff-01")
|
|
if tr.trace != caller || tr.parent != parent || tr.span == parent {
|
|
t.Fatalf("trace = %+v, want trace %s parent %s and a span of its own", tr, caller, parent)
|
|
}
|
|
}
|