Merge pull request #21 from whiteforestzx/add-set-args-method

Add SetArgs method

pool.go

@@ -290,6 +290,14 @@ func (p *Pool) SetRange(ctx context.Context, key string, offset int64, value str
 	return conn.SetRange(ctx, key, offset, value)
 }
 
+func (p *Pool) SetArgs(ctx context.Context, key string, value interface{}, a redis.SetArgs) *redis.StatusCmd {
+	conn, err := p.connFactory.getMasterConn(key)
+	if err != nil {
+		return newErrorStatusCmd(err)
+	}
+	return conn.SetArgs(ctx, key, value, a)
+}
+
 func (p *Pool) StrLen(ctx context.Context, key string) *redis.IntCmd {
 	conn, err := p.connFactory.getSlaveConn(key)
 	if err != nil {

pool_test.go

@@ -373,6 +373,16 @@ var _ = Describe("Pool", func() {
 			}
 		})
 
+		It("setargs", func() {
+			key := "setargs_key"
+			for _, pool := range pools {
+				Expect(pool.SetArgs(ctx, key, "bar", redis.SetArgs{Mode: "XX"}).Val()).To(Equal(""))
+				Expect(pool.SetArgs(ctx, key, "bar", redis.SetArgs{}).Val()).To(Equal("OK"))
+				Expect(pool.SetArgs(ctx, key, "bar", redis.SetArgs{Mode: "NX"}).Val()).To(Equal(""))
+				_, _ = pool.Del(ctx, key)
+			}
+		})
+
 		It("setrange", func() {
 			key := "setrange_key"
 			for _, pool := range pools {
@@ -1191,7 +1201,6 @@ var _ = Describe("Pool", func() {
 			}
 		})
 
-
 		It("ZAddGT", func() {
 			for _, pool := range pools {
 				_, err := pool.ZAddGT(ctx, "key", redis.Z{