mirror of
https://github.com/openimsdk/open-im-server.git
synced 2026-05-19 08:19:01 +08:00
perf: redis block with keys command (#1423)
Signed-off-by: rfyiamcool <rfyiamcool@163.com>
This commit is contained in:
Vendored
+47
@@ -385,3 +385,50 @@ func testParallelDeleteMessagesMix(t *testing.T, cid string, seqs []int64, input
|
||||
assert.EqualValues(t, 1, val) // exists
|
||||
}
|
||||
}
|
||||
|
||||
func TestCleanUpOneConversationAllMsg(t *testing.T) {
|
||||
rdb := redis.NewClient(&redis.Options{})
|
||||
defer rdb.Close()
|
||||
|
||||
cacher := msgCache{rdb: rdb}
|
||||
count := 1000
|
||||
prefix := fmt.Sprintf("%v", rand.Int63())
|
||||
|
||||
ids := []string{}
|
||||
for i := 0; i < count; i++ {
|
||||
id := fmt.Sprintf("%v-cid-%v", prefix, rand.Int63())
|
||||
ids = append(ids, id)
|
||||
|
||||
key := cacher.allMessageCacheKey(id)
|
||||
rdb.Set(context.Background(), key, "openim", 0)
|
||||
}
|
||||
|
||||
// delete 100 keys with scan.
|
||||
for i := 0; i < 100; i++ {
|
||||
pickedKey := ids[i]
|
||||
err := cacher.CleanUpOneConversationAllMsg(context.Background(), pickedKey)
|
||||
assert.Nil(t, err)
|
||||
|
||||
ls, err := rdb.Keys(context.Background(), pickedKey).Result()
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, 0, len(ls))
|
||||
|
||||
rcode, err := rdb.Exists(context.Background(), pickedKey).Result()
|
||||
assert.Nil(t, err)
|
||||
assert.EqualValues(t, 0, rcode) // non-exists
|
||||
}
|
||||
|
||||
sid := fmt.Sprintf("%v-cid-*", prefix)
|
||||
ls, err := rdb.Keys(context.Background(), cacher.allMessageCacheKey(sid)).Result()
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, count-100, len(ls))
|
||||
|
||||
// delete fuzzy matching keys.
|
||||
err = cacher.CleanUpOneConversationAllMsg(context.Background(), sid)
|
||||
assert.Nil(t, err)
|
||||
|
||||
// don't contains keys matched `{prefix}-cid-{random}` on redis
|
||||
ls, err = rdb.Keys(context.Background(), cacher.allMessageCacheKey(sid)).Result()
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, 0, len(ls))
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user