Source code for cobald.controller.linear

import trio

from cobald.interfaces import Pool, Controller

from cobald.daemon import service

[docs]@service(flavour=trio) class LinearController(Controller): """ Controller that linearly increases or decreases demand :param target: the pool to manage :param low_utilisation: pool utilisation below which resources are decreased :param high_allocation: pool allocation above which resources are increased :param rate: maximum change of demand in resources per second :param interval: interval between adjustments in seconds """ def __init__( self, target: Pool, low_utilisation=0.5, high_allocation=0.5, rate=1, interval=1 ): super().__init__(target=target) assert rate > 0 self.rate = rate self.interval = interval assert low_utilisation <= high_allocation self.low_utilisation = low_utilisation self.high_allocation = high_allocation
[docs] async def run(self): while True: self.regulate(self.interval) await trio.sleep(self.interval)
[docs] def regulate(self, interval): if < self.low_utilisation: -= interval * self.rate elif > self.high_allocation: += interval * self.rate