rohit/probo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refacto document filter

Browse Source 
Signed-off-by: Sacha Al Himdani <sacha@getprobo.com>
This commit is contained in:
Sacha Al Himdani
2025-08-05 10:48:55 +02:00
parent 7e59504462
commit 036970afdd
1 changed files with 20 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
pkg/coredata/document_filter.go
View File

@@ -39,40 +39,28 @@ func NewDocumentTrustCenterFilter() *DocumentFilter {
}
func (f *DocumentFilter) SQLArguments() pgx.StrictNamedArgs {
args := pgx.StrictNamedArgs{}
if f.query != nil {
args["query"] = *f.query
} else {
args["query"] = (*string)(nil)
return pgx.StrictNamedArgs{
"query": f.query,
"show_on_trust_center": f.showOnTrustCenter,
}
if f.showOnTrustCenter != nil {
args["show_on_trust_center"] = *f.showOnTrustCenter
} else {
args["show_on_trust_center"] = (*bool)(nil)
}
return args
}
func (f *DocumentFilter) SQLFragment() string {
sql := `(
CASE
WHEN @query::text IS NOT NULL AND @query::text != '' THEN
search_vector @@ (
SELECT to_tsquery('simple', string_agg(lexeme || ':*', ' & '))
FROM unnest(regexp_split_to_array(trim(@query::text), '\s+')) AS lexeme
)
ELSE TRUE
END
AND
CASE
WHEN @show_on_trust_center::boolean IS NOT NULL THEN
show_on_trust_center = @show_on_trust_center::boolean
ELSE TRUE
END
)`
return sql
return `
(
CASE
WHEN @query::text IS NOT NULL AND @query::text != '' THEN
search_vector @@ (
SELECT to_tsquery('simple', string_agg(lexeme || ':*', ' & '))
FROM unnest(regexp_split_to_array(trim(@query::text), '\s+')) AS lexeme
)
ELSE TRUE
END
AND
CASE
WHEN @show_on_trust_center::boolean IS NOT NULL THEN
show_on_trust_center = @show_on_trust_center::boolean
ELSE TRUE
END
)`
}