class Credit(Thread):
def __init__(self):
Thread.__init__(self)
self.event = Event()
self.unit = 1
self.charging_unit = 60
# self.credit_step = 1
self.credit_step = 5
self.sanity_step = 15
self.count = 0
self.daemon = True
self.controller = Controller()
self.controller.setup_default()
init('/var/log/cpsdirector/debugging.log')
self._logger = create_logger(__name__)
self._logger.setLevel(logging.DEBUG)
def run(self):
while not self.event.is_set():
self.count += 1
if self.count % self.credit_step == 0:
self.check_credits()
if self.count % self.sanity_step == 0:
self.check_sanity()
if self.count == max(self.credit_step, self.sanity_step):
self.count = 0
self.event.wait(self.unit * 60)
def stop(self):
self.event.set()
def check_credits(self):
from cpsdirector.application import Application
from cpsdirector.application import _deleteapp as stop_application
self._logger.debug('Checking credits @%s' % datetime.now())
users = User.query.all()
for user in users:
to_charge = 0
rescs = Resource.query.join(Application).filter_by(user_id=user.uid)
for res in rescs:
m_f_c = self.mins_from_creation(res.created)
tot_cost = m_f_c / self.charging_unit
if tot_cost >= res.charged or res.charged == 0:
self._logger.debug('Charge for resource %s' % res.vmid)
to_charge += tot_cost - res.charged + 1
res.charged += to_charge
user.credit -= to_charge
if user.credit < 0:
user.credit == 0
if user.credit == 0:
self._logger.debug('Stop all the applications for user %s\n' % user.uid)
apps = Application.query.filter_by(user_id=user.uid)
for app in apps:
stop_application(user.uid, app.aid, False)
if to_charge != 0:
db.session.commit()
def check_sanity(self):
self._logger.debug('Sanity check @%s' % datetime.now())
vms = self.controller.list_vms()
res_vmids = [res.vmid for res in Resource.query.all()]
vms_to_remove = []
for vm in vms:
if vm.vmid not in res_vmids and self.mins_from_creation(vm.created) > 1:
vms_to_remove += [vm]
if len(vms_to_remove) > 0:
self._logger.debug('Removing the following VMs as they are not present in the list of resources: %s' % [vm.vmid for vm in vms_to_remove])
self.controller.delete_nodes(vms_to_remove)
def mins_from_creation(self, created):
now = datetime.now()
created_ts = time.mktime(created.timetuple())
now_ts = time.mktime(now.timetuple())
return int(now_ts-created_ts) / 60
