mirror of
https://github.com/knightcrawler-stremio/knightcrawler.git
synced 2024-12-20 03:29:51 +00:00
66609c2a46
* add new indexes, and change year column to int * Change gist to gin, and change year to int * Producer changes for new gin query * Fully map the rtn response using json dump from Pydantic Also updates Rtn to 0.1.9 * Add housekeeping script to reconcile imdb ids. * Join Torrent onto the ingested torrent table Ensure that a torrent can always find the details of where it came from, and how it was parsed. * Version bump for release * missing quote on table name
35 lines
1.8 KiB
PL/PgSQL
35 lines
1.8 KiB
PL/PgSQL
-- Purpose: Change the year column to integer and add a search function that allows for searching by year.
|
|
ALTER TABLE imdb_metadata
|
|
ALTER COLUMN year TYPE integer USING (CASE WHEN year = '\N' THEN 0 ELSE year::integer END);
|
|
|
|
-- Remove the old search function
|
|
DROP FUNCTION IF EXISTS search_imdb_meta(TEXT, TEXT, TEXT, INT);
|
|
|
|
-- Add the new search function that allows for searching by year with a plus/minus one year range
|
|
CREATE OR REPLACE FUNCTION search_imdb_meta(search_term TEXT, category_param TEXT DEFAULT NULL, year_param INT DEFAULT NULL, limit_param INT DEFAULT 10)
|
|
RETURNS TABLE(imdb_id character varying(16), title character varying(1000),category character varying(50),year INT, score REAL) AS $$
|
|
BEGIN
|
|
SET pg_trgm.similarity_threshold = 0.9;
|
|
RETURN QUERY
|
|
SELECT imdb_metadata.imdb_id, imdb_metadata.title, imdb_metadata.category, imdb_metadata.year, similarity(imdb_metadata.title, search_term) as score
|
|
FROM imdb_metadata
|
|
WHERE (imdb_metadata.title % search_term)
|
|
AND (imdb_metadata.adult = FALSE)
|
|
AND (category_param IS NULL OR imdb_metadata.category = category_param)
|
|
AND (year_param IS NULL OR imdb_metadata.year BETWEEN year_param - 1 AND year_param + 1)
|
|
ORDER BY score DESC
|
|
LIMIT limit_param;
|
|
END; $$
|
|
LANGUAGE plpgsql;
|
|
|
|
-- Drop the old indexes
|
|
DROP INDEX IF EXISTS idx_imdb_metadata_adult;
|
|
DROP INDEX IF EXISTS idx_imdb_metadata_category;
|
|
DROP INDEX IF EXISTS idx_imdb_metadata_year;
|
|
DROP INDEX IF EXISTS title_gist;
|
|
|
|
-- Add indexes for the new columns
|
|
CREATE INDEX idx_imdb_metadata_adult ON imdb_metadata(adult);
|
|
CREATE INDEX idx_imdb_metadata_category ON imdb_metadata(category);
|
|
CREATE INDEX idx_imdb_metadata_year ON imdb_metadata(year);
|
|
CREATE INDEX title_gin ON imdb_metadata USING gin(title gin_trgm_ops); |