Add option for custom shared libs

citusdata · Oct 3, 2022 · 29c99c9 · 29c99c9
1 parent 8dadd96
commit 29c99c9
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/citus_dev/citus_dev b/citus_dev/citus_dev
@@ -2,7 +2,7 @@
 """citus_dev
 
 Usage:
-  citus_dev make <name> [--size=<count>] [--port=<port>] [--use-ssl] [--no-extension] [--mx] [--destroy] [--init-with=<sql_file>] [--init-worker-with=<sql_file>] [--with-pgbouncer]
+  citus_dev make <name> [--size=<count>] [--port=<port>] [--use-ssl] [--no-extension] [--mx] [--destroy] [--init-with=<sql_file>] [--init-worker-with=<sql_file>] [--with-pgbouncer] [--with-shared-preloaded-libs=<lib1,lib2...>]
   citus_dev restart <name> [--watch] [--port=<port>]
   citus_dev (start|stop) <name> [--port=<port>] [--force]
 
@@ -18,6 +18,7 @@ Options:
   --init-worker-with=<sql_file>   A SQL script to run after creation of the cluster on the workers
   --with-pgbouncer         Setup pgbouncers between worker and coordinator (requires citus enterprise)
   --force                  Forceful shutdown
+  --with-shared-preloaded-libs=<lib1,lib2...>   Comma separated list of extra shared preload libraries to be added to postgresql.conf of each node
 
 """
 from docopt import docopt
@@ -45,14 +46,17 @@ def run(command, *args, **kwargs):
     return result
 
 
-def createNodeCommands(clustername, role, index=None, usessl=False, mx=False):
+def createNodeCommands(clustername, role, index=None, usessl=False, mx=False, extra_preloaded_libraries=None):
     nodename = role
     if index != None:
         nodename += "%d" % index
 
     run(f"initdb -D {clustername}/{nodename}")
 
     shared_preload_libraries = ['citus', 'pg_stat_statements']
+
+    if extra_preloaded_libraries:
+        shared_preload_libraries.extend(extra_preloaded_libraries)
 
     shared_preload_libraries = ','.join(shared_preload_libraries)
 
@@ -110,13 +114,17 @@ def main(arguments):
         if arguments['--destroy']:
             stopCluster(clustername, True)
             run(f'rm -rf {clustername}')
-
+
+        extra_preloaded_libraries = []
+        if arguments["--with-shared-preloaded-libs"]:
+            extra_preloaded_libraries = arguments["--with-shared-preloaded-libs"]
 
         createNodeCommands(
             clustername,
             "coordinator",
             usessl=arguments["--use-ssl"],
             mx=arguments["--mx"],
+            extra_preloaded_libraries=extra_preloaded_libraries
         )
 
         port = int(arguments["--port"])
@@ -133,6 +141,7 @@ def main(arguments):
                 i,
                 usessl=arguments["--use-ssl"],
                 mx=arguments["--mx"],
+                extra_preloaded_libraries=extra_preloaded_libraries
             )
             if pgbouncer:
                 createPgBouncerConfig(clustername, port, i)