External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-08-04 03:50:01 -02:30
Code Issues Packages Projects Releases Wiki Activity

rename field to store_facts and fix up tests

Browse Source
This commit is contained in:
Chris Meyers
2017-04-28 13:45:57 -04:00
parent 6dec9d1014
commit 0be384af18
7 changed files with 90 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
awx/main/management/commands/run_fact_cache_receiver.py
View File

@@ -53,6 +53,24 @@ class FactBrokerWorker(ConsumerMixin):
facts = self._extract_module_facts(module, facts)
return (module, facts)
def _do_fact_scan_create_update(self, host_obj, module_name, facts, timestamp):
try:
fact_obj = Fact.objects.get(host__id=host_obj.id, module=module_name, timestamp=timestamp)
fact_obj.facts = facts
fact_obj.save()
logger.info('Updated existing fact <%s>' % (fact_obj.id))
except Fact.DoesNotExist:
# Create new Fact entry
fact_obj = Fact.add_fact(host_obj.id, module_name, self.timestamp, facts)
logger.info('Created new fact <fact_id, module> <%s, %s>' % (fact_obj.id, module_name))
analytics_logger.info('Received message with fact data', extra=dict(
module_name=module_name, facts_data=facts))
return fact_obj
def _do_gather_facts_update(self, host_obj, module_name, facts, timestamp):
host_obj.update_ansible_facts(module=module_name, facts=facts, timestamp=self.timestamp)
return host_obj
def process_fact_message(self, body, message):
hostname = body['host']
inventory_id = body['inventory_id']
@@ -99,20 +117,10 @@ class FactBrokerWorker(ConsumerMixin):
ret = None
# Update existing Fact entry
if is_fact_scan is True:
try:
fact_obj = Fact.objects.get(host__id=host_obj.id, module=module_name, timestamp=self.timestamp)
fact_obj.facts = facts
fact_obj.save()
logger.info('Updated existing fact <%s>' % (fact_obj.id))
except Fact.DoesNotExist:
# Create new Fact entry
fact_obj = Fact.add_fact(host_obj.id, module_name, self.timestamp, facts)
logger.info('Created new fact <fact_id, module> <%s, %s>' % (fact_obj.id, module_name))
analytics_logger.info('Received message with fact data', extra=dict(
module_name=module_name, facts_data=facts))
ret = fact_obj
if job.gather_facts is True:
host_obj.update_ansible_facts(module=module_name, facts=facts, timestamp=self.timestamp)
ret = self._do_fact_scan_create_update(host_obj, module_name, facts, self.timestamp)
if job.store_facts is True:
self._do_gather_facts_update(host_obj, module_name, facts, self.timestamp)
message.ack()
return ret