pandajedi/jediorder/JobBroker.py

0001 from pandacommon.pandalogger.PandaLogger import PandaLogger
0002
0003 from pandajedi.jediconfig import jedi_config
0004 from pandajedi.jedicore.FactoryBase import FactoryBase
0005
0006 logger = PandaLogger().getLogger(__name__.split(".")[-1])
0007
0008
0009 # factory class for job brokerage
0010 class JobBroker(FactoryBase):
0011     # constructor
0012     def __init__(self, vo, sourceLabel):
0013         FactoryBase.__init__(self, vo, sourceLabel, logger, jedi_config.jobbroker.modConfig)
0014
0015     # main
0016     def doBrokerage(self, taskSpec, cloudName, inputChunk, taskParamMap):
0017         return self.getImpl(taskSpec.vo, taskSpec.prodSourceLabel).doBrokerage(taskSpec, cloudName, inputChunk, taskParamMap)
0018
0019     # set live counter
0020     def setLiveCounter(self, vo, sourceLabel, liveCounter):
0021         self.getImpl(vo, sourceLabel).setLiveCounter(liveCounter)
0022
0023     # set lock ID
0024     def setLockID(self, vo, sourceLabel, pid, tid):
0025         self.getImpl(vo, sourceLabel).setLockID(pid, tid)
0026
0027     # get base lock ID
0028     def getBaseLockID(self, vo, sourceLabel):
0029         return self.getImpl(vo, sourceLabel).getBaseLockID()
0030
0031     # set test mode
0032     def setTestMode(self, vo, sourceLabel):
0033         self.getImpl(vo, sourceLabel).setTestMode()