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

Send single value instead of full config object to CheckpointManager

Browse source
This commit is contained in:
Xavi Ramirez 2017-08-10 20:01:07 +00:00
parent 6102f9c752
commit 3a04dad4bf
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
batchconsumer/checkpointmanager.go
View file

@ -20,12 +20,12 @@ type checkpointManager struct {
} }
func newCheckpointManager( func newCheckpointManager(
checkpointer kcl.Checkpointer, config Config, log kv.KayveeLogger, checkpointer kcl.Checkpointer, checkpointFreq time.Duration, log kv.KayveeLogger,
) *checkpointManager { ) *checkpointManager {
cm := &checkpointManager{ cm := &checkpointManager{
log: log, log: log,
checkpointFreq: config.CheckpointFreq, checkpointFreq: checkpointFreq,
checkpoint: make(chan kcl.SequencePair), checkpoint: make(chan kcl.SequencePair),
shutdown: make(chan chan<- struct{}), shutdown: make(chan chan<- struct{}),

2
batchconsumer/writer.go
View file

@ -43,7 +43,7 @@ func NewBatchedWriter(config Config, sender Sender, log kv.KayveeLogger) *batche
func (b *batchedWriter) Initialize(shardID string, checkpointer kcl.Checkpointer) error { func (b *batchedWriter) Initialize(shardID string, checkpointer kcl.Checkpointer) error {
b.shardID = shardID b.shardID = shardID
b.chkpntManager = newCheckpointManager(checkpointer, b.config, b.log) b.chkpntManager = newCheckpointManager(checkpointer, b.config.CheckpointFreq, b.log)
b.batcherManager = newBatcherManager(b.sender, b.chkpntManager, b.config, b.log) b.batcherManager = newBatcherManager(b.sender, b.chkpntManager, b.config, b.log)
return nil return nil