mirror of
https://github.com/ansible/awx.git
synced 2026-06-21 06:37:45 -02:30
49e21d7c1c
* Move PG version check to check_db command Move to utils, check in pre_migrate signal * Add back in environment var skip * Add tests for compliance tests Assisted-By: claude
62 lines
2.3 KiB
Python
62 lines
2.3 KiB
Python
# Copyright (c) 2017 Ansible by Red Hat
|
|
# All Rights Reserved.
|
|
|
|
from typing import Optional
|
|
import os
|
|
|
|
from awx.settings.application_name import set_application_name
|
|
from awx import MODE
|
|
|
|
from django.conf import settings
|
|
from django.db import connection
|
|
|
|
|
|
def set_connection_name(function):
|
|
set_application_name(settings.DATABASES, settings.CLUSTER_HOST_ID, function=function)
|
|
|
|
|
|
def bulk_update_sorted_by_id(model, objects, fields, batch_size=1000):
|
|
"""
|
|
Perform a sorted bulk update on model instances to avoid database deadlocks.
|
|
|
|
This function was introduced to prevent deadlocks observed in the AWX Controller
|
|
when concurrent jobs attempt to update different fields on the same `main_hosts` table.
|
|
Specifically, deadlocks occurred when one process updated `last_job_id` while another
|
|
simultaneously updated `ansible_facts`.
|
|
|
|
By sorting updates ID, we ensure a consistent update order,
|
|
which helps avoid the row-level locking contention that can lead to deadlocks
|
|
in PostgreSQL when multiple processes are involved.
|
|
|
|
Returns:
|
|
int: The number of rows affected by the update.
|
|
"""
|
|
objects = [obj for obj in objects if obj.id is not None]
|
|
if not objects:
|
|
return 0 # Return 0 when nothing is updated
|
|
|
|
sorted_objects = sorted(objects, key=lambda obj: obj.id)
|
|
return model.objects.bulk_update(sorted_objects, fields, batch_size=batch_size)
|
|
|
|
|
|
MIN_PG_VERSION = 12
|
|
|
|
|
|
def db_requirement_violations() -> Optional[str]:
|
|
if os.getenv('SKIP_PG_VERSION_CHECK', False):
|
|
return None
|
|
if connection.vendor == 'postgresql':
|
|
|
|
# enforce the postgres version is a minimum of 12 (we need this for partitioning); if not, then terminate program with exit code of 1
|
|
# In the future if we require a feature of a version of postgres > 12 this should be updated to reflect that.
|
|
# The return of connection.pg_version is something like 12013
|
|
major_version = connection.pg_version // 10000
|
|
if major_version < MIN_PG_VERSION:
|
|
return f"At a minimum, postgres version {MIN_PG_VERSION} is required, found {major_version}\n"
|
|
|
|
return None
|
|
else:
|
|
if MODE == 'production':
|
|
return f"Running server with '{connection.vendor}' type database is not supported\n"
|
|
return None
|