具有可变输入/输出类型的通用函数

Question

只是玩 aws sdk for go。当列出不同类型的资源时，我倾向于使用很多非常相似的函数，例如下面示例中的两个。有没有办法将它们重写为一个通用函数，该函数将根据作为参数传递的内容返回特定类型？

就像是：

func generic(session, funcToCall, t, input) (interface{}, error) {}

目前我必须这样做（功能是相同的，只是类型发生了变化）：

func getVolumes(s *session.Session) ([]*ec2.Volume, error) {

    client := ec2.New(s)

    t := []*ec2.Volume{}

    input := ec2.DescribeVolumesInput{}

    for {

        result, err := client.DescribeVolumes(&input)

        if err != nil {

            return nil, err

        }

        t = append(t, result.Volumes...)

        if result.NextToken != nil {

            input.NextToken = result.NextToken

        } else {

            break

        }

    }

    return t, nil

}

func getVpcs(s *session.Session) ([]*ec2.Vpc, error) {

    client := ec2.New(s)

    t := []*ec2.Vpc{}

    input := ec2.DescribeVpcsInput{}

    for {

        result, err := client.DescribeVpcs(&input)

        if err != nil {

            return nil, err

        }

        t = append(t, result.Vpcs...)

        if result.NextToken != nil {

            input.NextToken = result.NextToken

        } else {

            break

        }

    }

    return t, nil

} 

Answer 1

因为您只处理函数，所以可以使用 reflect 包在运行时生成函数。使用对象类型 (Volume, Vpc) 可以导出所有后续信息，以提供一个完全通用的实现，该实现非常枯燥，甚至更复杂、更慢。package mainimport (    "errors"    "fmt"    "reflect")func main() {    fmt.Printf("%T\n", getter(Volume{}))    fmt.Printf("%T\n", getter(Vpc{}))}type DescribeVolumesInput struct{}type DescribeVpcs struct{}type Volume struct{}type Vpc struct{}type Session struct{}type Client struct{}func New(s *Session) Client { return Client{} }var typeRegistry = make(map[string]reflect.Type)func init() {    some := []interface{}{DescribeVolumesInput{}, DescribeVpcs{}}    for _, v := range some {        typeRegistry[fmt.Sprintf("%T", v)] = reflect.TypeOf(v)    }}var errV = errors.New("")var errType = reflect.ValueOf(&errV).Elem().Type()var zeroErr = reflect.Zero(reflect.TypeOf((*error)(nil)).Elem())var nilErr = []reflect.Value{zeroErr}func getter(of interface{}) interface{} {    outType := reflect.SliceOf(reflect.PtrTo(reflect.TypeOf(of)))    fnType := reflect.FuncOf([]reflect.Type{reflect.TypeOf(new(Session))}, []reflect.Type{outType, errType}, false)    fnBody := func(input []reflect.Value) []reflect.Value {        client := reflect.ValueOf(New).Call(input)[0]        t := reflect.MakeSlice(outType, 0, 0)        name := fmt.Sprintf("Describe%TsInput", of)        descInput := reflect.New(typeRegistry[name]).Elem()        mName := fmt.Sprintf("Describe%Ts", of)        meth := client.MethodByName(mName)        if !meth.IsValid() {            return []reflect.Value{                t,                reflect.ValueOf(fmt.Errorf("no such method %q", mName)),            }        }        for {            out := meth.Call([]reflect.Value{descInput.Addr()})            if len(out) > 0 {                errOut := out[len(out)-1]                if errOut.Type().Implements(errType) && errOut.IsNil() == false {                    return []reflect.Value{t, errOut}                }            }            result := out[1]            fName := fmt.Sprintf("%Ts", of)            if x := result.FieldByName(fName); x.IsValid() {                t = reflect.AppendSlice(t, x)            } else {                return []reflect.Value{                    t,                    reflect.ValueOf(fmt.Errorf("field not found %q", fName)),                }            }            if x := result.FieldByName("NextToken"); x.IsValid() {                descInput.FieldByName("NextToken").Set(x)            } else {                break            }        }        return []reflect.Value{t, zeroErr}    }    fn := reflect.MakeFunc(fnType, fnBody)    return fn.Interface()}

Answer 2

代理第 3 方 API，用 go 实现非常简单，下面是它是如何用 endly e2e test runner AWS 代理实现的我会说 AWS API 是代理的完美候选者，只要反射性能价格不是问题。像kubernetes这样的其他一些第三方 API 更具挑战性，但仍然很容易用 go 代理，这是反射和代码生成的结合。