SharedKernal/feat: #72 Exposing Redis Eval Function

nammayatri · Apr 11, 2023 · 690b8ab · 690b8ab
1 parent 09e1119
commit 690b8ab
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/lib/mobility-core/src/Kernel/Storage/Hedis/Queries.hs b/lib/mobility-core/src/Kernel/Storage/Hedis/Queries.hs
@@ -19,7 +19,7 @@ import qualified Data.ByteString as BS
 import qualified Data.ByteString.Lazy as BSL
 import Data.String.Conversions
 import qualified Data.Text as Text
-import Database.Redis (Queued, Redis, RedisTx, Reply, TxResult (..))
+import Database.Redis (Queued, Redis, RedisTx, Reply, TxResult (..),RedisResult)
 import qualified Database.Redis as Hedis
 import GHC.Records.Extra
 import Kernel.Prelude
@@ -202,6 +202,11 @@ delByPattern ptrn = do
   runWithPrefix_ ptrn $ \prefKey ->
     Hedis.eval @_ @_ @Reply "for i, name in ipairs(redis.call('KEYS', ARGV[1])) do redis.call('DEL', name); end" ["0"] [prefKey]
 
+eval ::(RedisResult a, HedisFlow m env) => Text -> [Text] -> [Text] -> m a
+eval script keys args = do
+  nKeys <- mapM buildKey keys
+  runHedis $ Hedis.eval (cs script) nKeys (map cs args)
+
 tryLockRedis :: HedisFlow m env => Text -> ExpirationTime -> m Bool
 tryLockRedis key timeout = setNxExpire (buildLockResourceName key) timeout ()