如何高效分离Thrift客户端的连接、协议和调用？

优化Thrift客户端代码：分离连接、协议和调用

在Thrift客户端开发中，重复编写连接和协议设置代码是常见的低效现象。为了提高代码可维护性和效率，我们可以将连接、协议设置与实际业务调用分离。

一种有效的方案是构建一个通用的客户端工厂，负责处理连接和协议的初始化工作。以下示例展示了这种方法：

fundinfoserviceclient.go (客户端工厂)

PhotoScissors

免费自动图片背景去除

下载

package util

import (
    "fmt"
    "net"
    "os"

    "github.com/apache/thrift/lib/go/thrift"
    "your_project/fundinfo" // 替换为你的 fundinfo 包路径
)

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

    transportSocket, err := thrift.NewTSocket(net.JoinHostPort(ip, port))
    if err != nil {
        return nil, nil, fmt.Errorf("创建socket失败: %w", err)
    }
    transport, err := transportFactory.GetTransport(transportSocket)
    if err != nil {
        return nil, nil, fmt.Errorf("创建transport失败: %w", err)
    }

    client := fundinfo.NewFundInfoServiceClientFactory(transport, protocolFactory)

    if err := transport.Open(); err != nil {
        return nil, nil, fmt.Errorf("连接服务器失败: %w", err)
    }

    return client, transport, nil
}

在实际调用方法中，只需调用此工厂方法即可获取客户端和传输对象：

main.go (业务调用)

package main

import (
    "context"
    "fmt"
    "your_project/util" // 替换为你的 util 包路径
    "your_project/fundinfo" // 替换为你的 fundinfo 包路径
)

func main() {
    ctx := context.Background()
    ip := "your_server_ip" // 替换为你的服务器IP
    port := "your_server_port" // 替换为你的服务器端口
    fundcode := "your_fund_code" // 替换为你的基金代码

    client, transport, err := util.FundInfoServiceClient(ip, port)
    if err != nil {
        fmt.Printf("创建客户端失败: %v\n", err)
        return
    }
    defer transport.Close()

    res, err := client.GetFundInfo(ctx, fundcode)
    if err != nil {
        fmt.Printf("调用GetFundInfo失败: %v\n", err)
        return
    }

    fmt.Printf("基金信息: %+v\n", res)
}

通过这种方式，连接、协议和调用逻辑被清晰地分离，使代码更简洁、易于维护和扩展。 记得替换代码中的占位符为你的实际项目信息。