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. 示例使用：

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

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