undo tmp change

Author: whummer

paradedb/localstack_paradedb/extension.py

@@ -3,6 +3,7 @@
 
 from localstack_extensions.utils.docker import ProxiedDockerContainerExtension
 from localstack import config
+from localstack.extensions.api import http
 
 # Environment variables for configuration
 ENV_POSTGRES_USER = "PARADEDB_POSTGRES_USER"
@@ -56,22 +57,22 @@ def _tcp_health_check():
             tcp_ports=[postgres_port],  # Enable TCP proxying through gateway
         )
 
-    # TODO: uncomment
-    # def update_gateway_routes(self, router: http.Router[http.RouteHandler]):
-    #     """
-    #     Override to set up only TCP routing without HTTP proxy.
-    #
-    #     ParadeDB uses the native PostgreSQL wire protocol (not HTTP), so we
-    #     only need TCP protocol routing - not HTTP proxying. Adding an HTTP
-    #     proxy without a host restriction would cause all HTTP requests to be
-    #     forwarded to the PostgreSQL container, breaking other services.
-    #     """
-    #     # Start the container
-    #     self.start_container()
-    #
-    #     # Set up only TCP protocol routing (skip HTTP proxy from base class)
-    #     if self.tcp_ports:
-    #         self._setup_tcp_protocol_routing()
+    # TODO: this should be migrated into the base class directly ..!
+    def update_gateway_routes(self, router: http.Router[http.RouteHandler]):
+        """
+        Override to set up only TCP routing without HTTP proxy.
+
+        ParadeDB uses the native PostgreSQL wire protocol (not HTTP), so we
+        only need TCP protocol routing - not HTTP proxying. Adding an HTTP
+        proxy without a host restriction would cause all HTTP requests to be
+        forwarded to the PostgreSQL container, breaking other services.
+        """
+        # Start the container
+        self.start_container()
+
+        # Set up only TCP protocol routing (skip HTTP proxy from base class)
+        if self.tcp_ports:
+            self._setup_tcp_protocol_routing()
 
     def tcp_connection_matcher(self, data: bytes) -> bool:
         """