From b9e3fc65a5741851891e54a1ade45787584e6a77 Mon Sep 17 00:00:00 2001
From: Oscar Beaumont <oscar@otbeaumont.me>
Date: Mon, 26 Aug 2024 13:40:25 +0800
Subject: [PATCH] rspc_axum: feature flag `with_websocket` in `new`

---
 integrations/axum/src/endpoint.rs | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/integrations/axum/src/endpoint.rs b/integrations/axum/src/endpoint.rs
index 37132c00..d495c171 100644
--- 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.