Function lambdaWrapWithClient

cfn/wrap.go:27–74

(lambdaFunction CustomResourceFunction, client httpClient)

type CustomResourceFunction func(context.Context, Event) (physicalResourceID string, data map[string]interface{}, err error)

func lambdaWrapWithClient(lambdaFunction CustomResourceFunction, client httpClient) (fn CustomResourceLambdaFunction) {
	fn = func(ctx context.Context, event Event) (reason string, err error) {
		r := NewResponse(&event)

		// A previous physical resource id exists unless this is a create request.
		fallbackPhysicalResourceID := event.PhysicalResourceID
		if event.RequestType == RequestCreate {
			// If this is a create request, the fallback should be the request ID
			fallbackPhysicalResourceID = event.RequestID
		}

		funcDidPanic := true
		defer func() {
			if funcDidPanic {
				r.Status = StatusFailed
				r.Reason = "Function panicked, see log stream for details"
				r.PhysicalResourceID = fallbackPhysicalResourceID
				// FIXME: something should be done if an error is returned here
				_ = r.sendWith(client)
			}
		}()

		r.PhysicalResourceID, r.Data, err = lambdaFunction(ctx, event)
		funcDidPanic = false

		if r.PhysicalResourceID == "" {
			r.PhysicalResourceID = fallbackPhysicalResourceID
			log.Printf("PhysicalResourceID not set. Using fallback PhysicalResourceID: %s\n", r.PhysicalResourceID)
		}

		if err != nil {
			r.Status = StatusFailed
			r.Reason = err.Error()
			log.Printf("sending status failed: %s\n", r.Reason)
		} else {
			r.Status = StatusSuccess
		}

		err = r.sendWith(client)
		if err != nil {
			reason = err.Error()
		}

		return
	}

	return
}

// LambdaWrap returns a CustomResourceLambdaFunction which is something lambda.Start()
// will understand. The purpose of doing this is so that Response Handling boiler