feat: support multiple versions of the pgmq extension

ansible/files/postgresql_extension_custom_scripts/pgmq/after-create.sql

@@ -18,8 +18,8 @@ begin
     physical backups everywhere
 */
 -- Detach and delete the official function
-alter extension pgmq drop function pgmq.drop_queue;
-drop function pgmq.drop_queue;
+alter extension pgmq drop function pgmq.drop_queue(TEXT);
+drop function pgmq.drop_queue(TEXT);
 
 -- Create and reattach the patched function
 CREATE FUNCTION pgmq.drop_queue(queue_name TEXT)
@@ -134,7 +134,7 @@ BEGIN
 END;
 $func$ LANGUAGE plpgsql;
 
-alter extension pgmq add function pgmq.drop_queue;
+alter extension pgmq add function pgmq.drop_queue(TEXT);
 
 
   update pg_extension set extowner = 'postgres'::regrole where extname = 'pgmq';

ansible/vars.yml

@@ -10,9 +10,9 @@ postgres_major:
 
 # Full version strings for each major version
 postgres_release:
-  postgresorioledb-17: 17.5.1.036-orioledb
-  postgres17: 17.6.1.015
-  postgres15: 15.14.1.015
+  postgresorioledb-17: 17.5.1.037-orioledb
+  postgres17: 17.6.1.016
+  postgres15: 15.14.1.016
 
 # Non Postgres Extensions
 pgbouncer_release: 1.19.0

nix/ext/pgmq.nix

@@ -3,36 +3,88 @@
   stdenv,
   fetchFromGitHub,
   postgresql,
+  buildEnv,
 }:
