class PG::FTS::Index::OneToMany

one document, many sources

Attributes

catalog[R]
document[R]
fields[R]
foreign_key[R]
key[R]
name[R]
source[R]
table[R]

Public Class Methods

new(table, *fields, document: nil, key: nil, foreign_key: nil, catalog: nil, as: nil) click to toggle source

rubocop:disable Metrics/ParameterLists

# File lib/pg/fts/index/one_to_many.rb, line 11
def initialize(table, *fields,
               document: nil, key: nil, foreign_key: nil,
               catalog: nil,
               as: nil)
  @table = table
  @fields = fields
  @catalog = catalog || PG::FTS.catalog
  @document = document || fail(ArgumentError, ':document required')
  @key = key || :id
  @foreign_key = foreign_key || :"#{document}_id"
  @name = as
end

Public Instance Methods

build_query() click to toggle source

rubocop:enable Metrics/ParameterLists

# File lib/pg/fts/index/one_to_many.rb, line 25
  def build_query
    <<-SQL.gsub(/^ {4}/, '')
    INSERT INTO "#{PG::FTS.table}" (
      "document_id",
      "document_table",
      "source_id",
      "source_table",
      "tsv")
    SELECT "#{document}"."#{key}",
           '#{document}',
           "#{source}"."#{key}",
           '#{source}',
           #{ts_vector(source)}
    FROM "#{document}", "#{source}"
    WHERE "#{document}"."#{key}" = "#{source}"."#{foreign_key}";
    SQL
  end
on_document_delete_procedure() click to toggle source
# File lib/pg/fts/index/one_to_many.rb, line 247
  def on_document_delete_procedure
    <<-SQL.gsub(/^ {4}/, '')
    CREATE FUNCTION "#{on_document_delete_procedure_name}"()
    RETURNS TRIGGER AS $$
      BEGIN
        #{on_document_delete_query.gsub(/^/, '    ')}
        RETURN OLD;
      END;
    $$ LANGUAGE plpgsql;
    SQL
  end
on_document_delete_query() click to toggle source
# File lib/pg/fts/index/one_to_many.rb, line 238
  def on_document_delete_query
    <<-SQL.gsub(/^ {4}/, '')
    DELETE FROM "#{PG::FTS.table}"
    WHERE "#{PG::FTS.table}"."source_table" = '#{source}'
      AND "#{PG::FTS.table}"."document_id" = OLD."#{key}"
      AND "#{PG::FTS.table}"."document_table" = '#{document}';
    SQL
  end
on_document_delete_trigger() click to toggle source
# File lib/pg/fts/index/one_to_many.rb, line 297
  def on_document_delete_trigger
    <<-SQL.gsub(/^ {4}/, '')
    CREATE TRIGGER "#{on_document_delete_trigger_name}"
    AFTER DELETE ON "#{document}"
    FOR EACH ROW
    EXECUTE PROCEDURE "#{on_document_delete_procedure_name}"();
    SQL
  end
on_document_insert_procedure() click to toggle source
# File lib/pg/fts/index/one_to_many.rb, line 209
  def on_document_insert_procedure
    <<-SQL.gsub(/^ {4}/, '')
    CREATE FUNCTION "#{on_document_insert_procedure_name}"()
    RETURNS TRIGGER AS $$
      BEGIN
        #{on_document_insert_query.gsub(/^/, '    ')}
        RETURN NEW;
      END;
    $$ LANGUAGE plpgsql;
    SQL
  end
on_document_insert_query() click to toggle source
# File lib/pg/fts/index/one_to_many.rb, line 190
  def on_document_insert_query
    <<-SQL.gsub(/^ {4}/, '')
    INSERT INTO "#{PG::FTS.table}" (
      "document_id",
      "document_table",
      "source_id",
      "source_table",
      "tsv")
    SELECT
      NEW."#{key}",
      '#{document}',
      "#{source}"."#{key}",
      '#{source}',
      #{ts_vector(source)}
    FROM "#{source}"
    WHERE "#{source}"."#{foreign_key}" = NEW."#{key}";
    SQL
  end
on_document_insert_trigger() click to toggle source
# File lib/pg/fts/index/one_to_many.rb, line 279
  def on_document_insert_trigger
    <<-SQL.gsub(/^ {4}/, '')
    CREATE TRIGGER "#{on_document_insert_trigger_name}"
    AFTER INSERT ON "#{document}"
    FOR EACH ROW
    EXECUTE PROCEDURE "#{on_document_insert_procedure_name}"();
    SQL
  end
