glassechidna / awshoney Goto Github PK

package honeyflush

import (
	"context"
	"github.com/aws/aws-lambda-go/lambda"
	"github.com/honeycombio/beeline-go"
)

type wrapper struct {
	inner lambda.Handler
}

func (w *wrapper) Invoke(ctx context.Context, payload []byte) ([]byte, error) {
	ctx, _ = beeline.StartSpan(ctx, "lambda")
	// we could also include the aws lambda "request id" for correlation with cloudwatch logs
	defer beeline.Flush(ctx)
	return w.inner.Invoke(ctx, payload)
}

func Wrap(inner lambda.Handler) lambda.Handler {
	return &wrapper{inner: inner}
}

package main

import (
	"context"
	"github.com/aws/aws-lambda-go/events"
	"github.com/aws/aws-lambda-go/lambda"
	"github.com/glassechidna/flow/pkg/honeyflush"
)

func mainx() {
	handler := lambda.NewHandler(handlerFn)
	wrapped := honeyflush.Wrap(handler)
	lambda.StartHandler(wrapped)
}

func handlerFn(ctx context.Context, event events.SQSEvent) error {
	return nil
}