Add Pipeline support

Author: shacharPash

src/NRedisStack/Pipeline.cs

@@ -0,0 +1,50 @@
+using NRedisStack.RedisStackCommands;
+using StackExchange.Redis;
+
+namespace NRedisStack
+{
+    public class Pipeline
+    {
+        private readonly IDatabase _db;
+        private readonly List<SerializedCommand> _commands = new List<SerializedCommand>();
+
+        public Pipeline(IDatabase db)
+        {
+            _db = db;
+        }
+
+        public void AddCommand(SerializedCommand command)
+        {
+            _commands.Add(command);
+        }
+
+        public RedisResult[] Execute()
+        {
+            var transaction = _db.CreateTransaction();
+            var tasks = new List<Task<RedisResult>>();
+            foreach (var command in _commands)
+            {
+                tasks.Add(transaction.ExecuteAsync(command.Command, command.Args));
+            }
+
+            transaction.Execute();
+            Task.WhenAll(tasks).Wait();
+            return tasks.Select(x => x.Result).ToArray();
+        }
+
+        public async Task<RedisResult[]> ExecuteAsync()
+        {
+            var transaction = _db.CreateTransaction();
+            var tasks = new List<Task<RedisResult>>();
+            foreach (var command in _commands)
+            {
+                tasks.Add(transaction.ExecuteAsync(command.Command, command.Args));
+            }
+
+            transaction.Execute();
+            await Task.WhenAll(tasks);
+            return tasks.Select(x => x.Result).ToArray();
+        }
+
+    }
+}

tests/NRedisStack.Tests/Json/JsonTests.cs

@@ -20,6 +20,30 @@ public void Dispose()
         redisFixture.Redis.GetDatabase().KeyDelete(_testName);
     }
 
+    [Fact]
+    public void TestPipeline()
+    {
+        var conn = redisFixture.Redis;
+        var db = conn.GetDatabase();
+        IJsonCommands json = new JsonCommands(db);
+        var keys = CreateKeyNames(1);
+        var pipeline = new Pipeline(db);
+        pipeline.AddCommand(json.Set(keys[0], "$", new Person { Name = "Shachar", Age = 23 }));
+        pipeline.AddCommand(json.Get(keys[0]));
+        pipeline.AddCommand(json.Set(keys[0], "$.Name", "Shachar2"));
+        pipeline.AddCommand(json.Get(keys[0]));
+        pipeline.AddCommand(json.Set(keys[0], "$.Age", 24));
+        pipeline.AddCommand(json.Get(keys[0]));
+        var results = pipeline.Execute();
+        Assert.Equal(6, results.Length);
+        Assert.Equal("OK", results[0].ToString());
+        Assert.Equal("{\"Name\":\"Shachar\",\"Age\":23}", results[1].ToString());
+        Assert.Equal("OK", results[2].ToString());
+        Assert.Equal("{\"Name\":\"Shachar2\",\"Age\":23}", results[3].ToString());
+        Assert.Equal("OK", results[4].ToString());
+        Assert.Equal("{\"Name\":\"Shachar2\",\"Age\":24}", results[5].ToString());
+    }
+
     [Fact]
     public void TestSetFromFile()
     {