阿里云权重管理
背景
我们的公司采用的是阿里云环境，发布体系中需要滚动更新功能，需要编写脚本来完成负载均衡对应主机的权重为0，然后发布代码到这个服务器上， 重启成功后修改回原有权重，然后一个一个处理完毕集群的所有主机。
阿里云负载均衡简介
负载均衡的几个常见概念。
| 名词 | 说明 |
|---|---|
| 负载均衡服务 (Server Load Balancer) | 阿里云计算提供的一种网络负载均衡服务，结合阿里云提供的ECS服务，提供四层和七层负载均衡服务。 |
| 负载均衡实例(Server Load Balancer Instances) | 负载均衡实例是一个运行的负载均衡服务。要使用负载均衡服务，必须先创建一个负载均衡实例。 |
| 服务地址 (SLB IP address) | 系统为创建的负载均衡实例分配的服务IP地址。根据创建的负载均衡实例的类型，服务地址可能是公网IP也可能是私网IP。您可以将域名解析到公网IP地址提供对外服务。 |
| 监听 (Listeners) | 负载均衡服务监听规定了如何将请求转发给后端服务器。一个负载均衡实例至少添加一个监听。 |
| 后端服务器 (Backend Servers) | 处理负载均衡分发的前端请求的ECS实例。 |
| 虚拟服务器组 (Backend Servers Groups) | 一组处理负载均衡分发的前端请求的ECS实例。不同的监听可以关联不同的虚拟服务器组，实现监听维度的请求转发。 |
阿里云提供2种服务器组， 一个是默认服务器组，一个是虚拟服务器组， 如果一个负载均衡上面只挂一个域名的多个主机，建议直接将后端的服务器加入到默认服务器组即可，
如果你的负载均衡挂多个域名的话， 建议以域名为虚拟服务器的名字，创建多个虚拟服务器组，每个组下面挂载对应的后端服务器。
我们的线上环境建议采用一个域名一个负载均衡， 测试环境可以考虑一个负载下弄多个域名，省点钱，我们简单算下使用多负载均衡和一个负载负载多域名方案价格差异。
阿里云官方的负载均衡配置价格不贵，配置费用为0.02元每小时。公网流量费用0.48元/GB，对于2种方案来说流量都是一样的。采用多负载均衡会多出多个配置费用， 1个负载均衡的配置费用1年下来大概为0.02*24*365≈175元。可以根据自身公司情况选择一种方案。
管理脚本
使用这个脚本，需要安装python3环境，并安装如下包
pip install aliyun-python-sdk-core-v3pip install aliyun-python-sdk-ecspip install aliyun-python-sdk-slb
slb_manager.py内容如下：
# !/usr/bin/env python# -*-coding:utf-8 -*-from aliyunsdkcore.client import AcsClientfrom aliyunsdkcore.request import CommonRequestimport jsonimport osimport sysfrom enum import Enumdef byte_to_json(response): response_str = str(response, encoding='utf-8') response_json = json.loads(response_str) return response_jsonclass SlbManagerInfo(): def __init__(self, exit_code=0, exit_message=""): self.exit_code = exit_code self.exit_message = exit_message self.extra_message = {} def __str__(self): return str(self.__dict__)class SlbManager(): """ 阿里云负载均衡管理，支持对默认的组和虚拟服务器组的管理 """ def __init__(self, slb_config_file, region_id="cn-beijing"): """ 初始化client,slb_config_file内容为以下格式 { "slb_ak":"xxxxxxxxxxxxxx", "slb_sk":"xxxxxxxxxxxxxxxxxxxxxxxxxxxx" } """ self.slb_config_file = slb_config_file self.region_id = region_id self.slb_manager_info = SlbManagerInfo() self.ak, self.sk = self.get_aksk_from_config_file() self.client = AcsClient(self.ak, self.sk, self.region_id) def pre_check(self): if isinstance(self.weight,int): if (self.weight >100 or self.weight <0): self.slb_manager_info.exit_code = 1 self.slb_manager_info.exit_message = "权重设置不符合要求（0=100）" + "你的设置为" +str(self.weight) else: self.slb_manager_info.exit_code = 2 self.slb_manager_info.exit_message = "权重不是int的，请检查" def set_comm_request(self, request, product_type="slb"): request.set_accept_format('json') request.set_method('POST') request.set_domain(product_type + '.aliyuncs.com') def get_aksk_from_config_file(self): if not (os.path.exists(self.slb_config_file)): self.slb_manager_info.exit_code = 1 self.slb_manager_info.exit_message = self.slb_config_file + "文件找不到" try: with open(self.slb_config_file, 'r') as f: slb_config = json.load(f) except IOError as e: self.slb_manager_info.exit_code = 2 self.slb_manager_info.exit_message = "文件读取失败" + str(e) return slb_config["slb_ak"], slb_config["slb_sk"] def get_slb_id(self): request = CommonRequest() self.set_comm_request(request) request.set_version('2014-05-15') request.set_action_name('DescribeLoadBalancers') request.add_query_param('LoadBalancerName', self.slb_name) response = self.client.do_action_with_exception(request) response_json = byte_to_json(response) if response_json["TotalCount"] != 1 : self.slb_manager_info.exit_code = 3 self.slb_manager_info.exit_message ="根据负载均衡名字获取到的总个数不为1 ，请确认负载的名字，真实获取的个数为" +str(response_json["TotalCount"] ) else: self.slb_id = response_json["LoadBalancers"]["LoadBalancer"][0]["LoadBalancerId"] a=1 def get_ecs_id(self): request = CommonRequest() self.set_comm_request(request, product_type='ecs') request.set_version('2014-05-26') request.set_action_name('DescribeInstances') request.add_query_param('InstanceName', self.ecs_name) response = self.client.do_action_with_exception(request) response_json = byte_to_json(response) if response_json["TotalCount"] != 1 : self.slb_manager_info.exit_code = 4 self.slb_manager_info.exit_message = "获取不到ecsid,请检查ecs_name" else: self.ecs_id = response_json["Instances"]["Instance"][0]["InstanceId"] def set_weight_for_ecs(self): for backend_server in self.backend_servers: backend_server["ServerId"] = backend_server["VmName"] if backend_server["VmName"] == self.ecs_id: # 保留下设置前的权重 self.slb_manager_info.extra_message["old_weight"] = backend_server["Weight"] backend_server["Weight"] = self.weight def set_weight_for_default_group_backend_servers(self): request = CommonRequest() self.set_comm_request(request) request.set_version('2014-05-15') request.set_action_name('SetBackendServers') request.add_query_param('LoadBalancerId', self.slb_id) request.add_query_param('BackendServers', self.backend_servers) response = self.client.do_action_with_exception(request) response_json = byte_to_json(response) # 这个貌似没法对结果进行判定。 def set_weight_for_virtual_group_backend_servers(self): request = CommonRequest() self.set_comm_request(request) request.set_version('2014-05-15') request.set_action_name('SetVServerGroupAttribute') request.add_query_param('LoadBalancerId', self.slb_id) request.add_query_param('VServerGroupId', self.virtual_group_id) # request.add_query_param('VServerGroupName', 'test2') request.add_query_param('BackendServers', self.backend_servers) response = self.client.do_action_with_exception(request) response_json = byte_to_json(response) # 这个貌似没法对结果进行判定。 def get_slb_info(self): request = CommonRequest() self.set_comm_request(request) request.set_version('2014-05-15') request.set_action_name('DescribeLoadBalancersRelatedEcs') request.add_query_param('LoadBalancerId', self.slb_id) response = self.client.do_action_with_exception(request) response_json = byte_to_json(response) if not response_json["Success"]: self.slb_manager_info.exit_code = 4 self.slb_manager_info.exit_message = response_json["Message"] else: self.slb_info = response_json["LoadBalancers"]["LoadBalancer"][0] if self.is_default_group: self.backend_servers = self.slb_info["BackendServers"]["BackendServer"] else: self.virtual_server_group= None for virtual_group in self.slb_info["VServerGroups"]["VServerGroup"]: if virtual_group["GroupName"] == self.group_name: self.virtual_server_group = virtual_group break if self.virtual_server_group is None: self.slb_manager_info.exit_code = 6 self.slb_manager_info.exit_message = "你输入的虚拟服务器组，不在特定的负载均衡下，请检查" else: self.virtual_group_id = self.virtual_server_group["GroupId"] self.backend_servers = self.virtual_server_group["BackendServers"]["BackendServer"] # self.backend_servers = response_json["LoadBalancers"]["LoadBalancer"][0]["VServerGroups"]["VServerGroup"][0]["BackendServers"]["BackendServer"] def set_weight_for_default_group(self, slb_name, ecs_name, weight): # set self.slb_name = slb_name self.ecs_name = ecs_name self.weight = weight self.is_default_group = True if self.slb_manager_info.exit_code == 0: self.pre_check() if self.slb_manager_info.exit_code == 0: self.get_slb_id() if self.slb_manager_info.exit_code == 0: self.get_ecs_id() if self.slb_manager_info.exit_code == 0: self.get_slb_info() if self.slb_manager_info.exit_code == 0: self.set_weight_for_ecs() if self.slb_manager_info.exit_code == 0: self.set_weight_for_default_group_backend_servers() print(self.slb_manager_info) return self.slb_manager_info.exit_code def set_weight_for_virtual_group(self, slb_name, group_name, ecs_name, weight): # set self.slb_name = slb_name self.ecs_name = ecs_name self.group_name = group_name self.weight = weight self.is_default_group = False if self.slb_manager_info.exit_code == 0: self.pre_check() if self.slb_manager_info.exit_code == 0: self.get_slb_id() if self.slb_manager_info.exit_code == 0: self.get_ecs_id() if self.slb_manager_info.exit_code == 0: self.get_slb_info() if self.slb_manager_info.exit_code == 0: self.set_weight_for_ecs() if self.slb_manager_info.exit_code == 0: self.set_weight_for_virtual_group_backend_servers() print(self.slb_manager_info) return self.slb_manager_info.exit_codedef main(): """ python3 slb_manager.py """ print(sys.argv) slb_name = sys.argv[1] group_name = sys.argv[2] ecs_name = sys.argv[3] weight = int(sys.argv[4]) slb_config_file = r'c:\\slbconfig' region_id = 'cn-beijing' manager = SlbManager(slb_config_file=slb_config_file, region_id=region_id) ret =0 if group_name == "default": ret =manager.set_weight_for_default_group(slb_name, ecs_name, weight) else: ret =manager.set_weight_for_virtual_group(slb_name, group_name, ecs_name, weight) sys.exit(ret)if __name__ == "__main__": main() # manager = SlbManager(slb_config_file=r'c:\\slbconfig',region_id='cn-beijing') # manager.set_weight_for_default_group('lb_tmp','aliyun_test03',10) # manager.set_weight_for_virtual_group('lb_tmp','test2','aliyun_test02',0) 注意事项
- 脚本是支持默认服务器组和虚拟服务器的， 如果接受的第三个参数为default，表示设置的默认后端服务器的权重， 如果为其他的，表示设置虚拟服务器组，且组名为你传递过来的值，所有使用此脚本你的虚拟服务器组不能设置叫default的组名。
- 可以根据退出码来判定设置结果情况。
- 输出的结果信息里面有修改前的权重值，可以使用bash命令来提取这个脚本的输出来获取。
- 你提供的ecs名字在ecs在你所有的ec的名字中是唯一的
- 你提供的slb名字在你自己的slb的所有名字中是唯一的。
脚本使用
第一步： ram创建一个账号， 并授予该账号的slb所有权限，ecs所有权限，并为该用户生成ak,sk。
第二步： 创建一个文件写入ak,sk信息。
内容如下
{ "slb_ak":"xxxxxxxxxxxxxx", "slb_sk":"xxxxxxxxxxxxxxxxxxxxxxxxxxxx"} 第三步：
修改脚本的 slb_config_file的路径为你的aksk文件路径，修改region-id默认值。
第四步： 创建一个按量付费的负载均衡测试下。
使用样例
C:\Users\Administrator\Anaconda3\python.exe E:/haoweilai/Devops/scripts/slb_manager.py lb_tmp3 test23 aliyun_test02 13['E:/haoweilai/Devops/scripts/slb_manager.py', 'lb_tmp3', 'test23', 'aliyun_test02', '13']{ 'exit_code': 3, 'exit_message': '根据负载均衡名字获取到的总个数不为1 ，请确认负载的名字，真实获取的个数为0', 'extra_message': {}}C:\Users\Administrator\Anaconda3\python.exe E:/haoweilai/Devops/scripts/slb_manager.py lb_tmp test23 aliyun_test02 14['E:/haoweilai/Devops/scripts/slb_manager.py', 'lb_tmp', 'test23', 'aliyun_test02', '14']{ 'exit_code': 0, 'exit_message': '', 'extra_message': { 'old_weight': 100}}C:\Users\Administrator\Anaconda3\python.exe E:/haoweilai/Devops/scripts/slb_manager.py lb_tmp default aliyun_test02 50['E:/haoweilai/Devops/scripts/slb_manager.py', 'lb_tmp', 'default', 'aliyun_test02', '50']{ 'exit_code': 0, 'exit_message': '', 'extra_message': { 'old_weight': 100}} 参考
阿里云官方在线调试：
阿里云python sdk地址：
负载均衡产品api参考：