in main.go [40:88]
func mainWithError() error {
// Global context
ctx, cancel := signal.NotifyContext(context.Background(), syscall.SIGTERM, syscall.SIGINT)
defer cancel()
cfg, err := config.LoadDefaultConfig(ctx)
if err != nil {
return fmt.Errorf("failed to load AWS default config: %v", err)
}
appConfigs := []app.ConfigOption{
app.WithExtensionName(filepath.Base(os.Args[0])),
app.WithLambdaRuntimeAPI(os.Getenv("AWS_LAMBDA_RUNTIME_API")),
app.WithLogLevel(os.Getenv("ELASTIC_APM_LOG_LEVEL")),
app.WithAWSConfig(cfg),
}
rawDisableLogsAPI := os.Getenv("ELASTIC_APM_LAMBDA_DISABLE_LOGS_API")
if disableLogsAPI, _ := strconv.ParseBool(rawDisableLogsAPI); disableLogsAPI {
appConfigs = append(appConfigs, app.WithoutLogsAPI())
}
// ELASTIC_APM_LAMBDA_CAPTURE_LOGS indicate if the lambda extension
// should capture logs, the value defaults to true i.e. the extension
// will capture function logs by default
rawLambdaCaptureLogs := os.Getenv("ELASTIC_APM_LAMBDA_CAPTURE_LOGS")
captureLogs, err := strconv.ParseBool(rawLambdaCaptureLogs)
if err != nil {
if rawLambdaCaptureLogs != "" {
log.Printf("failed to parse env var ELASTIC_APM_LAMBDA_CAPTURE_LOGS, defaulting to true")
}
captureLogs = true
}
if captureLogs {
appConfigs = append(appConfigs, app.WithFunctionLogSubscription())
}
application, err := app.New(ctx, appConfigs...)
if err != nil {
return fmt.Errorf("failed to create the app: %v", err)
}
if err := application.Run(ctx); err != nil {
return fmt.Errorf("error while running: %v", err)
}
return nil
}