hyperloglog

Author: mgravell

src/RESPite.StackExchange.Redis/RedisCommands.HyperLogLogCommands.cs

@@ -0,0 +1,39 @@
+using System.Runtime.CompilerServices;
+using StackExchange.Redis;
+
+// ReSharper disable MemberCanBePrivate.Global
+// ReSharper disable InconsistentNaming
+namespace RESPite.StackExchange.Redis;
+
+internal static partial class RedisCommands
+{
+    // this is just a "type pun" - it should be an invisible/magic pointer cast to the JIT
+    public static ref readonly HyperLogLogCommands HyperLogLogs(this in RespContext context)
+        => ref Unsafe.As<RespContext, HyperLogLogCommands>(ref Unsafe.AsRef(in context));
+}
+
+public readonly struct HyperLogLogCommands(in RespContext context)
+{
+    public readonly RespContext Context = context; // important: this is the only field
+}
+
+internal static partial class HyperLogLogCommandsExtensions
+{
+    [RespCommand]
+    public static partial RespOperation<bool> PfAdd(this in HyperLogLogCommands context, RedisKey key, RedisValue value);
+
+    [RespCommand]
+    public static partial RespOperation<bool> PfAdd(this in HyperLogLogCommands context, RedisKey key, RedisValue[] values);
+
+    [RespCommand]
+    public static partial RespOperation<long> PfCount(this in HyperLogLogCommands context, RedisKey key);
+
+    [RespCommand]
+    public static partial RespOperation<long> PfCount(this in HyperLogLogCommands context, RedisKey[] keys);
+
+    [RespCommand]
+    public static partial RespOperation PfMerge(this in HyperLogLogCommands context, RedisKey destination, RedisKey first, RedisKey second);
+
+    [RespCommand]
+    public static partial RespOperation PfMerge(this in HyperLogLogCommands context, RedisKey destination, RedisKey[] sourceKeys);
+}

src/RESPite.StackExchange.Redis/RespContextDatabase.HyperLogLog.cs

@@ -1,58 +1,79 @@
-using StackExchange.Redis;
+using RESPite.Messages;
+using StackExchange.Redis;
 
 namespace RESPite.StackExchange.Redis;
 
 internal partial class RespContextDatabase
 {
     // Async HyperLogLog methods
-    public Task<bool> HyperLogLogAddAsync(RedisKey key, RedisValue value, CommandFlags flags = CommandFlags.None) =>
-        throw new NotImplementedException();
+    public Task<bool> HyperLogLogAddAsync(
+        RedisKey key,
+        RedisValue value,
+        CommandFlags flags = CommandFlags.None) =>
+        Context(flags).HyperLogLogs().PfAdd(key, value).AsTask();
 
-    public Task<bool> HyperLogLogAddAsync(RedisKey key, RedisValue[] values, CommandFlags flags = CommandFlags.None) =>
-        throw new NotImplementedException();
+    public Task<bool> HyperLogLogAddAsync(
+        RedisKey key,
+        RedisValue[] values,
+        CommandFlags flags = CommandFlags.None) =>
+        Context(flags).HyperLogLogs().PfAdd(key, values).AsTask();
 
-    public Task<long> HyperLogLogLengthAsync(RedisKey key, CommandFlags flags = CommandFlags.None) =>
-        throw new NotImplementedException();
+    public Task<long> HyperLogLogLengthAsync(
+        RedisKey key,
+        CommandFlags flags = CommandFlags.None) =>
+        Context(flags).HyperLogLogs().PfCount(key).AsTask();
 
-    public Task<long> HyperLogLogLengthAsync(RedisKey[] keys, CommandFlags flags = CommandFlags.None) =>
-        throw new NotImplementedException();
+    public Task<long> HyperLogLogLengthAsync(
+        RedisKey[] keys,
+        CommandFlags flags = CommandFlags.None) =>
+        Context(flags).HyperLogLogs().PfCount(keys).AsTask();
 
     public Task HyperLogLogMergeAsync(
         RedisKey destination,
         RedisKey first,
         RedisKey second,
         CommandFlags flags = CommandFlags.None) =>
-        throw new NotImplementedException();
+        Context(flags).HyperLogLogs().PfMerge(destination, first, second).AsTask();
 
     public Task HyperLogLogMergeAsync(
         RedisKey destination,
         RedisKey[] sourceKeys,
         CommandFlags flags = CommandFlags.None) =>
-        throw new NotImplementedException();
+        Context(flags).HyperLogLogs().PfMerge(destination, sourceKeys).AsTask();
 
     // Synchronous HyperLogLog methods
-    public bool HyperLogLogAdd(RedisKey key, RedisValue value, CommandFlags flags = CommandFlags.None) =>
-        throw new NotImplementedException();
+    public bool HyperLogLogAdd(
+        RedisKey key,
+        RedisValue value,
+        CommandFlags flags = CommandFlags.None) =>
+        Context(flags).HyperLogLogs().PfAdd(key, value).Wait(SyncTimeout);
 
-    public bool HyperLogLogAdd(RedisKey key, RedisValue[] values, CommandFlags flags = CommandFlags.None) =>
-        throw new NotImplementedException();
+    public bool HyperLogLogAdd(
+        RedisKey key,
+        RedisValue[] values,
+        CommandFlags flags = CommandFlags.None) =>
+        Context(flags).HyperLogLogs().PfAdd(key, values).Wait(SyncTimeout);
 
-    public long HyperLogLogLength(RedisKey key, CommandFlags flags = CommandFlags.None) =>
-        throw new NotImplementedException();
+    public long HyperLogLogLength(
+        RedisKey key,
+        CommandFlags flags = CommandFlags.None) =>
+        Context(flags).HyperLogLogs().PfCount(key).Wait(SyncTimeout);
 
-    public long HyperLogLogLength(RedisKey[] keys, CommandFlags flags = CommandFlags.None) =>
-        throw new NotImplementedException();
+    public long HyperLogLogLength(
+        RedisKey[] keys,
+        CommandFlags flags = CommandFlags.None) =>
+        Context(flags).HyperLogLogs().PfCount(keys).Wait(SyncTimeout);
 
     public void HyperLogLogMerge(
         RedisKey destination,
         RedisKey first,
         RedisKey second,
         CommandFlags flags = CommandFlags.None) =>
-        throw new NotImplementedException();
+        Context(flags).HyperLogLogs().PfMerge(destination, first, second).Wait(SyncTimeout);
 
     public void HyperLogLogMerge(
         RedisKey destination,
         RedisKey[] sourceKeys,
         CommandFlags flags = CommandFlags.None) =>
-        throw new NotImplementedException();
+        Context(flags).HyperLogLogs().PfMerge(destination, sourceKeys).Wait(SyncTimeout);
 }