feat: exposes additional worker configuration parameters

pyproject.toml

@@ -74,13 +74,14 @@ redis = { version = "*", optional = true }
 saq = { version = "^0.9.1", optional = true }
 sentry-sdk = { version = "*", optional = true }
 sqlalchemy = { version = "==2.0.0rc3", optional = true }
+greenlet = {version = "*", platform = "darwin", optional = true }
 
 [tool.poetry.extras]
 cache = ["redis", "hiredis"]
 worker = ["saq", "hiredis"]
 sentry = ["sentry-sdk"]
-sqlalchemy = ["sqlalchemy"]
-all = ["redis", "hiredis", "saq", "sentry-sdk", "sqlalchemy"]
+sqlalchemy = ["sqlalchemy","greenlet"]
+all = ["redis", "hiredis", "saq", "sentry-sdk", "sqlalchemy","greenlet"]
 
 [tool.poetry.plugins."pytest11"]
 pytest_starlite_saqlalchemy = "pytest_starlite_saqlalchemy"

src/starlite_saqlalchemy/settings.py

@@ -277,6 +277,11 @@ class Config:
         env_file = ".env"
         env_prefix = "WORKER_"
 
+    CONCURRENCY: int = 10
+    """The number of concurrent jobs allowed to execute per worker..
+
+    Default is set to 10.
+    """
     JOB_TIMEOUT: int = 10
     """Max time a job can run for, in seconds.
 

src/starlite_saqlalchemy/worker.py

@@ -23,6 +23,7 @@
     from starlite_saqlalchemy.service import Service
 
 __all__ = [
+    "CronJob",
     "JobConfig",
     "Queue",
     "Worker",
@@ -138,25 +139,48 @@ class JobConfig:
     """
 
 
+class CronJob(saq.CronJob):
+    """Cron Job."""
+
+
 default_job_config_dict = utils.dataclass_as_dict_shallow(JobConfig(), exclude_none=True)
 
 
 def create_worker_instance(
     functions: Collection[Callable[..., Any] | tuple[str, Callable]],
+    cron_jobs: Collection[saq.CronJob] = (),
+    concurrency: int | None = None,
+    startup: Callable[[dict[str, Any]], Awaitable[Any]] | None = None,
+    shutdown: Callable[[dict[str, Any]], Awaitable[Any]] | None = None,
     before_process: Callable[[dict[str, Any]], Awaitable[Any]] | None = None,
     after_process: Callable[[dict[str, Any]], Awaitable[Any]] | None = None,
 ) -> Worker:
-    """
+    """Create a worker instance.
 
     Args:
         functions: Functions to be called via the async workers.
+        cron_jobs: Cron configuration to schedule at startup.
+        concurrency: The number of jobs allowed to execute simultaneously per worker.
+        startup: Async function called on worker startup.
+        shutdown: Async function called on worker shutdown.
         before_process: Async function called before a job processes.
         after_process: Async function called after a job processes.
 
     Returns:
         The worker instance, instantiated with `functions`.
     """
-    return Worker(queue, functions, before_process=before_process, after_process=after_process)
+    if concurrency is None:
+        concurrency = settings.worker.CONCURRENCY
+    return Worker(
+        queue,
+        functions=functions,
+        cron_jobs=cron_jobs,
+        startup=startup,
+        concurrency=concurrency,
+        shutdown=shutdown,
+        before_process=before_process,
+        after_process=after_process,
+    )
 
 
 async def make_service_callback(