From dd72134ed3e2b5356885f23185c6e147e4cf7106 Mon Sep 17 00:00:00 2001 From: GoEdgeLab Date: Tue, 30 Aug 2022 18:49:21 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E4=BB=A3=E7=A0=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- internal/iplibrary/manager_ip_list.go | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/internal/iplibrary/manager_ip_list.go b/internal/iplibrary/manager_ip_list.go index 8458d6c..6901d7b 100644 --- a/internal/iplibrary/manager_ip_list.go +++ b/internal/iplibrary/manager_ip_list.go @@ -43,7 +43,7 @@ type IPListManager struct { func NewIPListManager() *IPListManager { return &IPListManager{ - pageSize: 500, + pageSize: 1000, listMap: map[int64]*IPList{}, } } @@ -111,15 +111,19 @@ func (this *IPListManager) init() { var size int64 = 1000 for { items, err := db.ReadItems(offset, size) + var l = len(items) if err != nil { remotelogs.Error("IP_LIST_MANAGER", "read ip list from local database failed: "+err.Error()) } else { - if len(items) == 0 { + if l == 0 { break } this.processItems(items, false) + if int64(l) < size { + break + } } - offset += int64(len(items)) + offset += int64(l) } } } @@ -155,7 +159,7 @@ func (this *IPListManager) fetch() (hasNext bool, err error) { } return false, err } - items := itemsResp.IpItems + var items = itemsResp.IpItems if len(items) == 0 { return false, nil }