Servers: Common implementation for both TCP and UDP servers

docs/source/_extensions/sphinx_easynetwork.py

@@ -3,32 +3,40 @@
 
 v0.1.0: Initial
 v0.1.1: Fix base is not replaced if the class is generic.
-v0.2.0 (current): Log when an object does not have a docstring.
+v0.2.0: Log when an object does not have a docstring.
+v0.2.1 (current): Add base class to replace.
 """
 
 from __future__ import annotations
 
 import logging
-from typing import TYPE_CHECKING, Any, get_origin
+from collections.abc import Callable
+from typing import TYPE_CHECKING, Any, get_args, get_origin
 
 if TYPE_CHECKING:
     from sphinx.application import Sphinx
 
-from easynetwork.servers._base import BaseStandaloneNetworkServerImpl
-from easynetwork.servers.abc import AbstractNetworkServer
+from easynetwork.servers._base import BaseAsyncNetworkServerImpl, BaseStandaloneNetworkServerImpl
+from easynetwork.servers.abc import AbstractAsyncNetworkServer, AbstractNetworkServer
 
 logger = logging.getLogger(__name__)
 
 
-def _replace_base_in_place(klass: type, bases: list[type], base_to_replace: type, base_to_set_instead: type) -> None:
+def _replace_base_in_place(
+    klass: type,
+    bases: list[type],
+    base_to_replace: type,
+    base_to_set_instead: Callable[[tuple[Any, ...]], Any],
+) -> None:
     if issubclass(klass, base_to_replace):
         for index, base in enumerate(bases):
-            if get_origin(base) is base_to_replace:
-                bases[index] = base_to_set_instead
+            if base is base_to_replace or get_origin(base) is base_to_replace:
+                bases[index] = base_to_set_instead(get_args(base))
 
 
 def autodoc_process_bases(app: Sphinx, name: str, obj: type, options: dict[str, Any], bases: list[type]) -> None:
-    _replace_base_in_place(obj, bases, BaseStandaloneNetworkServerImpl, AbstractNetworkServer)
+    _replace_base_in_place(obj, bases, BaseAsyncNetworkServerImpl, lambda _: AbstractAsyncNetworkServer)
+    _replace_base_in_place(obj, bases, BaseStandaloneNetworkServerImpl, lambda _: AbstractNetworkServer)
 
 
 def _is_magic_method(name: str) -> bool:
@@ -47,7 +55,7 @@ def setup(app: Sphinx) -> dict[str, Any]:
     app.connect("autodoc-process-docstring", autodoc_process_docstring)
 
     return {
-        "version": "0.2.0",
+        "version": "0.2.1",
         "parallel_read_safe": True,
         "parallel_write_safe": True,
     }

docs/source/_include/examples/howto/tcp_servers/background_server.py

@@ -63,6 +63,7 @@ async def main() -> None:
         await client(host, port, "Hello world 3")
 
         await server.shutdown()
+        await server_task
 
 
 if __name__ == "__main__":

docs/source/_include/examples/howto/tcp_servers/background_server_ssl.py

@@ -71,6 +71,7 @@ async def main() -> None:
         await client(host, port, "Hello world 3")
 
         await server.shutdown()
+        await server_task
 
 
 if __name__ == "__main__":

docs/source/_include/examples/howto/udp_servers/background_server.py

@@ -60,6 +60,7 @@ async def main() -> None:
         await client(host, port, "Hello world 3")
 
         await server.shutdown()
+        await server_task
 
 
 if __name__ == "__main__":

src/easynetwork/lowlevel/_utils.py

@@ -16,6 +16,7 @@
 
 __all__ = [
     "ElapsedTime",
+    "Flag",
     "WarnCallback",
     "adjust_leftover_buffer",
     "check_real_socket_state",
@@ -395,6 +396,22 @@ def __call__(
     ) -> None: ...
 
 
+class Flag:
+    __slots__ = ("__value", "__weakref__")
+
+    def __init__(self) -> None:
+        self.__value: bool = False
+
+    def is_set(self) -> bool:
+        return self.__value
+
+    def set(self) -> None:
+        self.__value = True
+
+    def clear(self) -> None:
+        self.__value = False
+
+
 class ElapsedTime:
     __slots__ = ("_current_time_func", "_start_time", "_end_time")