如何优雅地分离Thrift客户端连接、协议和调用逻辑？

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
}

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
}