Cutlery/Kyoo
1
0
Fork 0
mirror of https://github.com/zoriya/Kyoo.git synced 2025-06-20 14:00:35 -04:00
Code Issues Packages Projects Releases Wiki Activity

Add search vector & index

Browse Source
This commit is contained in:
Zoe Roux 2025-01-11 23:38:54 +01:00
parent 26470136af
commit 23dc0a2f88
2 changed files with 25 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
api/src/db/schema/shows.ts
View File

@ -1,4 +1,4 @@
import { relations, sql } from "drizzle-orm"; import { relations, SQL, sql } from "drizzle-orm";
import { import {
check, check,
date, date,
@ -12,7 +12,7 @@ import {
uuid, uuid,
varchar, varchar,
} from "drizzle-orm/pg-core"; } from "drizzle-orm/pg-core";
import { image, language, schema } from "./utils"; import { image, language, schema, tsvector } from "./utils";
export const showKind = schema.enum("show_kind", ["serie", "movie"]); export const showKind = schema.enum("show_kind", ["serie", "movie"]);
export const showStatus = schema.enum("show_status", [ export const showStatus = schema.enum("show_status", [
@ -110,8 +110,22 @@ export const showTranslations = schema.table(
banner: image(), banner: image(),
logo: image(), logo: image(),
trailerUrl: text(), trailerUrl: text(),
// TODO: use a real language instead of simple here (we could use the `language` column but dic names
// are `english` and not `en`.)
// we'll also need to handle fallback when the language has no dict available on pg.
search: tsvector().generatedAlwaysAs((): SQL => sql`
setweight(to_tsvector('simple', ${showTranslations.name}), 'A') ||
setweight(to_tsvector('simple', array_to_string(${showTranslations.aliases}, ' ')), 'B') ||
setweight(to_tsvector('simple', array_to_string(${showTranslations.tags}, ' ')), 'C') ||
setweight(to_tsvector('simple', coalese(${showTranslations.tagline}, '')), 'D') ||
setweight(to_tsvector('simple', coalese(${showTranslations.description}, '')), 'D')
`),
}, },
(t) => [primaryKey({ columns: [t.pk, t.language] })], (t) => [
primaryKey({ columns: [t.pk, t.language] }),
index("search").using("gin", t.search),
],
); );
export const showsRelations = relations(shows, ({ many, one }) => ({ export const showsRelations = relations(shows, ({ many, one }) => ({

8
api/src/db/schema/utils.ts
View File

@ -13,6 +13,7 @@ import type { CasingCache } from "drizzle-orm/casing";
import type { AnyMySqlSelect } from "drizzle-orm/mysql-core"; import type { AnyMySqlSelect } from "drizzle-orm/mysql-core";
import { import {
type AnyPgSelect, type AnyPgSelect,
customType,
jsonb, jsonb,
pgSchema, pgSchema,
varchar, varchar,
@ -82,3 +83,10 @@ export function conflictUpdateAllExcept<
export function sqlarr(array: unknown[]) { export function sqlarr(array: unknown[]) {
return `{${array.map((item) => `"${item}"`).join(",")}}`; return `{${array.map((item) => `"${item}"`).join(",")}}`;
} }
// idk why they didn't implement this one
export const tsvector = customType<{ data: string }>({
dataType() {
return "tsvector";
},
});