Skip to content

Commit

Permalink
Added inline annotations for getQueryBuilder
Browse files Browse the repository at this point in the history
  • Loading branch information
mpscholten committed Jun 22, 2024
1 parent d9ca29f commit 4fe8b56
Showing 1 changed file with 11 additions and 1 deletion.
12 changes: 11 additions & 1 deletion IHP/QueryBuilder.hs
Original file line number Diff line number Diff line change
Expand Up @@ -163,7 +163,8 @@ class HasQueryBuilder queryBuilderProvider joinRegister | queryBuilderProvider -
getQueryBuilder :: queryBuilderProvider table -> QueryBuilder table
injectQueryBuilder :: QueryBuilder table -> queryBuilderProvider table
getQueryIndex :: queryBuilderProvider table -> Maybe ByteString
getQueryIndex _ = Nothing
getQueryIndex _ = Nothing
{-# INLINE getQueryIndex #-}

-- Wrapper for QueryBuilders resulting from joins. Associates a joinRegister type.
newtype JoinQueryBuilderWrapper joinRegister table = JoinQueryBuilderWrapper (QueryBuilder table)
Expand All @@ -177,22 +178,31 @@ newtype LabeledQueryBuilderWrapper foreignTable indexColumn indexValue table = L
-- QueryBuilders have query builders and the join register is empty.
instance HasQueryBuilder QueryBuilder EmptyModelList where
getQueryBuilder = id
{-# INLINE getQueryBuilder #-}
injectQueryBuilder = id
{-# INLINE injectQueryBuilder #-}

-- JoinQueryBuilderWrappers have query builders
instance HasQueryBuilder (JoinQueryBuilderWrapper joinRegister) joinRegister where
getQueryBuilder (JoinQueryBuilderWrapper queryBuilder) = queryBuilder
{-# INLINE getQueryBuilder #-}
injectQueryBuilder = JoinQueryBuilderWrapper
{-# INLINE injectQueryBuilder #-}

-- NoJoinQueryBuilderWrapper have query builders and the join register does not allow any joins
instance HasQueryBuilder NoJoinQueryBuilderWrapper NoJoins where
getQueryBuilder (NoJoinQueryBuilderWrapper queryBuilder) = queryBuilder
{-# INLINE getQueryBuilder #-}
injectQueryBuilder = NoJoinQueryBuilderWrapper
{-# INLINE injectQueryBuilder #-}

instance (KnownSymbol foreignTable, foreignModel ~ GetModelByTableName foreignTable , KnownSymbol indexColumn, HasField indexColumn foreignModel indexValue) => HasQueryBuilder (LabeledQueryBuilderWrapper foreignTable indexColumn indexValue) NoJoins where
getQueryBuilder (LabeledQueryBuilderWrapper queryBuilder) = queryBuilder
{-# INLINE getQueryBuilder #-}
injectQueryBuilder = LabeledQueryBuilderWrapper
{-# INLINE injectQueryBuilder #-}
getQueryIndex _ = Just $ symbolToByteString @foreignTable <> "." <> (Text.encodeUtf8 . fieldNameToColumnName) (symbolToText @indexColumn)
{-# INLINE getQueryIndex #-}


data QueryBuilder (table :: Symbol) =
Expand Down

0 comments on commit 4fe8b56

Please sign in to comment.