From 93ecce36b5ff5b4c0562d131ae760392ddef1a90 Mon Sep 17 00:00:00 2001 From: TsMask <340112800@qq.com> Date: Wed, 4 Dec 2024 15:52:09 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20redis=20GetHashBatch=E6=A3=80=E6=9F=A5?= =?UTF-8?q?=E7=B4=A2=E5=BC=95=E6=98=AF=E5=90=A6=E8=B6=8A=E7=95=8C?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/framework/redis/redis.go | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/src/framework/redis/redis.go b/src/framework/redis/redis.go index 04901497..88e5cdd0 100644 --- a/src/framework/redis/redis.go +++ b/src/framework/redis/redis.go @@ -277,10 +277,6 @@ func GetHashBatch(source string, keys []string) (map[string]map[string]string, e } for i := 0; i < total; i += batchSize { - // 检查索引是否越界 - if i+batchSize > total { - batchSize = total - i - } wg.Add(1) go func(start int) { ctx := context.Background() @@ -292,8 +288,13 @@ func GetHashBatch(source string, keys []string) (map[string]map[string]string, e wg.Done() }() + // 检查索引是否越界 + end := start + batchSize + if end > total { + end = total + } pipe := rdb.Pipeline() - for _, key := range keys[start : start+batchSize] { + for _, key := range keys[start:end] { pipe.HGetAll(ctx, key) }