优化Thrift客户端代码，分离连接、协议和调用逻辑
为了避免Thrift客户端代码中连接、协议和调用逻辑的冗余，建议采用以下分离策略，提升代码可维护性和可重用性。
1. 创建工具类 (util 目录)
在util目录下新建thriftserviceclient.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() 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 mainimport ( "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为你的实际项目路径和服务器地址。
以上就是如何优雅地分离Thrift客户端连接、协议和调用逻辑？的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1383233.html
微信扫一扫 
支付宝扫一扫 
