valitydev/salt
14
0
Fork 0
mirror of https://github.com/valitydev/salt.git synced 2024-11-07 08:58:59 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Beacon interval scheduling

Browse Source
This commit is contained in:
Mike Place 2015-02-26 17:49:38 -07:00 committed by C. R. Oldham
parent 596dbc3ef2
commit 50588717a1
1 changed files with 41 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
salt/beacons/__init__.py
View File

@ -2,9 +2,17 @@
'''
This package contains the loader modules for the salt streams system
'''
# Import salt libs
# Import Python libs
from __future__ import absolute_import
import logging
import copy
# Import Salt libs
import salt.loader
log = logging.getLogger(__name__)
class Beacon(object):
'''
@ -13,6 +21,7 @@ class Beacon(object):
def __init__(self, opts):
self.opts = opts
self.beacons = salt.loader.beacons(opts)
self.interval_map = dict()
def process(self, config):
'''
@ -28,11 +37,19 @@ class Beacon(object):
- /var/cache/foo/*
'''
ret = []
b_config = copy.deepcopy(config)
for mod in config:
log.trace('Beacon processing: {0}'.format(mod))
fun_str = '{0}.beacon'.format(mod)
if fun_str in self.beacons:
interval = [arg for arg in config[mod] if 'interval' in arg]
if interval:
b_config[mod].remove(interval[0])
if not self._process_interval(mod, interval):
log.trace('Skipping beacon {0}. Interval not reached.'.format(mod))
continue
tag = 'salt/beacon/{0}/{1}/'.format(self.opts['id'], mod)
raw = self.beacons[fun_str](config[mod])
raw = self.beacons[fun_str](b_config[mod])
for data in raw:
if 'tag' in data:
tag += data.pop('tag')
@ -40,3 +57,25 @@ class Beacon(object):
data['id'] = self.opts['id']
ret.append({'tag': tag, 'data': data})
return ret
def _process_interval(self, mod, interval):
'''
Process beacons with intervals
Return True if a beacon should be run on this loop
'''
log.trace('Processing interval {0} for beacon mod {1}'.format(interval, mod))
loop_interval = self.opts['loop_interval']
if mod in self.interval_map:
log.trace('Processing interval in map')
counter = self.interval_map[mod]
log.trace('Interval counter: {0}'.format(counter))
if counter * loop_interval >= interval[0]['interval']:
self.interval_map[mod] = 1
return True
else:
self.interval_map[mod] += 1
else:
log.trace('Interval process inserting mod: {0}'.format(mod))
self.interval_map[mod] = 1
return False