Add more features to struct poll command

Author: vkottler

runtimepy/net/arbiter/housekeeping/__init__.py

@@ -52,7 +52,11 @@ async def dispatch(self) -> bool:
 
         # Handle any incoming commands.
         processors: list[Awaitable[None]] = []
-        for mapping in self.app.connections.values(), self.app.tasks.values():
+        for mapping in (
+            self.app.connections.values(),
+            self.app.tasks.values(),
+            self.app.structs.values(),
+        ):
             for item in mapping:
                 if isinstance(item, AsyncCommandProcessingMixin):
                     processors.append(item.process_command_queue())

runtimepy/struct/__init__.py

@@ -3,23 +3,31 @@
 """
 
 # built-in
+from argparse import Namespace
+import asyncio
 from logging import getLogger as _getLogger
+from typing import Optional
 
 # third-party
 from vcorelib.io import MarkdownMixin
 from vcorelib.io.types import JsonObject as _JsonObject
 
 # internal
 from runtimepy import PKG_NAME
+from runtimepy.channel.environment.command import FieldOrChannel
 from runtimepy.channel.environment.command.processor import (
     ChannelCommandProcessor,
 )
+from runtimepy.mixins.async_command import AsyncCommandProcessingMixin
 from runtimepy.mixins.environment import ChannelEnvironmentMixin
 from runtimepy.mixins.logging import LoggerMixinLevelControl
 
 
 class RuntimeStructBase(
-    LoggerMixinLevelControl, ChannelEnvironmentMixin, MarkdownMixin
+    LoggerMixinLevelControl,
+    ChannelEnvironmentMixin,
+    AsyncCommandProcessingMixin,
+    MarkdownMixin,
 ):
     """A base runtime structure."""
 
@@ -42,6 +50,23 @@ def __init__(
         self.command = ChannelCommandProcessor(self.env, self.logger)
         self.config = config
 
+        async def poll(args: Namespace, __: Optional[FieldOrChannel]) -> None:
+            """Handle a test command."""
+
+            count = 1
+            delay = 0.0
+
+            if args.extra:
+                count = int(args.extra[0])
+                if len(args.extra) > 1:
+                    delay = float(args.extra[1])
+
+            for _ in range(count):
+                self.poll()
+                await asyncio.sleep(delay)
+
+        self._setup_async_commands(poll)
+
     def poll(self) -> None:
         """
         A method that other runtime entities can call to perform canonical

tests/net/server/__init__.py

@@ -139,6 +139,17 @@ async def runtimepy_http_client_server(
             client.request_json(
                 RequestHeader(method="POST", target="/sample1/custom/asdf")
             ),
+            client.request_json(
+                RequestHeader(method="POST", target="/struct2/custom/poll")
+            ),
+            client.request_json(
+                RequestHeader(method="POST", target="/struct2/custom/poll/5")
+            ),
+            client.request_json(
+                RequestHeader(
+                    method="POST", target="/struct2/custom/poll/5/0.01"
+                )
+            ),
             client.request_json(
                 RequestHeader(
                     method="POST", target="/sample1/custom/test_command"