优化Thrift客户端代码:分离连接、协议和调用
在Thrift客户端开发中,重复编写连接和协议设置代码是常见的低效现象。为了提高代码可维护性和效率,我们可以将连接、协议设置与实际业务调用分离。
一种有效的方案是构建一个通用的客户端工厂,负责处理连接和协议的初始化工作。以下示例展示了这种方法:
fundinfoserviceclient.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() 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) }
通过这种方式,连接、协议和调用逻辑被清晰地分离,使代码更简洁、易于维护和扩展。 记得替换代码中的占位符为你的实际项目信息。
以上就是如何高效分离Thrift客户端的连接、协议和调用?的详细内容,更多请关注php中文网其它相关文章!
每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号