def map_growing(self, *, bids: ReservationSet):
"""
Maps reservations that are growing in this cycle (redeems or expanding
extends), and removes them from the bid set.
@param bids set of deferred operations for this cycle (non-null)
@throws Exception in case of error
"""
# self.logger.debug("Processing growing requests")
rids_to_remove = []
node_id_to_reservations = {}
for reservation in bids.values():
if reservation.is_terminal():
continue
adjust = reservation.get_deficit()
if adjust > 0:
if reservation.is_extending_lease():
self.logger.debug(
f"**Growing reservation by {adjust}:{reservation}")
else:
self.logger.debug(
f"**Redeeming reservation by {adjust}:{reservation}")
node_id_to_reservations = self.map(
reservation=reservation,
node_id_to_reservations=node_id_to_reservations)
rids_to_remove.append(reservation.get_reservation_id())
for rid in rids_to_remove:
bids.remove_by_rid(rid=rid)
def map_shrinking(self, *, bids: ReservationSet):
"""
Maps reservations that are shrinking or staying the same (extending with
no flex) in this cycle, and removes them from the bid set.
@param bids set of deferred operations for this cycle (non-null)
@raises Exception in case of error
"""
# self.logger.debug("Processing shrinking requests")
rids_to_remove = []
node_id_to_reservations = {}
for reservation in bids.values():
adjust = reservation.get_deficit()
if adjust > 0:
continue
if not reservation.is_terminal(
) and reservation.is_extending_lease():
if adjust < 0:
self.logger.debug(
f"**Shrinking reservation by {adjust}:{reservation}")
else:
self.logger.debug(
f"**Extending reservation (no flex): {reservation}")
node_id_to_reservations = self.map(
reservation=reservation,
node_id_to_reservations=node_id_to_reservations)
rids_to_remove.append(reservation.get_reservation_id())
for rid in rids_to_remove:
bids.remove_by_rid(rid=rid)