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

Fix workers count

Browse Source
This commit is contained in:
Kurbanov Bulat
2022-01-22 22:25:39 +03:00
parent 2f222ee55d
commit 924e5738a1
4 changed files with 16 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/app/alembic/versions/9b7cfb422191_.py
View File

@@ -29,10 +29,21 @@ def upgrade():
"object_id", "object_type", name="uc_cached_files_object_id_object_type" "object_id", "object_type", name="uc_cached_files_object_id_object_type"
), ),
) )
op.create_index(
op.f("ix_cached_files_object_id"), "cached_files", ["object_id"], unique=False
)
op.create_index(
op.f("ix_cached_files_object_type"),
"cached_files",
["object_type"],
unique=False,
)
# ### end Alembic commands ### # ### end Alembic commands ###
def downgrade(): def downgrade():
# ### commands auto generated by Alembic - please adjust! ### # ### commands auto generated by Alembic - please adjust! ###
op.drop_table("cached_files") op.drop_table("cached_files")
op.drop_index("ix_cached_files_object_id")
op.drop_index("ix_cached_files_object_type")
# ### end Alembic commands ### # ### end Alembic commands ###

4
src/app/models.py
View File

@@ -14,6 +14,6 @@ class CachedFile(ormar.Model):
constraints = [ormar.UniqueColumns("object_id", "object_type")] constraints = [ormar.UniqueColumns("object_id", "object_type")]
id: int = ormar.Integer(primary_key=True) # type: ignore id: int = ormar.Integer(primary_key=True) # type: ignore
object_id: int = ormar.Integer() # type: ignore object_id: int = ormar.Integer(index=True) # type: ignore
object_type: str = ormar.String(max_length=8) # type: ignore object_type: str = ormar.String(max_length=8, index=True) # type: ignore
data: dict = ormar.JSON() # type: ignore data: dict = ormar.JSON() # type: ignore

4
src/app/views.py
View File

@@ -29,7 +29,7 @@ async def get_cached_file(object_id: int, object_type: str):
) )
if not cached_file: if not cached_file:
cached_file = await cache_file_by_book_id(object_id, object_type) cached_file = await cache_file_by_book_id({}, object_id, object_type)
if not cached_file: if not cached_file:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND) raise HTTPException(status_code=status.HTTP_404_NOT_FOUND)
@@ -44,7 +44,7 @@ async def download_cached_file(object_id: int, object_type: str):
) )
if not cached_file: if not cached_file:
cached_file = await cache_file_by_book_id(object_id, object_type) cached_file = await cache_file_by_book_id({}, object_id, object_type)
if not cached_file: if not cached_file:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND) raise HTTPException(status_code=status.HTTP_404_NOT_FOUND)

2
src/core/setup_arq.py
View File

@@ -24,4 +24,4 @@ class WorkerSettings:
on_startup = startup on_startup = startup
on_shutdown = shutdown on_shutdown = shutdown
redis_settings = get_redis_settings() redis_settings = get_redis_settings()
max_jobs = 4 max_jobs = 2