Merge pull request #24 from supabase/or/citus

Citus extension

Author: olirice

ext/citus.nix

@@ -0,0 +1,35 @@
+{ lib, curl, lz4, zstd, krb5, icu, stdenv, fetchFromGitHub, postgresql }:
+
+stdenv.mkDerivation rec {
+  pname = "citus";
+  version = "12.0.0";
+
+  buildInputs = [ curl lz4 zstd krb5 icu.dev postgresql];
+
+  src = fetchFromGitHub {
+    owner  = "citusdata";
+    repo   = pname;
+    rev    = "refs/tags/v${version}";
+    hash   = "sha256-HH9/slsCRe2yIVIqwR8sDyqXFonf8BHhJhLzHNv1CF0=";
+  };
+
+  installPhase = ''
+    mkdir -p $out/{lib,share/postgresql/extension}
+
+    cp src/backend/columnar/citus_columnar.so      $out/lib
+    cp src/backend/columnar/citus_columnar.control $out/share/postgresql/extension
+    cp src/backend/columnar/build/sql/*.sql        $out/share/postgresql/extension
+
+    cp src/backend/distributed/citus.so            $out/lib
+    cp src/backend/distributed/citus.control       $out/share/postgresql/extension
+    cp src/backend/distributed/build/sql/*.sql     $out/share/postgresql/extension
+  '';
+
+  meta = with lib; {
+    description = "Distributed PostgreSQL as an extension";
+    homepage    = "https://github.com/citusdata/${pname}";
+    maintainers = with maintainers; [ olirice ];
+    platforms   = postgresql.meta.platforms;
+    license     = licenses.agpl3Plus;
+  };
+}

flake.nix

@@ -87,6 +87,7 @@
         # rollout new versions of these critical things easier without having to
         # go through the upstream release engineering process.
         ourExtensions = [
+          ./ext/citus.nix
           ./ext/pgsql-http.nix
           ./ext/pg_plan_filter.nix
           ./ext/pg_net.nix

tests/postgresql.conf.in

@@ -717,7 +717,7 @@ default_text_search_config = 'pg_catalog.english'
 
 #local_preload_libraries = ''
 #session_preload_libraries = ''
-shared_preload_libraries = 'auto_explain,pgsodium'
+shared_preload_libraries = 'citus,auto_explain,pgsodium'
 #jit_provider = 'llvmjit'		# JIT library to use
 
 # - Other Defaults -

tests/prime.sql

@@ -18,3 +18,5 @@ CREATE EXTENSION IF NOT EXISTS wrappers;
 CREATE EXTENSION IF NOT EXISTS http;
 CREATE EXTENSION IF NOT EXISTS pg_graphql;
 CREATE EXTENSION IF NOT EXISTS pg_jsonschema;
+
+CREATE EXTENSION IF NOT EXISTS citus;

tests/smoke/0004-citus.sql

@@ -0,0 +1,32 @@
+BEGIN;
+
+select plan(3);
+
+SELECT lives_ok($$
+  CREATE TABLE events (
+    device_id bigint,
+    event_id bigserial,
+    event_time timestamptz default now(),
+    data jsonb not null,
+    PRIMARY KEY (device_id, event_id)
+  );
+$$);
+
+-- citus distributed test
+SELECT lives_ok($$
+  SELECT create_distributed_table('events', 'device_id');
+$$);
+
+-- citus columnar test
+SELECT lives_ok($$
+  CREATE TABLE events_columnar (
+    device_id bigint,
+    event_id bigserial,
+    event_time timestamptz default now(),
+    data jsonb not null
+  )
+  USING columnar;
+$$);
+
+SELECT * FROM finish();
+ROLLBACK;