OMC/be.ems
1
0
Fork 0
Code Issues Releases 3 Activity

fix: redis批量获得缓存数据独立上下文

Browse Source
This commit is contained in:
TsMask
2024-11-08 16:54:27 +08:00
parent 26b950c919
commit 103af4a999
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/framework/redis/redis.go
View File

@@ -265,7 +265,6 @@ func GetHashBatch(source string, keys []string) (map[string]map[string]string, e
if source != "" { if source != "" {
rdb = RDB(source) rdb = RDB(source)
} }
ctx := context.Background()
// 创建一个有限的并发控制信号通道 // 创建一个有限的并发控制信号通道
sem := make(chan struct{}, 10) sem := make(chan struct{}, 10)
@@ -280,11 +279,13 @@ func GetHashBatch(source string, keys []string) (map[string]map[string]string, e
for i := 0; i < total; i += batchSize { for i := 0; i < total; i += batchSize {
wg.Add(1) wg.Add(1)
go func(start int) { go func(start int) {
ctx := context.Background()
// 并发控制，限制同时执行的 Goroutine 数量 // 并发控制，限制同时执行的 Goroutine 数量
sem <- struct{}{} sem <- struct{}{}
defer func() { defer func() {
wg.Done()
<-sem <-sem
ctx.Done()
wg.Done()
}() }()
pipe := rdb.Pipeline() pipe := rdb.Pipeline()