34 lines
906 B
Go
34 lines
906 B
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/harlow/go-etl"
|
|
"github.com/joho/godotenv"
|
|
"github.com/sendgridlabs/go-kinesis"
|
|
)
|
|
|
|
func putSampleDataOnStream(ksis *kinesis.Kinesis, streamName string, numRecords int) {
|
|
for i := 0; i < numRecords; i++ {
|
|
args := kinesis.NewArgs()
|
|
args.Add("StreamName", streamName)
|
|
args.AddData([]byte(fmt.Sprintf("Hello AWS Kinesis %d", i)))
|
|
args.Add("PartitionKey", fmt.Sprintf("partitionKey-%d", i))
|
|
resp, err := ksis.PutRecord(args)
|
|
|
|
if err != nil {
|
|
fmt.Printf("PutRecord err: %v\n", err)
|
|
} else {
|
|
fmt.Printf("SequenceNumber: %v\n", resp.SequenceNumber)
|
|
}
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
godotenv.Load()
|
|
streamName := "inputStream"
|
|
ksis := kinesis.New("", "", kinesis.Region{})
|
|
|
|
etl.CreateAndWaitForStreamToBecomeAvailable(ksis, streamName, 2)
|
|
putSampleDataOnStream(ksis, streamName, 50)
|
|
// deleteStream(ksis, streamName)
|
|
}
|