API обработчика AWS Lambda GoLang - PullRequest
Новая
       7

API обработчика AWS Lambda GoLang

0 голосов
/ 24 мая 2018

Можно ли реализовать обработчик функции Lambda с пользовательским API?

Для aws-lambda-go допустимый обработчик может быть одного из следующих типов: 

//  func ()
//  func () error
//  func (TIn) error
//  func () (TOut, error)
//  func (TIn) (TOut, error)
//  func (context.Context) error
//  func (context.Context, TIn) error
//  func (context.Context) (TOut, error)
//  func (context.Context, TIn) (TOut, error)

В моем случае TIn - это KinesisEvent, содержащее MySpecificType в качестве полезной нагрузки.Я ищу способ (возможно, перезаписать lambda.Start()), чтобы иметь возможность создать обработчик типа 

func(MySpecificType) error

Цель состоит в том, чтобы не извлекать полезную нагрузку из KinesisEvent в каждой лямбде и делать обработчикнезависимый от источника

1 Ответ

0 голосов
/ 24 мая 2018

Похоже, решение довольно простое.Просто перезапишите метод lambda.Start (): 

func Start(handler func(t * MySpecificType, i int) error) {

    lambda.Start(func(kinesisEvent events.KinesisEvent) (err error) {

        for i, record := range kinesisEvent.Records {

            mySpecType := new(MySpecificType)
            if err = json.Unmarshal(record.Kinesis.Data, mySpecType); err != nil {
                logrus.Errorf("records[%d]: %s", i, err)
                continue
            }

            err = handler(mySpecType, i)
        }
        return
    })
}

, а затем: 

// main.go
func Handler(m *MySpecificType, index int) error {
// ...
}

func main() {
    <wherever_you_put_it_package>.Start(Handler)
    // instead of lambda.Start(Handler)
}
...