mirror of
https://github.com/ansible/awx.git
synced 2026-03-20 02:17:37 -02:30
4f41b50a09
* AAP-57817 Add Redis connection retry using redis-py 7.0+ built-in mechanism * Refactor Redis client helpers to use settings and eliminate code duplication * Create awx/main/utils/redis.py and move Redis client functions to avoid circular imports * Fix subsystem_metrics to share Redis connection pool between client and pipeline * Cache Redis clients in RelayConsumer and RelayWebsocketStatsManager to avoid creating new connection pools on every call * Add cap and base config * Add Redis retry logic with exponential backoff to handle connection failures during long-running operations * Add REDIS_BACKOFF_CAP and REDIS_BACKOFF_BASE settings to allow adjustment of retry timing in worst-case scenarios without code changes * Simplify Redis retry tests by removing unnecessary reload logic
35 lines
1004 B
Python
35 lines
1004 B
Python
# Copyright (c) 2015 Ansible, Inc.
|
|
# All Rights Reserved.
|
|
|
|
# Python
|
|
import json
|
|
import logging
|
|
|
|
# Django
|
|
from django.conf import settings
|
|
|
|
# AWX
|
|
from awx.main.utils.redis import get_redis_client
|
|
|
|
__all__ = ['CallbackQueueDispatcher']
|
|
|
|
|
|
# use a custom JSON serializer so we can properly handle !unsafe and !vault
|
|
# objects that may exist in events emitted by the callback plugin
|
|
# see: https://github.com/ansible/ansible/pull/38759
|
|
class AnsibleJSONEncoder(json.JSONEncoder):
|
|
def default(self, o):
|
|
if getattr(o, 'yaml_tag', None) == '!vault':
|
|
return o.data
|
|
return super(AnsibleJSONEncoder, self).default(o)
|
|
|
|
|
|
class CallbackQueueDispatcher(object):
|
|
def __init__(self):
|
|
self.queue = getattr(settings, 'CALLBACK_QUEUE', '')
|
|
self.logger = logging.getLogger('awx.main.queue.CallbackQueueDispatcher')
|
|
self.connection = get_redis_client()
|
|
|
|
def dispatch(self, obj):
|
|
self.connection.rpush(self.queue, json.dumps(obj, cls=AnsibleJSONEncoder))
|