on_document_truncate_procedure() click to toggle source
# File lib/pg/fts/index/one_to_many.rb, line 267
  def on_document_truncate_procedure
    <<-SQL.gsub(/^ {4}/, '')
    CREATE FUNCTION "#{on_document_truncate_procedure_name}"()
    RETURNS TRIGGER AS $$
      BEGIN
        #{on_document_truncate_query.gsub(/^/, '    ')}
        RETURN NULL;
      END;
    $$ LANGUAGE plpgsql;
    SQL
  end
on_document_truncate_query() click to toggle source
# File lib/pg/fts/index/one_to_many.rb, line 259
  def on_document_truncate_query
    <<-SQL.gsub(/^ {4}/, '')
    DELETE FROM "#{PG::FTS.table}"
    WHERE "#{PG::FTS.table}"."source_table" = '#{source}'
      AND "#{PG::FTS.table}"."document_table" = '#{document}';
    SQL
  end
on_document_truncate_trigger() click to toggle source
# File lib/pg/fts/index/one_to_many.rb, line 306
  def on_document_truncate_trigger
    <<-SQL.gsub(/^ {4}/, '')
    CREATE TRIGGER "#{on_document_truncate_trigger_name}"
    AFTER TRUNCATE ON "#{document}"
    FOR EACH STATEMENT
    EXECUTE PROCEDURE "#{on_document_truncate_procedure_name}"();
    SQL
  end
on_document_update_procedure() click to toggle source
# File lib/pg/fts/index/one_to_many.rb, line 226
  def on_document_update_procedure
    <<-SQL.gsub(/^ {4}/, '')
    CREATE FUNCTION "#{on_document_update_procedure_name}"()
    RETURNS TRIGGER AS $$
      BEGIN
        #{on_document_update_query.gsub(/^/, '    ')}
        RETURN NEW;
      END;
    $$ LANGUAGE plpgsql;
    SQL
  end
on_document_update_query() click to toggle source
# File lib/pg/fts/index/one_to_many.rb, line 221
  def on_document_update_query
    <<-SQL.gsub(/^ {4}/, '')
    SQL
  end
on_document_update_trigger() click to toggle source
# File lib/pg/fts/index/one_to_many.rb, line 288
  def on_document_update_trigger
    <<-SQL.gsub(/^ {4}/, '')
    CREATE TRIGGER "#{on_document_update_trigger_name}"
    AFTER UPDATE ON "#{document}"
    FOR EACH ROW
    EXECUTE PROCEDURE "#{on_document_update_procedure_name}"();
    SQL
  end
on_source_delete_procedure() click to toggle source
# File lib/pg/fts/index/one_to_many.rb, line 122
  def on_source_delete_procedure
    <<-SQL.gsub(/^ {4}/, '')
    CREATE FUNCTION "#{on_source_delete_procedure_name}"()
    RETURNS TRIGGER AS $$
      BEGIN
        #{on_source_delete_query.gsub(/^/, '    ')}
        RETURN OLD;
      END;
    $$ LANGUAGE plpgsql;
    SQL
  end
on_source_delete_query() click to toggle source
# File lib/pg/fts/index/one_to_many.rb, line 112
  def on_source_delete_query
    <<-SQL.gsub(/^ {4}/, '')
    DELETE FROM "#{PG::FTS.table}"
    WHERE "#{PG::FTS.table}"."source_id" = OLD."#{key}"
      AND "#{PG::FTS.table}"."source_table" = '#{source}'
      AND "#{PG::FTS.table}"."document_id" = OLD."#{foreign_key}"
      AND "#{PG::FTS.table}"."document_table" = '#{document}';
    SQL
  end
on_source_delete_trigger() click to toggle source
# File lib/pg/fts/index/one_to_many.rb, line 172
  def on_source_delete_trigger
    <<-SQL.gsub(/^ {4}/, '')
    CREATE TRIGGER "#{on_source_delete_trigger_name}"
    AFTER DELETE ON "#{source}"
    FOR EACH ROW
    EXECUTE PROCEDURE "#{on_source_delete_procedure_name}"();
    SQL
  end
on_source_insert_procedure() click to toggle source
# File lib/pg/fts/index/one_to_many.rb, line 62
  def on_source_insert_procedure
    <<-SQL.gsub(/^ {4}/, '')
    CREATE FUNCTION "#{on_source_insert_procedure_name}"()
    RETURNS TRIGGER AS $$
      BEGIN
        IF NEW."#{foreign_key}" IS NULL THEN
          RETURN NEW;
        END IF;
        #{on_source_insert_query.gsub(/^/, '    ')}
        RETURN NEW;
      END;
    $$ LANGUAGE plpgsql;
    SQL
  end
