diff options
| -rw-r--r-- | CHANGES | 1 | ||||
| -rw-r--r-- | redis/cluster.py | 11 | ||||
| -rw-r--r-- | tests/test_cluster.py | 19 |
3 files changed, 31 insertions, 0 deletions
@@ -1,3 +1,4 @@ + * Support `.unlink()` in ClusterPipeline * Simplify synchronous SocketBuffer state management * Fix string cleanse in Redis Graph * Make PythonParser resumable in case of error (#2510) diff --git a/redis/cluster.py b/redis/cluster.py index 5f730a8..d6dc02d 100644 --- a/redis/cluster.py +++ b/redis/cluster.py @@ -2136,6 +2136,17 @@ class ClusterPipeline(RedisCluster): return self.execute_command("DEL", names[0]) + def unlink(self, *names): + """ + "Unlink a key specified by ``names``" + """ + if len(names) != 1: + raise RedisClusterException( + "unlinking multiple keys is not implemented in pipeline command" + ) + + return self.execute_command("UNLINK", names[0]) + def block_pipeline_command(name: str) -> Callable[..., Any]: """ diff --git a/tests/test_cluster.py b/tests/test_cluster.py index da6a8e4..1bf57a3 100644 --- a/tests/test_cluster.py +++ b/tests/test_cluster.py @@ -2703,6 +2703,25 @@ class TestClusterPipeline: with pytest.raises(RedisClusterException): pipe.delete("a", "b") + def test_unlink_single(self, r): + """ + Test a single unlink operation + """ + r["a"] = 1 + with r.pipeline(transaction=False) as pipe: + pipe.unlink("a") + assert pipe.execute() == [1] + + def test_multi_unlink_unsupported(self, r): + """ + Test that multi unlink operation is unsupported + """ + with r.pipeline(transaction=False) as pipe: + r["a"] = 1 + r["b"] = 2 + with pytest.raises(RedisClusterException): + pipe.unlink("a", "b") + def test_brpoplpush_disabled(self, r): """ Test that brpoplpush is disabled for ClusterPipeline |