首页 > 编程语言 >Python实现简单负载均衡

Python实现简单负载均衡

时间:2024-06-06 22:30:49浏览次数:20  
标签:负载 缓存 Python cache server 均衡 服务器 response pool

使用场景,最近在研究把废弃的安卓手机利用起来跑python代码,但是有好多台安卓手机。我就想着把其中一台拿来做负载均衡,然后分配给其他4台烂手机,哈哈哈,废物再利用。至于安卓怎么跑python代码,就自己查一下相关资料了哈,这里就不赘述了。
 

import requests
import threading
import time
from queue import Queue

# 定义服务器列表和缓存池
servers = ["http://serverA.com", "http://serverB.com", "http://serverC.com", "http://serverD.com"]
cache_pool = Queue(maxsize=len(servers))

# 初始化缓存池,将所有服务器加入缓存池
for server in servers:
    cache_pool.put(server)

# 检查服务器状态的函数
def check_server_status():
    while True:
        for server in servers:
            try:
                # 尝试请求服务器
                response = requests.get(server, timeout=5)
                # 如果服务器正常且不在缓存池中,将其加入缓存池
                if response.status_code == 200 and server not in list(cache_pool.queue):
                    cache_pool.put(server)
            except requests.RequestException:
                # 如果服务器不可用且在缓存池中,将其从缓存池移除
                if server in list(cache_pool.queue):
                    with cache_pool.mutex:
                        queue_list = list(cache_pool.queue)
                        queue_list.remove(server)
                        cache_pool.queue = Queue(len(queue_list))
                        for s in queue_list:
                            cache_pool.put(s)
        # 每10秒检查一次服务器状态
        time.sleep(10)

# 启动服务器状态检查线程
status_thread = threading.Thread(target=check_server_status)
status_thread.daemon = True
status_thread.start()

# 处理请求的函数
def handle_request(method, url, headers=None, cookies=None, data=None, retries=3):
    # 如果缓存池为空,返回错误信息
    if cache_pool.empty():
        return {"error": "No available servers"}

    # 从缓存池中获取一个服务器
    server = cache_pool.get()
    try:
        # 处理GET请求
        if method == 'GET':
            response = requests.get(f"{server}{url}", headers=headers, cookies=cookies)
        # 处理POST请求
        elif method == 'POST':
            response = requests.post(f"{server}{url}", headers=headers, cookies=cookies, data=data)
        else:
            response = {"error": "Unsupported HTTP method"}
        # 将服务器放回缓存池
        cache_pool.put(server)
        return response
    except requests.RequestException:
        # 如果请求失败,递归重试
        if retries > 0:
            return handle_request(method, url, headers, cookies, data, retries - 1)
        else:
            return {"error": "Failed to process request after retries"}

# 示例使用
if __name__ == "__main__":
    headers = {"Content-Type": "application/json"}
    cookies = {"session_id": "123456"}
    data = {"key": "value"}

    # GET请求示例
    get_response = handle_request('GET', '/api/test', headers=headers, cookies=cookies)
    print(get_response.text if isinstance(get_response, requests.Response) else get_response)

    # POST请求示例
    post_response = handle_request('POST', '/api/test', headers=headers, cookies=cookies, data=data)
    print(post_response.text if isinstance(post_response, requests.Response) else post_response)

代码注释说明

  1. 服务器列表和缓存池初始化

    • 服务器列表 servers 包含四个服务器的URL。
    • 使用 Queue 初始化缓存池 cache_pool,大小为服务器数量。
    • 将所有服务器加入缓存池。
  2. 检查服务器状态的函数 check_server_status

    • 该函数在一个独立的线程中运行,定期检查服务器状态。
    • 对每个服务器发送一个GET请求,超时时间为5秒。
    • 如果服务器响应正常且不在缓存池中,将其加入缓存池。
    • 如果服务器不可用且在缓存池中,将其从缓存池中移除。
    • 每10秒检查一次服务器状态。
  3. 处理请求的函数 handle_request

    • 从缓存池中获取一个可用服务器。
    • 根据请求方法(GET或POST)发送请求。
    • 请求完成后将服务器放回缓存池。
    • 如果请求失败,递归重试最多3次。
  4. 示例使用

    • 设置请求头 headerscookies
    • 发送一个GET请求并打印响应。
    • 发送一个POST请求并打印响应。

