优化Thrift客户端代码:分离连接、协议和调用
在Thrift客户端开发中,重复编写连接和协议设置代码是常见的低效现象。为了提高代码可维护性和效率,我们可以将连接、协议设置与实际业务调用分离。
一种有效的方案是构建一个通用的客户端工厂,负责处理连接和协议的初始化工作。以下示例展示了这种方法:
fundinfoserviceclient.go (客户端工厂)
package utilimport ( "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 mainimport ( "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", err) return } defer transport.Close() res, err := client.GetFundInfo(ctx, fundcode) if err != nil { fmt.Printf("调用GetFundInfo失败: %v", err) return } fmt.Printf("基金信息: %+v", res)}
登录后复制
通过这种方式,连接、协议和调用逻辑被清晰地分离,使代码更简洁、易于维护和扩展。 记得替换代码中的占位符为你的实际项目信息。
以上就是如何高效分离Thrift客户端的连接、协议和调用?的详细内容,更多请关注【创想鸟】其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至253000106@qq.com举报,一经查实,本站将立刻删除。
发布者:PHP中文网,转转请注明出处:https://www.chuangxiangniao.com/p/2309152.html