klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-07 21:29:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix database connections leakage (#980)

Browse Source
This commit is contained in:
Ilya Khrustalev
2024-11-17 19:15:05 +01:00
parent 69ef6829c1
commit 00b1af8f99
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/khoj/database/adapters/__init__.py
View File

@@ -27,6 +27,7 @@ from django.contrib.sessions.backends.db import SessionStore
from django.db.models import Prefetch, Q from django.db.models import Prefetch, Q
from django.db.models.manager import BaseManager from django.db.models.manager import BaseManager
from django.db.utils import IntegrityError from django.db.utils import IntegrityError
from django_apscheduler import util
from django_apscheduler.models import DjangoJob, DjangoJobExecution from django_apscheduler.models import DjangoJob, DjangoJobExecution
from fastapi import HTTPException from fastapi import HTTPException
from pgvector.django import CosineDistance from pgvector.django import CosineDistance
@@ -606,6 +607,7 @@ class ProcessLockAdapters:
logger.debug(f"Skip removing {operation} process lock as it was not set") logger.debug(f"Skip removing {operation} process lock as it was not set")
@util.close_old_connections
def run_with_process_lock(*args, **kwargs): def run_with_process_lock(*args, **kwargs):
"""Wrapper function used for scheduling jobs. """Wrapper function used for scheduling jobs.
Required as APScheduler can't discover the `ProcessLockAdapter.run_with_lock' method on its own. Required as APScheduler can't discover the `ProcessLockAdapter.run_with_lock' method on its own.