-
-stdenv.mkDerivation rec {
+let
   pname = "pgmq";
-  version = "1.4.4";
-  buildInputs = [ postgresql ];
-  src = fetchFromGitHub {
-    owner = "tembo-io";
-    repo = pname;
-    rev = "v${version}";
-    hash = "sha256-z+8/BqIlHwlMnuIzMz6eylmYbSmhtsNt7TJf/CxbdVw=";
-  };
 
-  buildPhase = ''
-    cd pgmq-extension
-  '';
+  # Load version configuration from external file
+  allVersions = (builtins.fromJSON (builtins.readFile ./versions.json)).${pname};
+
+  # Filter versions compatible with current PostgreSQL version
+  supportedVersions = lib.filterAttrs (
+    _: value: builtins.elem (lib.versions.major postgresql.version) value.postgresql
+  ) allVersions;
+
+  # Derived version information
+  versions = lib.naturalSort (lib.attrNames supportedVersions);
+  latestVersion = lib.last versions;
+  numberOfVersions = builtins.length versions;
+  packages = builtins.attrValues (
+    lib.mapAttrs (name: value: build name value.hash) supportedVersions
+  );
+
+  # Build function for individual versions
+  build =
+    version: hash:
+    stdenv.mkDerivation rec {
+      inherit pname version;
+      buildInputs = [ postgresql ];
+      src = fetchFromGitHub {
+        owner = "tembo-io";
+        repo = pname;
+        rev = "v${version}";
+        inherit hash;
+      };
+
+      buildPhase = ''
+        cd pgmq-extension
+      '';
+
+      installPhase = ''
+        runHook preInstall
+
+        mkdir -p $out/share/postgresql/extension
+
+        # Create versioned sql install script
+        cp sql/${pname}.sql $out/share/postgresql/extension/${pname}--${version}.sql
+
+        # Create versioned control file with modified module path
+        sed -e "/^default_version =/d" \
+            -e "s|^module_pathname = .*|module_pathname = '\$libdir/${pname}'|" \
+          ${pname}.control > $out/share/postgresql/extension/${pname}--${version}.control
+
+        # For the latest version, create default control file and symlink and copy SQL upgrade scripts
+        if [[ "${version}" == "${latestVersion}" ]]; then
+          {
+            echo "default_version = '${version}'"
+            cat $out/share/postgresql/extension/${pname}--${version}.control
+          } > $out/share/postgresql/extension/${pname}.control
+          cp sql/*.sql $out/share/postgresql/extension
+        fi
+
+        runHook postInstall
+      '';
 
-  installPhase = ''
-    mkdir -p $out/{lib,share/postgresql/extension}
+      meta = with lib; {
+        description = "A lightweight message queue. Like AWS SQS and RSMQ but on Postgres.";
+        homepage = "https://github.com/tembo-io/pgmq";
+        maintainers = with maintainers; [ olirice ];
+        inherit (postgresql.meta) platforms;
+        license = licenses.postgresql;
+      };
+    };
+in
+buildEnv {
+  name = pname;
+  paths = packages;
 
-    mv sql/pgmq.sql $out/share/postgresql/extension/pgmq--${version}.sql
-    cp sql/*.sql $out/share/postgresql/extension
-    cp *.control $out/share/postgresql/extension
-  '';
+  pathsToLink = [ "/share/postgresql/extension" ];
 
-  meta = with lib; {
-    description = "A lightweight message queue. Like AWS SQS and RSMQ but on Postgres.";
-    homepage = "https://github.com/tembo-io/pgmq";
-    maintainers = with maintainers; [ olirice ];
-    platforms = postgresql.meta.platforms;
-    license = licenses.postgresql;
+  passthru = {
+    inherit versions numberOfVersions;
+    pname = "${pname}-all";
+    version =
+      "multi-" + lib.concatStringsSep "-" (map (v: lib.replaceStrings [ "." ] [ "-" ] v) versions);
   };
 }

nix/ext/tests/default.nix

@@ -187,6 +187,7 @@ builtins.listToAttrs (
       "pg_cron"
       "pg_graphql"
       "pg_net"
+      "pgmq"
       "vector"
       "wrappers"
     ]

nix/ext/versions.json

@@ -277,6 +277,21 @@
       ],
       "hash": "sha256-Cpi2iASi1QJoED0Qs1dANqg/BNZTsz5S+pw8iYyW03Y="
     }
+  },
+    "pgmq": {
+    "1.4.4": {
+      "postgresql": [
+        "15"
+      ],
+      "hash": "sha256-z+8/BqIlHwlMnuIzMz6eylmYbSmhtsNt7TJf/CxbdVw="
+    },
+    "1.5.1": {
+      "postgresql": [
+        "15",
+        "17"
+      ],
+      "hash": "sha256-IU+i6ONPwtgsFKdzya6E+222ualR66gkbb0lDr+7Rb8="
+    }
   },
   "pgrouting": {
     "3.4.1": {
@@ -412,7 +427,7 @@
       "hash": "sha256-JsZV+I4eRMypXTjGmjCtMBXDVpqTIPHQa28ogXncE/Q="
     }
   },
-    "wrappers": {
+  "wrappers": {
     "0.5.4": {
       "postgresql": [
         "15",

nix/tests/expected/pgmq.out

@@ -160,6 +160,7 @@ order by
 -------------+-------------------------------+----------
  pgmq        | _belongs_to_pgmq              | postgres
  pgmq        | _ensure_pg_partman_installed  | postgres
+ pgmq        | _extension_exists             | postgres
  pgmq        | _get_partition_col            | postgres
  pgmq        | _get_pg_partman_major_version | postgres
  pgmq        | _get_pg_partman_schema        | postgres
@@ -174,6 +175,7 @@ order by
  pgmq        | delete                        | postgres
  pgmq        | detach_archive                | postgres
  pgmq        | drop_queue                    | postgres
+ pgmq        | drop_queue                    | postgres
  pgmq        | format_table_name             | postgres
  pgmq        | list_queues                   | postgres
  pgmq        | metrics                       | postgres
@@ -183,8 +185,18 @@ order by
  pgmq        | read                          | postgres
  pgmq        | read_with_poll                | postgres
  pgmq        | send                          | postgres
+ pgmq        | send                          | postgres
+ pgmq        | send                          | postgres
+ pgmq        | send                          | postgres
+ pgmq        | send                          | postgres
+ pgmq        | send                          | postgres
+ pgmq        | send_batch                    | postgres
+ pgmq        | send_batch                    | postgres
+ pgmq        | send_batch                    | postgres
+ pgmq        | send_batch                    | postgres
+ pgmq        | send_batch                    | postgres
  pgmq        | send_batch                    | postgres
  pgmq        | set_vt                        | postgres
  pgmq        | validate_queue_name           | postgres
-(28 rows)
+(40 rows)