Close #9: Support multiple concurrent worker processes

.coveragerc

@@ -0,0 +1,2 @@
+[run]
+source = src/

.vscode/launch.json

@@ -11,22 +11,38 @@
             "program": "./src/aiotaskq/main.py",
             "console": "integratedTerminal"
         },
+        {
+            "name": "Sample App",
+            "type": "python",
+            "request": "launch",
+            "module": "aiotaskq.tests.apps.simple_app",
+            "args": [],
+            "console": "integratedTerminal"
+        },
         {
             "name": "Test",
             "type": "python",
+            "module": "coverage",
+            "args": [
+                "run",
+                "-m",
+                "pytest",
+                "-vvv",
+                "-s",
+            ],
             "request": "launch",
-            "program": "./src/aiotaskq/tests/integration_test.py",
             "console": "integratedTerminal"
         },
         {
-            "name": "Worker",
+            "name": "Sample Worker (Simple App)",
             "type": "python",
             "request": "launch",
-            "program": "./src/aiotaskq/worker.py",
-            "console": "integratedTerminal",
+            "module": "aiotaskq",
             "args": [
-                "aiotaskq.main"
-            ]
+                "worker",
+                "aiotaskq.tests.apps.simple_app"
+            ],
+            "console": "integratedTerminal",
         }
     ]
 }

.vscode/settings.json

@@ -3,6 +3,11 @@
         "**/.git": true,
         "**/.venv": true,
         "**/.hg": true,
-        "**/.DS_Store": true
-    }
-}
+        "**/.DS_Store": true,
+        "**/*.egg-info": true,
+        "**/__pycache__": true,
+        "**/.mypy_cache": true
+    },
+    "python.testing.pytestEnabled": true,
+    "python.testing.unittestEnabled": false
+}

README.md

@@ -33,7 +33,7 @@ import asyncio
 import aiotaskq
 
 
-@aiotaskq.register_task
+@aiotaskq.task
 def some_task(b: int) -> int:
     # Some task with high cpu usage
     def _naive_fib(n: int) -> int:
@@ -130,25 +130,25 @@ Using `aiotaskq` we may end up with the following:
 ```python
 import asyncio
 
-from aiotaskq import register_task
+from aiotaskq import task
 
 
-@register_task
+@task
 def task_1(*args, **kwargs):
         pass
 
 
-@register_task
+@task
 def task_2(*args, **kwargs):
         pass
 
 
-@register_task
+@task
 def task_3(*args, **kwargs):
         pass
 
 
-@register_task
+@task
 def task_4(*args, **kwargs):
         pass
 

pyproject.toml

@@ -12,11 +12,15 @@ dependencies = [
     "click==8.0.4"
 ]
 name = "aiotaskq"
-version = "0.0.4"
+version = "0.0.5"
 [project.optional-dependencies]
 tests = [
     "mypy==0.931",
     "mypy-extensions==0.4.3",
     "typing_extensions==4.1.1",
-    "black==22.1.0"
+    "black==22.1.0",
+    "pytest==7.1.2",
+    "pytest-asyncio==0.19.0"
 ]
+[tool.black]
+line-length = 100

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = aiotaskq
-version = 0.0.4
+version = 0.0.5
 author = Imran Ariffin
 author_email = ariffin.imran@gmail.com
 description = A simple asynchronous task queue

src/aiotaskq/__init__.py

@@ -7,7 +7,7 @@
     import aiotaskq
 
 
-    @aiotaskq.register_task
+    @aiotaskq.task
     def some_task(b: int) -> int:
         # Some task with high cpu usage
         def _naive_fib(n: int) -> int:
@@ -29,7 +29,7 @@ async def main():
 
 """
 
-from .main import register_task
+from .main import task
 
 
-__all__ = ["register_task"]
+__all__ = ["task"]

src/aiotaskq/__main__.py

@@ -1,22 +1,27 @@
+"""Module to define the main commands for the cli."""
+
 #!/usr/bin/env python
 
 import asyncio
+import typing as t
 
 import typer
 
-from aiotaskq.worker import worker
+from aiotaskq.worker import Defaults, worker
 
 cli = typer.Typer()
 
 
 @cli.command(name="worker")
-def _worker_command(app: str):
+def worker_command(app: str, concurrency: t.Optional[int] = Defaults.concurrency):
+    """Command to start workers."""
     loop = asyncio.get_event_loop()
-    loop.run_until_complete(worker(app_import_path=app))
+    loop.run_until_complete(worker(app_import_path=app, concurrency=concurrency))
 
 
 @cli.command(name="metric")
-def _metric_server(app: str):
+def metric_server(app: str):
+    """Command to start server to collect and report tasks metrics (TODO)."""
     print(f"TODO: Running metrics server for app={app}")
 
 

src/aiotaskq/constants.py

@@ -1,7 +1,5 @@
+"""Module to define and store all constants used across the library."""
+
 REDIS_URL = "redis://127.0.0.1:6379"
 TASKS_CHANNEL = "channel:tasks"
 RESULTS_CHANNEL_TEMPLATE = "channel:results:{task_id}"
-
-# REDIS_URL = "redis://127.0.0.1:6379"
-# TASKS_CHANNEL = "channel:tasks"
-# RESULTS_CHANNEL_TEMPLATE = "channel:results:{task_id}"

src/aiotaskq/exceptions.py

@@ -1,2 +1,9 @@
-class WorkerNotReady(Exception):
-    """Attempt to send task to worker but no worker is subscribing to tasks channel."""
+"""
+Define all exceptions that are possibly raised by the package.
+
+Any raised thrown must be defined here.
+"""
+
+
+class ModuleInvalidForTask(Exception):
+    """Attempt to convert to task a function in an invalid module."""