mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-17 04:24:35 +03:00
fd30fb343b
PR-URL: https://github.com/hasura/graphql-engine-mono/pull/3982 Co-authored-by: Brandon Simmons <210815+jberryman@users.noreply.github.com> GitOrigin-RevId: f90b2e8f394e7bd69780f003d2d980475f104f42
92 lines
2.8 KiB
Haskell
92 lines
2.8 KiB
Haskell
module Hasura.Backends.BigQuery.DDL
|
|
( fetchAndValidateEnumValues,
|
|
buildFunctionInfo,
|
|
updateColumnInEventTrigger,
|
|
parseBoolExpOperations,
|
|
parseCollectableType,
|
|
module M,
|
|
)
|
|
where
|
|
|
|
import Data.Aeson
|
|
import Hasura.Backends.BigQuery.DDL.BoolExp
|
|
import Hasura.Backends.BigQuery.DDL.ComputedField as M
|
|
import Hasura.Backends.BigQuery.DDL.Source as M
|
|
import Hasura.Backends.BigQuery.Types qualified as BigQuery
|
|
import Hasura.Base.Error
|
|
import Hasura.Prelude
|
|
import Hasura.RQL.IR.BoolExp
|
|
import Hasura.RQL.Types.Backend
|
|
import Hasura.RQL.Types.Column
|
|
import Hasura.RQL.Types.Common
|
|
import Hasura.RQL.Types.EventTrigger
|
|
import Hasura.RQL.Types.Function
|
|
import Hasura.RQL.Types.SchemaCache
|
|
import Hasura.RQL.Types.SourceCustomization (NamingCase)
|
|
import Hasura.RQL.Types.Table
|
|
import Hasura.SQL.Backend
|
|
import Hasura.SQL.Types
|
|
import Hasura.Server.Utils
|
|
import Hasura.Session
|
|
|
|
fetchAndValidateEnumValues ::
|
|
(Monad m) =>
|
|
SourceConfig 'BigQuery ->
|
|
TableName 'BigQuery ->
|
|
Maybe (PrimaryKey 'BigQuery (RawColumnInfo 'BigQuery)) ->
|
|
[RawColumnInfo 'BigQuery] ->
|
|
m (Either QErr EnumValues)
|
|
fetchAndValidateEnumValues _ _ _ _ =
|
|
runExceptT $
|
|
throw400 NotSupported "Enum tables are not supported for BigQuery sources"
|
|
|
|
buildFunctionInfo ::
|
|
(MonadError QErr m) =>
|
|
SourceName ->
|
|
FunctionName 'BigQuery ->
|
|
SystemDefined ->
|
|
FunctionConfig ->
|
|
FunctionPermissionsMap ->
|
|
RawFunctionInfo 'BigQuery ->
|
|
Maybe Text ->
|
|
NamingCase ->
|
|
m (FunctionInfo 'BigQuery, SchemaDependency)
|
|
buildFunctionInfo _ _ _ _ _ _ _ _ =
|
|
throw400 NotSupported "SQL Functions are not supported for BigQuery source"
|
|
|
|
updateColumnInEventTrigger ::
|
|
TableName 'BigQuery ->
|
|
Column 'BigQuery ->
|
|
Column 'BigQuery ->
|
|
TableName 'BigQuery ->
|
|
EventTriggerConf 'BigQuery ->
|
|
EventTriggerConf 'BigQuery
|
|
updateColumnInEventTrigger _ _ _ _ = id
|
|
|
|
parseCollectableType ::
|
|
(MonadError QErr m) =>
|
|
CollectableType (ColumnType 'BigQuery) ->
|
|
Value ->
|
|
m (PartialSQLExp 'BigQuery)
|
|
parseCollectableType collectableType = \case
|
|
String t
|
|
| isSessionVariable t -> pure $ mkTypedSessionVar collectableType $ mkSessionVariable t
|
|
| isReqUserId t -> pure $ mkTypedSessionVar collectableType userIdHeader
|
|
val -> case collectableType of
|
|
CollectableTypeScalar scalarType ->
|
|
PSESQLExp . BigQuery.ValueExpression <$> parseScalarValueColumnType scalarType val
|
|
CollectableTypeArray _ ->
|
|
throw400 NotSupported "Array types are not supported in BigQuery backend"
|
|
|
|
mkTypedSessionVar ::
|
|
CollectableType (ColumnType 'BigQuery) ->
|
|
SessionVariable ->
|
|
PartialSQLExp 'BigQuery
|
|
mkTypedSessionVar columnType =
|
|
PSESessVar (msColumnTypeToScalarType <$> columnType)
|
|
|
|
msColumnTypeToScalarType :: ColumnType 'BigQuery -> ScalarType 'BigQuery
|
|
msColumnTypeToScalarType = \case
|
|
ColumnScalar scalarType -> scalarType
|
|
ColumnEnumReference _ -> BigQuery.StringScalarType
|