X-Git-Url: https://git.openstreetmap.org./osqa.git/blobdiff_plain/b484fb008ae574dd4ef10a12d4e530ab96743acc..143150bddf1a02498da0dcbdf0ad59d540c1519a:/forum_modules/pgfulltext/pg_fts_install.sql?ds=sidebyside diff --git a/forum_modules/pgfulltext/pg_fts_install.sql b/forum_modules/pgfulltext/pg_fts_install.sql index 9ac5db9..c046870 100644 --- a/forum_modules/pgfulltext/pg_fts_install.sql +++ b/forum_modules/pgfulltext/pg_fts_install.sql @@ -1,86 +1,129 @@ - CREATE OR REPLACE FUNCTION public.create_plpgsql_language () - RETURNS TEXT - AS $$ - CREATE LANGUAGE plpgsql; - SELECT 'language plpgsql created'::TEXT; - $$ - LANGUAGE 'sql'; - - SELECT CASE WHEN - (SELECT true::BOOLEAN - FROM pg_language - WHERE lanname='plpgsql') - THEN - (SELECT 'language already installed'::TEXT) - ELSE - (SELECT public.create_plpgsql_language()) - END; - - DROP FUNCTION public.create_plpgsql_language (); - - CREATE OR REPLACE FUNCTION node_ranking(node_id int, srch text) RETURNS float AS $$ - declare - v tsvector; - cv tsvector; - rev_id int; - child_count int; - r record; - begin - SELECT active_revision_id INTO rev_id FROM forum_node WHERE id = node_id; - SELECT tsv INTO v FROM forum_noderevision WHERE id = rev_id; - - SELECT count(*) INTO child_count FROM forum_node WHERE abs_parent_id = node_id AND NOT deleted; - - IF child_count > 0 THEN - FOR r in SELECT * FROM forum_node WHERE abs_parent_id = node_id AND NOT deleted LOOP - SELECT tsv INTO cv FROM forum_noderevision WHERE id = r.active_revision_id; - v :=(v || cv); +CREATE OR REPLACE FUNCTION public.create_plpgsql_language () + RETURNS TEXT + AS $$ + CREATE LANGUAGE plpgsql; + SELECT 'language plpgsql created'::TEXT; + $$ +LANGUAGE 'sql'; + +SELECT CASE WHEN + (SELECT true::BOOLEAN + FROM pg_language + WHERE lanname='plpgsql') + THEN + (SELECT 'language already installed'::TEXT) + ELSE + (SELECT public.create_plpgsql_language()) + END; + +DROP FUNCTION public.create_plpgsql_language (); + +CREATE OR REPLACE FUNCTION public.drop_tsv_noderevision_column () RETURNS VOID AS $$ +begin + ALTER TABLE forum_noderevision DROP COLUMN tsv; + DROP TRIGGER IF EXISTS tsvectorupdate ON forum_noderevision; +end +$$ LANGUAGE plpgsql; + +CREATE OR REPLACE FUNCTION public.tsv_noderevision_column_exists() RETURNS int AS $$ + SELECT COUNT(attname)::int FROM pg_attribute WHERE attrelid = (SELECT oid FROM pg_class WHERE relname = 'forum_noderevision') AND attname = 'tsv'; +$$ LANGUAGE 'sql'; + +select case when public.tsv_noderevision_column_exists()>0 then public.drop_tsv_noderevision_column()end; + +drop function drop_tsv_noderevision_column(); +drop function tsv_noderevision_column_exists(); + +CREATE OR REPLACE FUNCTION set_doctable_tsv() RETURNS TRIGGER AS $$ +declare + root_id int; + doc tsvector; + rcount int; + cv tsvector; +begin + SELECT abs_parent_id INTO root_id FROM forum_node WHERE id = new.node_id; + + IF root_id IS NULL THEN + root_id := new.node_id; + END IF; + + SELECT count(*)::int INTO rcount FROM forum_node WHERE id = root_id; + + IF rcount = 0 THEN + return new; + END IF; + + SELECT + setweight(to_tsvector('english', coalesce(tagnames,'')), 'A') || + setweight(to_tsvector('english', coalesce(title,'')), 'B') || + setweight(to_tsvector('english', coalesce(body,'')), 'C') INTO doc + FROM forum_node WHERE id = root_id; + + SELECT count(*)::int INTO rcount FROM forum_node WHERE abs_parent_id = root_id AND (NOT state_string LIKE '%%deleted%%'); + + IF rcount > 0 THEN + FOR cv in SELECT setweight(to_tsvector('english', coalesce(body,'')), 'C') FROM forum_node WHERE abs_parent_id = root_id AND (NOT state_string LIKE '%%deleted%%') LOOP + doc :=(doc || cv); END LOOP; END IF; - RAISE NOTICE '%', v; - RETURN ts_rank_cd(v, plainto_tsquery('english', srch), 32); - end - $$ LANGUAGE plpgsql; - -select node_ranking(50, 'free'); - - - - CREATE OR REPLACE FUNCTION set_node_tsv() RETURNS TRIGGER AS $$ - begin - new.tsv := - setweight(to_tsvector('english', coalesce(new.tagnames,'')), 'A') || - setweight(to_tsvector('english', coalesce(new.title,'')), 'B') || - setweight(to_tsvector('english', coalesce(new.body,'')), 'C'); - - RETURN new; - end - $$ LANGUAGE plpgsql; - - CREATE OR REPLACE FUNCTION public.create_tsv_noderevision_column () - RETURNS TEXT - AS $$ - ALTER TABLE forum_noderevision ADD COLUMN tsv tsvector; - - CREATE TRIGGER tsvectorupdate BEFORE INSERT OR UPDATE - ON forum_noderevision FOR EACH ROW EXECUTE PROCEDURE set_node_tsv(); - - CREATE INDEX noderevision_tsv ON forum_noderevision USING gin(tsv); - - SELECT 'tsv column created'::TEXT; - $$ - LANGUAGE 'sql'; - - SELECT CASE WHEN - (SELECT true::BOOLEAN FROM pg_attribute WHERE attrelid = (SELECT oid FROM pg_class WHERE relname = 'forum_noderevision') AND attname = 'tsv') - THEN - (SELECT 'Tsv column already exists'::TEXT) - ELSE - (SELECT public.create_tsv_noderevision_column()) - - END; - - DROP FUNCTION public.create_tsv_noderevision_column(); - - UPDATE forum_noderevision SET id=id WHERE TRUE; \ No newline at end of file + SELECT count(*)::int INTO rcount FROM forum_rootnode_doc WHERE node_id = root_id; + + IF rcount > 0 THEN + UPDATE forum_rootnode_doc SET document = doc WHERE node_id = root_id; + ELSE + INSERT INTO forum_rootnode_doc (node_id, document) VALUES (root_id, doc); + END IF; + + RETURN new; +end +$$ LANGUAGE plpgsql; + +CREATE OR REPLACE FUNCTION public.build_doc_table() RETURNS VOID as $$ + CREATE TABLE forum_rootnode_doc + ( + node_id integer, + "document" tsvector, + PRIMARY KEY (node_id), + FOREIGN KEY (node_id) REFERENCES forum_node (id) ON UPDATE NO ACTION ON DELETE NO ACTION + ) WITH (OIDS=FALSE); + + DROP TRIGGER IF EXISTS tsvectorupdate ON forum_noderevision; + + CREATE TRIGGER tsvectorupdate BEFORE INSERT OR UPDATE + ON forum_noderevision FOR EACH ROW EXECUTE PROCEDURE set_doctable_tsv(); + + CREATE INDEX doctable_tsv ON forum_rootnode_doc USING gin(document); +$$ LANGUAGE 'sql'; + +CREATE OR REPLACE FUNCTION public.doc_table_exists() RETURNS int AS $$ + SELECT COUNT(table_name)::int FROM information_schema.tables WHERE table_name = 'forum_rootnode_doc'; +$$ LANGUAGE 'sql'; + +select case when public.doc_table_exists()=0 then public.build_doc_table()end; + +drop function build_doc_table(); +drop function doc_table_exists(); + +CREATE OR REPLACE FUNCTION rank_exact_matches(rank float) RETURNS float AS $$ +begin + IF rank = 0 THEN + return 1; + ELSE + return rank; + END IF; + +end +$$ LANGUAGE plpgsql; + +CREATE OR REPLACE FUNCTION public.rebuild_index() RETURNS VOID as $$ + DECLARE + r integer; + BEGIN + FOR r IN SELECT active_revision_id FROM forum_node WHERE node_type = 'question' LOOP + UPDATE forum_noderevision SET id = id WHERE id = r; + END LOOP; + END +$$ LANGUAGE 'plpgsql'; + +SELECT rebuild_index();