mirror of
https://github.com/ansible/awx.git
synced 2026-04-05 01:59:25 -02:30
Fix up usage of django-guid
It has replaced the class-based middleware, everything is function-based now.
This commit is contained in:
@@ -6,7 +6,8 @@ from multiprocessing import Process
|
|||||||
from django.conf import settings
|
from django.conf import settings
|
||||||
from django.db import connections
|
from django.db import connections
|
||||||
from schedule import Scheduler
|
from schedule import Scheduler
|
||||||
from django_guid.middleware import GuidMiddleware
|
from django_guid import set_guid
|
||||||
|
from django_guid.utils import generate_guid
|
||||||
|
|
||||||
from awx.main.dispatch.worker import TaskWorker
|
from awx.main.dispatch.worker import TaskWorker
|
||||||
|
|
||||||
@@ -32,7 +33,7 @@ class Scheduler(Scheduler):
|
|||||||
# If the database connection has a hiccup, re-establish a new
|
# If the database connection has a hiccup, re-establish a new
|
||||||
# connection
|
# connection
|
||||||
conn.close_if_unusable_or_obsolete()
|
conn.close_if_unusable_or_obsolete()
|
||||||
GuidMiddleware.set_guid(GuidMiddleware._generate_guid())
|
set_guid(generate_guid())
|
||||||
self.run_pending()
|
self.run_pending()
|
||||||
except Exception:
|
except Exception:
|
||||||
logger.exception('encountered an error while scheduling periodic tasks')
|
logger.exception('encountered an error while scheduling periodic tasks')
|
||||||
|
|||||||
@@ -16,7 +16,7 @@ from queue import Full as QueueFull, Empty as QueueEmpty
|
|||||||
from django.conf import settings
|
from django.conf import settings
|
||||||
from django.db import connection as django_connection, connections
|
from django.db import connection as django_connection, connections
|
||||||
from django.core.cache import cache as django_cache
|
from django.core.cache import cache as django_cache
|
||||||
from django_guid.middleware import GuidMiddleware
|
from django_guid import set_guid
|
||||||
from jinja2 import Template
|
from jinja2 import Template
|
||||||
import psutil
|
import psutil
|
||||||
|
|
||||||
@@ -436,7 +436,7 @@ class AutoscalePool(WorkerPool):
|
|||||||
|
|
||||||
def write(self, preferred_queue, body):
|
def write(self, preferred_queue, body):
|
||||||
if 'guid' in body:
|
if 'guid' in body:
|
||||||
GuidMiddleware.set_guid(body['guid'])
|
set_guid(body['guid'])
|
||||||
try:
|
try:
|
||||||
# when the cluster heartbeat occurs, clean up internally
|
# when the cluster heartbeat occurs, clean up internally
|
||||||
if isinstance(body, dict) and 'cluster_node_heartbeat' in body['task']:
|
if isinstance(body, dict) and 'cluster_node_heartbeat' in body['task']:
|
||||||
|
|||||||
@@ -5,7 +5,7 @@ import json
|
|||||||
from uuid import uuid4
|
from uuid import uuid4
|
||||||
|
|
||||||
from django.conf import settings
|
from django.conf import settings
|
||||||
from django_guid.middleware import GuidMiddleware
|
from django_guid import get_guid
|
||||||
|
|
||||||
from . import pg_bus_conn
|
from . import pg_bus_conn
|
||||||
|
|
||||||
@@ -76,7 +76,7 @@ class task:
|
|||||||
logger.error(msg)
|
logger.error(msg)
|
||||||
raise ValueError(msg)
|
raise ValueError(msg)
|
||||||
obj = {'uuid': task_id, 'args': args, 'kwargs': kwargs, 'task': cls.name}
|
obj = {'uuid': task_id, 'args': args, 'kwargs': kwargs, 'task': cls.name}
|
||||||
guid = GuidMiddleware.get_guid()
|
guid = get_guid()
|
||||||
if guid:
|
if guid:
|
||||||
obj['guid'] = guid
|
obj['guid'] = guid
|
||||||
obj.update(**kw)
|
obj.update(**kw)
|
||||||
|
|||||||
@@ -9,7 +9,7 @@ from django.conf import settings
|
|||||||
from django.utils.timezone import now as tz_now
|
from django.utils.timezone import now as tz_now
|
||||||
from django.db import DatabaseError, OperationalError, connection as django_connection
|
from django.db import DatabaseError, OperationalError, connection as django_connection
|
||||||
from django.db.utils import InterfaceError, InternalError
|
from django.db.utils import InterfaceError, InternalError
|
||||||
from django_guid.middleware import GuidMiddleware
|
from django_guid import set_guid
|
||||||
|
|
||||||
import psutil
|
import psutil
|
||||||
|
|
||||||
@@ -184,7 +184,7 @@ class CallbackBrokerWorker(BaseWorker):
|
|||||||
if body.get('event') == 'EOF':
|
if body.get('event') == 'EOF':
|
||||||
try:
|
try:
|
||||||
if 'guid' in body:
|
if 'guid' in body:
|
||||||
GuidMiddleware.set_guid(body['guid'])
|
set_guid(body['guid'])
|
||||||
final_counter = body.get('final_counter', 0)
|
final_counter = body.get('final_counter', 0)
|
||||||
logger.info('Event processing is finished for Job {}, sending notifications'.format(job_identifier))
|
logger.info('Event processing is finished for Job {}, sending notifications'.format(job_identifier))
|
||||||
# EOF events are sent when stdout for the running task is
|
# EOF events are sent when stdout for the running task is
|
||||||
@@ -208,7 +208,7 @@ class CallbackBrokerWorker(BaseWorker):
|
|||||||
logger.exception('Worker failed to emit notifications: Job {}'.format(job_identifier))
|
logger.exception('Worker failed to emit notifications: Job {}'.format(job_identifier))
|
||||||
finally:
|
finally:
|
||||||
self.subsystem_metrics.inc('callback_receiver_events_in_memory', -1)
|
self.subsystem_metrics.inc('callback_receiver_events_in_memory', -1)
|
||||||
GuidMiddleware.set_guid('')
|
set_guid('')
|
||||||
return
|
return
|
||||||
|
|
||||||
skip_websocket_message = body.pop('skip_websocket_message', False)
|
skip_websocket_message = body.pop('skip_websocket_message', False)
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ import traceback
|
|||||||
from kubernetes.config import kube_config
|
from kubernetes.config import kube_config
|
||||||
|
|
||||||
from django.conf import settings
|
from django.conf import settings
|
||||||
from django_guid.middleware import GuidMiddleware
|
from django_guid import set_guid
|
||||||
|
|
||||||
from awx.main.tasks.system import dispatch_startup, inform_cluster_of_shutdown
|
from awx.main.tasks.system import dispatch_startup, inform_cluster_of_shutdown
|
||||||
|
|
||||||
@@ -54,7 +54,7 @@ class TaskWorker(BaseWorker):
|
|||||||
args = body.get('args', [])
|
args = body.get('args', [])
|
||||||
kwargs = body.get('kwargs', {})
|
kwargs = body.get('kwargs', {})
|
||||||
if 'guid' in body:
|
if 'guid' in body:
|
||||||
GuidMiddleware.set_guid(body.pop('guid'))
|
set_guid(body.pop('guid'))
|
||||||
_call = TaskWorker.resolve_callable(task)
|
_call = TaskWorker.resolve_callable(task)
|
||||||
if inspect.isclass(_call):
|
if inspect.isclass(_call):
|
||||||
# the callable is a class, e.g., RunJob; instantiate and
|
# the callable is a class, e.g., RunJob; instantiate and
|
||||||
|
|||||||
@@ -8,7 +8,7 @@ import stat
|
|||||||
# Django
|
# Django
|
||||||
from django.utils.timezone import now
|
from django.utils.timezone import now
|
||||||
from django.conf import settings
|
from django.conf import settings
|
||||||
from django_guid.middleware import GuidMiddleware
|
from django_guid import get_guid
|
||||||
|
|
||||||
# AWX
|
# AWX
|
||||||
from awx.main.redact import UriCleaner
|
from awx.main.redact import UriCleaner
|
||||||
@@ -25,7 +25,7 @@ class RunnerCallback:
|
|||||||
def __init__(self, model=None):
|
def __init__(self, model=None):
|
||||||
self.parent_workflow_job_id = None
|
self.parent_workflow_job_id = None
|
||||||
self.host_map = {}
|
self.host_map = {}
|
||||||
self.guid = GuidMiddleware.get_guid()
|
self.guid = get_guid()
|
||||||
self.job_created = None
|
self.job_created = None
|
||||||
self.recent_event_timings = deque(maxlen=settings.MAX_WEBSOCKET_EVENT_RATE)
|
self.recent_event_timings = deque(maxlen=settings.MAX_WEBSOCKET_EVENT_RATE)
|
||||||
self.dispatcher = CallbackQueueDispatcher()
|
self.dispatcher = CallbackQueueDispatcher()
|
||||||
|
|||||||
@@ -15,8 +15,8 @@ from django.apps import apps
|
|||||||
from django.db import models
|
from django.db import models
|
||||||
from django.conf import settings
|
from django.conf import settings
|
||||||
|
|
||||||
|
from django_guid import get_guid
|
||||||
from django_guid.log_filters import CorrelationId
|
from django_guid.log_filters import CorrelationId
|
||||||
from django_guid.middleware import GuidMiddleware
|
|
||||||
|
|
||||||
from awx import MODE
|
from awx import MODE
|
||||||
from awx.main.constants import LOGGER_BLOCKLIST
|
from awx.main.constants import LOGGER_BLOCKLIST
|
||||||
@@ -366,7 +366,7 @@ class SmartFilter(object):
|
|||||||
|
|
||||||
class DefaultCorrelationId(CorrelationId):
|
class DefaultCorrelationId(CorrelationId):
|
||||||
def filter(self, record):
|
def filter(self, record):
|
||||||
guid = GuidMiddleware.get_guid() or '-'
|
guid = get_guid() or '-'
|
||||||
if MODE == 'development':
|
if MODE == 'development':
|
||||||
guid = guid[:8]
|
guid = guid[:8]
|
||||||
record.guid = guid
|
record.guid = guid
|
||||||
|
|||||||
@@ -935,7 +935,7 @@ AWX_CLEANUP_PATHS = True
|
|||||||
RECEPTOR_RELEASE_WORK = True
|
RECEPTOR_RELEASE_WORK = True
|
||||||
|
|
||||||
MIDDLEWARE = [
|
MIDDLEWARE = [
|
||||||
'django_guid.middleware.GuidMiddleware',
|
'django_guid.middleware.guid_middleware',
|
||||||
'awx.main.middleware.TimingMiddleware',
|
'awx.main.middleware.TimingMiddleware',
|
||||||
'django.contrib.sessions.middleware.SessionMiddleware',
|
'django.contrib.sessions.middleware.SessionMiddleware',
|
||||||
'awx.main.middleware.MigrationRanCheckMiddleware',
|
'awx.main.middleware.MigrationRanCheckMiddleware',
|
||||||
|
|||||||
Reference in New Issue
Block a user