Adding queue name and change the database schema

Sources/JobsPostgres/Migrations/CreateSwiftJobsMigrations.swift

@@ -0,0 +1,89 @@
+//===----------------------------------------------------------------------===//
+//
+// This source file is part of the Hummingbird server framework project
+//
+// Copyright (c) 2024 the Hummingbird authors
+// Licensed under Apache License v2.0
+//
+// See LICENSE.txt for license information
+// See hummingbird/CONTRIBUTORS.txt for the list of Hummingbird authors
+//
+// SPDX-License-Identifier: Apache-2.0
+//
+//===----------------------------------------------------------------------===//
+
+import Logging
+import PostgresMigrations
+import PostgresNIO
+
+struct CreateSwiftJobsMigrations: DatabaseMigration {
+
+    func apply(connection: PostgresNIO.PostgresConnection, logger: Logging.Logger) async throws {
+
+        try await connection.query("CREATE SCHEMA IF NOT EXISTS swift_jobs;", logger: logger)
+
+        try await connection.query(
+            """
+            CREATE TABLE IF NOT EXISTS swift_jobs.jobs (
+                id uuid PRIMARY KEY,
+                job bytea NOT NULL,
+                last_modified TIMESTAMPTZ NOT NULL DEFAULT now(),
+                queue_name TEXT NOT NULL DEFAULT 'default',
+                status smallint NOT NULL
+            );
+            """,
+            logger: logger
+        )
+
+        try await connection.query(
+            """
+            CREATE TABLE IF NOT EXISTS swift_jobs.queues(
+                job_id uuid PRIMARY KEY,
+                created_at TIMESTAMPTZ NOT NULL,
+                delayed_until TIMESTAMPTZ NOT NULL DEFAULT now(),
+                queue_name TEXT NOT NULL DEFAULT 'default'
+            );
+            """,
+            logger: logger
+        )
+
+        try await connection.query(
+            """
+            CREATE INDEX CONCURRENTLY IF NOT EXISTS queues_delayed_until_queue_name_idx 
+            ON swift_jobs.queues(delayed_until, queue_name)
+            """,
+            logger: logger
+        )
+
+        try await connection.query(
+            """
+            CREATE TABLE IF NOT EXISTS swift_jobs.queues_metadata(
+                key text PRIMARY KEY,
+                value bytea NOT NULL,
+                queue_name TEXT NOT NULL DEFAULT 'default'
+            )
+            """,
+            logger: logger
+        )
+
+        try await connection.query(
+            """
+            CREATE INDEX CONCURRENTLY IF NOT EXISTS queues_metadata_key_queue_name_idx
+            ON swift_jobs.queues_metadata(key, queue_name)
+            """,
+            logger: logger
+        )
+    }
+
+    func revert(connection: PostgresNIO.PostgresConnection, logger: Logging.Logger) async throws {
+        try await connection.query(
+            """
+            DROP SCHEMA swift_jobs CASCADE;
+            """,
+            logger: logger
+        )
+    }
+
+    var description: String { "__CreateSwiftJobsMigrations__" }
+    var group: DatabaseMigrationGroup { .jobQueue }
+}

Sources/JobsPostgres/Migrations/Database+Migrations.swift

@@ -0,0 +1,20 @@
+//===----------------------------------------------------------------------===//
+//
+// This source file is part of the Hummingbird server framework project
+//
+// Copyright (c) 2024 the Hummingbird authors
+// Licensed under Apache License v2.0
+//
+// See LICENSE.txt for license information
+// See hummingbird/CONTRIBUTORS.txt for the list of Hummingbird authors
+//
+// SPDX-License-Identifier: Apache-2.0
+//
+//===----------------------------------------------------------------------===//
+
+import PostgresMigrations
+
+extension DatabaseMigrationGroup {
+    /// JobQueue migration group
+    public static var jobQueue: Self { .init("_hb_jobqueue") }
+}