通过这些注释,应该可以更清楚地理解代码的实现和工作原理。

标签:负载,缓存,Python,cache,server,均衡,服务器,response,pool
From: https://blog.csdn.net/m0_65089078/article/details/139511374

相关文章

  • Python中的“点阵字体”
    “点阵字体”是个啥?,在python中怎么使?在现在全面高清的5G5G5G时代,它还有用“武”之地?(笔记模板由python脚本于2024年06月01日18:44:31创......
  • 基于Python的实验室管理系统的设计与实现(论文+源码)_kaic
    摘 要随着实验室设备越来越多,实验室及其设备管理工作变得越来越繁重,还存在些管理模式仍旧处于手工管理模式和一些抢占实验室的不文明现象,传统的手工模式已经满足不了日益增长的管理需求,而本系统摒弃传统模式,开启新式的实验室管理模式。在需求进行分析的基础上,采用Python语言......
  • 基于Python的街区医院管理系统的设计与实现(论文+源码)_kaic
    基于Python的街区医院管理系统的设计与实现摘 要采用Python语言、Mysql数据库,在IDEA平台下实现了街区医院管理系统,利用街道医疗机构的管理系统,不仅能够有效地进行信息管理,促进各部门之间的有序合作,还能够大幅改善医疗环境,极大地改善病人的就诊体验,并且能够更加有效地满......
  • 运筹学习题Python精解——线性规划
    题1某企业有三个车间生产同一种产品。每件产品由四个零件1和三个零件2组成。两个零件需耗用两种原材料A和B。已知这两种原材料的供应量分别为300kg和500kg。由于三个车间拥有的设备及工艺条件不同,每个工班原材料耗用量和零件产量也不同。见下表(三个车间每班用料和生产......
  • Python-GEE遥感云大数据分析、管理与可视化
    随着航空、航天、近地空间遥感平台的持续发展,遥感技术近年来取得显著进步。遥感数据的空间、时间、光谱分辨率及数据量均大幅提升,呈现出大数据特征。这为相关研究带来了新机遇,但同时也带来巨大挑战。传统的工作站和服务器已无法满足大区域、多尺度海量遥感数据处理需求。为解......
  • 基于FPGA的图像直方图均衡化处理verilog实现,包含tb测试文件和MATLAB辅助验证
    1.算法运行效果图预览FPGA的仿真图如下:   将数据导入MATLAB,对比结果如下:  2.算法运行软件版本MATLAB2022a vivado2019.2 3.部分核心程序  `timescale1ns/1ps////Company://Engineer:////CreateDate:2023/01/1901:47:15//DesignNa......
  • python爬虫学习路径
    python爬虫学习路径阶段一:Python基础(预计1-2周)里程碑1:掌握Python基础语法数据类型(字符串、列表、字典等)控制流(条件语句、循环语句)函数定义与使用模块导入与使用文件读写操作学习资源:廖雪峰Python教程Python官方文档CodecademyPython课程练习......
  • 【2024最新】Python 学习路线分享
    学习资料已打包,需要的小伙伴可以戳这里学习资料整理了一份Python学习路线。内容依然是从入门到进阶,既有教程,也有经典书籍推荐,还有实战开源项目。Python的发展方向还是挺多的,比如服务端开发,爬虫,数据分析,机器学习等,本文推荐的内容全部是服务端开发,Web开发方向。主......
  • python: 如何将方法对象和类实例对象序列化
    文章目录1.Json序列化2.Pickle序列化1.Json序列化在python中对于一般数据的序列化保存经常使用的就是json序列化json序列化可序列的对象:一般数据类型列表,字典,字符串,数字,布尔值,None等json可以序列化的对象类型: 基本数据类型: None 布尔值:True,False 数......
  • python的datetime使用时区转换的注意事项
    由于做美股的开始,经常用到需要将utc时间转换成美东时间,或者美东时间转换为utc时间。美东时间涉及到一个夏令时,一个冬令时,不同的阶段与utc时间的时区相差不同的时间段 以前我一直用pytz的三方模块给datetime赋值时区信息,然后生成的对象用astimezone的方法,转换到对应的时区信息......