コード例 #1
0
ファイル: dispatcher.py プロジェクト: flowlsj/task_dispatcher

if __name__ == "__main__":
    arguments = get_arguments()
    if arguments is None:
        print """
        Usage: python dispatcher.py --build_type=<build_type> --build_number=<build_number> --task_list=<task_list_file>
        """
    executor_manager = ExecutorManager()
    executor_manager.load_executors(build_type=arguments.build_type,
                                    build_number=arguments.build_number)

    task_manager = TaskManager()
    task_manager.load_tasks(task_file=arguments.task_list_file)

    task = task_manager.get_next_task()
    while task is not None:
        sleep(1)
        if executor_manager.has_valid_executor_for_task(task):
            executor = executor_manager.get_executor_for_task(task)
            if executor is not None:
                worker_thread = Thread(target=executor_run_task,
                                       args=(executor, task))
                worker_thread.start()
            else:
                # print "Put task [%s] back to task pool" % task.name
                task_manager.put_back_to_task_pool(task)
            task = task_manager.get_next_task()
        else:
            print "Mark task [%s] as cannot be executed" % task.name
            task_manager.mark_task_as_cannot_be_executed(task)