luciano/kinesis-consumer
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Shorten the CreateStream function name

Browse source
This commit is contained in:
Harlow Ward 2014-12-20 18:55:13 -08:00
parent 05920cc5d3
commit 3d290c4512
1 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
utils.go
View file

@ -89,9 +89,9 @@ func LoadConfig(config interface{}, filename string) error {
return nil
}
// CreateAndWaitForStreamToBecomeAvailable creates a new Kinesis stream (uses
// existing stream if exists) and waits for it to become available.
func CreateAndWaitForStreamToBecomeAvailable(k *kinesis.Kinesis, streamName string, shardCount int) {
// CreateStream creates a new Kinesis stream (uses existing stream if exists) and
// waits for it to become available.
func CreateStream(k *kinesis.Kinesis, streamName string, shardCount int) {
if !StreamExists(k, streamName) {
err := k.CreateStream(streamName, shardCount)
@ -123,7 +123,11 @@ func CreateAndWaitForStreamToBecomeAvailable(k *kinesis.Kinesis, streamName stri
// StreamExists checks if a Kinesis stream exists.
func StreamExists(k *kinesis.Kinesis, streamName string) bool {
args := kinesis.NewArgs()
resp, _ := k.ListStreams(args)
resp, err := k.ListStreams(args)
if err != nil {
fmt.Printf("ListStream ERROR: %v\n", err)
return false
}
for _, s := range resp.StreamNames {
if s == streamName {
return true