1 files changed, 114 insertions, 0 deletions
diff --git a/lib/sql/clone_schema.sql b/lib/sql/clone_schema.sql
new file mode 100644
index 000000000..9fb88466b
--- /dev/null
+++ b/lib/sql/clone_schema.sql
@@ -0,0 +1,114 @@
+CREATE OR REPLACE FUNCTION clone_schema( source_schema text, dest_schema text, include_recs boolean) RETURNS void AS
+$BODY$
+  DECLARE
+    src_oid oid;
+    tbl_oid oid;
+    func_oid oid;
+    object text;
+    buffer text;
+    srctbl text;
+    default_ text;
+    column_ text;
+    qry text;
+    dest_qry text;
+    v_def text;
+    seqval bigint;
+    sq_last_value bigint;
+    sq_max_value bigint;
+    sq_start_value bigint;
+    sq_increment_by bigint;
+    sq_min_value bigint;
+    sq_cache_value bigint;
+    sq_log_cnt bigint;
+    sq_is_called boolean;
+    sq_is_cycled boolean;
+    sq_cycled char(10);
+  BEGIN
+    -- Assure that source_schema exists
+    SELECT oid INTO src_oid FROM pg_namespace WHERE nspname = quote_ident(source_schema);
+    IF NOT FOUND THEN
+      RAISE NOTICE 'source schema % does not exist!', source_schema;
+      RETURN;
+    END IF;
+
+    -- Refute that dest_schema exists and then create it
+    PERFORM nspname FROM pg_namespace WHERE nspname = quote_ident(dest_schema);
+    IF FOUND THEN
+      RAISE NOTICE 'dest schema % already exists!', dest_schema;
+      RETURN;
+    END IF;
+    EXECUTE 'CREATE SCHEMA ' || quote_ident(dest_schema) ;
+
+    -- loop over sequences, creating the same one in the destination namespace, ...
+    FOR object IN
+      SELECT sequence_name::text FROM information_schema.sequences WHERE sequence_schema = quote_ident(source_schema)
+    LOOP
+      EXECUTE 'CREATE SEQUENCE ' || quote_ident(dest_schema) || '.' || quote_ident(object);
+
+      -- ...storing the attributes of the sequence from the source namespace in local variables, ...
+      srctbl := quote_ident(source_schema) || '.' || quote_ident(object);
+      EXECUTE 'SELECT last_value, max_value, start_value, increment_by, min_value, cache_value, log_cnt, is_cycled, is_called FROM ' || srctbl || ';' INTO sq_last_value, sq_max_value, sq_start_value, sq_increment_by, sq_min_value, sq_cache_value, sq_log_cnt, sq_is_cycled, sq_is_called;
+      IF sq_is_cycled THEN
+        sq_cycled := 'CYCLE';
+      ELSE
+        sq_cycled := 'NO CYCLE';
+      END IF;
+      EXECUTE 'ALTER SEQUENCE ' || (dest_schema) || '.' || quote_ident(object) || ' INCREMENT BY ' || sq_increment_by || ' MINVALUE ' || sq_min_value || ' MAXVALUE ' || sq_max_value || ' START WITH ' || sq_start_value || ' RESTART ' || sq_min_value || ' CACHE ' || sq_cache_value || sq_cycled || ' ;' ;
+
+      buffer := quote_ident(dest_schema) || '.' || quote_ident(object);
+      IF include_recs THEN
+        EXECUTE 'SELECT setval( ''' || buffer || ''', ' || sq_last_value || ', ' || sq_is_called || ');' ;
+      ELSE
+        EXECUTE 'SELECT setval( ''' || buffer || ''', ' || sq_start_value || ', ' || sq_is_called || ');' ;
+      END IF;
+    END LOOP;
+
+    -- loop over tables in source schema,...
+    FOR object IN
+      SELECT TABLE_NAME::text FROM information_schema.tables WHERE table_schema = quote_ident(source_schema) AND table_type = 'BASE TABLE'
+    LOOP
+      -- ...creating the table in the destination schema, potentially including the records
+      buffer := dest_schema || '.' || quote_ident(object);
+      EXECUTE 'CREATE TABLE ' || buffer || '(LIKE ' || quote_ident(source_schema) || '.' || quote_ident(object) || ' INCLUDING ALL)';
+      IF include_recs THEN
+        EXECUTE 'INSERT INTO ' || buffer || ' SELECT * FROM ' || quote_ident(source_schema) || '.' || quote_ident(object) || ';';
+      END IF;
+
+      -- alter table, assuring the destination schema's table has:
+      --    * the same defaults
+      FOR column_, default_
+        IN SELECT column_name::text, REPLACE(column_default::text, source_schema, dest_schema)
+        FROM information_schema.COLUMNS
+        WHERE table_schema = dest_schema AND TABLE_NAME = object AND column_default LIKE 'nextval(%' || quote_ident(source_schema) || '%::regclass)'
+      LOOP
+        EXECUTE 'ALTER TABLE ' || buffer || ' ALTER COLUMN ' || column_ || ' SET DEFAULT ' || default_;
+      END LOOP;
+    END LOOP;
+
+      /* SELECT 'ALTER TABLE ' || quote_ident(dest_schema) || '.' || quote_ident(rn.relname) || ' ADD CONSTRAINT ' || quote_ident(ct.conname) || ' ' || pg_get_constraintdef(ct.oid) || ';' FROM pg_constraint ct JOIN pg_class rn ON rn.oid = ct.conrelid WHERE connamespace = src_oid AND rn.relkind = 'r' AND ct.contype = 'f' LIMIT 2; */
+    -- apply all constraints on tables in destination schema
+    FOR qry IN
+      SELECT 'ALTER TABLE ' || quote_ident(dest_schema) || '.' || quote_ident(rn.relname) || ' ADD CONSTRAINT ' || quote_ident(ct.conname) || ' ' || REPLACE(pg_get_constraintdef(ct.oid), source_schema || '.', dest_schema || '.') || ';'
+      FROM pg_constraint ct JOIN pg_class rn ON rn.oid = ct.conrelid WHERE connamespace = src_oid AND rn.relkind = 'r' AND ct.contype = 'f'
+    LOOP
+      EXECUTE qry;
+    END LOOP;
+
+    -- create views from source schema in destination schema
+    FOR object IN
+      SELECT table_name::text, view_definition FROM information_schema.views WHERE table_schema = quote_ident(source_schema)
+    LOOP
+      buffer := dest_schema || '.' || quote_ident(object);
+      SELECT view_definition INTO v_def FROM information_schema.views WHERE table_schema = quote_ident(source_schema) AND table_name = quote_ident(object);
+      EXECUTE 'CREATE OR REPLACE VIEW ' || buffer || ' AS ' || v_def || ';' ;
+    END LOOP;
+
+    FOR func_oid IN SELECT oid FROM pg_proc WHERE pronamespace = src_oid
+    LOOP
+      SELECT pg_get_functiondef(func_oid) INTO qry;
+      SELECT replace(qry, source_schema, dest_schema) INTO dest_qry;
+      EXECUTE dest_qry;
+    END LOOP;
+    RETURN;
+  END;
+$BODY$ LANGUAGE plpgsql VOLATILE COST 100;