add hkeys

client.go

@@ -5,6 +5,7 @@ import (
 	"net"
 	"time"
 
+	"github.com/bytedance/sonic"
 	cache "github.com/xgzlucario/GigaCache"
 	"github.com/xgzlucario/rotom/base"
 )
@@ -90,6 +91,17 @@ func (c *Client) HLen(key string) (int, error) {
 	return base.ParseInt[int](args), nil
 }
 
+// HKeys
+func (c *Client) HKeys(key string) ([]string, error) {
+	args, err := c.do(NewCodec(OpHKeys).Str(key))
+	if err != nil {
+		return nil, err
+	}
+	var keys []string
+	err = sonic.Unmarshal(args, &keys)
+	return keys, err
+}
+
 // HRemove
 func (c *Client) HRemove(key, field string) (bool, error) {
 	args, err := c.do(NewCodec(OpHRemove).Str(key).Str(field))

rotom.go

@@ -13,6 +13,7 @@ import (
 	"sync"
 	"time"
 
+	"github.com/bytedance/sonic"
 	"github.com/panjf2000/gnet/v2"
 	cache "github.com/xgzlucario/GigaCache"
 	"github.com/xgzlucario/rotom/base"
@@ -33,6 +34,7 @@ const (
 	OpHSet
 	OpHGet
 	OpHLen
+	OpHKeys
 	OpHRemove
 	// set
 	OpSAdd
@@ -172,6 +174,19 @@ var cmdTable = []Cmd{
 		_, err = w.Write(res)
 		return err
 	}},
+	{OpHKeys, 1, func(e *Engine, args [][]byte, w base.Writer) error {
+		// key
+		m, err := e.fetchMap(*b2s(args[0]))
+		if err != nil {
+			return err
+		}
+		src, err := sonic.Marshal(m.Keys())
+		if err != nil {
+			return err
+		}
+		_, err = w.Write(src)
+		return err
+	}},
 	{OpHRemove, 2, func(e *Engine, args [][]byte, w base.Writer) error {
 		// key, field
 		ok, err := e.HRemove(*b2s(args[0]), *b2s(args[1]))

rotom_test.go

@@ -590,14 +590,14 @@ func TestClient(t *testing.T) {
 		assert.Nil(err)
 		assert.Equal(num, 1)
 
+		// HKeys
+		keys, err := cli.HKeys("exmap")
+		assert.Nil(err)
+		assert.ElementsMatch(keys, []string{key})
+
 		// HRemove
 		ok, err := cli.HRemove("exmap", key)
 		assert.Nil(err)
 		assert.True(ok)
-
-		// HLen
-		num, err = cli.HLen("exmap")
-		assert.Nil(err)
-		assert.Equal(num, 0)
 	}
 }