Skip to content

Avoid generating WITH clauses for 0 CTEs #761

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 2 commits into
base: master
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 3 additions & 2 deletions beam-core/Database/Beam/Backend/SQL/SQL99.hs
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@ module Database.Beam.Backend.SQL.SQL99
import Database.Beam.Backend.SQL.SQL92

import Data.Kind ( Type )
import Data.List.NonEmpty ( NonEmpty )
import Data.Text ( Text )

class IsSql92SelectSyntax select =>
Expand Down Expand Up @@ -56,12 +57,12 @@ class IsSql92SelectSyntax syntax =>
IsSql99CommonTableExpressionSelectSyntax syntax where
type Sql99SelectCTESyntax syntax :: Type

withSyntax :: [ Sql99SelectCTESyntax syntax ] -> syntax -> syntax
withSyntax :: NonEmpty (Sql99SelectCTESyntax syntax) -> syntax -> syntax

class IsSql99CommonTableExpressionSelectSyntax syntax
=> IsSql99RecursiveCommonTableExpressionSelectSyntax syntax where

withRecursiveSyntax :: [ Sql99SelectCTESyntax syntax ] -> syntax -> syntax
withRecursiveSyntax :: NonEmpty (Sql99SelectCTESyntax syntax) -> syntax -> syntax

class IsSql99CommonTableExpressionSyntax syntax where
type Sql99CTESelectSyntax syntax :: Type
Expand Down
16 changes: 11 additions & 5 deletions beam-core/Database/Beam/Query.hs
Original file line number Diff line number Diff line change
Expand Up @@ -123,6 +123,7 @@ import Control.Monad.State.Strict

import Data.Kind (Type)
import Data.Functor.Const (Const(..))
import Data.List.NonEmpty (nonEmpty)
import Data.Text (Text)
import Data.Proxy

Expand Down Expand Up @@ -159,11 +160,16 @@ selectWith :: forall be db res
=> With be db (Q be db QBaseScope res) -> SqlSelect be (QExprToIdentity res)
selectWith (CTE.With mkQ) =
let (q, (recursiveness, ctes)) = evalState (runWriterT mkQ) 0
in case recursiveness of
CTE.Nonrecursive -> SqlSelect (withSyntax ctes
(buildSqlQuery "t" q))
CTE.Recursive -> SqlSelect (withRecursiveSyntax ctes
(buildSqlQuery "t" q))
in case (recursiveness, nonEmpty ctes) of
(CTE.Nonrecursive, Just ctes') -> SqlSelect (withSyntax ctes'
(buildSqlQuery "t" q))
(CTE.Recursive, Just ctes') -> SqlSelect (withRecursiveSyntax ctes'
(buildSqlQuery "t" q))
{- `WITH` clauses should not be used when no CTEs are created.

See: https://github.com/haskell-beam/beam/issues/760
-}
(_, Nothing) -> SqlSelect (buildSqlQuery "t" q)

-- | Convenience function to generate a 'SqlSelect' that looks up a table row
-- given a primary key.
Expand Down
8 changes: 5 additions & 3 deletions beam-postgres/Database/Beam/Postgres/Full.hs
Original file line number Diff line number Diff line change
Expand Up @@ -72,6 +72,7 @@ import Control.Monad.State.Strict (evalState)
import Control.Monad.Writer (runWriterT)

import Data.Kind (Type)
import Data.List.NonEmpty (nonEmpty)
import Data.Proxy (Proxy(..))
import qualified Data.Text as T

Expand Down Expand Up @@ -298,9 +299,10 @@ pgSelectWith :: forall db s res
pgSelectWith (CTE.With mkQ) =
let (q, (recursiveness, ctes)) = evalState (runWriterT mkQ) 0
fromSyntax tblPfx =
case recursiveness of
CTE.Nonrecursive -> withSyntax ctes (buildSqlQuery tblPfx q)
CTE.Recursive -> withRecursiveSyntax ctes (buildSqlQuery tblPfx q)
case (recursiveness, nonEmpty ctes) of
(CTE.Nonrecursive, Just ctes') -> withSyntax ctes' (buildSqlQuery tblPfx q)
(CTE.Recursive, Just ctes') -> withRecursiveSyntax ctes' (buildSqlQuery tblPfx q)
(_, Nothing) -> buildSqlQuery tblPfx q
in Q (liftF (QAll (\tblPfx tName ->
let (_, names) = mkFieldNames @Postgres @res (qualifiedField tName)
in fromTable (PgTableSourceSyntax $
Expand Down
5 changes: 3 additions & 2 deletions beam-postgres/Database/Beam/Postgres/Syntax.hs
Original file line number Diff line number Diff line change
Expand Up @@ -110,6 +110,7 @@ import Data.Coerce
import Data.Functor.Classes
import Data.Hashable
import Data.Int
import qualified Data.List.NonEmpty as NonEmpty
import Data.Maybe
import Data.Scientific (Scientific)
import Data.String (IsString(..), fromString)
Expand Down Expand Up @@ -589,14 +590,14 @@ instance IsSql99CommonTableExpressionSelectSyntax PgSelectSyntax where
withSyntax ctes (PgSelectSyntax select) =
PgSelectSyntax $
emit "WITH " <>
pgSepBy (emit ", ") (map fromPgCommonTableExpression ctes) <>
pgSepBy (emit ", ") (map fromPgCommonTableExpression $ NonEmpty.toList ctes) <>
select

instance IsSql99RecursiveCommonTableExpressionSelectSyntax PgSelectSyntax where
withRecursiveSyntax ctes (PgSelectSyntax select) =
PgSelectSyntax $
emit "WITH RECURSIVE " <>
pgSepBy (emit ", ") (map fromPgCommonTableExpression ctes) <>
pgSepBy (emit ", ") (map fromPgCommonTableExpression $ NonEmpty.toList ctes) <>
select

instance IsSql99CommonTableExpressionSyntax PgCommonTableExpressionSyntax where
Expand Down
Loading