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

Log file closed when kclProcessor is done running

Browse source
This commit is contained in:
Xavi Ramirez 2017-07-18 19:12:46 +00:00
parent 4c67f39c99
commit 8bcef6cd91
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
batchconsumer/consumer.go
View file

@ -35,12 +35,11 @@ type Config struct {
CheckpointRetries int CheckpointRetries int
// CheckpointRetrySleep the amount of time between checkpoint save attempts // CheckpointRetrySleep the amount of time between checkpoint save attempts
CheckpointRetrySleep time.Duration CheckpointRetrySleep time.Duration
logOutput io.Writer
} }
type BatchConsumer struct { type BatchConsumer struct {
kclProcess *kcl.KCLProcess kclProcess *kcl.KCLProcess
logfile *os.File
} }
func withDefaults(config Config) Config { func withDefaults(config Config) Config {
@ -91,7 +90,6 @@ func NewBatchConsumerFromFiles(
if err != nil { if err != nil {
log.Fatalf("Unable to create log file: %s", err.Error()) log.Fatalf("Unable to create log file: %s", err.Error())
} }
defer file.Close()
kvlog := logger.New("amazon-kinesis-client-go") kvlog := logger.New("amazon-kinesis-client-go")
kvlog.SetOutput(file) kvlog.SetOutput(file)
@ -105,6 +103,7 @@ func NewBatchConsumerFromFiles(
return &BatchConsumer{ return &BatchConsumer{
kclProcess: kclProcess, kclProcess: kclProcess,
logfile: file,
} }
} }
@ -114,4 +113,5 @@ func NewBatchConsumer(config Config, sender Sender) *BatchConsumer {
func (b *BatchConsumer) Start() { func (b *BatchConsumer) Start() {
b.kclProcess.Run() b.kclProcess.Run()
b.logfile.Close()
} }