From 8bcef6cd91835aa627598d1c85b85a4f0b5f3e83 Mon Sep 17 00:00:00 2001 From: Xavi Ramirez Date: Tue, 18 Jul 2017 19:12:46 +0000 Subject: [PATCH] Log file closed when kclProcessor is done running --- batchconsumer/consumer.go | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/batchconsumer/consumer.go b/batchconsumer/consumer.go index 5b47a2a..1218ab4 100644 --- a/batchconsumer/consumer.go +++ b/batchconsumer/consumer.go @@ -35,12 +35,11 @@ type Config struct { CheckpointRetries int // CheckpointRetrySleep the amount of time between checkpoint save attempts CheckpointRetrySleep time.Duration - - logOutput io.Writer } type BatchConsumer struct { kclProcess *kcl.KCLProcess + logfile *os.File } func withDefaults(config Config) Config { @@ -91,7 +90,6 @@ func NewBatchConsumerFromFiles( if err != nil { log.Fatalf("Unable to create log file: %s", err.Error()) } - defer file.Close() kvlog := logger.New("amazon-kinesis-client-go") kvlog.SetOutput(file) @@ -105,6 +103,7 @@ func NewBatchConsumerFromFiles( return &BatchConsumer{ kclProcess: kclProcess, + logfile: file, } } @@ -114,4 +113,5 @@ func NewBatchConsumer(config Config, sender Sender) *BatchConsumer { func (b *BatchConsumer) Start() { b.kclProcess.Run() + b.logfile.Close() }