如何优化Go语言Thrift客户端代码，分离连接、协议和服务调用？

import (
    "net"
    "os"
    "fmt"

    "github.com/apache/thrift/lib/go/thrift"
)

func newFundInfoServiceClient(ip string, port int) (*fundinfo.FundInfoServiceClient, thrift.TTransport, error) {
    transportFactory := thrift.NewTFramedTransportFactory(thrift.NewTTransportFactory())
    protocolFactory := thrift.NewTBinaryProtocolFactoryDefault()

    transportSocket, err := thrift.NewTSocket(net.JoinHostPort(ip, fmt.Sprint(port)))
    if err != nil {
        return nil, nil, fmt.Errorf("无法解析地址: %w", err)
    }

    transport, err := transportFactory.GetTransport(transportSocket)
    if err != nil {
        return nil, nil, fmt.Errorf("创建传输失败: %w", err)
    }

    client := fundinfo.NewFundInfoServiceClientFactory(transport, protocolFactory)

    if err := transport.Open(); err != nil {
        return nil, nil, fmt.Errorf("打开套接字失败: %w", err)
    }

    return client, transport, nil
}

func main() {
    client, transport, err := newFundInfoServiceClient("localhost", 9090)
    if err != nil {
        fmt.Fprintf(os.Stderr, "连接失败: %v\n", err)
        os.Exit(1)
    }
    defer transport.Close()

    res, err := client.GetFundInfo(context.Background(), fundcode)
    if err != nil {
        fmt.Fprintf(os.Stderr, "调用服务失败: %v\n", err)
        os.Exit(1)
    }

    // 处理结果 res
}