mirror of
https://github.com/ZwareBear/awx.git
synced 2026-04-24 08:51:48 -05:00
89 lines
3.0 KiB
Python
89 lines
3.0 KiB
Python
# Copyright (c) 2015 Ansible, Inc. (formerly AnsibleWorks, Inc.)
|
|
# All Rights Reserved
|
|
|
|
import logging
|
|
from datetime import datetime
|
|
|
|
from django.core.management.base import NoArgsCommand
|
|
|
|
from awx.fact.models.fact import * # noqa
|
|
from awx.main.socket import Socket
|
|
|
|
logger = logging.getLogger('awx.main.commands.run_fact_cache_receiver')
|
|
|
|
class FactCacheReceiver(object):
|
|
def __init__(self):
|
|
self.timestamp = None
|
|
|
|
def _determine_module(self, facts):
|
|
# Symantically determine the module type
|
|
if len(facts) == 1:
|
|
return facts.iterkeys().next()
|
|
return 'ansible'
|
|
|
|
def _extract_module_facts(self, module, facts):
|
|
if module in facts:
|
|
f = facts[module]
|
|
return f
|
|
return facts
|
|
|
|
def process_facts(self, facts):
|
|
module = self._determine_module(facts)
|
|
facts = self._extract_module_facts(module, facts)
|
|
return (module, facts)
|
|
|
|
def process_fact_message(self, message):
|
|
hostname = message['host']
|
|
facts_data = message['facts']
|
|
date_key = message['date_key']
|
|
|
|
# TODO: in ansible < v2 module_setup is emitted for "smart" fact caching.
|
|
# ansible v2 will not emit this message. Thus, this can be removed at that time.
|
|
if 'module_setup' in facts_data and len(facts_data) == 1:
|
|
return
|
|
|
|
try:
|
|
host = FactHost.objects.get(hostname=hostname)
|
|
except FactHost.DoesNotExist:
|
|
host = FactHost(hostname=hostname)
|
|
host.save()
|
|
except FactHost.MultipleObjectsReturned:
|
|
query = "db['fact_host'].find(hostname=%s)" % hostname
|
|
logger.warn('Database inconsistent. Multiple FactHost "%s" exist. Try the query %s to find the records.' % (hostname, query))
|
|
return
|
|
except Exception, e:
|
|
logger.error("Exception communicating with Fact Cache Database: %s" % str(e))
|
|
return
|
|
|
|
(module, facts) = self.process_facts(facts_data)
|
|
self.timestamp = datetime.fromtimestamp(date_key, None)
|
|
|
|
try:
|
|
# Update existing Fact entry
|
|
version_obj = FactVersion.objects.get(timestamp=self.timestamp, host=host, module=module)
|
|
Fact.objects(id=version_obj.fact.id).update_one(fact=facts)
|
|
except FactVersion.DoesNotExist:
|
|
# Create new Fact entry
|
|
(fact_obj, version_obj) = Fact.add_fact(self.timestamp, facts, host, module)
|
|
|
|
def run_receiver(self):
|
|
with Socket('fact_cache', 'r') as facts:
|
|
for message in facts.listen():
|
|
if 'host' not in message or 'facts' not in message or 'date_key' not in message:
|
|
continue
|
|
self.process_fact_message(message)
|
|
|
|
class Command(NoArgsCommand):
|
|
'''
|
|
blah blah
|
|
'''
|
|
help = 'Launch the Fact Cache Receiver'
|
|
|
|
def handle_noargs(self, **options):
|
|
fcr = FactCacheReceiver()
|
|
try:
|
|
fcr.run_receiver()
|
|
except KeyboardInterrupt:
|
|
pass
|
|
|