added support for new kinesis event (#24)

* added support for new kinesis event we need spring-media/terraform-aws-lambda#49 merged and released as v4.4.0 before * removed deprecated table name from config map * added stream name to kinesis blueprint
moritzzimmer · Feb 3, 2020 · c757f77 · c757f77
1 parent e9f7d28
commit c757f77
Show file tree

Hide file tree

Showing 4 changed files with 27 additions and 5 deletions.
diff --git a/cmd/new.go b/cmd/new.go
@@ -78,7 +78,7 @@ func init() {
 	rootCmd.AddCommand(newCmd)
 
 	newCmd.Flags().BoolP("dry-run", "d", false, "dry run")
-	newCmd.Flags().StringP("event", "e", "cloudwatch-event", "event type triggering the Lambda function [cloudwatch-event, dynamodb, s3, sns]")
+	newCmd.Flags().StringP("event", "e", "cloudwatch-event", "event type triggering the Lambda function [cloudwatch-event, dynamodb, kinesis, s3, sns]")
 	newCmd.Flags().String("ci", "none", "ci provider config file to generate [none, travis]")
 	viper.BindPFlags(newCmd.Flags())
 }
diff --git a/generate/core/options.go b/generate/core/options.go
@@ -42,10 +42,10 @@ func DefaultOpts() *Options {
 		},
 		Aws: &Aws{Region: "eu-west-1"},
 		Terraform: &Terraform{
-			Version: "0.12.12",
+			Version: "0.12.20",
 			Module: &Module{
 				Source:  "spring-media/lambda/aws",
-				Version: "4.3.0",
+				Version: "4.4.0",
 			},
 		},
 	}

diff --git a/generate/core/templates/main.go.tmpl b/generate/core/templates/main.go.tmpl
@@ -1,12 +1,12 @@
 package main
 
 import (
-	{{if or (eq .opts.App.Event "dynamodb") (eq .opts.App.Event "sns") (eq .opts.App.Event "s3") -}}
+	{{if or (eq .opts.App.Event "dynamodb") (eq .opts.App.Event "kinesis") (eq .opts.App.Event "sns") (eq .opts.App.Event "s3") -}}
 	"context"
 	{{end -}}
 	"fmt"
 
-	{{if or (eq .opts.App.Event "dynamodb") (eq .opts.App.Event "sns") (eq .opts.App.Event "s3") -}}
+	{{if or (eq .opts.App.Event "dynamodb") (eq .opts.App.Event "kinesis") (eq .opts.App.Event "sns") (eq .opts.App.Event "s3") -}}
 	"github.com/aws/aws-lambda-go/events"
 	{{end -}}
 	"github.com/aws/aws-lambda-go/lambda"
@@ -37,6 +37,19 @@ func handler(ctx context.Context, e events.DynamoDBEvent) {
 }
 {{end -}}
 
+{{if eq .opts.App.Event "kinesis" -}}
+func handler(ctx context.Context, kinesisEvent events.KinesisEvent) error {
+	for _, record := range kinesisEvent.Records {
+		kinesisRecord := record.Kinesis
+		dataBytes := kinesisRecord.Data
+		dataText := string(dataBytes)
+
+		fmt.Printf("%s Data = %s \n", record.EventName, dataText)
+	}
+	return nil
+}
+{{end -}}
+
 {{if eq .opts.App.Event "sns" -}}
 func handler(ctx context.Context, snsEvent events.SNSEvent) {
 	for _, record := range snsEvent.Records {

diff --git a/generate/core/templates/terraform/main.tf.tmpl b/generate/core/templates/terraform/main.tf.tmpl
@@ -19,13 +19,22 @@ module "{{.opts.App.Name}}" {
   }
   {{end -}}
 
+
   {{if eq .opts.App.Event "dynamodb" -}}
   event = {
     type                    = "dynamodb"
     stream_event_source_arn = "arn:aws:dynamodb:eu-west-1:647379381847:table/some-table/stream/some-identifier"
     table_name              = "some-table"
   }
   {{end -}}
+
+  {{if eq .opts.App.Event "kinesis" -}}
+  event = {
+    type             = "kinesis"
+    event_source_arn = "arn:aws:kinesis:eu-central-1:647379381847:stream/some-stream"
+    stream_name      = "some-stream"
+  }
+  {{end -}}
 
   {{if eq .opts.App.Event "sns" -}}
   event = {