ff1cff0293
Requiring the Logger be passed in to each of the structs was causing Panics if the Logger was omitted. * Add function for overriding the default logger * Remove panics caused by initialization
35 lines
959 B
Go
35 lines
959 B
Go
package connector
|
|
|
|
import (
|
|
"github.com/sendgridlabs/go-kinesis"
|
|
)
|
|
|
|
// An implementation of Emitter that puts event data on S3 file, and then puts the
|
|
// S3 file path onto the output stream for processing by manifest application.
|
|
type S3ManifestEmitter struct {
|
|
OutputStream string
|
|
S3Bucket string
|
|
Ksis *kinesis.Kinesis
|
|
}
|
|
|
|
func (e S3ManifestEmitter) Emit(b Buffer, t Transformer) {
|
|
|
|
// Emit buffer contents to S3 Bucket
|
|
s3Emitter := S3Emitter{S3Bucket: e.S3Bucket}
|
|
s3Emitter.Emit(b, t)
|
|
s3File := s3Emitter.S3FileName(b.FirstSequenceNumber(), b.LastSequenceNumber())
|
|
|
|
// Emit the file path to Kinesis Output stream
|
|
args := kinesis.NewArgs()
|
|
args.Add("StreamName", e.OutputStream)
|
|
args.Add("PartitionKey", s3File)
|
|
args.AddData([]byte(s3File))
|
|
|
|
_, err := e.Ksis.PutRecord(args)
|
|
|
|
if err != nil {
|
|
logger.Printf("PutRecord ERROR: %v", err)
|
|
} else {
|
|
logger.Printf("[%s] emitted to [%s]", b.FirstSequenceNumber(), e.OutputStream)
|
|
}
|
|
}
|