feat: add Pipeline to rueidiscompat (#599)

* feat: add Pipeline to rueidiscompat Signed-off-by: Rueian <[email protected]> * test: add Pipeline to rueidiscompat Signed-off-by: Rueian <[email protected]> --------- Signed-off-by: Rueian <[email protected]>
valkey-io · Aug 11, 2024 · 44c7724 · 44c7724
1 parent 97bb0e5
commit 44c7724
Show file tree

Hide file tree

Showing 4 changed files with 3,505 additions and 267 deletions.
diff --git a/valkeycompat/adapter.go b/valkeycompat/adapter.go
@@ -473,6 +473,9 @@ type ProbabilisticCmdable interface {
 	Subscribe(ctx context.Context, channels ...string) PubSub
 	PSubscribe(ctx context.Context, patterns ...string) PubSub
 	SSubscribe(ctx context.Context, channels ...string) PubSub
+
+	Pipeline() Pipeliner
+	Pipelined(ctx context.Context, fn func(Pipeliner) error) ([]Cmder, error)
 }
 
 // Align with go-redis
@@ -4617,6 +4620,14 @@ func (c *Compat) PSubscribe(ctx context.Context, patterns ...string) PubSub {
 	return p
 }
 
+func (c *Compat) Pipelined(ctx context.Context, fn func(Pipeliner) error) ([]Cmder, error) {
+	return newPipeline(c.client).Pipelined(ctx, fn)
+}
+
+func (c *Compat) Pipeline() Pipeliner {
+	return newPipeline(c.client)
+}
+
 func (c CacheCompat) BitCount(ctx context.Context, key string, bitCount *BitCount) *IntCmd {
 	var resp valkey.ValkeyResult
 	if bitCount == nil {