flibusta-apps/book_library_server
1
0
Fork 0
mirror of https://github.com/flibusta-apps/book_library_server.git synced 2025-12-06 15:15:36 +01:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Add search limits

Browse Source
This commit is contained in:
Kurbanov Bulat
2021-12-10 15:40:07 +03:00
parent 7e825d099b
commit 962650175e
3 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
fastapi_book_server/app/services/author.py
View File

@@ -3,7 +3,7 @@ from app.models import Author
from app.services.common import TRGMSearchService
GET_OBJECTS_IDS_QUERY = """
GET_OBJECT_IDS_QUERY = """
SELECT ARRAY(
WITH filtered_authors AS (
SELECT
@@ -32,6 +32,7 @@ SELECT ARRAY(
)
SELECT fauthors.id FROM filtered_authors as fauthors
ORDER BY fauthors.sml DESC, fauthors.books_count DESC
LIMIT 210
);
"""
@@ -39,4 +40,4 @@ SELECT ARRAY(
class AuthorTGRMSearchService(TRGMSearchService):
MODEL_CLASS = Author
PREFETCH_RELATED = ["source", "annotations"]
GET_OBJECT_IDS_QUERY = GET_OBJECTS_IDS_QUERY
GET_OBJECT_IDS_QUERY = GET_OBJECT_IDS_QUERY

5
fastapi_book_server/app/services/book.py
View File

@@ -8,7 +8,7 @@ from app.services.common import TRGMSearchService
from app.serializers.book import CreateBook, CreateRemoteBook
GET_OBJECTS_IDS_QUERY = """
GET_OBJECT_IDS_QUERY = """
SELECT ARRAY(
WITH filtered_books AS (
SELECT id, similarity(title, :query) as sml FROM books
@@ -16,6 +16,7 @@ SELECT ARRAY(
)
SELECT fbooks.id FROM filtered_books as fbooks
ORDER BY fbooks.sml DESC, fbooks.id
LIMIT 210
);
"""
@@ -23,7 +24,7 @@ SELECT ARRAY(
class BookTGRMSearchService(TRGMSearchService):
MODEL_CLASS = BookDB
PREFETCH_RELATED = ["source", "authors", "annotations"]
GET_OBJECT_IDS_QUERY = GET_OBJECTS_IDS_QUERY
GET_OBJECT_IDS_QUERY = GET_OBJECT_IDS_QUERY
class BookCreator:

7
fastapi_book_server/app/services/sequence.py
View File

@@ -3,8 +3,8 @@ from app.models import Sequence
from app.services.common import TRGMSearchService
GET_OBJECTS_IDS_QUERY = """
EXPLAIN ANALYZE SELECT ARRAY (
GET_OBJECT_IDS_QUERY = """
SELECT ARRAY (
WITH filtered_sequences AS (
SELECT
id,
@@ -24,6 +24,7 @@ EXPLAIN ANALYZE SELECT ARRAY (
)
SELECT fsequences.id FROM filtered_sequences as fsequences
ORDER BY fsequences.sml DESC, fsequences.books_count DESC
LIMIT 210
);
"""
@@ -31,4 +32,4 @@ EXPLAIN ANALYZE SELECT ARRAY (
class SequenceTGRMSearchService(TRGMSearchService):
MODEL_CLASS = Sequence
PREFETCH_RELATED = ["source"]
GET_OBJECTS_IDS_QUERY = GET_OBJECTS_IDS_QUERY
GET_OBJECT_IDS_QUERY = GET_OBJECT_IDS_QUERY