优化Thrift客户端代码，分离连接、协议和调用逻辑
为了避免Thrift客户端代码中连接、协议和调用逻辑的冗余，建议采用以下分离策略，提升代码可维护性和可重用性。
1. 创建工具类 (util 目录)
在util目录下新建thriftserviceclient.go文件，封装通用的连接和协议创建逻辑：
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()
transport, err := thrift.NewTSocket(net.JoinHostPort(ip, port))
if err != nil {
return nil, nil, fmt.Errorf("error resolving address: %w", err)
}
tTransport := transportFactory.GetTransport(transport)
client := fundinfo.NewFundInfoServiceClientFactory(tTransport, protocolFactory)
if err := tTransport.Open(); err != nil {
return nil, nil, fmt.Errorf("error opening socket to %s:%s: %w", ip, port, err)
}
return client, tTransport, nil
}
2. 简化客户端调用逻辑
在客户端方法中，只需调用工具类获取客户端对象和传输对象，然后执行所需的方法：
package main
import (
"context"
"your_project/util" // 替换为你的 util 包路径
)
func main() {
client, tTransport, err := util.FundInfoServiceClient("your_ip", "your_port") // 替换为你的IP和端口
if err != nil {
// 处理错误
}
defer tTransport.Close()
ctx := context.Background()
res, err := client.GetFundInfo(ctx, "fundcode") // 替换为实际的 fundcode
if err != nil {
// 处理错误
}
// 处理结果 res
}
改进说明:
- 使用了
net.JoinHostPort函数，避免手动拼接IP和端口。 - 使用了
fmt.Errorf包装错误信息，方便调试。 - 添加了错误处理，提高代码健壮性。
- 将IP和端口作为参数传入，提升了工具类的复用性。
通过这种方式，连接、协议和调用逻辑被清晰地分离，提高了代码的可读性、可维护性和可重用性。 当需要调用其他Thrift服务方法时，只需修改调用client对象的相应方法即可，无需重复编写连接和协议创建代码。 记住替换your_project/fundinfo和your_ip, your_port为你的实际项目路径和服务器地址。
