(self, worker_name: str, check_heart_beat: bool,
worker_status: dict)
| 67 | logger.info("Init controller") |
| 68 | |
| 69 | def register_worker(self, worker_name: str, check_heart_beat: bool, |
| 70 | worker_status: dict): |
| 71 | if worker_name not in self.worker_info: |
| 72 | logger.info(f"Register a new worker: {worker_name}") |
| 73 | else: |
| 74 | logger.info(f"Register an existing worker: {worker_name}") |
| 75 | |
| 76 | if not worker_status: |
| 77 | worker_status = self.get_worker_status(worker_name) |
| 78 | if not worker_status: |
| 79 | return False |
| 80 | |
| 81 | self.worker_info[worker_name] = WorkerInfo( |
| 82 | worker_status["model_names"], worker_status["speed"], worker_status["queue_length"], |
| 83 | check_heart_beat, time.time()) |
| 84 | |
| 85 | logger.info(f"Register done: {worker_name}, {worker_status}") |
| 86 | return True |
| 87 | |
| 88 | def get_worker_status(self, worker_name: str): |
| 89 | try: |
no test coverage detected