def cleanTaskQueues(): tqDB = TaskQueueDB() jobDB = JobDB() logDB = JobLoggingDB() result = tqDB.enableAllTaskQueues() if not result['OK']: return result result = tqDB.findOrphanJobs() if not result['OK']: return result for jid in result['Value']: result = tqDB.deleteJob(jid) if not result['OK']: gLogger.error("Cannot delete from TQ job %s" % jid, result['Message']) continue result = jobDB.rescheduleJob(jid) if not result['OK']: gLogger.error("Cannot reschedule in JobDB job %s" % jid, result['Message']) continue result = logDB.addLoggingRecord(jid, JobStatus.RECEIVED, "", "", source="JobState") if not result['OK']: gLogger.error("Cannot add logging record in JobLoggingDB %s" % jid, result['Message']) continue return S_OK()
class TaskQueueAgent(OptimizerModule): """ The specific Optimizer must provide the following methods: - initializeOptimizer() before each execution cycle - checkJob() - the main method called for each job """ ############################################################################# def initializeOptimizer(self): """Initialize specific parameters for TaskQueueAgent. """ self.waitingStatus = self.am_getOption('WaitingStatus', 'Waiting') self.waitingMinorStatus = self.am_getOption('WaitingMinorStatus', 'Pilot Agent Submission') try: self.taskQueueDB = TaskQueueDB() result = self.taskQueueDB.enableAllTaskQueues() if not result['OK']: raise Exception("Can't enable TaskQueues: %s" % result['Message']) except Exception, e: self.log.exception() return S_ERROR("Cannot initialize taskqueueDB: %s" % str(e)) return S_OK()
class TaskQueueAgent( OptimizerModule ): """ The specific Optimizer must provide the following methods: - initializeOptimizer() before each execution cycle - checkJob() - the main method called for each job """ ############################################################################# def initializeOptimizer( self ): """Initialize specific parameters for TaskQueueAgent. """ self.waitingStatus = self.am_getOption( 'WaitingStatus', 'Waiting' ) self.waitingMinorStatus = self.am_getOption( 'WaitingMinorStatus', 'Pilot Agent Submission' ) try: self.taskQueueDB = TaskQueueDB() result = self.taskQueueDB.enableAllTaskQueues() if not result[ 'OK' ]: raise Exception( "Can't enable TaskQueues: %s" % result[ 'Message' ] ) except Exception, e: self.log.exception() return S_ERROR( "Cannot initialize taskqueueDB: %s" % str( e ) ) return S_OK()