rspc_axum: feature flag with_websocket in new

specta-rs · Aug 26, 2024 · b9e3fc6 · b9e3fc6
1 parent 6ed156b
commit b9e3fc6
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/integrations/axum/src/endpoint.rs b/integrations/axum/src/endpoint.rs
@@ -46,11 +46,12 @@ impl<TCtx: Send + Sync + 'static> Endpoint<TCtx> {
         // F: Future<Output = Result<TCtx, ()>> + Send + Sync + 'static,
         TCtx: Clone,
     {
-        Self::builder(router)
-            .with_endpoints()
-            .with_websocket()
-            .with_batching()
-            .build(ctx_fn)
+        let mut t = Self::builder(router).with_endpoints();
+        #[cfg(feature = "ws")]
+        {
+            t = t.with_websocket();
+        }
+        t.with_batching().build(ctx_fn)
     }
 
     /// Construct a new [`Endpoint`](Endpoint) with no features enabled.