mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-19 21:41:44 +03:00
fc7ea9213c
Insert trigger function: If query affects no rows then return `null` Insert trigger function is modified to have `IF r IS NULL THEN RETURN null; ELSE RETURN r; END IF;` in return statement.
55 lines
1.8 KiB
Haskell
55 lines
1.8 KiB
Haskell
{-# LANGUAGE FlexibleContexts #-}
|
|
{-# LANGUAGE MultiParamTypeClasses #-}
|
|
{-# LANGUAGE OverloadedStrings #-}
|
|
{-# LANGUAGE TemplateHaskell #-}
|
|
|
|
module Hasura.RQL.DDL.Permission.Triggers where
|
|
|
|
import Hasura.Prelude
|
|
import Hasura.RQL.Types
|
|
import Hasura.Server.Utils
|
|
import Hasura.SQL.Types
|
|
|
|
import qualified Database.PG.Query as Q
|
|
import qualified Hasura.SQL.DML as S
|
|
|
|
import qualified Data.Aeson as J
|
|
import qualified Data.ByteString.Builder as BB
|
|
import qualified Data.FileEmbed as FE
|
|
import qualified Data.Text as T
|
|
|
|
buildInsTrig :: QualifiedTable -> Q.Query
|
|
buildInsTrig qt@(QualifiedTable _ tn) =
|
|
Q.fromBuilder $ mconcat
|
|
[ BB.string7 "CREATE TRIGGER " <> toSQL tn
|
|
, BB.string7 " INSTEAD OF INSERT ON " <> toSQL qt
|
|
, BB.string7 " FOR EACH ROW EXECUTE PROCEDURE "
|
|
, toSQL qt <> BB.string7 "();"
|
|
]
|
|
|
|
dropInsTrigFn :: QualifiedTable -> Q.Query
|
|
dropInsTrigFn fn =
|
|
Q.fromBuilder $ BB.string7 "DROP FUNCTION " <> toSQL fn <> "()"
|
|
|
|
getInsTrigTmplt :: (MonadError QErr m) => m GingerTmplt
|
|
getInsTrigTmplt =
|
|
either throwErr return $ parseGingerTmplt trigFnSrc
|
|
where
|
|
trigFnSrc = $(FE.embedStringFile "src-rsr/insert_trigger.sql.j2")
|
|
|
|
throwErr e = throw500 $ "cannot render insert trigger function template: "
|
|
<> T.pack e
|
|
|
|
buildInsTrigFn
|
|
:: (MonadError QErr m)
|
|
=> QualifiedTable -> QualifiedTable -> S.BoolExp -> m Q.Query
|
|
buildInsTrigFn fn tn be = do
|
|
insTmplt <- getInsTrigTmplt
|
|
return $ Q.fromBuilder $ BB.string7 $ T.unpack $
|
|
renderGingerTmplt tmpltVals insTmplt
|
|
where
|
|
tmpltVals = J.object [ "function_name" J..= toSQLTxt fn
|
|
, "table_name" J..= toSQLTxt tn
|
|
, "check_expression" J..= toSQLTxt be
|
|
]
|