Go语言Thrift客户端分离连接与协议技巧

Go语言Thrift客户端开发：提升代码可重用性与可维护性

在Go语言中开发Thrift客户端时，如果将网络连接、协议配置和服务调用逻辑混杂在一起，代码会变得冗长且难以维护。为了提高代码的可重用性和可维护性，建议将这三个部分分离。

一种有效的方案是创建一个独立的函数或方法（例如newFundInfoServiceClient），负责建立Thrift客户端连接并配置传输协议。示例如下：

func newFundInfoServiceClient(ip, port string) (*fundinfo.FundInfoServiceClient, thrift.TTransport, error) {
    transportFactory := thrift.NewTFramedTransportFactory(thrift.NewTTransportFactory())
    protocolFactory := thrift.NewTBinaryProtocolFactoryDefault()

    transport, err := thrift.NewTSocket(net.JoinHostPort(ip, port))
    if err != nil {
        return nil, nil, fmt.Errorf("无法解析地址: %w", err)
    }

    transport = transportFactory.GetTransport(transport)
    client := fundinfo.NewFundInfoServiceClientFactory(transport, protocolFactory)

    if err := transport.Open(); err != nil {
        return nil, nil, fmt.Errorf("连接到 %s:%s 失败: %w", ip, port, err)
    }

    return client, transport, nil
}

在实际调用服务方法时，只需调用newFundInfoServiceClient函数创建客户端，然后调用所需的服务方法即可。使用完毕后，记得关闭传输对象。示例如下：

func main() {
    client, transport, err := newFundInfoServiceClient("127.0.0.1", "9090")
    if err != nil {
        log.Fatal(err)
    }
    defer transport.Close()

    ctx := context.Background()
    res, err := client.GetFundInfo(ctx, "fundcode")
    if err != nil {
        log.Fatal(err)
    }
    // 处理结果 res
}

这种设计模式使代码更加模块化，易于测试和维护，并提高了代码的可重用性。 当需要调用其他Thrift服务时，只需创建相应的客户端函数即可，无需重复编写连接和协议配置代码。