From 4bdbbefa346df85b9b0ea2454082b32e490597e7 Mon Sep 17 00:00:00 2001
From: Tim Bart <tim@fewagainstmany.com>
Date: Fri, 27 Apr 2018 22:12:11 -0700
Subject: [PATCH] Use functional options to configure KinesisClient (#46)

---
 client.go | 25 ++++++++++++++++++++++---
 1 file changed, 22 insertions(+), 3 deletions(-)

diff --git a/client.go b/client.go
index 8938175..7d752c9 100644
--- a/client.go
+++ b/client.go
@@ -9,10 +9,29 @@ import (
 	"github.com/aws/aws-sdk-go/service/kinesis"
 )
 
+// ClientOption is used to override defaults when creating a KinesisClient
+type ClientOption func(*KinesisClient)
+
+// WithKinesis overrides the default Kinesis client
+func WithKinesis(svc *kinesis.Kinesis) ClientOption {
+	return func(kc *KinesisClient) {
+		kc.svc = svc
+	}
+}
+
 // NewKinesisClient returns client to interface with Kinesis stream
-func NewKinesisClient() *KinesisClient {
-	svc := kinesis.New(session.New(aws.NewConfig()))
-	return &KinesisClient{svc}
+func NewKinesisClient(opts ...ClientOption) *KinesisClient {
+	kc := &KinesisClient{}
+
+	for _, opt := range opts {
+		opt(kc)
+	}
+
+	if kc.svc == nil {
+		kc.svc = kinesis.New(session.New(aws.NewConfig()))
+	}
+
+	return kc
 }
 
 // KinesisClient acts as wrapper around Kinesis client