pg_fts 0.0.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 3458e83b224d9191b68c2f32ce1172cf5f5280a0
-  data.tar.gz: a945387bb0a70cfaa15b9059b522293cdefda53f
+  metadata.gz: 1ea680738925689db1fc42b39c9c093cb8614dc7
+  data.tar.gz: d7d82abffb9221c04a4a3ff8941adfc936b7c4e6
 SHA512:
-  metadata.gz: f72e926e83c7b218f79ee42d7d1e9d8b38b73eac9214db1253229e3c876943de75edbd7975bc80b68dcdbef3ea9a9605a5cad2913308c41b8905d54dbb240cc0
-  data.tar.gz: 80793c65aeb3342d6209205a5064dd775f756deb14441f2e60be32513b8ccd07405dedd1ef87d92c81871d5ddb935886cbdf8efa97005f4faa0c6c1759de02c1
+  metadata.gz: 0ee0f73f4700ca766afc752ebe714f90f93a2ac4b6d63cb1baceea623b9cc80fa418d9afad231f3d63d9ad3be1ff678c4de223a396226bd7ba7fc18eecc2ba32
+  data.tar.gz: 5a5584885fa6f196bb43a313a21716cc814a25272d198cbc22695d78d4b2eb99e0e32cc57e37c6f577a9f395cccf4ce478ce8b89a262028f634ecd0f1c1ccb8c

data/lib/pg/fts/index/many_to_one.rb

@@ -0,0 +1,325 @@
+# many documents, one source
+class PG::FTS::Index::ManyToOne
+  attr_reader :table, :fields, :catalog, :document, :key, :foreign_key, :name
+  alias source table
+
+  include PG::FTS::Index
+  include PG::FTS::Naming
+  include PG::FTS::TSVector
+
+  # rubocop:disable Metrics/ParameterLists
+  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 || :"#{table}_id"
+    @name = as
+  end
+  # rubocop:enable Metrics/ParameterLists
+
+  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}"."#{foreign_key}" = "#{source}"."#{key}";
+    SQL
+  end
+
+  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}"."#{foreign_key}" = NEW."#{key}";
+    SQL
+  end
+
+  def on_source_insert_procedure
+    <<-SQL.gsub(/^ {4}/, '')
+    CREATE FUNCTION "#{on_source_insert_procedure_name}"()
+    RETURNS TRIGGER AS $$
+      BEGIN
+        #{on_source_insert_query.gsub(/^/, '    ')}
+        RETURN NEW;
+      END;
+    $$ LANGUAGE plpgsql;
+    SQL
+  end
+
+  def on_source_update_query
+    <<-SQL.gsub(/^ {4}/, '')
+    UPDATE "#{PG::FTS.table}"
+    SET
+      "tsv" = (#{ts_vector}),
+      "document_table" = '#{document}',
+      "document_id" = "#{document}"."#{key}"
+    FROM "#{document}"
+    WHERE "#{document}"."#{foreign_key}" = NEW."#{key}"
+      AND "#{PG::FTS.table}"."source_id" = OLD."#{key}"
+      AND "#{PG::FTS.table}"."source_table" = '#{source}'
+      AND "#{PG::FTS.table}"."document_id" = "#{document}"."#{key}"
+      AND "#{PG::FTS.table}"."document_table" = '#{document}';
+    SQL
+  end
+
+  def on_source_update_procedure
+    <<-SQL.gsub(/^ {4}/, '')
+    CREATE FUNCTION "#{on_source_update_procedure_name}"()
+    RETURNS TRIGGER AS $$
+      BEGIN
+        #{on_source_update_query.gsub(/^/, '    ')}
+        RETURN NEW;
+      END;
+    $$ LANGUAGE plpgsql;
+    SQL
+  end
+
+  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_table" = '#{document}';
+    SQL
+  end
+
+  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
+
+  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
+
+  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
+
+  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
+
+  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
+
+  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
+
+  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
+
+  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}',
+      NEW."#{foreign_key}",
+      '#{source}',
+      #{ts_vector(source)}
+    FROM "#{source}"
+    WHERE "#{source}"."#{key}" = NEW."#{foreign_key}";
+    SQL
+  end
+
+  def on_document_insert_procedure
+    <<-SQL.gsub(/^ {4}/, '')
+    CREATE FUNCTION "#{on_document_insert_procedure_name}"()
+    RETURNS TRIGGER AS $$
+      BEGIN
+        IF NEW."#{foreign_key}" IS NULL THEN
+          RETURN NEW;
+        END IF;
+        #{on_document_insert_query.gsub(/^/, '    ')}
+        RETURN NEW;
+      END;
+    $$ LANGUAGE plpgsql;
+    SQL
+  end
+
+  def on_document_update_query
+    <<-SQL.gsub(/^ {4}/, '')
+    UPDATE "#{PG::FTS.table}"
+    SET
+      "tsv" = (#{ts_vector(source)}),
+      "source_table" = '#{source}',
+      "source_id" = NEW."#{foreign_key}"
+    FROM "#{source}"
+    WHERE "#{source}"."#{key}" = NEW."#{foreign_key}"
+      AND "#{PG::FTS.table}"."source_id" = OLD."#{foreign_key}"
+      AND "#{PG::FTS.table}"."source_table" = '#{source}'
+      AND "#{PG::FTS.table}"."document_id" = OLD."#{key}"
+      AND "#{PG::FTS.table}"."document_table" = '#{document}';
+    SQL
+  end
+
+  def on_document_update_procedure
+    <<-SQL.gsub(/^ {4}/, '')
+    CREATE FUNCTION "#{on_document_update_procedure_name}"()
+    RETURNS TRIGGER AS $$
+      BEGIN
+        #{on_document_update_query.gsub(/^/, '    ')}
+        IF NOT FOUND THEN
+          #{on_document_insert_query.gsub(/^/, '      ')}
+          IF NOT FOUND THEN
+            #{on_document_delete_query.gsub(/^/, '        ')}
+          END IF;
+          RETURN NEW;
+        END IF;
+        RETURN NEW;
+      END;
+    $$ LANGUAGE plpgsql;
+    SQL
+  end
+
+  def on_document_delete_query
+    <<-SQL.gsub(/^ {4}/, '')
+    DELETE FROM "#{PG::FTS.table}"
+    WHERE "#{PG::FTS.table}"."source_id" = OLD."#{foreign_key}"
+      AND "#{PG::FTS.table}"."source_table" = '#{source}'
+      AND "#{PG::FTS.table}"."document_id" = OLD."#{key}"
+      AND "#{PG::FTS.table}"."document_table" = '#{document}';
+    SQL
+  end
+
+  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
+
+  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
+
+  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
+
+  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
+
+  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
+
+  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
+
+  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
+end