From f1d2cbb0eeef734b6046737c5e5202acb8892bcc Mon Sep 17 00:00:00 2001 From: HuangJiaLuo <1820799930@qq.com> Date: Sat, 9 Oct 2021 10:55:12 +0800 Subject: [PATCH] feat(event): feat the io of judgement function --- pkg/event/define.go | 2 +- pkg/event/driver.go | 8 ++++++-- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/pkg/event/define.go b/pkg/event/define.go index 8921901..3604c9c 100644 --- a/pkg/event/define.go +++ b/pkg/event/define.go @@ -18,7 +18,7 @@ type EventWorkFunc func() (interface{}, error) type DriverInterface interface { Get() *Event Put(event *Event) - GetLength() int + GetLengthJudge() bool } type ProduceInterface interface { diff --git a/pkg/event/driver.go b/pkg/event/driver.go index 764ff22..78797ea 100644 --- a/pkg/event/driver.go +++ b/pkg/event/driver.go @@ -137,8 +137,12 @@ func (d *Driver) Put(event *Event) { d.queue <- event } -func (d *Driver) GetLength() int { - return len(d.queue) +func (d *Driver) GetLengthJudge() bool { + // 自动对当前的最大io数量进行判断 + if len(d.queue) > d.maxQueueSize * 1/2{ + return true + } + return false } // NewDriver 新建 Driver