Add get_available_stream_capacity to conn interface

Author: VictorPrins

httpcore/_async/connection.py

@@ -199,6 +199,11 @@ def is_closed(self) -> bool:
             return self._connect_failed
         return self._connection.is_closed()
 
+    def get_available_stream_capacity(self) -> int:
+        if self._connection is None:
+            return 1
+        return self._connection.get_available_stream_capacity()
+
     def info(self) -> str:
         if self._connection is None:
             return "CONNECTION FAILED" if self._connect_failed else "CONNECTING"

httpcore/_async/connection_pool.py

@@ -15,7 +15,7 @@
 
 try:
     from .http2 import AsyncHTTP2Connection
-except ImportError:
+except ImportError:  # pragma: nocover
     # ImportError happens when the user installed httpcore without the optional http2 dependency
     AsyncHTTP2Connection = None  # type: ignore[assignment, misc]
 
@@ -303,7 +303,7 @@ def _assign_requests_to_connections(self) -> list[AsyncConnectionInterface]:
                 # Available connections
                 available_conns.append(conn)
                 # Track HTTP/2 connection capacity
-                if self._http2 and isinstance(conn, AsyncHTTP2Connection):
+                if self._http2:
                     # Get the actual available stream count from the connection
                     http2_conn_stream_capacity[conn] = (
                         conn.get_available_stream_capacity()

httpcore/_async/http11.py

@@ -291,6 +291,13 @@ def is_idle(self) -> bool:
     def is_closed(self) -> bool:
         return self._state == HTTPConnectionState.CLOSED
 
+    def get_available_stream_capacity(self) -> int:
+        """
+        For HTTP/1.1, return 1 if the connection is idle (can accept a request),
+        0 otherwise (connection is busy).
+        """
+        return 1 if self._state == HTTPConnectionState.IDLE else 0
+
     def info(self) -> str:
         origin = str(self._origin)
         return (

httpcore/_async/http_proxy.py

@@ -363,5 +363,8 @@ def is_idle(self) -> bool:
     def is_closed(self) -> bool:
         return self._connection.is_closed()
 
+    def get_available_stream_capacity(self) -> int:
+        return self._connection.get_available_stream_capacity()
+
     def __repr__(self) -> str:
         return f"<{self.__class__.__name__} [{self.info()}]>"

httpcore/_async/interfaces.py

@@ -135,3 +135,12 @@ def is_closed(self) -> bool:
         returned to the connection pool or not.
         """
         raise NotImplementedError()  # pragma: nocover
+
+    def get_available_stream_capacity(self) -> int:
+        """
+        Return the number of additional streams that can be handled by this connection.
+
+        For HTTP/1.1 connections, this is 1 if the connection is idle, 0 otherwise.
+        For HTTP/2 connections, this is the number of available concurrent streams.
+        """
+        raise NotImplementedError()  # pragma: nocover

httpcore/_sync/connection.py

@@ -199,6 +199,11 @@ def is_closed(self) -> bool:
             return self._connect_failed
         return self._connection.is_closed()
 
+    def get_available_stream_capacity(self) -> int:
+        if self._connection is None:
+            return 1
+        return self._connection.get_available_stream_capacity()
+
     def info(self) -> str:
         if self._connection is None:
             return "CONNECTION FAILED" if self._connect_failed else "CONNECTING"

httpcore/_sync/connection_pool.py

@@ -15,7 +15,7 @@
 
 try:
     from .http2 import HTTP2Connection
-except ImportError:
+except ImportError:  # pragma: nocover
     # ImportError happens when the user installed httpcore without the optional http2 dependency
     HTTP2Connection = None  # type: ignore[assignment, misc]
 
@@ -303,7 +303,7 @@ def _assign_requests_to_connections(self) -> list[ConnectionInterface]:
                 # Available connections
                 available_conns.append(conn)
                 # Track HTTP/2 connection capacity
-                if self._http2 and isinstance(conn, HTTP2Connection):
+                if self._http2:
                     # Get the actual available stream count from the connection
                     http2_conn_stream_capacity[conn] = (
                         conn.get_available_stream_capacity()

httpcore/_sync/http11.py

@@ -291,6 +291,13 @@ def is_idle(self) -> bool:
     def is_closed(self) -> bool:
         return self._state == HTTPConnectionState.CLOSED
 
+    def get_available_stream_capacity(self) -> int:
+        """
+        For HTTP/1.1, return 1 if the connection is idle (can accept a request),
+        0 otherwise (connection is busy).
+        """
+        return 1 if self._state == HTTPConnectionState.IDLE else 0
+
     def info(self) -> str:
         origin = str(self._origin)
         return (

httpcore/_sync/http_proxy.py

@@ -363,5 +363,8 @@ def is_idle(self) -> bool:
     def is_closed(self) -> bool:
         return self._connection.is_closed()
 
+    def get_available_stream_capacity(self) -> int:
+        return self._connection.get_available_stream_capacity()
+
     def __repr__(self) -> str:
         return f"<{self.__class__.__name__} [{self.info()}]>"

httpcore/_sync/interfaces.py

@@ -135,3 +135,12 @@ def is_closed(self) -> bool:
         returned to the connection pool or not.
         """
         raise NotImplementedError()  # pragma: nocover
+
+    def get_available_stream_capacity(self) -> int:
+        """
+        Return the number of additional streams that can be handled by this connection.
+
+        For HTTP/1.1 connections, this is 1 if the connection is idle, 0 otherwise.
+        For HTTP/2 connections, this is the number of available concurrent streams.
+        """
+        raise NotImplementedError()  # pragma: nocover