|
@@ -0,0 +1,244 @@
|
|
|
+#!/bin/env python
|
|
|
+# coding=utf-8
|
|
|
+import logging
|
|
|
+import time
|
|
|
+import docker
|
|
|
+import sys
|
|
|
+import requests
|
|
|
+import json
|
|
|
+import queue
|
|
|
+import threading
|
|
|
+from aliyunsdkcore import client
|
|
|
+from aliyunsdkecs.request.v20140526.DescribeNetworkInterfacesRequest import DescribeNetworkInterfacesRequest
|
|
|
+from aliyunsdkcore.request import CommonRequest
|
|
|
+from aliyunsdkslb.request.v20140515.DescribeLoadBalancerAttributeRequest import DescribeLoadBalancerAttributeRequest
|
|
|
+from concurrent.futures import ThreadPoolExecutor
|
|
|
+
|
|
|
+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')
|
|
|
+
|
|
|
+# ##### 修改负载均衡权限
|
|
|
+AccessKey = 'LTAIuPbTPL3LDDKN'
|
|
|
+AccessSecret = 'ORcNedKwWuwVtcq4IRFtUDZgS0b1le'
|
|
|
+RegionId = 'cn-hangzhou'
|
|
|
+clt = client.AcsClient(AccessKey, AccessSecret, RegionId)
|
|
|
+
|
|
|
+slb_id_1 = 'lb-bp1werfophtsjzfr76njm'
|
|
|
+
|
|
|
+
|
|
|
+def getInstanceId(slb_id):
|
|
|
+ """获取slb下所有服务器信息"""
|
|
|
+ request = DescribeLoadBalancerAttributeRequest()
|
|
|
+ request.set_accept_format('json')
|
|
|
+ request.set_LoadBalancerId(slb_id)
|
|
|
+ response = clt.do_action_with_exception(request)
|
|
|
+ return json.loads(response)
|
|
|
+
|
|
|
+
|
|
|
+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 = clt.do_action_with_exception(request)
|
|
|
+ except Exception as e:
|
|
|
+ logging.error(e)
|
|
|
+
|
|
|
+
|
|
|
+def getIpadd(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 checkHealth(ipadd):
|
|
|
+ """健康检查"""
|
|
|
+ while True:
|
|
|
+ health_url = 'http://%s:5001/healthcheck' %(ipadd)
|
|
|
+ header = {"Content-Type": "application/json"}
|
|
|
+ try:
|
|
|
+ health_code = requests.get(health_url).status_code
|
|
|
+ except Exception as e:
|
|
|
+ continue
|
|
|
+ if health_code == 200:
|
|
|
+ return False
|
|
|
+
|
|
|
+
|
|
|
+def setInstanceWeightProcess(instance_id):
|
|
|
+ """服务更新完之后逐步修改服务器的权重值"""
|
|
|
+ # 直接加载100会出现502。权重值每次增加20,每5s修改一次
|
|
|
+ for i in range(1, 6):
|
|
|
+ weight = i * 20
|
|
|
+ setWeight(slb_id_1, instance_id, weight)
|
|
|
+ time.sleep(5)
|
|
|
+
|
|
|
+
|
|
|
+# def update():
|
|
|
+# """更新服务"""
|
|
|
+# time.sleep(10)
|
|
|
+# global success_count
|
|
|
+# apps = 'rov-server'
|
|
|
+# version = sys.argv[1]
|
|
|
+# registry = 'registry-vpc.cn-hangzhou.aliyuncs.com/stuuudy/rov-server:{}'.format(version)
|
|
|
+# while not q2.empty():
|
|
|
+# instance_id = q2.get()
|
|
|
+# ipadd = getIpadd(instance_id)
|
|
|
+# logging.info(f"服务器信息:{instance_id}/{ipadd}")
|
|
|
+# client = docker.DockerClient(base_url=f'tcp://{ipadd}:2375', timeout=60)
|
|
|
+# try:
|
|
|
+# # 更新前移除旧的容器
|
|
|
+# id = client.containers.get(apps)
|
|
|
+# id.remove(force=True)
|
|
|
+# except Exception as e:
|
|
|
+# print("容器不存在或者无法删除当前容器")
|
|
|
+#
|
|
|
+# try:
|
|
|
+# # 更新前删除旧的镜像
|
|
|
+# images = client.images.list(all)
|
|
|
+# for i in range(len(images)):
|
|
|
+# if images[i]:
|
|
|
+# client.images.remove(force=True, image=images[i].tags[0])
|
|
|
+# time.sleep(2)
|
|
|
+# # images = client.images.get(registry)
|
|
|
+# # client.images.remove(force=True, image=registry)
|
|
|
+# except Exception as e:
|
|
|
+# print(e)
|
|
|
+# print("镜像不存在,无法获取到镜像ID")
|
|
|
+# sys.exit(0)
|
|
|
+#
|
|
|
+# try:
|
|
|
+# # 登录镜像仓库
|
|
|
+# client.login(username='stuuudys', password='Qingqu@2019', registry='registry-vpc.cn-hangzhou.aliyuncs.com')
|
|
|
+# # 启动一个容器
|
|
|
+# client.containers.run(registry, detach=True, cap_add='SYS_PTRACE', network_mode='host', name=apps,
|
|
|
+# volumes={'/datalog/': {'bind': '/datalog/', 'mode': 'rw'}})
|
|
|
+# checkHealth(ipadd)
|
|
|
+# print(f"{ipadd}: 权重修改中......")
|
|
|
+# # 修改权重
|
|
|
+# setInstanceWeightProcess(instance_id)
|
|
|
+# success_count = success_count + 1
|
|
|
+# print(f"更新进度: {success_count}/{total}")
|
|
|
+# except Exception as e:
|
|
|
+# raise e
|
|
|
+# sys.exit() # 容器启动失败立即退出更新
|
|
|
+
|
|
|
+
|
|
|
+class MyThread(threading.Thread):
|
|
|
+ def __init__(self, func):
|
|
|
+ threading.Thread.__init__(self)
|
|
|
+ self.func = func
|
|
|
+
|
|
|
+ def run(self):
|
|
|
+ self.func()
|
|
|
+
|
|
|
+
|
|
|
+def update_sever(instance_id):
|
|
|
+ """更新服务"""
|
|
|
+ # time.sleep(20)
|
|
|
+ # print(slb_id_1, instance_id)
|
|
|
+
|
|
|
+ # 设置slb权重为0
|
|
|
+ setWeight(slb_id_1, instance_id, 0)
|
|
|
+ # 等待10s,处理剩余请求
|
|
|
+ time.sleep(10)
|
|
|
+
|
|
|
+ apps = 'rov-server'
|
|
|
+ version = sys.argv[1]
|
|
|
+ registry = f'registry-vpc.cn-hangzhou.aliyuncs.com/stuuudy/rov-server:{version}'
|
|
|
+ ipadd = getIpadd(instance_id)
|
|
|
+ logging.info(f"服务器信息:{instance_id}/{ipadd}")
|
|
|
+ client = docker.DockerClient(base_url=f'tcp://{ipadd}:2375', timeout=60)
|
|
|
+ try:
|
|
|
+ # 更新前移除旧的容器
|
|
|
+ id = client.containers.get(apps)
|
|
|
+ id.remove(force=True)
|
|
|
+ except Exception as e:
|
|
|
+ print("容器不存在或者无法删除当前容器")
|
|
|
+
|
|
|
+ try:
|
|
|
+ # 更新前删除旧的镜像
|
|
|
+ images = client.images.list(all)
|
|
|
+ for i in range(len(images)):
|
|
|
+ if images[i]:
|
|
|
+ client.images.remove(force=True, image=images[i].tags[0])
|
|
|
+ time.sleep(2)
|
|
|
+ # images = client.images.get(registry)
|
|
|
+ # client.images.remove(force=True, image=registry)
|
|
|
+ except Exception as e:
|
|
|
+ print(e)
|
|
|
+ print("镜像不存在,无法获取到镜像ID")
|
|
|
+ sys.exit(0)
|
|
|
+
|
|
|
+ try:
|
|
|
+ # 登录镜像仓库
|
|
|
+ client.login(username='stuuudys', password='Qingqu@2019', registry='registry-vpc.cn-hangzhou.aliyuncs.com')
|
|
|
+ # 启动一个容器
|
|
|
+ client.containers.run(registry, detach=True, cap_add='SYS_PTRACE', network_mode='host', name=apps,
|
|
|
+ volumes={'/datalog/': {'bind': '/datalog/', 'mode': 'rw'}})
|
|
|
+ checkHealth(ipadd)
|
|
|
+ print(f"{ipadd}: 权重修改中......")
|
|
|
+ # 修改权重
|
|
|
+ setInstanceWeightProcess(instance_id)
|
|
|
+ except Exception as e:
|
|
|
+ raise e
|
|
|
+ sys.exit() # 容器启动失败立即退出更新
|
|
|
+
|
|
|
+
|
|
|
+if __name__ == '__main__':
|
|
|
+ # 更新完成计数
|
|
|
+ success_count = 0
|
|
|
+ threads = []
|
|
|
+ # 线程数
|
|
|
+ thread_num = 2
|
|
|
+ # 获取slb下所有服务器信息
|
|
|
+ res = getInstanceId(slb_id_1)
|
|
|
+ # slb下服务器总数
|
|
|
+ total = len(res["BackendServers"]["BackendServer"])
|
|
|
+ # q1 = queue.Queue()
|
|
|
+ # q2 = queue.Queue()
|
|
|
+ thread_pool = ThreadPoolExecutor(max_workers=thread_num)
|
|
|
+ # 获取slb下所有服务器的 instance_id
|
|
|
+ if res["BackendServers"]["BackendServer"]:
|
|
|
+ for i in range((len(res["BackendServers"]["BackendServer"]))):
|
|
|
+ instance_id = res["BackendServers"]["BackendServer"][i]["ServerId"]
|
|
|
+ # q1.put(instance_id)
|
|
|
+ thread_pool.submit(update_sever, instance_id)
|
|
|
+ thread_pool.shutdown(wait=True)
|
|
|
+
|
|
|
+ # while not q1.empty():
|
|
|
+ # # 摘流量
|
|
|
+ # for i in range(thread_num):
|
|
|
+ # if q1.empty():
|
|
|
+ # break
|
|
|
+ # instance_id = q1.get()
|
|
|
+ # # 设置slb权重为0
|
|
|
+ # setWeight(slb_id_1, instance_id, 0)
|
|
|
+ # q2.put(instance_id)
|
|
|
+ #
|
|
|
+ # # 挂流量
|
|
|
+ # for i in range(thread_num):
|
|
|
+ # # 开启线程
|
|
|
+ # thread = MyThread(update)
|
|
|
+ # thread.start()
|
|
|
+ # threads.append(thread)
|
|
|
+ # for thread in threads:
|
|
|
+ # thread.join()
|