优化Thrift客户端代码:分离连接、协议和调用逻辑
Thrift客户端代码通常包含连接建立、协议定义和服务调用三个部分。为了提升代码的可读性、可维护性和复用性,建议将这三个部分分离。
一种有效的策略是创建一个独立的工具类模块(例如 util.go
),该模块包含一个用于创建Thrift客户端的函数。此函数负责建立连接、配置协议工厂,并返回客户端实例和底层传输对象。示例如下:
func fundinfoserviceclient() (*fundinfo.fundinfoserviceclient, thrift.TTransport) {
transportFactory := thrift.NewTFramedTransportFactory(thrift.NewTTransportFactory())
protocolFactory := thrift.NewTBinaryProtocolFactoryDefault()
transportSocket, err := thrift.NewTSocket(net.JoinHostPort(ip, port))
if err != nil {
fmt.Fprintln(os.Stderr, "无法解析地址:", err)
os.Exit(1)
}
transport, _ := transportFactory.GetTransport(transportSocket)
client := fundinfo.NewFundInfoServiceClientFactory(transport, protocolFactory)
if err := transport.Open(); err != nil {
fmt.Fprintf(os.Stderr, "连接%s:%s失败: %vn", ip, port, err)
os.Exit(1)
}
return client, transport
}
在客户端方法中,可以直接调用这个函数获取客户端实例和传输对象。例如:
func GetFundInfo() {
client, transport := util.FundInfoServiceClient()
defer transport.Close()
res, err := client.GetFundInfo(context.Background(), fundcode)
if err != nil {
// 处理错误
}
// 处理结果 res
}
这种分离方法使客户端方法更加简洁,专注于具体的业务逻辑,而连接、协议等公共部分则被封装在工具类中,提高了代码的可重用性和可维护性,并避免了代码冗余。