controller.py 3.56 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
    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)
66
67
        if target == 'i':
            for k in pl:
68
                r = range(int(pl[k]['curW'])+1, int(pl[k]['maxW']))
69
70
71
                actions.extend([Action(k, s, s - r[0]) for s in r])
        elif target == 'd':
            for k in pl:
72
73
                r = range(1, int(pl[k]['curW']))
                actions.extend([Action(k, s, r[-1] - s) for s in r])
74
75
76
        return actions

    def execute(self, action):
77
        logger.info("changing power limit: %r, %r", action.command, action.delta)
78
79
80
81
82
        self.sensor_manager.set_powerlimit(action.target, action.command)

    def update(self, action):
        pass

83
84
85
86
87

class Controller(object):

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

88
89
    def __init__(self, actuators):
        self.actuators = actuators
90
91
92
93

    def planify(self, target, machineinfo):
        """Plan the next action for the control loop."""
        total_power = machineinfo['energy']['power']['total']
94
        direction = None
95
        if total_power < target:
96
            direction = 'i'
97
        elif total_power > target:
98
99
100
            direction = 'd'

        if direction:
101
102
103
104
            actions = []
            for act in self.actuators:
                newactions = act.available_actions(direction)
                actions.extend([(a, act) for a in newactions])
105
106
            if actions:
                # TODO: better choice
107
                actions.sort(key=lambda x: x[0].delta)
108
                return actions.pop(0)
109
            else:
110
                return (None, None)
111

112
    def execute(self, action, actuator):
113
        """Build the action for the appropriate manager."""
114
        actuator.execute(action)
115

116
    def update(self, action, actuator):
117
        """Update tracking across the board to reflect the last action."""
118
        actuator.update(action)