heyudev 2 years ago
parent
commit
9c786dace0
1 changed files with 21 additions and 21 deletions
  1. 21 21
      ess_longvideo.py

+ 21 - 21
ess_longvideo.py

@@ -85,27 +85,27 @@ async def ess_instance(create_client, slb_client, ess_count, max_workers):
     ]
     await asyncio.wait(tasks)
     logging.info(f"health instances count: {len(health_instances)}, {health_instances}")
-    # # 5. 挂载流量
-    # if len(health_instances) > 0:
-    #     # 所有机器探活成功
-    #     time.sleep(60)
-    #     utils.add_backend_servers_with_slbs(client=slb_client,
-    #                                         slb_id_list=longvideo_config.slb_id_list,
-    #                                         instances=health_instances)
-    #     add_weight_list = [(10, 30), (20, 20), (40, 10), (60, 10), (80, 10), (100, 10)]
-    #     health_instance_ids = [instance_id for instance_id, _ in health_instances]
-    #     utils.set_instance_weight_process_with_slbs(client=slb_client,
-    #                                                 slb_id_list=longvideo_config.slb_id_list,
-    #                                                 instance_id_list=health_instance_ids,
-    #                                                 weight_list=add_weight_list)
-    #     logging.info(f"ess count: {ess_count}, "
-    #                  f"create count: {len(ess_instance_ids)}, "
-    #                  f"finished count: {len(health_instance_ids)}")
-    # else:
-    #     logging.info(f"ess count: {ess_count}, "
-    #                  f"create count: {len(ess_instance_ids)}, "
-    #                  f"health count: {len(health_instances)}")
-    #     sys.exit()
+    # 5. 挂载流量
+    if len(health_instances) > 0:
+        # 所有机器探活成功
+        time.sleep(60)
+        utils.add_backend_servers_with_slbs(client=slb_client,
+                                            slb_id_list=longvideo_config.slb_id_list,
+                                            instances=health_instances)
+        add_weight_list = [(10, 10), (20, 10), (40, 10), (60, 10), (80, 10), (100, 10)]
+        health_instance_ids = [instance_id for instance_id, _ in health_instances]
+        utils.set_instance_weight_process_with_slbs(client=slb_client,
+                                                    slb_id_list=longvideo_config.slb_id_list,
+                                                    instance_id_list=health_instance_ids,
+                                                    weight_list=add_weight_list)
+        logging.info(f"ess count: {ess_count}, "
+                     f"create count: {len(ess_instance_ids)}, "
+                     f"finished count: {len(health_instance_ids)}")
+    else:
+        logging.info(f"ess count: {ess_count}, "
+                     f"create count: {len(ess_instance_ids)}, "
+                     f"health count: {len(health_instances)}")
+        sys.exit()
 
 
 def main():