From 0deb7719a1c7acadf4485f71ce36e5a4217a7e83 Mon Sep 17 00:00:00 2001 From: Ehsan Noureddin Moosa Date: Thu, 16 Nov 2023 09:04:39 +0300 Subject: [PATCH] [kit] expose more batcher methods --- kit/utils/batch/batcher.go | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/kit/utils/batch/batcher.go b/kit/utils/batch/batcher.go index b1371020..2d54c27a 100644 --- a/kit/utils/batch/batcher.go +++ b/kit/utils/batch/batcher.go @@ -57,11 +57,11 @@ func (fp *MultiBatcher[IN, OUT]) getBatcher(tagID string) *Batcher[IN, OUT] { } func (fp *MultiBatcher[IN, OUT]) Enter(targetID string, entry Entry[IN, OUT]) { - fp.getBatcher(targetID).enter(entry) + fp.getBatcher(targetID).Enter(entry) } func (fp *MultiBatcher[IN, OUT]) EnterAndWait(targetID string, entry Entry[IN, OUT]) { - fp.getBatcher(targetID).enterAndWait(entry) + fp.getBatcher(targetID).EnterAndWait(entry) } type Batcher[IN, OUT any] struct { @@ -109,16 +109,17 @@ func (f *Batcher[IN, OUT]) startWorker() { f: f, bs: int(f.batchSize), } + go w.run() } -func (f *Batcher[IN, OUT]) enter(entry Entry[IN, OUT]) { +func (f *Batcher[IN, OUT]) Enter(entry Entry[IN, OUT]) { f.entryChan <- entry f.startWorker() } -func (f *Batcher[IN, OUT]) enterAndWait(entry Entry[IN, OUT]) { - f.enter(entry) +func (f *Batcher[IN, OUT]) EnterAndWait(entry Entry[IN, OUT]) { + f.Enter(entry) entry.wait() }