|
@@ -0,0 +1,245 @@
|
|
|
+# #!/usr/bin/env python
|
|
|
+# #coding=utf-8
|
|
|
+
|
|
|
+import json
|
|
|
+import logging
|
|
|
+import time
|
|
|
+import requests
|
|
|
+import sys
|
|
|
+from aliyunsdkcore import client
|
|
|
+from aliyunsdkcore.client import AcsClient
|
|
|
+from aliyunsdkcore.acs_exception.exceptions import ClientException
|
|
|
+from aliyunsdkcore.acs_exception.exceptions import ServerException
|
|
|
+from aliyunsdkecs.request.v20140526.CreateInstanceRequest import CreateInstanceRequest
|
|
|
+from aliyunsdkecs.request.v20140526.StartInstanceRequest import StartInstanceRequest
|
|
|
+from aliyunsdkecs.request.v20140526.DescribeInstancesRequest import DescribeInstancesRequest
|
|
|
+from aliyunsdkecs.request.v20140526.RunInstancesRequest import RunInstancesRequest
|
|
|
+from aliyunsdkecs.request.v20140526.DescribeNetworkInterfacesRequest import DescribeNetworkInterfacesRequest
|
|
|
+from aliyunsdkslb.request.v20140515.AddBackendServersRequest import AddBackendServersRequest
|
|
|
+from aliyunsdkecs.request.v20140526.RunCommandRequest import RunCommandRequest
|
|
|
+from aliyunsdkcore.request import CommonRequest
|
|
|
+from aliyunsdkecs.request.v20140526.SendFileRequest import SendFileRequest
|
|
|
+
|
|
|
+
|
|
|
+logging.basicConfig(level=logging.INFO,
|
|
|
+ format='%(asctime)s %(filename)s[line:%(lineno)d] %(levelname)s %(message)s',
|
|
|
+ datefmt='%a, %d %b %Y %H:%M:%S')
|
|
|
+# ##### 购买服务器权限
|
|
|
+# access key Id
|
|
|
+ak_id = "LTAI4GBWbFvvXoXsSVBe1o9f"
|
|
|
+# access key secret
|
|
|
+ak_secret = "kRAikWitb4kDxaAyBqNrmLmllMEDO3"
|
|
|
+# 设置地域
|
|
|
+region_id = "cn-hangzhou"
|
|
|
+clt = client.AcsClient(ak_id, ak_secret, region_id)
|
|
|
+
|
|
|
+# 获取批量创建ECS实例的数量, 取值范围:1-100, 默认值:1
|
|
|
+amount = int(sys.argv[1])
|
|
|
+# 使用的镜像信息
|
|
|
+image_id = "m-bp1e5jx8eqhq22l91xw7"
|
|
|
+
|
|
|
+# 设置实例规格
|
|
|
+instance_type = "ecs.ic5.xlarge"
|
|
|
+# 选择的交换机
|
|
|
+vswitch_id = "vsw-bp19lpjwtc6j0p0m9mdc2"
|
|
|
+# 当前VPC类型的安全组
|
|
|
+security_group_id = "sg-bp1irhrkr4vfj272hk4y"
|
|
|
+# 硬盘的大小,单位:G
|
|
|
+disk_size = "200"
|
|
|
+# 服务器命名
|
|
|
+instance_name = "ESS-rov-server-[1,2]"
|
|
|
+# 服务器所在区域
|
|
|
+zone_id = "cn-hangzhou-h"
|
|
|
+# 磁盘类型:云盘
|
|
|
+disk_category = "cloud_efficiency"
|
|
|
+# 密钥
|
|
|
+key_pair_name = "stuuudy"
|
|
|
+# 负载均衡ID
|
|
|
+slb_id_2 = "lb-bp1werfophtsjzfr76njm"
|
|
|
+
|
|
|
+# ##### 修改负载均衡权限
|
|
|
+AccessKey = 'LTAIuPbTPL3LDDKN'
|
|
|
+AccessSecret = 'ORcNedKwWuwVtcq4IRFtUDZgS0b1le'
|
|
|
+RegionId = 'cn-hangzhou'
|
|
|
+client = AcsClient(AccessKey, AccessSecret, RegionId)
|
|
|
+
|
|
|
+# server_start
|
|
|
+start_target_dir = '/home/sh'
|
|
|
+start_name = 'rov_server_start_new.sh'
|
|
|
+start_content = """#!/bin/bash
|
|
|
+docker login --username=stuuudys --password=Qingqu@2019 registry.piaoquantv.com
|
|
|
+docker run -d -it --network=host --name=rov-server registry.piaoquantv.com/piaoquan/rov-server:latest
|
|
|
+"""
|
|
|
+
|
|
|
+
|
|
|
+def build_request():
|
|
|
+ """购买服务器参数配置"""
|
|
|
+ request = RunInstancesRequest()
|
|
|
+ request.set_ImageId(image_id)
|
|
|
+ request.set_VSwitchId(vswitch_id)
|
|
|
+ request.set_SecurityGroupId(security_group_id)
|
|
|
+ request.set_ZoneId(zone_id)
|
|
|
+ request.set_InstanceType(instance_type)
|
|
|
+ # request.set_UserData(init_data)
|
|
|
+ request.set_InstanceName(instance_name)
|
|
|
+ request.set_SystemDiskSize(disk_size)
|
|
|
+ request.set_SystemDiskCategory(disk_category)
|
|
|
+ request.set_KeyPairName(key_pair_name)
|
|
|
+ request.set_Tags([
|
|
|
+ {
|
|
|
+ "Key": "ecs",
|
|
|
+ "Value": "rov-server.prod"
|
|
|
+ }
|
|
|
+ ])
|
|
|
+ return request
|
|
|
+
|
|
|
+
|
|
|
+def _send_request(request):
|
|
|
+ """发送API请求"""
|
|
|
+ request.set_accept_format('json')
|
|
|
+ try:
|
|
|
+ response_str = clt.do_action_with_exception(request)
|
|
|
+ logging.info(response_str)
|
|
|
+ response_detail = json.loads(response_str)
|
|
|
+ return response_detail
|
|
|
+ except Exception as e:
|
|
|
+ logging.error(e)
|
|
|
+
|
|
|
+
|
|
|
+def check_instance_running(instance_ids):
|
|
|
+ """检查服务器运行状态"""
|
|
|
+ request = DescribeInstancesRequest()
|
|
|
+ request.set_InstanceIds(json.dumps(instance_ids))
|
|
|
+ response = _send_request(request)
|
|
|
+ if response.get('Code') is None:
|
|
|
+ instances_list = response.get('Instances').get('Instance')
|
|
|
+ running_count = 0
|
|
|
+ for instance_detail in instances_list:
|
|
|
+ if instance_detail.get('Status') == "Running":
|
|
|
+ running_count += 1
|
|
|
+ return running_count
|
|
|
+
|
|
|
+
|
|
|
+def runCommand(instance_ids):
|
|
|
+ """批量执行shell脚本:启动服务"""
|
|
|
+ request = RunCommandRequest()
|
|
|
+ request.set_accept_format('json')
|
|
|
+ request.set_Type("RunShellScript")
|
|
|
+ request.set_CommandContent("sh /home/sh/rov_server_start_new.sh")
|
|
|
+ request.set_InstanceIds(instance_ids)
|
|
|
+ response = clt.do_action_with_exception(request)
|
|
|
+ logging.info(response)
|
|
|
+
|
|
|
+
|
|
|
+def healthCheck(instance_id):
|
|
|
+ """服务健康检查 & 服务器挂载到负载均衡"""
|
|
|
+ ipaddr = getIpaddr(instance_id)
|
|
|
+ while True:
|
|
|
+ health_url = 'http://%s:5001/healthcheck' %(ipaddr)
|
|
|
+ try:
|
|
|
+ http_code = requests.get(health_url).status_code
|
|
|
+ except:
|
|
|
+ logging.info("images is downloading")
|
|
|
+ http_code = 0
|
|
|
+
|
|
|
+ if http_code == 200:
|
|
|
+ break
|
|
|
+ else:
|
|
|
+ time.sleep(20)
|
|
|
+
|
|
|
+ # 服务预热60s, 服务刚启动时CPU很高
|
|
|
+ time.sleep(60)
|
|
|
+ # 挂载到负载均衡
|
|
|
+ addBackendServers(slb_id_2, instance_id)
|
|
|
+
|
|
|
+
|
|
|
+def setWeight(slb_id, instance_id, weight):
|
|
|
+ """
|
|
|
+ 设置slb权重
|
|
|
+ :param slb_id: slb_id
|
|
|
+ :param instance_id: 服务器id
|
|
|
+ :param weight: 权重值
|
|
|
+ :return: None
|
|
|
+ """
|
|
|
+ BackendServers = [{"ServerId": instance_id, "Weight": weight}]
|
|
|
+ request = CommonRequest()
|
|
|
+ request.set_accept_format('json')
|
|
|
+ request.set_domain('slb.aliyuncs.com')
|
|
|
+ request.set_version('2014-05-15')
|
|
|
+ request.set_method('POST')
|
|
|
+ request.set_action_name('SetBackendServers')
|
|
|
+ request.add_query_param('BackendServers', BackendServers)
|
|
|
+ request.add_query_param('LoadBalancerId', slb_id)
|
|
|
+ try:
|
|
|
+ response = client.do_action_with_exception(request)
|
|
|
+ except Exception as e:
|
|
|
+ logging.error(e)
|
|
|
+
|
|
|
+
|
|
|
+def addBackendServers(slb_id, instance_id):
|
|
|
+ """服务器挂载到负载均衡"""
|
|
|
+ request = AddBackendServersRequest()
|
|
|
+ request.set_accept_format('json')
|
|
|
+ ipaddr = getIpaddr(instance_id)
|
|
|
+ request.set_LoadBalancerId(slb_id)
|
|
|
+ request.set_BackendServers([{"ServerId": instance_id, "Weight": "0", "Type": "ecs", "ServerIp": ipaddr}])
|
|
|
+ response = client.do_action_with_exception(request)
|
|
|
+ logging.info(response)
|
|
|
+ for i in range(1, 6):
|
|
|
+ time.sleep(5)
|
|
|
+ weight = i * 20
|
|
|
+ logging.info(f"instance_id: {instance_id}, weight: {weight}")
|
|
|
+ setWeight(slb_id=slb_id, instance_id=instance_id, weight=weight)
|
|
|
+
|
|
|
+
|
|
|
+def getIpaddr(instance_id):
|
|
|
+ """根据 instance_id 获取实例IP地址"""
|
|
|
+ request = DescribeNetworkInterfacesRequest()
|
|
|
+ request.set_accept_format('json')
|
|
|
+ request.set_InstanceId(instance_id)
|
|
|
+ response = clt.do_action_with_exception(request)
|
|
|
+ request_content = json.loads(response)
|
|
|
+ ipaddr = request_content['NetworkInterfaceSets']['NetworkInterfaceSet'][0]['PrivateIpAddress']
|
|
|
+ return ipaddr
|
|
|
+
|
|
|
+
|
|
|
+def send_file_to_ecs(instance_id_list, target_dir, name, content):
|
|
|
+ request = SendFileRequest()
|
|
|
+ request.set_Content(content)
|
|
|
+ request.set_TargetDir(target_dir)
|
|
|
+ request.set_Name(name)
|
|
|
+ request.set_Overwrite(True)
|
|
|
+ request.set_InstanceIds(instance_id_list)
|
|
|
+ response = clt.do_action_with_exception(request)
|
|
|
+
|
|
|
+
|
|
|
+def _execute_request(request):
|
|
|
+ response = _send_request(request)
|
|
|
+ if response.get('Code') is None:
|
|
|
+ instance_ids = response.get('InstanceIdSets').get('InstanceIdSet')
|
|
|
+ running_amount = 0
|
|
|
+ while running_amount < amount:
|
|
|
+ time.sleep(10)
|
|
|
+ running_amount = check_instance_running(instance_ids)
|
|
|
+
|
|
|
+ logging.info(f"{instance_ids} is running")
|
|
|
+ time.sleep(60)
|
|
|
+ send_file_to_ecs(instance_id_list=instance_ids,
|
|
|
+ target_dir=start_target_dir, name=start_name, content=start_content)
|
|
|
+ runCommand(instance_ids)
|
|
|
+
|
|
|
+ for instance_id in instance_ids:
|
|
|
+ healthCheck(instance_id)
|
|
|
+
|
|
|
+
|
|
|
+def create_multiple_instances():
|
|
|
+ """创建ECS实例并启动"""
|
|
|
+ request = build_request()
|
|
|
+ request.set_Amount(amount)
|
|
|
+ _execute_request(request)
|
|
|
+
|
|
|
+
|
|
|
+if __name__ == '__main__':
|
|
|
+ logging.info("Start Create Instances")
|
|
|
+ # 创建ECS实例并启动
|
|
|
+ create_multiple_instances()
|