Общая функция с переменными типами ввода / вывода - PullRequest
Новая
       42

Общая функция с переменными типами ввода / вывода

0 голосов
/ 12 мая 2019

Просто играю с aws sdk на ходу. При перечислении ресурсов разных типов я склонен иметь много очень похожих функций, таких как две в примере ниже. Есть ли способ переписать их как одну универсальную функцию, которая будет возвращать определенный тип в зависимости от того, что передается как param?

Что-то вроде: 

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 ]

0 голосов
/ 13 мая 2019

Прокси-сторонний API, довольно прост в реализации вот, как это было реализовано с помощью конечного бегуна e2e AWS proxy

Я бы сказал, что AWS API является идеальным кандидатом для прокси, если цена за отражение не является проблемой.

Некоторые другие сторонние API, такие как kubernetes гораздо сложнее, но все же довольно легко прокси с Go, который представляет собой сочетание отражения и генерации кода:

0 голосов
/ 12 мая 2019

Поскольку вы имеете дело только с функциями, можно использовать пакет отражения для генерации функций во время выполнения.

Используя тип объекта (Volume, Vpc), можно получить всю информацию о последующих запросах, чтобы получить полностью общую реализацию, которая действительно сухая, в той степени, в которой она более сложна и медленнее.

Это не проверено, вы можете помочь в тестировании и исправлении, но что-то вроде этого должно поставить вас на путь

https://play.golang.org/p/mGjtYVG2OZS

Идея реестра возникла из этого ответа https://stackoverflow.com/a/23031445/4466350

для справки: документация golang пакета отражений находится на https://golang.org/pkg/reflect/

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()
}
...