Merge pull request #9 from La-404-Devinci/feat/redis-sets

feat: add sorted sets functions to redis
La-404-Devinci · Nov 4, 2024 · 9638b29 · 9638b29
2 parents 65168b7 + dc4400f
commit 9638b29
Showing 1 changed file with 29 additions and 0 deletions.
diff --git a/database/redis.ts b/database/redis.ts
@@ -38,4 +38,33 @@ export default abstract class Redis {
     public static async del(key: string): Promise<void> {
         await Redis.client.del(key);
     }
+
+    public static async sortedSet<T>(setName: string, score: number, data: T): Promise<void> {
+        const stringValue = JSON.stringify(data);
+
+        await Redis.client.zAdd(setName, {
+            value: stringValue,
+            score
+        });
+    }
+
+    public static async sortedRemove<T>(setName: string, data: T): Promise<void> {
+        const stringValue = JSON.stringify(data);
+        await Redis.client.zRem(setName, stringValue);
+    }
+
+    public static async sortedAll<T>(setName: string): Promise<{ value: T; score: number }[]> {
+        const data = await Redis.client.zRangeWithScores(setName, 0, -1);
+
+        return data.map(({ value, score }) => ({
+            value: JSON.parse(value) as T,
+            score
+        }));
+    }
+
+    public static async sortedGetScore<T>(setName: string, data: T): Promise<number> {
+        const stringValue = JSON.stringify(data);
+        const score = await Redis.client.zScore(setName, stringValue);
+        return score ? score : 0;
+    }
 }