backend : added kvDbflow in dsl

nammayatri · May 16, 2024 · 99d3378 · 99d3378
1 parent 7ca566f
commit 99d3378
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/lib/namma-dsl/src/NammaDSL/Generator/Haskell/BeamQueries.hs b/lib/namma-dsl/src/NammaDSL/Generator/Haskell/BeamQueries.hs
@@ -232,7 +232,7 @@ generateDefaultCreateQuery storageRead = do
   let name = tableNameHaskell tableDef
   let withIdFields = getAllFieldsWithIdRelation (fields tableDef)
   let dName = domainTypeModulePrefix ++ name ++ "." ++ name
-  let fungSign = maybe ([_EsqDBFlow, _MonadFlow, _CacheFlow] <> [_HasSchemaName name | isHasSchemaNameRequired' ]) (pure . vT) tableDef.defaultQueryTypeConstraint
+  let fungSign = maybe ([_KvDbFlow] <> [_HasSchemaName name | isHasSchemaNameRequired' ]) (pure . vT) tableDef.defaultQueryTypeConstraint
   TH.decsW $ do
     TH.sigDW "create" $ do
       TH.forallT [] fungSign $
@@ -271,6 +271,9 @@ generateDefaultCreateQuery storageRead = do
 _EsqDBFlow :: Q TH.Type
 _EsqDBFlow = cT "EsqDBFlow" ~~ vT "m" ~~ vT "r"
 
+_KvDbFlow :: Q TH.Type
+_KvDbFlow = cT "KvDbFlow" ~~ vT "m" ~~ vT "r"
+
 _MonadFlow :: Q TH.Type
 _MonadFlow = cT "MonadFlow" ~~ vT "m"
 

diff --git a/lib/namma-dsl/tests/dsl-config.dhall b/lib/namma-dsl/tests/dsl-config.dhall
@@ -142,7 +142,7 @@ let defaultImports =
           [ "Kernel.Beam.Functions"
           , "Kernel.Prelude"
           , "Kernel.External.Encryption"
-          , "Kernel.Utils.Common (MonadFlow, CacheFlow, EsqDBFlow, getCurrentTime, fromMaybeM)"
+          , "Kernel.Utils.Common (MonadFlow, CacheFlow, KvDbFlow, EsqDBFlow, getCurrentTime, fromMaybeM)"
           , "Kernel.Types.Error"
           ]
         , _qualifiedImports = [ "Sequelize as Se" ]