#Python 分布式进程快速入门与实战（附完整代码）

import random
import time
from multiprocessing import Queue
from multiprocessing.managers import BaseManager

# 自定义管理器类（必须继承 BaseManager）
class QueueManager(BaseManager):
    pass

def main():
    # 1. 初始化本地队列
    task_queue = Queue()   # 存放待计算的随机数
    result_queue = Queue() # 存放计算结果

    # 2. 把本地队列注册为网络可调用的接口
    # 使用 lambda 匿名函数返回队列实例，避免提前实例化问题
    QueueManager.register('get_task_queue', callable=lambda: task_queue)
    QueueManager.register('get_result_queue', callable=lambda: result_queue)

    # 3. 启动网络管理器
    # address: ('0.0.0.0', 端口) 表示监听本机所有网卡上的该端口
    # authkey: 字节串密钥，Master 和 Worker 必须完全一致，防止非法连接
    manager = QueueManager(
        address=('0.0.0.0', 56789),
        authkey=b'python_distributed_test_202x'
    )
    manager.start()
    print(f"✅ Master 启动成功！监听端口 56789")

    try:
        # 4. 获取代理队列对象（和本地 Queue 用法完全一致）
        proxy_task = manager.get_task_queue()
        proxy_result = manager.get_result_queue()

        # 5. 分发模拟任务：放 10 个 0～10000 的随机数
        print("\n📦 开始分发任务...")
        for _ in range(10):
            n = random.randint(0, 10000)
            print(f"→ 放入任务：计算 {n} 的平方")
            proxy_task.put(n)

        # 6. 阻塞等待 Worker 返回结果（设置 30 秒超时避免死等）
        print("\n⏳ 等待 Worker 返回结果...")
        for _ in range(10):
            res = proxy_result.get(timeout=30)
            print(f"← 收到结果：{res}")

    except KeyboardInterrupt:
        print("\n⚠️  Master 被手动中断")
    finally:
        # 7. 清理资源，关闭网络管理器
        manager.shutdown()
        print("\n👋 Master 已安全退出")

if __name__ == "__main__":
    main()

import time
import sys
from multiprocessing.managers import BaseManager

# 自定义管理器类（和 Master 保持一致，否则注册会失败）
class QueueManager(BaseManager):
    pass

def main():
    # 1. 注册要调用的 Master 接口（只需要名称，不需要具体实现）
    QueueManager.register('get_task_queue')
    QueueManager.register('get_result_queue')

    # 2. 从命令行或配置获取 Master 的 IP（默认 127.0.0.1，方便单测）
    server_ip = sys.argv[1] if len(sys.argv) > 1 else '127.0.0.1'
    print(f"🔗 正在连接 Master: {server_ip}:56789...")

    try:
        # 3. 连接到 Master 的网络管理器
        manager = QueueManager(
            address=(server_ip, 56789),
            authkey=b'python_distributed_test_202x'
        )
        manager.connect()
        print("✅ 连接 Master 成功！")

        # 4. 获取代理队列对象
        proxy_task = manager.get_task_queue()
        proxy_result = manager.get_result_queue()

        # 5. 循环取任务（直到超时 10 秒没有新任务，自动退出）
        print("\n🏃 Worker 开始工作...按 Ctrl+C 可手动停止")
        while True:
            try:
                n = proxy_task.get(timeout=10)
                print(f"→ 收到任务：计算 {n} 的平方")
                # 模拟 1 秒的真实耗时任务
                time.sleep(1)
                res = f"{n} * {n} = {n*n}"
                print(f"← 完成任务：{res}")
                proxy_result.put(res)
            except Exception as e:
                print(f"\n⚠️  Worker 异常 / 无新任务：{str(e)}")
                break

    except ConnectionRefusedError:
        print("❌ 连接失败：请检查 Master 是否启动、IP/端口/密钥是否正确")
    finally:
        print("\n👋 Worker 已安全退出")

if __name__ == "__main__":
    main()