sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2024-11-23 23:48:56 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 16

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
Show stats 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.