为了账号安全,请及时绑定邮箱和手机立即绑定

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

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

Go
江户川乱折腾 2023-06-19 14:01:46
只是玩 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} 
查看完整描述

2 回答

?
慕哥9229398

TA贡献1877条经验 获得超6个赞

因为您只处理函数,所以可以使用 reflect 包在运行时生成函数。

使用对象类型 (Volume, Vpc) 可以导出所有后续信息,以提供一个完全通用的实现,该实现非常枯燥,甚至更复杂、更慢。


package main


import (

    "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()

}


查看完整回答
反对 回复 2023-06-19
?
月关宝盒

TA贡献1772条经验 获得超5个赞

代理第 3 方 API,用 go 实现非常简单,下面是它是如何用 endly e2e test runner AWS 代理实现的

我会说 AWS API 是代理的完美候选者,只要反射性能价格不是问题。

像kubernetes这样的其他一些第三方 API 更具挑战性,但仍然很容易用 go 代理,这是反射和代码生成的结合。


查看完整回答
反对 回复 2023-06-19
  • 2 回答
  • 0 关注
  • 119 浏览
慕课专栏
更多

添加回答

举报

0/150
提交
取消
意见反馈 帮助中心 APP下载
官方微信