This commit is contained in:
pablodanswer 2024-10-20 11:52:24 -07:00
parent 277d37e0ee
commit 04f965e656
5 changed files with 16 additions and 13 deletions

View File

@ -91,7 +91,7 @@ def on_task_postrun(
task_id: str | None = None, task_id: str | None = None,
task: Task | None = None, task: Task | None = None,
args: tuple | None = None, args: tuple | None = None,
kwargs: dict | None = None, kwargs: dict[str, Any] | None = None,
retval: Any | None = None, retval: Any | None = None,
state: str | None = None, state: str | None = None,
**kwds: Any, **kwds: Any,
@ -111,10 +111,16 @@ def on_task_postrun(
return return
# Get tenant_id directly from kwargs- each celery task has a tenant_id kwarg # Get tenant_id directly from kwargs- each celery task has a tenant_id kwarg
if "tenant_id" not in kwargs: if not kwargs:
logger.error(f"Task {task.name} (ID: {task_id}) is missing tenant_id in kwargs") logger.error(f"Task {task.name} (ID: {task_id}) is missing kwargs")
tenant_id = None
else:
tenant_id = kwargs.get("tenant_id")
tenant_id = kwargs.get("tenant_id", None) task_logger.debug(
f"Task {task.name} (ID: {task_id}) completed with state: {state} "
f"{f'for tenant_id={tenant_id}' if tenant_id else ''}"
)
if state not in READY_STATES: if state not in READY_STATES:
return return
@ -226,11 +232,8 @@ def on_worker_init(sender: Any, **kwargs: Any) -> None:
time_start = time.monotonic() time_start = time.monotonic()
logger.notice("Redis: Readiness check starting.") logger.notice("Redis: Readiness check starting.")
while True: while True:
try: if r.exists(DanswerRedisLocks.PRIMARY_WORKER):
if r.ping():
break break
except Exception:
logger.noticeinfo("Redis: Ping failed.")
time_elapsed = time.monotonic() - time_start time_elapsed = time.monotonic() - time_start
logger.info( logger.info(

View File

@ -140,7 +140,7 @@ POSTGRES_PASSWORD = urllib.parse.quote_plus(
os.environ.get("POSTGRES_PASSWORD") or "password" os.environ.get("POSTGRES_PASSWORD") or "password"
) )
POSTGRES_HOST = os.environ.get("POSTGRES_HOST") or "localhost" POSTGRES_HOST = os.environ.get("POSTGRES_HOST") or "localhost"
POSTGRES_PORT = os.environ.get("POSTGRES_PORT") or "5433" POSTGRES_PORT = os.environ.get("POSTGRES_PORT") or "5432"
POSTGRES_DB = os.environ.get("POSTGRES_DB") or "postgres" POSTGRES_DB = os.environ.get("POSTGRES_DB") or "postgres"
POSTGRES_API_SERVER_POOL_SIZE = int( POSTGRES_API_SERVER_POOL_SIZE = int(

View File

@ -313,7 +313,7 @@ services:
- POSTGRES_USER=${POSTGRES_USER:-postgres} - POSTGRES_USER=${POSTGRES_USER:-postgres}
- POSTGRES_PASSWORD=${POSTGRES_PASSWORD:-password} - POSTGRES_PASSWORD=${POSTGRES_PASSWORD:-password}
ports: ports:
- "5433:5432" - "5432:5432"
volumes: volumes:
- db_volume:/var/lib/postgresql/data - db_volume:/var/lib/postgresql/data

View File

@ -313,7 +313,7 @@ services:
- POSTGRES_USER=${POSTGRES_USER:-postgres} - POSTGRES_USER=${POSTGRES_USER:-postgres}
- POSTGRES_PASSWORD=${POSTGRES_PASSWORD:-password} - POSTGRES_PASSWORD=${POSTGRES_PASSWORD:-password}
ports: ports:
- "5433:5432" - "5432:5432"
volumes: volumes:
- db_volume:/var/lib/postgresql/data - db_volume:/var/lib/postgresql/data

View File

@ -157,7 +157,7 @@ services:
- POSTGRES_USER=${POSTGRES_USER:-postgres} - POSTGRES_USER=${POSTGRES_USER:-postgres}
- POSTGRES_PASSWORD=${POSTGRES_PASSWORD:-password} - POSTGRES_PASSWORD=${POSTGRES_PASSWORD:-password}
ports: ports:
- "5433" - "5432"
volumes: volumes:
- db_volume:/var/lib/postgresql/data - db_volume:/var/lib/postgresql/data