Skip to content

feat: add schedule source for psqlpy and aiopg#18

Merged
danfimov merged 1 commit intomainfrom
add-schedule-source
Oct 10, 2025
Merged

feat: add schedule source for psqlpy and aiopg#18
danfimov merged 1 commit intomainfrom
add-schedule-source

Conversation

@danfimov
Copy link
Owner

closes #16 and #15

@danfimov danfimov self-assigned this Oct 10, 2025
Copilot AI review requested due to automatic review settings October 10, 2025 08:43
Copy link

Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull Request Overview

This PR adds schedule source functionality for PostgreSQL-based TaskIQ brokers, supporting both psqlpy and aiopg database adapters alongside the existing asyncpg implementation. The implementation allows for persistent storage and retrieval of scheduled tasks from PostgreSQL databases.

  • Implements PSQLPyScheduleSource and AiopgScheduleSource classes with full CRUD operations for task schedules
  • Refactors existing AsyncpgScheduleSource to inherit from a new shared base class BasePostgresScheduleSource
  • Adds comprehensive integration tests covering all three database adapters

Reviewed Changes

Copilot reviewed 11 out of 12 changed files in this pull request and generated 6 comments.

Show a summary per file
File Description
tests/integration/test_schedule_source.py Integration tests for all three schedule source implementations
src/taskiq_pg/psqlpy/schedule_source.py PSQLPy-based schedule source implementation
src/taskiq_pg/psqlpy/queries.py SQL queries for PSQLPy schedule operations
src/taskiq_pg/psqlpy/init.py Export PSQLPyScheduleSource in public API
src/taskiq_pg/asyncpg/schedule_source.py Refactored to use shared base class
src/taskiq_pg/aiopg/schedule_source.py Aiopg-based schedule source implementation
src/taskiq_pg/aiopg/queries.py SQL queries for aiopg schedule operations
src/taskiq_pg/aiopg/init.py Export AiopgScheduleSource in public API
src/taskiq_pg/_internal/schedule_source.py Shared base class for PostgreSQL schedule sources
src/taskiq_pg/_internal/init.py Export shared base class
pyproject.toml Linting configuration updates

Tip: Customize your code reviews with copilot-instructions.md. Create the file or learn how to get started.

@danfimov danfimov force-pushed the add-schedule-source branch from f5a7c49 to fc3c45e Compare October 10, 2025 08:46
@danfimov danfimov merged commit e1bf60f into main Oct 10, 2025
4 checks passed
@danfimov danfimov deleted the add-schedule-source branch October 10, 2025 08:48
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Add schedule source for aiopg driver

2 participants