sql/base/base.sql

   1 --------------------------------------------------------------------------
   2 -- An example of how to use:
   3 --
   4 -- DROP SCHEMA foo CASCADE; CREATE SCHEMA foo;
   5 -- \i base.sql
   6 -- SELECT migration_tools.init('foo');
   7 -- SELECT migration_tools.build('foo');
   8 -- SELECT * FROM foo.fields_requiring_mapping;
   9 -- \d foo.actor_usr
  10 -- create some incoming ILS specific staging tables, like CREATE foo.legacy_items ( l_barcode TEXT, .. ) INHERITS foo.asset_copy;
  11 -- Do some mapping, like UPDATE foo.legacy_items SET barcode = TRIM(BOTH ' ' FROM l_barcode);
  12 -- Then, to move into production, do: select migration_tools.insert_base_into_production('foo')
  13
  14 CREATE SCHEMA migration_tools;
  15
  16 CREATE OR REPLACE FUNCTION migration_tools.production_tables (TEXT) RETURNS TEXT[] AS $$
  17     DECLARE
  18         migration_schema ALIAS FOR $1;
  19         output  RECORD;
  20     BEGIN
  21         FOR output IN
  22             EXECUTE 'SELECT string_to_array(value,'','') AS tables FROM ' || migration_schema || '.config WHERE key = ''production_tables'';'
  23         LOOP
  24             RETURN output.tables;
  25         END LOOP;
  26     END;
  27 $$ LANGUAGE PLPGSQL STRICT STABLE;
  28
  29 CREATE OR REPLACE FUNCTION migration_tools.country_code (TEXT) RETURNS TEXT AS $$
  30     DECLARE
  31         migration_schema ALIAS FOR $1;
  32         output TEXT;
  33     BEGIN
  34         FOR output IN
  35             EXECUTE 'SELECT value FROM ' || migration_schema || '.config WHERE key = ''country_code'';'
  36         LOOP
  37             RETURN output;
  38         END LOOP;
  39     END;
  40 $$ LANGUAGE PLPGSQL STRICT STABLE;
  41
  42
  43 CREATE OR REPLACE FUNCTION migration_tools.log (TEXT,TEXT,INTEGER) RETURNS VOID AS $$
  44     DECLARE
  45         migration_schema ALIAS FOR $1;
  46         sql ALIAS FOR $2;
  47         nrows ALIAS FOR $3;
  48     BEGIN
  49         EXECUTE 'INSERT INTO ' || migration_schema || '.sql_log ( sql, row_count ) VALUES ( ' || quote_literal(sql) || ', ' || nrows || ' );';
  50     END;
  51 $$ LANGUAGE PLPGSQL STRICT VOLATILE;
  52
  53 CREATE OR REPLACE FUNCTION migration_tools.exec (TEXT,TEXT) RETURNS VOID AS $$
  54     DECLARE
  55         migration_schema ALIAS FOR $1;
  56         sql ALIAS FOR $2;
  57         nrows INTEGER;
  58     BEGIN
  59         EXECUTE 'UPDATE ' || migration_schema || '.sql_current SET sql = ' || quote_literal(sql) || ';';
  60         EXECUTE sql;
  61         GET DIAGNOSTICS nrows = ROW_COUNT;
  62         PERFORM migration_tools.log(migration_schema,sql,nrows);
  63     EXCEPTION
  64         WHEN OTHERS THEN
  65             RAISE EXCEPTION '!!!!!!!!!!! state = %, msg = %, sql = %', SQLSTATE, SQLERRM, sql;
  66     END;
  67 $$ LANGUAGE PLPGSQL STRICT VOLATILE;
  68
  69 CREATE OR REPLACE FUNCTION migration_tools.init (TEXT) RETURNS VOID AS $$
  70     DECLARE
  71         migration_schema ALIAS FOR $1;
  72         sql TEXT;
  73     BEGIN
  74         EXECUTE 'DROP TABLE IF EXISTS ' || migration_schema || '.sql_current;';
  75         EXECUTE 'CREATE TABLE ' || migration_schema || '.sql_current ( sql TEXT);';
  76         EXECUTE 'INSERT INTO ' || migration_schema || '.sql_current ( sql ) VALUES ( '''' );';
  77         BEGIN
  78             SELECT 'CREATE TABLE ' || migration_schema || '.sql_log ( time TIMESTAMP NOT NULL DEFAULT NOW(), row_count INTEGER, sql TEXT );' INTO STRICT sql;
  79             EXECUTE sql;
  80         EXCEPTION
  81             WHEN OTHERS THEN
  82                 RAISE INFO '!!!!!!!!!!! state = %, msg = %, sql = %', SQLSTATE, SQLERRM, sql;
  83         END;
  84         PERFORM migration_tools.exec( $1, 'DROP TABLE IF EXISTS ' || migration_schema || '.config;' );
  85         PERFORM migration_tools.exec( $1, 'CREATE TABLE ' || migration_schema || '.config ( key TEXT UNIQUE, value TEXT);' );
  86         PERFORM migration_tools.exec( $1, 'INSERT INTO ' || migration_schema || '.config (key,value) VALUES ( ''production_tables'', ''asset.call_number,asset.copy_location,asset.copy,asset.stat_cat,asset.stat_cat_entry,asset.stat_cat_entry_copy_map,asset.copy_note,actor.usr,actor.card,actor.usr_address,actor.stat_cat,actor.stat_cat_entry,actor.stat_cat_entry_usr_map,actor.usr_note,action.circulation,action.hold_request,money.grocery,money.billing,money.cash_payment,money.forgive_payment'' );' );
  87         PERFORM migration_tools.exec( $1, 'INSERT INTO ' || migration_schema || '.config (key,value) VALUES ( ''country_code'', ''USA'' );' );
  88         PERFORM migration_tools.exec( $1, 'DROP TABLE IF EXISTS ' || migration_schema || '.fields_requiring_mapping;' );
  89         PERFORM migration_tools.exec( $1, 'CREATE TABLE ' || migration_schema || '.fields_requiring_mapping( table_schema TEXT, table_name TEXT, column_name TEXT, data_type TEXT);' );
  90         BEGIN
  91             PERFORM migration_tools.exec( $1, 'INSERT INTO ' || migration_schema || '.config (key,value) VALUES ( ''last_init'', now() );' );
  92         EXCEPTION
  93             WHEN OTHERS THEN PERFORM migration_tools.exec( $1, 'UPDATE ' || migration_schema || '.config SET value = now() WHERE key = ''last_init'';' );
  94         END;
  95     END;
  96 $$ LANGUAGE PLPGSQL STRICT VOLATILE;
  97
  98 CREATE OR REPLACE FUNCTION migration_tools.build (TEXT) RETURNS VOID AS $$
  99     DECLARE
 100         migration_schema ALIAS FOR $1;
 101         production_tables TEXT[];
 102     BEGIN
 103         --RAISE INFO 'In migration_tools.build(%)', migration_schema;
 104         SELECT migration_tools.production_tables(migration_schema) INTO STRICT production_tables;
 105         PERFORM migration_tools.build_base_staging_tables(migration_schema,production_tables);
 106     END;
 107 $$ LANGUAGE PLPGSQL STRICT VOLATILE;
 108
 109 CREATE OR REPLACE FUNCTION migration_tools.build_base_staging_tables (TEXT,TEXT[]) RETURNS VOID AS $$
 110     DECLARE
 111         migration_schema ALIAS FOR $1;
 112         production_tables ALIAS FOR $2;
 113     BEGIN
 114         --RAISE INFO 'In migration_tools.build_base_staging_tables(%,%)', migration_schema, production_tables;
 115         FOR i IN array_lower(production_tables,1) .. array_upper(production_tables,1) LOOP
 116             PERFORM migration_tools.build_specific_base_staging_table(migration_schema,production_tables[i]);
 117         END LOOP;
 118     END;
 119 $$ LANGUAGE PLPGSQL STRICT VOLATILE;
 120
 121 CREATE OR REPLACE FUNCTION migration_tools.build_specific_base_staging_table (TEXT,TEXT) RETURNS VOID AS $$
 122     DECLARE
 123         migration_schema ALIAS FOR $1;
 124         production_table ALIAS FOR $2;
 125         base_staging_table TEXT;
 126         columns RECORD;
 127     BEGIN
 128         base_staging_table = REPLACE( production_table, '.', '_' );
 129         --RAISE INFO 'In migration_tools.build_specific_base_staging_table(%,%) -> %', migration_schema, production_table, base_staging_table;
 130         PERFORM migration_tools.exec( $1, 'CREATE TABLE ' || migration_schema || '.' || base_staging_table || ' ( LIKE ' || production_table || ' INCLUDING DEFAULTS EXCLUDING CONSTRAINTS );' );
 131         PERFORM migration_tools.exec( $1, '
 132             INSERT INTO ' || migration_schema || '.fields_requiring_mapping
 133                 SELECT table_schema, table_name, column_name, data_type
 134                 FROM information_schema.columns
 135                 WHERE table_schema = ''' || migration_schema || ''' AND table_name = ''' || base_staging_table || ''' AND is_nullable = ''NO'' AND column_default IS NULL;
 136         ' );
 137         FOR columns IN
 138             SELECT table_schema, table_name, column_name, data_type
 139             FROM information_schema.columns
 140             WHERE table_schema = migration_schema AND table_name = base_staging_table AND is_nullable = 'NO' AND column_default IS NULL
 141         LOOP
 142             PERFORM migration_tools.exec( $1, 'ALTER TABLE ' || columns.table_schema || '.' || columns.table_name || ' ALTER COLUMN ' || columns.column_name || ' DROP NOT NULL;' );
 143         END LOOP;
 144     END;
 145 $$ LANGUAGE PLPGSQL STRICT VOLATILE;
 146
 147 CREATE OR REPLACE FUNCTION migration_tools.insert_base_into_production (TEXT) RETURNS VOID AS $$
 148     DECLARE
 149         migration_schema ALIAS FOR $1;
 150         production_tables TEXT[];
 151     BEGIN
 152         --RAISE INFO 'In migration_tools.insert_into_production(%)', migration_schema;
 153         SELECT migration_tools.production_tables(migration_schema) INTO STRICT production_tables;
 154         FOR i IN array_lower(production_tables,1) .. array_upper(production_tables,1) LOOP
 155             PERFORM migration_tools.insert_into_production(migration_schema,production_tables[i]);
 156         END LOOP;
 157     END;
 158 $$ LANGUAGE PLPGSQL STRICT VOLATILE;
 159
 160 CREATE OR REPLACE FUNCTION migration_tools.insert_into_production (TEXT,TEXT) RETURNS VOID AS $$
 161     DECLARE
 162         migration_schema ALIAS FOR $1;
 163         production_table ALIAS FOR $2;
 164         base_staging_table TEXT;
 165         columns RECORD;
 166     BEGIN
 167         base_staging_table = REPLACE( production_table, '.', '_' );
 168         --RAISE INFO 'In migration_tools.insert_into_production(%,%) -> %', migration_schema, production_table, base_staging_table;
 169         PERFORM migration_tools.exec( $1, 'INSERT INTO ' || production_table || ' SELECT * FROM ' || migration_schema || '.' || base_staging_table || ';' );
 170     END;
 171 $$ LANGUAGE PLPGSQL STRICT VOLATILE;
 172
 173 CREATE OR REPLACE FUNCTION migration_tools.address_parse_out_citystatezip (TEXT) RETURNS TEXT[] AS $$
 174     DECLARE
 175         city_state_zip TEXT := $1;
 176         city TEXT := '';
 177         state TEXT := '';
 178         zip TEXT := '';
 179     BEGIN
 180         zip := CASE WHEN city_state_zip ~ E'\\d\\d\\d\\d\\d' THEN REGEXP_REPLACE( city_state_zip, E'^.*(\\d\\d\\d\\d\\d-?\\d*).*$', E'\\1' ) ELSE '' END;
 181         city_state_zip := REGEXP_REPLACE( city_state_zip, E'^(.*)\\d\\d\\d\\d\\d-?\\d*(.*)$', E'\\1\\2');
 182         IF city_state_zip ~ ',' THEN
 183             state := REGEXP_REPLACE( city_state_zip, E'^(.*),(.*)$', E'\\2');
 184             city := REGEXP_REPLACE( city_state_zip, E'^(.*),(.*)$', E'\\1');
 185         ELSE
 186             IF city_state_zip ~ E'\\s+[A-Z][A-Z]\\s*' THEN
 187                 state := REGEXP_REPLACE( city_state_zip, E'^.*,?\\s+([A-Z][A-Z])\\s*.*$', E'\\1' );
 188                 city := REGEXP_REPLACE( city_state_zip, E'^(.*?),?\\s+[A-Z][A-Z](\\s*.*)$', E'\\1\\2' );
 189             ELSE
 190                 IF city_state_zip ~ E'^\\S+$'  THEN
 191                     city := city_state_zip;
 192                     state := 'N/A';
 193                 ELSE
 194                     state := REGEXP_REPLACE( city_state_zip, E'^(.*?),?\\s*(\\S+)\\s*$', E'\\2');
 195                     city := REGEXP_REPLACE( city_state_zip, E'^(.*?),?\\s*(\\S+)\\s*$', E'\\1');
 196                 END IF;
 197             END IF;
 198         END IF;
 199         RETURN ARRAY[ TRIM(BOTH ' ' FROM city), TRIM(BOTH ' ' FROM state), TRIM(BOTH ' ' FROM zip) ];
 200     END;
 201 $$ LANGUAGE PLPGSQL STRICT VOLATILE;
 202
 203 CREATE OR REPLACE FUNCTION migration_tools.rebarcode (o TEXT, t BIGINT) RETURNS TEXT AS $$
 204     DECLARE
 205         n TEXT := o;
 206     BEGIN
 207         IF o ~ E'^\\d+$' AND o !~ E'^0' THEN
 208             IF o::BIGINT < t THEN
 209                 n = o::BIGINT + t;
 210             END IF;
 211         END IF;
 212
 213         RETURN n;
 214     END;
 215 $$ LANGUAGE PLPGSQL STRICT IMMUTABLE;
 216