From 5588c6c1c488cd2a7009028cc32e2773e4417151 Mon Sep 17 00:00:00 2001 From: "aiden.ma" Date: Mon, 23 Sep 2024 10:06:29 +0800 Subject: [PATCH] feat: add redis bigkey hook --- core/stores/redis/bigkeyhook_test.go | 34 +--------------------------- 1 file changed, 1 insertion(+), 33 deletions(-) diff --git a/core/stores/redis/bigkeyhook_test.go b/core/stores/redis/bigkeyhook_test.go index a6505674f970..5593f4042bb6 100644 --- a/core/stores/redis/bigkeyhook_test.go +++ b/core/stores/redis/bigkeyhook_test.go @@ -4,10 +4,10 @@ import ( "bytes" "context" "testing" - "time" "github.com/alicebob/miniredis/v2" "github.com/stretchr/testify/assert" + "github.com/zeromicro/go-zero/core/logx" ) @@ -320,35 +320,3 @@ func TestBigKeyHook_AfterProcess_Zrange(t *testing.T) { _, _ = r.ZrangebyscoreWithScoresAndLimit("foo", 0, 100, 0, 10) assert.Contains(t, buf.String(), "BigKey limit") } - -func TestBigKeyHook_stat(t *testing.T) { - var buf bytes.Buffer - logx.Reset() - logx.SetLevel(logx.InfoLevel) - logx.SetWriter(logx.NewWriter(&buf)) - defer logx.Reset() - - r := MustNewRedis(RedisConf{ - Host: miniredis.RunT(t).Addr(), - Type: "node", - VerifyBigKey: BigKeyHookConfig{ - Enable: true, - LimitSize: 5, - LimitCount: 1, - BufferLen: 100, - StatInterval: time.Millisecond * 100, - }, - }) - - err := r.Set("foo", "123456") - assert.NoError(t, err) - - for i := 0; i < 99; i++ { - _, _ = r.Get("foo") - } - - time.Sleep(time.Second) - - assert.Contains(t, buf.String(), "[REDIS] BigKey limit, key: foo, size: 6, count: 100") - -}