Python实现简单负载均衡

使用场景,最近在研究把废弃的安卓手机利用起来跑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请求并打印响应。

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

只想安静的搬砖

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值