insert proxy also in context for proxydb

plabayo · Sep 5, 2024 · 01b25c3 · 01b25c3
1 parent a5e55ad
commit 01b25c3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/rama-proxy/src/proxydb/layer.rs b/rama-proxy/src/proxydb/layer.rs
@@ -295,7 +295,10 @@ where
             ctx.insert(proxy_address);
 
             // insert the id of the selected proxy
-            ctx.insert(super::ProxyID::from(proxy.id));
+            ctx.insert(super::ProxyID::from(proxy.id.clone()));
+
+            // insert the entire proxy also in there, for full "Context"
+            ctx.insert(proxy);
         }
 
         self.inner.serve(ctx, req).await.map_err(Into::into)