| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061 |
- import time
- from concurrent.futures import ThreadPoolExecutor
- import requests
- from .ipFind import ipFind # 假设ipFind是一个从本地模块导入的函数
- def task(i):
- """发送HTTP请求检查服务器状态的任务函数"""
- #sleep_seconds = random.uniform(0, 0.1)
- #print(sleep_seconds)
- # 可选：取消注释以显示线程执行信息
- # print('线程名称：{}，参数：{}，睡眠时间：{}'.format(threading.current_thread().name, i, sleep_seconds))
- try:
- # 随机延迟，减少请求风暴
- #time.sleep(sleep_seconds)
- # 发送请求并设置合理的超时时间
- r = requests.get(f'http://{i}:5001/findserver', timeout=0.5, headers={'Content-Type': 'application/json'})
- r = r.json()['msg']
- print(r)
- except requests.exceptions.ConnectTimeout:
- # 连接超时处理
- r = '0'
- except requests.exceptions.ReadTimeout:
- # 读取超时处理
- r = '0'
- except requests.exceptions.ConnectionError:
- # 连接错误处理
- r = '0'
- except Exception as e:
- # 其他异常处理
- #print(f"请求处理异常: {e}")
- r = '0'
- return [i, r]
- def findServer():
- """查找可用服务器的主函数"""
- time1 = time.time()
- # 创建线程池，设置合理的最大工作线程数
- with ThreadPoolExecutor(max_workers=500) as pool:
- # 获取所有IP地址
- allIP = ipFind()
- # 使用线程池并行执行任务
- result = list(pool.map(task, allIP))
- #print(f'耗时：{time.time() - time1}')
- # 筛选可用服务器
- serverList = []
- for item in result:
- if "天府综合平台一期:" in item[1]:
- item[1] = item[1].replace("天府综合平台一期:", "")
- serverList.append(item)
- return serverList
- if __name__ == '__main__':
- l = findServer()
- #print("可用服务器列表:", l)
|