on_source_insert_query() click to toggle source
# File lib/pg/fts/index/one_to_many.rb, line 43
  def on_source_insert_query
    <<-SQL.gsub(/^ {4}/, '')
    INSERT INTO "#{PG::FTS.table}" (
      "document_id",
      "document_table",
      "source_id",
      "source_table",
      "tsv")
    SELECT
      "#{document}"."#{key}",
      '#{document}',
      NEW."#{key}",
      '#{source}',
      #{ts_vector}
    FROM "#{document}"
    WHERE "#{document}"."#{key}" = NEW."#{foreign_key}";
    SQL
  end
on_source_insert_trigger() click to toggle source
# File lib/pg/fts/index/one_to_many.rb, line 154
  def on_source_insert_trigger
    <<-SQL.gsub(/^ {4}/, '')
    CREATE TRIGGER "#{on_source_insert_trigger_name}"
    AFTER INSERT ON "#{source}"
    FOR EACH ROW
    EXECUTE PROCEDURE "#{on_source_insert_procedure_name}"();
    SQL
  end
on_source_truncate_procedure() click to toggle source
# File lib/pg/fts/index/one_to_many.rb, line 142
  def on_source_truncate_procedure
    <<-SQL.gsub(/^ {4}/, '')
    CREATE FUNCTION "#{on_source_truncate_procedure_name}"()
    RETURNS TRIGGER AS $$
      BEGIN
        #{on_source_truncate_query.gsub(/^/, '    ')}
        RETURN NULL;
      END;
    $$ LANGUAGE plpgsql;
    SQL
  end
on_source_truncate_query() click to toggle source
# File lib/pg/fts/index/one_to_many.rb, line 134
  def on_source_truncate_query
    <<-SQL.gsub(/^ {4}/, '')
    DELETE FROM "#{PG::FTS.table}"
    WHERE "#{PG::FTS.table}"."source_table" = '#{source}'
      AND "#{PG::FTS.table}"."document_table" = '#{document}';
    SQL
  end
on_source_truncate_trigger() click to toggle source
# File lib/pg/fts/index/one_to_many.rb, line 181
  def on_source_truncate_trigger
    <<-SQL.gsub(/^ {4}/, '')
    CREATE TRIGGER "#{on_source_truncate_trigger_name}"
    AFTER TRUNCATE ON "#{source}"
    FOR EACH STATEMENT
    EXECUTE PROCEDURE "#{on_source_truncate_procedure_name}"();
    SQL
  end
on_source_update_procedure() click to toggle source
# File lib/pg/fts/index/one_to_many.rb, line 93
  def on_source_update_procedure
    <<-SQL.gsub(/^ {4}/, '')
    CREATE FUNCTION "#{on_source_update_procedure_name}"()
    RETURNS TRIGGER AS $$
      BEGIN
        #{on_source_update_query.gsub(/^/, '    ')}
        IF NOT FOUND THEN
          #{on_source_insert_query.gsub(/^/, '    ')}
          IF NOT FOUND THEN
            #{on_source_delete_query.gsub(/^/, '    ')}
          END IF;
          RETURN NEW;
        END IF;
        RETURN NEW;
      END;
    $$ LANGUAGE plpgsql;
    SQL
  end
on_source_update_query() click to toggle source
# File lib/pg/fts/index/one_to_many.rb, line 77
  def on_source_update_query
    <<-SQL.gsub(/^ {4}/, '')
    UPDATE "#{PG::FTS.table}"
    SET
      "tsv" = (#{ts_vector}),
      "document_table" = '#{document}',
      "document_id" = NEW."#{foreign_key}"
    FROM "#{document}"
    WHERE "#{document}"."#{key}" = NEW."#{foreign_key}"
      AND "#{PG::FTS.table}"."source_id" = OLD."#{key}"
      AND "#{PG::FTS.table}"."source_table" = '#{source}'
      AND "#{PG::FTS.table}"."document_id" = OLD."#{foreign_key}"
      AND "#{PG::FTS.table}"."document_table" = '#{document}';
    SQL
  end
on_source_update_trigger() click to toggle source
# File lib/pg/fts/index/one_to_many.rb, line 163
  def on_source_update_trigger
    <<-SQL.gsub(/^ {4}/, '')
    CREATE TRIGGER "#{on_source_update_trigger_name}"
    AFTER UPDATE ON "#{source}"
    FOR EACH ROW
    EXECUTE PROCEDURE "#{on_source_update_procedure_name}"();
    SQL
  end