From 0dc5478bc6b3acdd3e2790d619eb2c5f32d218c3 Mon Sep 17 00:00:00 2001 From: GoEdgeLab Date: Thu, 20 Oct 2022 16:12:35 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E8=AE=BF=E9=97=AE=E6=97=A5?= =?UTF-8?q?=E5=BF=97=E9=87=87=E6=A0=B7=E7=8E=87=E5=8F=AF=E8=83=BD=E4=BC=9A?= =?UTF-8?q?=E8=A2=AB=E6=94=BE=E5=A4=A7=E7=9A=84Bug?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- internal/db/models/http_access_log_dao.go | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/internal/db/models/http_access_log_dao.go b/internal/db/models/http_access_log_dao.go index 520a6416..c87f764a 100644 --- a/internal/db/models/http_access_log_dao.go +++ b/internal/db/models/http_access_log_dao.go @@ -51,6 +51,10 @@ var ( accessLogRowsPerTable int64 = 500_000 // 自动分表的单表最大值 ) +func AccessLogQueuePercent() int { + return accessLogQueuePercent +} + type accessLogTableQuery struct { daoWrapper *HTTPAccessLogDAOWrapper name string @@ -131,13 +135,13 @@ func (this *HTTPAccessLogDAO) CreateHTTPAccessLogs(tx *dbs.Tx, accessLogs []*pb. // 写入队列 var queue = accessLogQueue // 这样写非常重要，防止在写入过程中队列有切换 for _, accessLog := range accessLogs { - if accessLog.FirewallPolicyId == 0 { // 如果是WAF记录，则采取采样率 + if accessLog.FirewallPolicyId == 0 { // 如果是非WAF记录，则采取采样率 // 采样率 if accessLogQueuePercent <= 0 { return nil } if accessLogQueuePercent < 100 && rands.Int(1, 100) > accessLogQueuePercent { - return nil + continue } }