on ca7aaef help sign in
rel8/src/Rel8.hs
682 lines 19 kB history raw
1{-# language DuplicateRecordFields #-}
3module Rel8
4 ( -- * Getting Started
5 -- $setup
7 -- ** Writing Queries
8 -- $guideQueries
10 -- ** Joins
11 -- $guideJoins
13 -- ** Aggregation
14 -- $guideAggregation
16 -- * Database types
17 -- ** @DBType@
18 DBType(..)
20 -- *** Deriving-via helpers
21 -- **** @JSONEncoded@
22 , JSONEncoded(..)
23 , JSONBEncoded(..)
25 -- **** @ReadShow@
26 , ReadShow(..)
28 -- *** @TypeInformation@
29 , TypeInformation(..)
30 , mapTypeInformation
31 , parseTypeInformation
33 -- ** The @DBType@ hierarchy
34 , DBSemigroup(..)
35 , DBMonoid(..)
36 , DBNum
37 , DBIntegral
38 , DBFractional
40 -- * Tables and higher-kinded tables
41 , Rel8able
42 , Column
43 , Default, Label
44 , HMaybe
45 , HList
46 , HNonEmpty
47 , HThese
49 , Table
50 , AltTable((<|>:))
51 , EqTable, (==:), (/=:)
52 , OrdTable, ascTable, descTable
53 , lit
54 , bool
55 , case_
57 -- ** @MaybeTable@
58 , MaybeTable
59 , maybeTable, ($?), nothingTable, justTable
60 , isNothingTable, isJustTable
61 , optional
62 , catMaybeTable
63 , bindMaybeTable
64 , traverseMaybeTable
66 -- ** @EitherTable@
67 , EitherTable
68 , eitherTable, leftTable, rightTable
69 , isLeftTable, isRightTable
70 , keepLeftTable
71 , keepRightTable
72 , bindEitherTable
73 , bitraverseEitherTable
75 -- ** @TheseTable@
76 , TheseTable
77 , theseTable, thisTable, thatTable, thoseTable
78 , isThisTable, isThatTable, isThoseTable
79 , hasHereTable, hasThereTable
80 , justHereTable, justThereTable
81 , alignBy
82 , keepHereTable, loseHereTable
83 , keepThereTable, loseThereTable
84 , keepThisTable, loseThisTable
85 , keepThatTable, loseThatTable
86 , keepThoseTable, loseThoseTable
87 , bindTheseTable
88 , bitraverseTheseTable
90 -- ** @ListTable@
91 , ListTable
92 , many
94 -- ** @NonEmptyTable@
95 , NonEmptyTable
96 , some
98 -- ** Table schemas
99 , TableSchema(..)
100 , Name
101 , namesFromLabels
102 , namesFromLabelsWith
104 -- * Expressions
105 , Expr
106 , Sql
107 , litExpr
108 , unsafeCastExpr
110 -- ** @null@
111 , null
112 , nullify
113 , nullable
114 , isNull
115 , isNonNull
116 , mapNullable
117 , liftOpNullable
118 , catNullable
119 , coalesce
121 -- ** Boolean operations
122 , DBEq
123 , true, false, not_
124 , (&&.), and_
125 , (||.), or_
126 , (==.), (/=.), (==?), (/=?)
127 , in_
128 , boolExpr, caseExpr
130 -- ** Ordering
131 , DBOrd
132 , (<.), (<=.), (>.), (>=.)
133 , (<?), (<=?), (>?), (>=?)
134 , leastExpr, greatestExpr
136 -- ** Functions
137 , Function
138 , function
139 , nullaryFunction
140 , binaryOperator
142 -- * Queries
143 , Query
144 , showQuery
146 -- ** Selecting rows
147 , Selects
148 , each
149 , values
151 -- ** Filtering
152 , filter
153 , where_
154 , whereExists
155 , whereNotExists
156 , distinct
157 , distinctOn
158 , distinctOnBy
160 -- ** @LIMIT@/@OFFSET@
161 , limit
162 , offset
164 -- ** @UNION@
165 , union
166 , unionAll
168 -- ** @INTERSECT@
169 , intersect
170 , intersectAll
172 -- ** @EXCEPT@
173 , except
174 , exceptAll
176 -- ** @EXISTS@
177 , exists
178 , with
179 , withBy
180 , without
181 , withoutBy
183 -- ** Aggregation
184 , Aggregate
185 , aggregate
186 , countRows
187 , listAgg, listAggExpr
188 , nonEmptyAgg, nonEmptyAggExpr
189 , groupBy
190 , DBMax, max
191 , DBMin, min
192 , DBSum, sum, sumWhere
193 , DBString, stringAgg
194 , count
195 , countStar
196 , countDistinct
197 , countWhere
198 , and
199 , or
201 , Aggregates
202 , sequenceAggregate
203 , distributeAggregate
206 -- ** Ordering
207 , orderBy
208 , Order
209 , asc
210 , desc
211 , nullsFirst
212 , nullsLast
214 -- * IO
215 , Serializable
217 -- * Running statements
218 -- ** @SELECT@
219 , select
221 -- ** @INSERT@
222 , Insert(..)
223 , OnConflict(..)
224 , insert
225 , toInsert
226 , toInsertDefaults
228 -- ** @DELETE@
229 , Delete(..)
230 , delete
232 -- ** @UPDATE@
233 , update
234 , Update(..)
236 -- ** @.. RETURNING@
237 , Returning(..)
239 -- ** @CREATE VIEW@
240 , createView
242 -- * TODO
243 -- TODO
244 -- These need organizing, but are reachable from Rel8's documentation so we
245 -- do need to export and document them.
246 , HasNullability
247 , NotNull
248 , HTable
249 , Labelable
250 , Context
251 ) where
253-- base
254import Prelude ()
256-- rel8
257import Rel8.Aggregate
258import Rel8.Expr
259import Rel8.Expr.Aggregate
260import Rel8.Expr.Bool
261import Rel8.Expr.Eq
262import Rel8.Expr.Function
263import Rel8.Expr.Null
264import Rel8.Expr.Ord
265import Rel8.Expr.Order
266import Rel8.Expr.Serialize
267import Rel8.Order
268import Rel8.Query
269import Rel8.Query.Aggregate
270import Rel8.Query.Distinct
271import Rel8.Query.Each
272import Rel8.Query.Either
273import Rel8.Query.Exists
274import Rel8.Query.Filter
275import Rel8.Query.Limit
276import Rel8.Query.List
277import Rel8.Query.Maybe
278import Rel8.Query.Null
279import Rel8.Query.Order
280import Rel8.Query.SQL (showQuery)
281import Rel8.Query.Set
282import Rel8.Query.These
283import Rel8.Query.Values
284import Rel8.Schema.Column
285import Rel8.Schema.Context.Label
286import Rel8.Schema.Generic
287import Rel8.Schema.HTable
288import Rel8.Schema.Name
289import Rel8.Schema.Nullability
290import Rel8.Schema.Table
291import Rel8.Statement.Delete
292import Rel8.Statement.Insert
293import Rel8.Statement.Returning
294import Rel8.Statement.Select
295import Rel8.Statement.Update
296import Rel8.Statement.View
297import Rel8.Table
298import Rel8.Table.Aggregate
299import Rel8.Table.Alternative
300import Rel8.Table.Bool
301import Rel8.Table.Either
302import Rel8.Table.Eq
303import Rel8.Table.Insert
304import Rel8.Table.List
305import Rel8.Table.Maybe
306import Rel8.Table.Name
307import Rel8.Table.NonEmpty
308import Rel8.Table.Ord
309import Rel8.Table.Order
310import Rel8.Table.Serialize
311import Rel8.Table.These
312import Rel8.Type
313import Rel8.Type.Eq
314import Rel8.Type.Information
315import Rel8.Type.JSONEncoded
316import Rel8.Type.JSONBEncoded
317import Rel8.Type.Monoid
318import Rel8.Type.Num
319import Rel8.Type.Ord
320import Rel8.Type.ReadShow
321import Rel8.Type.Semigroup
322import Rel8.Type.String
323import Rel8.Type.Sum
324import Rel8.Expr.Opaleye (unsafeCastExpr)
327-- $setup
328--
329-- In this section, we'll take a look at using Rel8 to work with a small
330-- database for Haskell packages. We'll take a look at idiomatic usage of Rel8,
331-- mapping tables to Haskell, and then look at writing some simple queries.
332--
333-- The documentation in this library uses the following language extensions:
334--
335-- >>> :set -XBlockArguments
336-- >>> :set -XDeriveAnyClass
337-- >>> :set -XDeriveGeneric
338-- >>> :set -XDerivingStrategies
339-- >>> :set -XDerivingVia
340-- >>> :set -XDuplicateRecordFields
341-- >>> :set -XGeneralizedNewtypeDeriving
342-- >>> :set -XOverloadedStrings
343-- >>> :set -XStandaloneDeriving
344-- >>> :set -XTypeApplications
345-- >>> :set -XTypeFamilies
346--
347-- Before we start writing any Haskell, let's take a look at the schema we'll
348-- work with.
349--
350-- > author: project:
351-- >
352-- > Column │ Type │ Nullable Column │ Type │ Nullable
353-- > ═══════════╪═════════╪══════════ ═══════════╪═════════╪══════════
354-- > author_id │ integer │ not null author_id │ integer │ not null
355-- > name │ text │ not null name │ text │ not null
356-- > url │ text │
357--
358-- Our schema consists of two tables - @author@ and @project@. A @project@
359-- always has an @author@, but not all @author@s have projects. Each @author@
360-- has a name and (maybe) an associated website, and each project has a name.
361--
362-- Now that we've seen our schema, we can begin writing a mapping in Rel8. The
363-- idiomatic way to map a table is to use a record that is parameterised what
364-- Rel8 calls an /interpretation functor/, and to define each field with
365-- 'Column'. For this type to be usable with Rel8 we need it to be an instance
366-- of 'HigherKindedTable', which can be derived with a combination of
367-- @DeriveAnyClass@ and @DeriveGeneric@.
368--
369-- Following these steps for @author@, we have:
370--
371-- > data Author f = Author
372-- > { authorId :: Column f Int64
373-- > , name :: Column f Text
374-- > , url :: Column f (Maybe Text)
375-- > } deriving (Generic, HigherKindedTable)
376--
377-- This is a perfectly reasonable definition, but cautious readers might notice
378-- a problem - in particular, with the type of the @authorId@ field. While
379-- @Int64@ is correct, it's not the best type. If we had other identifier types
380-- in our project, it would be too easy to accidentally mix them up and create
381-- nonsensical joins. As Haskell programmers, we often solve this problem by
382-- creating @newtype@ wrappers, and we can also use this technique with Rel8:
383--
384-- >>> :{
385-- newtype AuthorId = AuthorId { toInt64 :: Int64 }
386-- deriving newtype (DBEq, DBType, Eq, Show)
387-- :}
388--
389-- Now we can write our final schema mapping. First, the @author@ table:
390--
391-- >>> :{
392-- data Author f = Author
393-- { authorId :: Column f AuthorId
394-- , authorName :: Column f Text
395-- , authorUrl :: Column f (Maybe Text)
396-- }
397-- deriving stock Generic
398-- deriving anyclass HigherKindedTable
399-- :}
400--
401-- And similarly, the @project@ table:
402--
403-- >>> :{
404-- data Project f = Project
405-- { projectAuthorId :: Column f AuthorId
406-- , projectName :: Column f Text
407-- }
408-- deriving stock Generic
409-- deriving anyclass HigherKindedTable
410-- :}
411--
412-- To show query results in this documentation, we'll also need @Show@
413-- instances: Unfortunately these definitions look a bit scary, but they are
414-- essentially just @deriving (Show)@:
415--
416-- >>> deriving stock instance f ~ Identity => Show (Author f)
417-- >>> deriving stock instance f ~ Identity => Show (Project f)
418--
419-- These data types describe the structural mapping of the tables, but we also
420-- need to specify a 'TableSchema' for each table. A @TableSchema@ contains the
421-- name of the table and the name of all columns in the table, which will
422-- ultimately allow us to @SELECT@ and @INSERT@ rows for these tables.
423--
424-- To define a @TableSchema@, we just need to fill construct appropriate
425-- @TableSchema@ values. When it comes to the @tableColumns@ field, we
426-- construct values of our data types above, and set each field to the name of
427-- the column that it maps to:
428--
429-- First, @authorSchema@ describes the column names of the @author@ table when
430-- associated with the @Author@ type:
431--
432-- >>> :{
433-- authorSchema :: TableSchema (Author ColumnSchema)
434-- authorSchema = TableSchema
435-- { tableName = "author"
436-- , tableSchema = Nothing
437-- , tableColumns = Author
438-- { authorId = "author_id"
439-- , authorName = "name"
440-- , authorUrl = "url"
441-- }
442-- }
443-- :}
444--
445-- And likewise for @project@ and @Project@:
446--
447-- >>> :{
448-- projectSchema :: TableSchema (Project ColumnSchema)
449-- projectSchema = TableSchema
450-- { tableName = "project"
451-- , tableSchema = Nothing
452-- , tableColumns = Project
453-- { projectAuthorId = "author_id"
454-- , projectName = "name"
455-- }
456-- }
457-- :}
458--
459-- Aside: you might be wondering why this information isn't in the definitions
460-- of @Author@ and @Project@ above. Rel8 decouples @TableSchema@ from the data
461-- types themselves, as not all tables you define will necessarily have a
462-- schema. For example, Rel8 allows you to define helper types to simplify the
463-- types of queries - these tables only exist at query time, but there is no
464-- corresponding base table. We'll see more on this idea later!
465--
466-- With these table definitions, we can now start writing some queries! To
467-- actually run a query, we'll need a database connection. Rel8 uses the
468-- @hasql@ library, and for this documentation we'll get the connection string
469-- from the @$TEST_DATABASE_URL@ environment variable.
470--
471-- >>> connectionString <- System.Environment.getEnv "TEST_DATABASE_URL"
472-- >>> Right c <- Hasql.Connection.acquire (Data.ByteString.Char8.pack connectionString)
473-- >>> Control.Monad.void $ Hasql.Session.run (Hasql.Session.sql "BEGIN") c
475-- $guideQueries
476--
477-- First, we'll take a look at @SELECT@ statements - usually the bulk of most
478-- database heavy applications.
479--
480-- In Rel8, @SELECT@ statements are built using the 'Query' monad. You can
481-- think of this monad like the ordinary @[]@ (List) monad - but this isn't
482-- required knowledge.
483--
484-- To start, we'll look at one of the simplest queries possible - a basic
485-- @SELECT * FROM@ statement. To select all rows from a table, we use 'each',
486-- and supply a 'TableSchema'. So to select all @project@ rows, we can write:
487--
488-- >>> :t each projectSchema
489-- each projectSchema :: Query (Project Expr)
490--
491-- Notice that @each@ gives us a @Query@ that yields @Project Expr@ rows. To
492-- see what this means, let's have a look at a single field of a
493-- @Project Expr@:
494--
495-- >>> let aProjectExpr = undefined :: Project Expr
496-- >>> :t projectAuthorId aProjectExpr
497-- projectAuthorId aProjectExpr :: Expr AuthorId
498--
499-- Recall we defined @projectAuthorId@ as @Column f AuthorId@. Now we have @f@
500-- is @Expr@, and @Column Expr AuthorId@ reduces to @Expr AuthorId@. We'll see
501-- more about @Expr@ soon, but you can think of @Expr a@ as "SQL expressions of
502-- type @a@".
503--
504-- To execute this @Query@, we pass it to 'select':
505--
506-- >>> :t select c (each projectSchema)
507-- select c (each projectSchema) :: MonadIO m => m [Project Identity]
508--
509-- When we @select@ things containing @Expr@s, Rel8 builds a new response table
510-- with the @Identity@ interpretation. This means you'll get back plain Haskell
511-- values. Studying @projectAuthorId@ again, we have:
512--
513-- >>> let aProjectIdentity = undefined :: Project Identity
514-- >>> :t projectAuthorId aProjectIdentity
515-- projectAuthorId aProjectIdentity :: AuthorId
516--
517-- Here @Column Identity AuthorId@ reduces to just @AuthorId@, with no
518-- wrappping type at all.
519--
520-- Putting this all together, we can run our first query:
521--
522-- >>> select c (each projectSchema) >>= mapM_ print
523-- Project {projectAuthorId = 1, projectName = "rel8"}
524-- Project {projectAuthorId = 2, projectName = "aeson"}
525-- Project {projectAuthorId = 2, projectName = "text"}
526--
527-- Cool!
528--
529-- We now know that 'each' is the equivalent of a @SELECT *@ query, but
530-- sometimes we're only interested in a subset of the columns of a table. To
531-- restrict the returned columns, we can specify a projection by using 'Query's
532-- @Functor@ instance:
533--
534-- >>> select c $ projectName <$> each projectSchema
535-- ["rel8","aeson","text"]
537-- $guideJoins
538--
539-- Another common operation in relational databases is to take the @JOIN@ of
540-- multiple tables. Rel8 doesn't have a specific join operation, but we can
541-- recover the functionality of a join by selecting all rows of two tables, and
542-- then using 'where_' to filter them.
543--
544-- To see how this works, first let's look at taking the product of two tables.
545-- We can do this by simply calling 'each' twice, and then returning a tuple of
546-- their results.
547--
548-- >>> :{
549-- mapM_ print =<< select c do
550-- author <- each authorSchema
551-- project <- each projectSchema
552-- return (projectName project, authorName author)
553-- :}
554-- ("rel8","Ollie")
555-- ("rel8","Bryan O'Sullivan")
556-- ("rel8","Emily Pillmore")
557-- ("aeson","Ollie")
558-- ("aeson","Bryan O'Sullivan")
559-- ("aeson","Emily Pillmore")
560-- ("text","Ollie")
561-- ("text","Bryan O'Sullivan")
562-- ("text","Emily Pillmore")
563--
564-- This isn't quite right, though, as we have ended up pairing up the wrong
565-- projects and authors. To fix this, we can use 'where_' to restrict the
566-- returned rows. We could write:
567--
568-- > select c $ do
569-- > author <- each authorSchema
570-- > project <- each projectSchema
571-- > where_ $ projectAuthorId project ==. authorId author
572-- > return (project, author)
573--
574-- but doing this every time you need a join can obscure the meaning of the
575-- query you're writing. A good practice is to introduce specialised functions
576-- for the particular joins in your database. In our case, this would be:
577--
578-- >>> :{
579-- projectsForAuthor :: Author Expr -> Query (Project Expr)
580-- projectsForAuthor a = each projectSchema >>= filter \p ->
581-- projectAuthorId p ==. authorId a
582-- :}
583--
584-- Our final query is then:
585--
586-- >>> :{
587-- mapM_ print =<< select c do
588-- author <- each authorSchema
589-- project <- projectsForAuthor author
590-- return (projectName project, authorName author)
591-- :}
592-- ("rel8","Ollie")
593-- ("aeson","Bryan O'Sullivan")
594-- ("text","Bryan O'Sullivan")
595--
596-- == @LEFT JOIN@s
597--
598-- Rel8 is also capable of performing @LEFT JOIN@s. To perform @LEFT JOIN@s, we
599-- follow the same approach as before, but use the 'optional' query transformer
600-- to allow for the possibility of the join to fail.
601--
602-- In our test database, we can see that there's another author who we haven't
603-- seen yet:
604--
605-- >>> select c $ authorName <$> each authorSchema
606-- ["Ollie","Bryan O'Sullivan","Emily Pillmore"]
607--
608-- Emily wasn't returned in our earlier query because - in our database - she
609-- doesn't have any registered projects. We can account for this partiality in
610-- our original query by wrapping the @projectsForAuthor@ call with 'optional':
611--
612-- >>> :{
613-- mapM_ print =<< select c do
614-- author <- each authorSchema
615-- mproject <- optional $ projectsForAuthor author
616-- return (authorName author, projectName <$> mproject)
617-- :}
618-- ("Ollie",Just "rel8")
619-- ("Bryan O'Sullivan",Just "aeson")
620-- ("Bryan O'Sullivan",Just "text")
621-- ("Emily Pillmore",Nothing)
623-- $guideAggregation
624--
625-- Aggregations are operations like @sum@ and @count@ - operations that reduce
626-- multiple rows to single values. To perform aggregations in Rel8, we can use
627-- the 'aggregate' function, which takes a 'Query' of aggregated expressions,
628-- runs the aggregation, and returns aggregated rows.
629--
630-- To start, let's look at a simple aggregation that tells us how many projects
631-- exist:
632--
633-- >>> error "TODO"
634--
635-- Rel8 is also capable of aggregating multiple rows into a single row by
636-- concatenating all rows as a list. This aggregation allows us to break free
637-- of the row-orientated nature of SQL and write queries that return tree-like
638-- structures. Earlier we saw an example of returning authors with their
639-- projects, but the query didn't do a great job of describing the one-to-many
640-- relationship between authors and their projects.
641--
642-- Let's look again at a query that returns authors and their projects, and
643-- focus on the /type/ of that query.
644--
645-- >>> :{
646-- projectsForAuthor a = each projectSchema >>= filter \p ->
647-- projectAuthorId p ==. authorId a
648-- :}
649--
650-- >>> :{
651-- let authorsAndProjects = do
652-- author <- each authorSchema
653-- project <- projectsForAuthor author
654-- return (author, project)
655-- where
656-- :}
657--
658-- >>> :t select c authorsAndProjects
659-- select c authorsAndProjects
660-- :: MonadIO m => m [(Author Identity, Project Identity)]
661--
662-- Our query gives us a single list of pairs of authors and projects. However,
663-- with our domain knowledge of the schema, this isn't a great type - what we'd
664-- rather have is a list of pairs of authors and /lists/ of projects. That is,
665-- what we'd like is:
666--
667-- > [(Author Identity, [Project Identity])]
668--
669-- This would be a much better type! Rel8 can produce a query with this type by
670-- simply wrapping the call to @projectsForAuthor@ with either 'some' or
671-- 'many'. Here we'll use 'many', which allows for the possibility of an author
672-- to have no projects:
673--
674-- >>> :{
675-- mapM_ print =<< select c do
676-- author <- each authorSchema
677-- projectNames <- many $ projectName <$> projectsForAuthor author
678-- return (authorName author, projectNames)
679-- :}
680-- ("Ollie",["rel8"])
681-- ("Bryan O'Sullivan",["aeson","text"])
682-- ("Emily Pillmore",[])