controller.py 3.27 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11
from __future__ import print_function

import logging

logger = logging.getLogger('nrm')


class Action(object):

    """Information about a control action."""

12
    def __init__(self, target, command, delta):
13 14
        self.target = target
        self.command = command
15 16 17 18 19 20 21
        self.delta = delta


class ApplicationActuator(object):

    """Actuator in charge of application thread control."""

22
    def __init__(self, am, pubstream):
23
        self.application_manager = am
24
        self.pubstream = pubstream
25 26 27 28 29 30 31 32 33

    def available_actions(self, target):
        ret = []
        for identity, application in \
                self.application_manager.applications.iteritems():
            if target in application.get_allowed_thread_requests():
                delta = application.get_thread_request_impact(target)
                ret.append(Action(application, target, delta))
        return ret
34

35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76
    def execute(self, action):
        target_threads = action.target.threads
        update = {'type': 'application',
                  'command': 'threads',
                  'uuid': action.target.uuid,
                  'event': 'threads',
                  }
        if action.command == 'i':
            payload = target_threads['cur'] + 1
        elif action.command == 'd':
            payload = target_threads['cur'] - 1
        else:
            assert False, "impossible command"
        update['payload'] = payload
        self.pubstream.send_json(update)

    def update(self, action):
        action.target.do_thread_transition(action.command)


class PowerActuator(object):

    """Actuator in charge of power control."""

    def __init__(self, sm):
        self.sensor_manager = sm

    def available_actions(self, target):
        actions = []
        pl = self.sensor_manager.get_powerlimits()
        logger.info("power limits: %r:", pl)
        for k in pl:
            r = range(int(pl[k]['curW']), int(pl[k]['maxW']))
            actions.extend([Action(k, s, s - r[0]) for s in r])
        return actions

    def execute(self, action):
        self.sensor_manager.set_powerlimit(action.target, action.command)

    def update(self, action):
        pass

77 78 79 80 81

class Controller(object):

    """Implements a control loop for resource management."""

82 83
    def __init__(self, actuators):
        self.actuators = actuators
84 85 86 87

    def planify(self, target, machineinfo):
        """Plan the next action for the control loop."""
        total_power = machineinfo['energy']['power']['total']
88
        direction = None
89
        if total_power < target:
90
            direction = 'i'
91
        elif total_power > target:
92 93 94
            direction = 'd'

        if direction:
95 96 97 98
            actions = []
            for act in self.actuators:
                newactions = act.available_actions(direction)
                actions.extend([(a, act) for a in newactions])
99 100
            if actions:
                # TODO: better choice
101
                actions.sort(key=lambda x: x[0].delta)
102 103
                return actions.pop()
            else:
104
                return (None, None)
105

106
    def execute(self, action, actuator):
107
        """Build the action for the appropriate manager."""
108
        actuator.execute(action)
109

110
    def update(self, action, actuator):
111
        """Update tracking across the board to reflect the last action."""
112
        actuator.update(action)