Move role cache into the play to avoid roles crossing play boundaries
This commit is contained in:
parent
b520d5bc60
commit
f8ddf2eb04
4 changed files with 15 additions and 24 deletions
|
@ -25,7 +25,6 @@ from ansible import constants as C
|
||||||
from ansible.errors import *
|
from ansible.errors import *
|
||||||
from ansible.executor.task_queue_manager import TaskQueueManager
|
from ansible.executor.task_queue_manager import TaskQueueManager
|
||||||
from ansible.playbook import Playbook
|
from ansible.playbook import Playbook
|
||||||
from ansible.playbook.role import role_reset_has_run
|
|
||||||
from ansible.plugins import module_loader
|
from ansible.plugins import module_loader
|
||||||
from ansible.template import Templar
|
from ansible.template import Templar
|
||||||
|
|
||||||
|
@ -84,9 +83,6 @@ class PlaybookExecutor:
|
||||||
self._display.vv('%d plays in %s' % (len(plays), playbook_path))
|
self._display.vv('%d plays in %s' % (len(plays), playbook_path))
|
||||||
|
|
||||||
for play in plays:
|
for play in plays:
|
||||||
# clear out the flag on all roles indicating they had any tasks run
|
|
||||||
role_reset_has_run()
|
|
||||||
|
|
||||||
# clear any filters which may have been applied to the inventory
|
# clear any filters which may have been applied to the inventory
|
||||||
self._inventory.remove_restriction()
|
self._inventory.remove_restriction()
|
||||||
|
|
||||||
|
|
|
@ -88,6 +88,8 @@ class Play(Base, Taggable, Become):
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
super(Play, self).__init__()
|
super(Play, self).__init__()
|
||||||
|
|
||||||
|
self.ROLE_CACHE = {}
|
||||||
|
|
||||||
def __repr__(self):
|
def __repr__(self):
|
||||||
return self.get_name()
|
return self.get_name()
|
||||||
|
|
||||||
|
@ -322,3 +324,8 @@ class Play(Base, Taggable, Become):
|
||||||
setattr(self, 'roles', roles)
|
setattr(self, 'roles', roles)
|
||||||
del data['roles']
|
del data['roles']
|
||||||
|
|
||||||
|
def copy(self):
|
||||||
|
new_me = super(Play, self).copy()
|
||||||
|
new_me.ROLE_CACHE = self.ROLE_CACHE.copy()
|
||||||
|
return new_me
|
||||||
|
|
||||||
|
|
|
@ -41,7 +41,7 @@ from ansible.plugins import get_all_plugin_loaders, push_basedir
|
||||||
from ansible.utils.vars import combine_vars
|
from ansible.utils.vars import combine_vars
|
||||||
|
|
||||||
|
|
||||||
__all__ = ['Role', 'ROLE_CACHE', 'hash_params', 'role_reset_has_run']
|
__all__ = ['Role', 'hash_params']
|
||||||
|
|
||||||
# FIXME: this should be a utility function, but can't be a member of
|
# FIXME: this should be a utility function, but can't be a member of
|
||||||
# the role due to the fact that it would require the use of self
|
# the role due to the fact that it would require the use of self
|
||||||
|
@ -64,17 +64,6 @@ def hash_params(params):
|
||||||
s.update((k, v))
|
s.update((k, v))
|
||||||
return frozenset(s)
|
return frozenset(s)
|
||||||
|
|
||||||
# The role cache is used to prevent re-loading roles, which
|
|
||||||
# may already exist. Keys into this cache are the SHA1 hash
|
|
||||||
# of the role definition (for dictionary definitions, this
|
|
||||||
# will be based on the repr() of the dictionary object)
|
|
||||||
ROLE_CACHE = dict()
|
|
||||||
|
|
||||||
def role_reset_has_run():
|
|
||||||
for (role_name, cached_roles) in ROLE_CACHE.iteritems():
|
|
||||||
for (hashed_params, role) in cached_roles.iteritems():
|
|
||||||
role._had_task_run = False
|
|
||||||
|
|
||||||
class Role(Base, Become, Conditional, Taggable):
|
class Role(Base, Become, Conditional, Taggable):
|
||||||
|
|
||||||
def __init__(self, play=None):
|
def __init__(self, play=None):
|
||||||
|
@ -111,13 +100,12 @@ class Role(Base, Become, Conditional, Taggable):
|
||||||
# specified for a role as the key and the Role() object itself.
|
# specified for a role as the key and the Role() object itself.
|
||||||
# We use frozenset to make the dictionary hashable.
|
# We use frozenset to make the dictionary hashable.
|
||||||
|
|
||||||
#hashed_params = frozenset(role_include.get_role_params().iteritems())
|
|
||||||
params = role_include.get_role_params()
|
params = role_include.get_role_params()
|
||||||
params['tags'] = role_include.tags
|
params['tags'] = role_include.tags
|
||||||
params['when'] = role_include.when
|
params['when'] = role_include.when
|
||||||
hashed_params = hash_params(params)
|
hashed_params = hash_params(params)
|
||||||
if role_include.role in ROLE_CACHE:
|
if role_include.role in play.ROLE_CACHE:
|
||||||
for (entry, role_obj) in ROLE_CACHE[role_include.role].iteritems():
|
for (entry, role_obj) in play.ROLE_CACHE[role_include.role].iteritems():
|
||||||
if hashed_params == entry:
|
if hashed_params == entry:
|
||||||
if parent_role:
|
if parent_role:
|
||||||
role_obj.add_parent(parent_role)
|
role_obj.add_parent(parent_role)
|
||||||
|
@ -126,10 +114,10 @@ class Role(Base, Become, Conditional, Taggable):
|
||||||
r = Role(play=play)
|
r = Role(play=play)
|
||||||
r._load_role_data(role_include, parent_role=parent_role)
|
r._load_role_data(role_include, parent_role=parent_role)
|
||||||
|
|
||||||
if role_include.role not in ROLE_CACHE:
|
if role_include.role not in play.ROLE_CACHE:
|
||||||
ROLE_CACHE[role_include.role] = dict()
|
play.ROLE_CACHE[role_include.role] = dict()
|
||||||
|
|
||||||
ROLE_CACHE[role_include.role][hashed_params] = r
|
play.ROLE_CACHE[role_include.role][hashed_params] = r
|
||||||
return r
|
return r
|
||||||
|
|
||||||
except RuntimeError:
|
except RuntimeError:
|
||||||
|
|
|
@ -28,7 +28,7 @@ from ansible.inventory.host import Host
|
||||||
from ansible.inventory.group import Group
|
from ansible.inventory.group import Group
|
||||||
from ansible.playbook.handler import Handler
|
from ansible.playbook.handler import Handler
|
||||||
from ansible.playbook.helpers import load_list_of_blocks
|
from ansible.playbook.helpers import load_list_of_blocks
|
||||||
from ansible.playbook.role import ROLE_CACHE, hash_params
|
from ansible.playbook.role import hash_params
|
||||||
from ansible.plugins import _basedirs, filter_loader, lookup_loader, module_loader
|
from ansible.plugins import _basedirs, filter_loader, lookup_loader, module_loader
|
||||||
from ansible.template import Templar
|
from ansible.template import Templar
|
||||||
from ansible.utils.debug import debug
|
from ansible.utils.debug import debug
|
||||||
|
@ -193,7 +193,7 @@ class StrategyBase:
|
||||||
if task_result._task._role is not None and result[0] in ('host_task_ok', 'host_task_failed'):
|
if task_result._task._role is not None and result[0] in ('host_task_ok', 'host_task_failed'):
|
||||||
# lookup the role in the ROLE_CACHE to make sure we're dealing
|
# lookup the role in the ROLE_CACHE to make sure we're dealing
|
||||||
# with the correct object and mark it as executed
|
# with the correct object and mark it as executed
|
||||||
for (entry, role_obj) in ROLE_CACHE[task_result._task._role._role_name].iteritems():
|
for (entry, role_obj) in iterator._play.ROLE_CACHE[task_result._task._role._role_name].iteritems():
|
||||||
hashed_entry = hash_params(task_result._task._role._role_params)
|
hashed_entry = hash_params(task_result._task._role._role_params)
|
||||||
if entry == hashed_entry:
|
if entry == hashed_entry:
|
||||||
role_obj._had_task_run = True
|
role_obj._had_task_run = True
|
||||||
|
|
Loading…
Reference in a new issue