revert examples

This commit is contained in:
Harlow Ward 2019-06-02 17:26:24 -07:00
parent bd42663013
commit 2ab5ec4031
4 changed files with 12 additions and 12 deletions

View file

@ -19,7 +19,7 @@ import (
"github.com/aws/aws-sdk-go/service/dynamodb" "github.com/aws/aws-sdk-go/service/dynamodb"
"github.com/aws/aws-sdk-go/service/kinesis" "github.com/aws/aws-sdk-go/service/kinesis"
consumer "github.com/harlow/kinesis-consumer" consumer "github.com/harlow/kinesis-consumer"
storage "github.com/harlow/kinesis-consumer/checkpoint/ddb" checkpoint "github.com/harlow/kinesis-consumer/checkpoint/ddb"
) )
// kick off a server for exposing scan metrics // kick off a server for exposing scan metrics
@ -69,8 +69,8 @@ func main() {
myKsis := kinesis.New(sess) myKsis := kinesis.New(sess)
myDdbClient := dynamodb.New(sess) myDdbClient := dynamodb.New(sess)
// ddb persitance // ddb checkpoint
ddb, err := storage.New(*app, *table, storage.WithDynamoClient(myDdbClient), storage.WithRetryer(&MyRetryer{})) ck, err := checkpoint.New(*app, *table, checkpoint.WithDynamoClient(myDdbClient), checkpoint.WithRetryer(&MyRetryer{}))
if err != nil { if err != nil {
log.Log("checkpoint error: %v", err) log.Log("checkpoint error: %v", err)
} }
@ -81,7 +81,7 @@ func main() {
// consumer // consumer
c, err := consumer.New( c, err := consumer.New(
*stream, *stream,
consumer.WithStorage(ddb), consumer.WithCheckpoint(ck),
consumer.WithLogger(log), consumer.WithLogger(log),
consumer.WithCounter(counter), consumer.WithCounter(counter),
consumer.WithClient(myKsis), consumer.WithClient(myKsis),
@ -111,17 +111,17 @@ func main() {
log.Log("scan error: %v", err) log.Log("scan error: %v", err)
} }
if err := ddb.Shutdown(); err != nil { if err := ck.Shutdown(); err != nil {
log.Log("storage shutdown error: %v", err) log.Log("checkpoint shutdown error: %v", err)
} }
} }
// MyRetryer used for storage // MyRetryer used for checkpointing
type MyRetryer struct { type MyRetryer struct {
storage.Retryer checkpoint.Retryer
} }
// ShouldRetry implements custom logic for when errors should retry // ShouldRetry implements custom logic for when a checkpont should retry
func (r *MyRetryer) ShouldRetry(err error) bool { func (r *MyRetryer) ShouldRetry(err error) bool {
if awsErr, ok := err.(awserr.Error); ok { if awsErr, ok := err.(awserr.Error); ok {
switch awsErr.Code() { switch awsErr.Code() {

View file

@ -33,7 +33,7 @@ func main() {
// consumer // consumer
c, err := consumer.New( c, err := consumer.New(
*stream, *stream,
consumer.WithStorage(ck), consumer.WithCheckpoint(ck),
consumer.WithCounter(counter), consumer.WithCounter(counter),
) )
if err != nil { if err != nil {

View file

@ -33,7 +33,7 @@ func main() {
// consumer // consumer
c, err := consumer.New( c, err := consumer.New(
*stream, *stream,
consumer.WithStorage(ck), consumer.WithCheckpoint(ck),
consumer.WithCounter(counter), consumer.WithCounter(counter),
) )
if err != nil { if err != nil {

View file

@ -43,7 +43,7 @@ func main() {
// consumer // consumer
c, err := consumer.New( c, err := consumer.New(
*stream, *stream,
consumer.WithStorage(ck), consumer.WithCheckpoint(ck),
consumer.WithLogger(logger), consumer.WithLogger(logger),
) )
if err != nil { if err != nil {