def __init__(self, bus, hostname, port, options): plugins.SimplePlugin.__init__(self, bus) self.id = None self.hostname = hostname self.port = port self.master_url = options.master self.master_proxy = MasterProxy(self, bus, self.master_url) self.master_proxy.subscribe() if options.blockstore is None: block_store_dir = tempfile.mkdtemp( prefix=os.getenv('TEMP', default='/tmp/sw-files-')) else: block_store_dir = options.blockstore self.block_store = BlockStore(cherrypy.engine, self.hostname, self.port, block_store_dir, ignore_blocks=options.ignore_blocks) self.block_store.build_pin_set() self.upload_deferred_work = DeferredWorkPlugin(bus, 'upload_work') self.upload_deferred_work.subscribe() self.upload_manager = UploadManager(self.block_store, self.upload_deferred_work) self.execution_features = ExecutionFeatures() self.task_executor = TaskExecutorPlugin(bus, self.block_store, self.master_proxy, self.execution_features, 1) self.task_executor.subscribe() self.server_root = WorkerRoot(self) self.pinger = Pinger(bus, self.master_proxy, None, 30) self.pinger.subscribe() self.stopping = False self.event_log = [] self.log_lock = Lock() self.log_condition = Condition(self.log_lock) self.cherrypy_conf = {} cherrypy.config.update({"server.thread_pool": 20}) if options.staticbase is not None: self.cherrypy_conf["/skyweb"] = { "tools.staticdir.on": True, "tools.staticdir.dir": options.staticbase } if options.lib is not None: self.cherrypy_conf["/stdlib"] = { "tools.staticdir.on": True, "tools.staticdir.dir": options.lib } self.subscribe()
def worker_process_main(base_dir, task_queue, response_queue): master_proxy = QueueMasterProxy(response_queue) execution_features = ExecutionFeatures() block_store = BlockStore(ciel.engine, 'localhost', 8000, base_dir, True) # XXX: Broken because we now need a pseudoworker in place of a block_store. thread_task_executor = TaskExecutorPlugin(ciel.engine, PseudoWorker(block_store), master_proxy, execution_features, 1) while True: task = task_queue.get() if isinstance(task, ThreadTerminator): return task_descriptor = task.as_descriptor(False) thread_task_executor.handle_input(task_descriptor)