X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=accounts.py;h=74afcffe90d06e0138d503d2017cccd8755c64bf;hb=c6259a1349497252fac6ba0bbe96512ccd47738b;hp=c064ec65210c1bd944d85e640aec2a37f70d0671;hpb=fc854ce9e3bb24319df3f95545c5764647af6f73;p=nodemanager.git diff --git a/accounts.py b/accounts.py index c064ec6..74afcff 100644 --- a/accounts.py +++ b/accounts.py @@ -1,11 +1,10 @@ -# $Id$ -# $URL$ +### """Functionality common to all account classes. -Each subclass of Account must provide five methods: - (*) create() and destroy(), which are static; - (*) configure(), start(), and stop(), which are not. +Each subclass of Account must provide five methods: + (*) create() and destroy(), which are static; + (*) configure(), start(), and stop(), which are not. configure(), which takes a record as its only argument, does things like set up ssh keys. In addition, an Account subclass must @@ -25,7 +24,6 @@ numbers of accounts, this may cause the NM process to run out of maximum stack size. """ -#import Queue import os import pwd, grp import threading @@ -34,8 +32,6 @@ import logger import tools -# When this variable is true, start after any ensure_created -Startingup = False # shell path -> account class association shell_acct_class = {} # account type -> account class association @@ -78,7 +74,6 @@ class Account: logger.verbose('accounts: Initing account %s'%rec['name']) self.name = rec['name'] self.keys = '' - self.initscriptchanged = False self.configure(rec) @staticmethod @@ -132,9 +127,11 @@ class Worker: self.name = name # username self._acct = None # the account object currently associated with this worker - def ensure_created(self, rec, startingup = Startingup): - """Check account type is still valid. If not, recreate sliver. + def ensure_created(self, rec): + """Check account type is still valid. If not, recreate sliver. If still valid, check if running and configure/start if not.""" + logger.log_data_in_file(rec,"/var/lib/nodemanager/%s.rec.txt"%rec['name'], + 'raw rec captured in ensure_created',logger.LOG_VERBOSE) curr_class = self._get_class() next_class = type_acct_class[rec['type']] if next_class != curr_class: @@ -143,22 +140,38 @@ If still valid, check if running and configure/start if not.""" try: next_class.create(self.name, rec['vref']) finally: create_sem.release() if not isinstance(self._acct, next_class): self._acct = next_class(rec) - if startingup or \ - not self.is_running() or \ - next_class != curr_class or \ - self._acct.initscriptchanged: - self.start(rec) - else: self._acct.configure(rec) + logger.verbose("accounts.ensure_created: %s, running=%r"%(self.name,self.is_running())) + + # reservation_alive is set on reervable nodes, and its value is a boolean + if 'reservation_alive' in rec: + # reservable nodes + if rec['reservation_alive']: + # this sliver has the lease, it is safe to start it + if not self.is_running(): self.start(rec) + else: self.configure(rec) + else: + # not having the lease, do not start it + self.configure(rec) + # usual nodes - preserve old code + # xxx it's not clear what to do when a sliver changes type/class + # in a reservable node + else: + if not self.is_running() or next_class != curr_class: + self.start(rec) + else: self.configure(rec) def ensure_destroyed(self): self._destroy(self._get_class()) - def start(self, rec, d = 0): + def start(self, rec, d = 0): self._acct.configure(rec) self._acct.start(delay=d) + def configure(self, rec): + self._acct.configure(rec) + def stop(self): self._acct.stop() - def is_running(self): + def is_running(self): if (self._acct != None) and self._acct.is_running(): status = True else: