aboutsummaryrefslogtreecommitdiffstats
path: root/lib
diff options
context:
space:
mode:
Diffstat (limited to 'lib')
-rw-r--r--lib/af83/stored_procedures.rb44
-rw-r--r--lib/sql/clone_schema.sql114
-rw-r--r--lib/stif/reflex_synchronization.rb1
-rw-r--r--lib/tasks/ci.rake2
4 files changed, 160 insertions, 1 deletions
diff --git a/lib/af83/stored_procedures.rb b/lib/af83/stored_procedures.rb
new file mode 100644
index 000000000..698f3861d
--- /dev/null
+++ b/lib/af83/stored_procedures.rb
@@ -0,0 +1,44 @@
+module StoredProcedures extend self
+
+ def invoke_stored_procedure(name, *params)
+ name = name.to_s
+ raise ArgumentError, "no such stored procedure #{name.inspect}" unless stored_procedures[name]
+ invocation = "#{name}(#{quote_params(params)})"
+ ActiveRecord::Base.connection.execute "SELECT #{invocation}"
+ end
+
+ def create_stored_procedure(name)
+ name = name.to_s
+ sql_file = File.expand_path("../../sql/#{name}.sql", __FILE__)
+ raise ArgumentError, "missing sql file #{sql_file.inspect}" unless File.readable? sql_file
+
+ # We could store the file's content for reload without application restart if desired.
+ stored_procedures[name] = true
+
+ ActiveRecord::Base.connection.execute File.read(sql_file)
+ end
+
+ private
+ def quote_params(params)
+ params
+ .map(&method(:quote_param))
+ .join(", ")
+ end
+
+ def quote_param(param)
+ case param
+ when String
+ "'#{param}'"
+ when TrueClass
+ "TRUE"
+ when FalseClass
+ "FALSE"
+ else
+ param
+ end
+ end
+
+ def stored_procedures
+ @__stored_procedures__ ||= {}
+ end
+end
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;
diff --git a/lib/stif/reflex_synchronization.rb b/lib/stif/reflex_synchronization.rb
index fee54f4c1..a7dda7e30 100644
--- a/lib/stif/reflex_synchronization.rb
+++ b/lib/stif/reflex_synchronization.rb
@@ -153,6 +153,7 @@ module Stif
:city_name => 'Town',
:stif_type => 'OBJECT_STATUS'
}.each do |k, v| stop[k] = entry[v] end
+ # TODO: use stop.update_attributes instead of the above
if entry['gml:pos']
stop['longitude'] = entry['gml:pos'][:lng]
diff --git a/lib/tasks/ci.rake b/lib/tasks/ci.rake
index 52bab7aa4..85d1cbdf9 100644
--- a/lib/tasks/ci.rake
+++ b/lib/tasks/ci.rake
@@ -29,7 +29,7 @@ namespace :ci do
desc "Deploy after CI"
task :deploy do
- sh "cap #{deploy_env} deploy:migrations deploy:seed"
+ sh "cap #{deploy_env} deploy:migrations"
end
desc "Clean test files"