mirror of
https://github.com/ansible/awx.git
synced 2026-01-28 23:04:41 -03:30
39 lines
1.0 KiB
Python
39 lines
1.0 KiB
Python
import redis
|
|
import logging
|
|
|
|
from django.conf.urls import url
|
|
from django.conf import settings
|
|
|
|
from channels.auth import AuthMiddlewareStack
|
|
from channels.routing import ProtocolTypeRouter, URLRouter
|
|
|
|
from . import consumers
|
|
|
|
|
|
logger = logging.getLogger('awx.main.routing')
|
|
|
|
|
|
class AWXProtocolTypeRouter(ProtocolTypeRouter):
|
|
def __init__(self, *args, **kwargs):
|
|
try:
|
|
r = redis.Redis.from_url(settings.BROKER_URL)
|
|
for k in r.scan_iter('asgi:*', 500):
|
|
logger.debug(f"cleaning up Redis key {k}")
|
|
r.delete(k)
|
|
except redis.exceptions.RedisError as e:
|
|
logger.warn("encountered an error communicating with redis.")
|
|
raise e
|
|
super().__init__(*args, **kwargs)
|
|
|
|
|
|
websocket_urlpatterns = [
|
|
url(r'websocket/$', consumers.EventConsumer),
|
|
url(r'websocket/broadcast/$', consumers.BroadcastConsumer),
|
|
]
|
|
|
|
application = AWXProtocolTypeRouter(
|
|
{
|
|
'websocket': AuthMiddlewareStack(URLRouter(websocket_urlpatterns)),
|
|
}
|
|
)
|