Detect CitusDB by checking for its catalogs

More obvious than checking for some GUC.
citusdata · May 23, 2015 · 77d0839 · 77d0839
1 parent 42749a6
commit 77d0839
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/pg_shard--1.2.sql b/pg_shard--1.2.sql
@@ -11,14 +11,13 @@ LANGUAGE C STABLE STRICT;
 
 DO $$
 DECLARE
-	replication_factor integer;
 	use_citus_metadata boolean := false;
 BEGIN
 	BEGIN
-		replication_factor = current_setting('shard_replication_factor')::integer;
-		use_citus_metadata = (replication_factor IS NOT NULL);
+		PERFORM 'pg_catalog.pg_dist_partition'::regclass;
+		use_citus_metadata = true;
 	EXCEPTION
-		WHEN undefined_object THEN
+		WHEN undefined_table THEN
 			use_citus_metadata = false;
 	END;