基于gRPC的最小连接负载均衡实现出现TypeError报错排查

问题场景 #

在实现基于gRPC的最小连接负载均衡时，运行node2.py触发负载均衡流程时出现如下错误：

Node overloaded -- starting load balancing process
Traceback (most recent call last):
File "D:\lab7p2\least connection\node2.py", line 73, in <module>
node.check_load()
File "D:\lab7p2\least connection\node2.py", line 46, in check_load
self.balance_load()
File "D:\lab7p2\least connection\node2.py", line 36, in balance_load
request = least_connection_pb2.LoadTransferMessage(server=least_connections_address, load=int(transfer_load))
              ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
TypeError: Message must be initialized with a dict: least_connection.LoadTransferMessage

错误原因 #

查看least_connection.proto定义可知，LoadTransferMessage的server字段是嵌套的Server消息类型，而非字符串类型：

message LoadTransferMessage {
  Server server = 1;
  int32 load = 2;
}

但代码中直接传入了字符串类型的least_connections_address作为server字段的值，不符合gRPC消息初始化规则——嵌套消息必须用对应的消息对象或字典来初始化，因此触发类型错误。

修复方案 #

有两种正确的初始化方式：

方式1：创建Server消息对象再传入 #

修改balance_load方法中初始化LoadTransferMessage的代码，先构造对应的Server对象：

def balance_load(self):
    least_connections_server = self.get_least_connections_server()
    if least_connections_server is None:
        print("No server available for load transfer")
        return

    least_connections_address, least_connections_count = least_connections_server

    transfer_load = self.connections - self.threshold
    if transfer_load <= 0:
        print("No load to transfer")
        return

    # 修正：连接目标服务器地址而非自身地址
    channel = grpc.insecure_channel(least_connections_address)
    stub = least_connection_pb2_grpc.LeastConnectionLoadBalancerStub(channel)

    # 先创建Server消息对象
    server_msg = least_connection_pb2.Server(name=least_connections_address, current_connections=least_connections_count)
    # 初始化LoadTransferMessage
    request = least_connection_pb2.LoadTransferMessage(server=server_msg, load=int(transfer_load))
    response = stub.TransferLoad(request)
    if response.success:
        print(f"Node {self.node_id} --> {transfer_load} Units --> Server {least_connections_address}")
        self.connections -= transfer_load

方式2：用字典直接初始化嵌套字段 #

直接通过字典结构初始化嵌套的server字段：

def balance_load(self):
    least_connections_server = self.get_least_connections_server()
    if least_connections_server is None:
        print("No server available for load transfer")
        return

    least_connections_address, least_connections_count = least_connections_server

    transfer_load = self.connections - self.threshold
    if transfer_load <= 0:
        print("No load to transfer")
        return

    # 修正：连接目标服务器地址而非自身地址
    channel = grpc.insecure_channel(least_connections_address)
    stub = least_connection_pb2_grpc.LeastConnectionLoadBalancerStub(channel)

    # 用字典初始化嵌套的server字段
    request = least_connection_pb2.LoadTransferMessage(
        server={"name": least_connections_address, "current_connections": least_connections_count},
        load=int(transfer_load)
    )
    response = stub.TransferLoad(request)
    if response.success:
        print(f"Node {self.node_id} --> {transfer_load} Units --> Server {least_connections_address}")
        self.connections -= transfer_load

额外注意点 #

原代码中channel = grpc.insecure_channel(self.server_address)是连接到自身节点的地址，实际负载转移应该连接到目标服务器（least_connections_address），否则请求会发送给自身，不符合负载均衡逻辑，修复时建议一并修正此问题。

内容的提问来源于stack exchange，提问作者Yash Pahlani