feat: 增加RawCmdable接口

sandwich-go · Nov 9, 2022 · c9b16e0 · c9b16e0
1 parent 7964e3d
commit c9b16e0
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 1 deletion.
diff --git a/cmd_pipeline.go b/cmd_pipeline.go
@@ -13,4 +13,5 @@ type Pipeliner interface {
 	Exec(ctx context.Context) ([]interface{}, error)
 }
 
-func (c *client) Pipeline() Pipeliner { return c.cmdable.Pipeline() }
+func (c *client) Pipeline() Pipeliner     { return c.cmdable.Pipeline() }
+func (c *client) RawCmdable() interface{} { return c.cmdable.RawCmdable() }
diff --git a/redis.go b/redis.go
@@ -9,6 +9,7 @@ type Cmdable interface {
 	PoolStats() PoolStats
 	RegisterCollector(RegisterCollectorFunc)
 	Close() error
+	RawCmdable() interface{}
 
 	CacheCmdable
 	BitmapCmdable

diff --git a/redis_resp2.go b/redis_resp2.go
@@ -618,6 +618,8 @@ func (p *pipelineResp2) Exec(ctx context.Context) ([]interface{}, error) {
 	return result, err
 }
 
+func (r *resp2) RawCmdable() interface{} { return r.cmd }
+
 func (r *resp2) Publish(ctx context.Context, channel string, message interface{}) IntCmd {
 	return r.cmd.Publish(ctx, channel, message)
 }

diff --git a/redis_resp3.go b/redis_resp3.go
@@ -964,6 +964,8 @@ func (p *pipelineResp3) Exec(_ context.Context) ([]interface{}, error) {
 	return p.res, p.firstError
 }
 
+func (r *resp3) RawCmdable() interface{} { return r.cmd }
+
 func (r *resp3) Publish(ctx context.Context, channel string, message interface{}) IntCmd {
 	return newIntCmd(r.cmd.Do(ctx, r.cmd.B().Publish().Channel(channel).Message(str(message)).Build()))
 }