go-pkg/asynq
2
0
Fork 0
mirror of https://github.com/hibiken/asynq.git synced 2024-09-20 11:05:58 +08:00
Code Issues Projects Releases Wiki Activity

Replaced blocking KEYS XXX:* command to non-blocking SCAN XXX:*

Browse Source 
More details: https://redis.io/commands/KEYS
This commit is contained in:
disc 2021-03-20 11:56:17 +02:00 committed by Ken Hibino
parent 6a9d9fd717
commit 4722ca2d3d
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
internal/rdb/inspect.go
View File

@ -172,10 +172,18 @@ func (r *RDB) CurrentStats(qname string) (*Stats, error) {
} }
func (r *RDB) memoryUsage(qname string) (int64, error) { func (r *RDB) memoryUsage(qname string) (int64, error) {
keys, err := r.client.Keys(fmt.Sprintf("asynq:{%s}*", qname)).Result() var cursor uint64
var keys []string
for {
data, cursor, err := r.client.Scan(cursor, fmt.Sprintf("asynq:{%s}*", qname), 100).Result()
if err != nil { if err != nil {
return 0, err return 0, err
} }
keys = append(keys, data...)
if cursor == 0 {
break
}
}
var usg int64 var usg int64
for _, k := range keys { for _, k := range keys {
n, err := r.client.MemoryUsage(k).Result() n, err := r.client.MemoryUsage(k).Result()