Support tail workers over rpc

danlapid · web-flow · commit 8f3b24d3bbbd · 2025-03-12T14:02:26.000-04:00
diff --git a/src/workerd/server/server.c++ b/src/workerd/server/server.c++
@@ -3709,7 +3709,14 @@ class Server::HttpListener final: public kj::Refcounted {
     }
 
     kj::Promise<void> sendTraces(SendTracesContext context) override {
-      throwUnsupported();
+      auto traces =
+          KJ_MAP(trace, context.getParams().getTraces()){ return kj::refcounted<Trace>(trace); };
+      auto event = kj::heap<api::TraceCustomEventImpl>(
+        api::TraceCustomEventImpl::TYPE, kj::mv(traces));
+      auto worker = getWorker();
+      auto result = co_await worker->customEvent(kj::mv(event));
+      auto resp = context.getResults().getResult();
+      resp.setOutcome(result.outcome);
     }
 
     kj::Promise<void> prewarm(PrewarmContext context) override {

Original file line number	Diff line number	Diff line change
`@@ -3709,7 +3709,14 @@ class Server::HttpListener final: public kj::Refcounted {`
`3709`	`3709`	`}`
`3710`	`3710`
`3711`	`3711`	`kj::Promise<void> sendTraces(SendTracesContext context) override {`
`3712`		`- throwUnsupported();`
	`3712`	`+ auto traces =`
	`3713`	`+ KJ_MAP(trace, context.getParams().getTraces()){ return kj::refcounted<Trace>(trace); };`
	`3714`	`+ auto event = kj::heap<api::TraceCustomEventImpl>(`
	`3715`	`+ api::TraceCustomEventImpl::TYPE, kj::mv(traces));`
	`3716`	`+ auto worker = getWorker();`
	`3717`	`+ auto result = co_await worker->customEvent(kj::mv(event));`
	`3718`	`+ auto resp = context.getResults().getResult();`
	`3719`	`+ resp.setOutcome(result.outcome);`
`3713`	`3720`	`}`
`3714`	`3721`
`3715`	`3722`	`kj::Promise<void> prewarm(PrewarmContext context) override {`