add Pgmq class

Author: Muhammad-Magdi

.vscode/settings.json

@@ -0,0 +1,3 @@
+{
+  "cSpell.words": ["Pgmq"]
+}

index.ts

@@ -4,16 +4,18 @@
   "description": "",
   "main": "index.js",
   "scripts": {
-    "start": "ts-node index.ts",
-    "test": "echo \"Error: no test specified\" && exit 1"
+    "start": "ts-node src/index.ts"
   },
   "keywords": [],
   "author": "",
   "license": "ISC",
   "devDependencies": {
+    "@types/pg-pool": "^2.0.6",
     "typescript": "^5.3.3"
   },
   "dependencies": {
+    "pg": "^8.11.3",
+    "pg-pool": "^3.6.1",
     "ts-node": "^10.9.2"
   }
 }

package-lock.json

@@ -0,0 +1,5 @@
+export const DEFAULT_MAX_POOL_SIZE = 20;
+export const DEFAULT_IDLE_TIMEOUT_MILLIS = 1000;
+export const DEFAULT_CONNECTION_TIMEOUT_MILLIS = 1000;
+export const DEFAULT_MAX_USES = 7500;
+export const DEFAULT_SSL = true;

package.json

@@ -0,0 +1,35 @@
+import { Client } from "pg";
+import Pool from "pg-pool";
+import { PgPoolConfig } from "./types";
+import {
+  DEFAULT_CONNECTION_TIMEOUT_MILLIS,
+  DEFAULT_IDLE_TIMEOUT_MILLIS,
+  DEFAULT_MAX_POOL_SIZE,
+  DEFAULT_MAX_USES,
+  DEFAULT_SSL,
+} from "./constants";
+
+export class Pgmq {
+  private constructor(private readonly pool: Pool<Client>) {
+  }
+
+  public static async new(c: PgPoolConfig) {
+    if (c.max === undefined) c.max = DEFAULT_MAX_POOL_SIZE;
+    if (c.idleTimeoutMillis === undefined) c.max = DEFAULT_IDLE_TIMEOUT_MILLIS;
+    if (c.connectionTimeoutMillis === undefined)
+      c.connectionTimeoutMillis = DEFAULT_CONNECTION_TIMEOUT_MILLIS;
+    if (c.maxUses === undefined) c.maxUses = DEFAULT_MAX_USES;
+    if (c.ssl === undefined) c.ssl = DEFAULT_SSL;
+
+    const pool = new Pool(c);
+
+    await pool.connect();
+    const pgmq = new Pgmq(pool);
+    await pgmq.prepare();
+    return pgmq;
+  }
+
+  private async prepare() {
+    await this.pool.query("CREATE EXTENSION IF NOT EXISTS pgmq CASCADE");
+  }
+}

src/constants.ts

@@ -0,0 +1,12 @@
+export type PgPoolConfig = {
+  host: string;
+  database: string;
+  user: string;
+  password: string;
+  port: number;
+  ssl?: boolean;
+  max?: number;
+  idleTimeoutMillis?: number;
+  connectionTimeoutMillis?: number;
+  maxUses?: number;
+};

src/index.ts

@@ -26,7 +26,7 @@
 
     /* Modules */
     "module": "commonjs" /* Specify what module code is generated. */,
-    // "rootDir": "./",                                  /* Specify the root folder within your source files. */
+    "rootDir": "./src" /* Specify the root folder within your source files. */,
     // "moduleResolution": "node10",                     /* Specify how TypeScript looks up a file from a given module specifier. */
     "baseUrl": "./" /* Specify the base directory to resolve non-relative module names. */,
     // "paths": {},                                      /* Specify a set of entries that re-map imports to additional lookup locations. */