add client oper for set

client.go

@@ -111,6 +111,63 @@ func (c *Client) HRemove(key, field string) (bool, error) {
 	return args[0] == _true, nil
 }
 
+// SAdd
+func (c *Client) SAdd(key, item string) error {
+	_, err := c.do(NewCodec(OpSAdd).Str(key).Str(item))
+	return err
+}
+
+// SRemove
+func (c *Client) SRemove(key, item string) error {
+	_, err := c.do(NewCodec(OpSRemove).Str(key).Str(item))
+	return err
+}
+
+// SHas
+func (c *Client) SHas(key, item string) (bool, error) {
+	args, err := c.do(NewCodec(OpSHas).Str(key).Str(item))
+	if err != nil {
+		return false, err
+	}
+	return args[0] == _true, nil
+}
+
+// SCard
+func (c *Client) SCard(key string) (int, error) {
+	args, err := c.do(NewCodec(OpSCard).Str(key))
+	if err != nil {
+		return 0, err
+	}
+	return base.ParseInt[int](args), nil
+}
+
+// SMembers
+func (c *Client) SMembers(key string) ([]string, error) {
+	args, err := c.do(NewCodec(OpSMembers).Str(key))
+	if err != nil {
+		return nil, err
+	}
+	return base.ParseStrSlice(args), nil
+}
+
+// SUnion
+func (c *Client) SUnion(dstKey string, srcKeys ...string) error {
+	_, err := c.do(NewCodec(OpSUnion).Str(dstKey).StrSlice(srcKeys))
+	return err
+}
+
+// SInter
+func (c *Client) SInter(dstKey string, srcKeys ...string) error {
+	_, err := c.do(NewCodec(OpSInter).Str(dstKey).StrSlice(srcKeys))
+	return err
+}
+
+// SDiff
+func (c *Client) SDiff(dstKey string, srcKeys ...string) error {
+	_, err := c.do(NewCodec(OpSDiff).Str(dstKey).StrSlice(srcKeys))
+	return err
+}
+
 // Close
 func (c *Client) Close() error {
 	return c.c.Close()

rotom.go

@@ -39,6 +39,9 @@ const (
 	// set
 	OpSAdd
 	OpSRemove
+	OpSHas
+	OpSCard
+	OpSMembers
 	OpSUnion
 	OpSInter
 	OpSDiff
@@ -202,17 +205,43 @@ var cmdTable = []Cmd{
 		// key, item
 		return e.SRemove(*b2s(args[0]), *b2s(args[1]))
 	}},
-	{OpSUnion, 2, func(e *Engine, args [][]byte, w base.Writer) error {
+	{OpSHas, 2, func(e *Engine, args [][]byte, w base.Writer) error {
+		// key, item
+		ok, err := e.SHas(*b2s(args[0]), *b2s(args[1]))
+		if err != nil {
+			return err
+		}
+		return w.WriteByte(bool2byte(ok))
+	}},
+	{OpSCard, 1, func(e *Engine, args [][]byte, w base.Writer) error {
+		// key
+		n, err := e.SCard(*b2s(args[0]))
+		if err != nil {
+			return err
+		}
+		_, err = w.Write(base.FormatInt(n))
+		return err
+	}},
+	{OpSMembers, 1, func(e *Engine, args [][]byte, w base.Writer) error {
+		// key
+		s, err := e.fetchSet(*b2s(args[0]))
+		if err != nil {
+			return err
+		}
+		_, err = w.Write(base.FormatStrSlice(s.ToSlice()))
+		return err
+	}},
+	{OpSUnion, 2, func(e *Engine, args [][]byte, _ base.Writer) error {
 		// dstKey, srcKeys...
 		srcKeys := base.ParseStrSlice(args[1])
 		return e.SUnion(*b2s(args[0]), srcKeys...)
 	}},
-	{OpSInter, 2, func(e *Engine, args [][]byte, w base.Writer) error {
+	{OpSInter, 2, func(e *Engine, args [][]byte, _ base.Writer) error {
 		// dstKey, srcKeys...
 		srcKeys := base.ParseStrSlice(args[1])
 		return e.SInter(*b2s(args[0]), srcKeys...)
 	}},
-	{OpSDiff, 2, func(e *Engine, args [][]byte, w base.Writer) error {
+	{OpSDiff, 2, func(e *Engine, args [][]byte, _ base.Writer) error {
 		// dstKey, srcKeys...
 		srcKeys := base.ParseStrSlice(args[1])
 		return e.SDiff(*b2s(args[0]), srcKeys...)

rotom_test.go

@@ -104,7 +104,7 @@ func TestDB(t *testing.T) {
 	assert.Nil(db)
 }
 
-func TestSet(t *testing.T) {
+func TestCSet(t *testing.T) {
 	assert := assert.New(t)
 
 	cfg := DefaultConfig
@@ -285,6 +285,36 @@ func TestHMap(t *testing.T) {
 	}
 }
 
+func TestSet(t *testing.T) {
+	assert := assert.New(t)
+
+	cfg := DefaultConfig
+	cfg.Path = gofakeit.UUID() + ".db"
+
+	db, err := Open(cfg)
+	assert.Nil(err)
+
+	port := gofakeit.Number(10000, 20000)
+	addr := "localhost:" + strconv.Itoa(port)
+
+	// listen
+	go db.Listen(addr)
+	time.Sleep(time.Second / 10)
+
+	cli, err := NewClient(addr)
+	assert.Nil(err)
+
+	for i := 0; i < 1000; i++ {
+		assert.Nil(cli.SAdd("set", strconv.Itoa(i)))
+	}
+
+	for i := 200; i < 1200; i++ {
+		ok, err := cli.SHas("set", strconv.Itoa(i))
+		assert.Nil(err)
+		assert.Equal(ok, i < 1000, i)
+	}
+}
+
 func TestZSet(t *testing.T) {
 	assert := assert.New(t)