[haskell] refactoring
This commit is contained in:
parent
d668eac4a6
commit
7e27a0a7ea
26 changed files with 289 additions and 311 deletions
|
@ -1,24 +1,24 @@
|
|||
{-# LANGUAGE OverloadedStrings #-}
|
||||
{-# LANGUAGE QuasiQuotes #-}
|
||||
{-# LANGUAGE TemplateHaskell #-}
|
||||
{-# LANGUAGE QuasiQuotes #-}
|
||||
{-# LANGUAGE TemplateHaskell #-}
|
||||
|
||||
module SpaceTraders.Database
|
||||
( close
|
||||
, open
|
||||
) where
|
||||
|
||||
import Control.Exception
|
||||
import qualified Data.ByteString as B
|
||||
import Data.FileEmbed
|
||||
import Control.Exception
|
||||
import qualified Data.ByteString as B
|
||||
import Data.FileEmbed
|
||||
import qualified Data.Text as T
|
||||
import qualified Data.Text.Encoding as T
|
||||
import qualified Database.SQLite.Simple as S
|
||||
import qualified Data.Text as T
|
||||
import qualified Data.Text.Encoding as T
|
||||
|
||||
migrations :: [B.ByteString]
|
||||
migrations = [ $(embedFile "src/SpaceTraders/Database/000_init.sql") ]
|
||||
|
||||
close :: S.Connection -> IO ()
|
||||
close conn = S.close conn
|
||||
close = S.close
|
||||
|
||||
open :: IO S.Connection
|
||||
open = do
|
||||
|
@ -27,7 +27,7 @@ open = do
|
|||
S.execute_ conn "PRAGMA journal_mode = WAL;"
|
||||
S.withTransaction conn $ do
|
||||
version <- getSchemaVersion conn `catch` defaultVersion
|
||||
mapM_ (S.execute_ conn) $ S.Query <$> (filter (/= "\n") . concat . map ((T.splitOn ";") . T.decodeUtf8) $ drop version migrations)
|
||||
mapM_ (S.execute_ conn) $ S.Query <$> concatMap (filter (/= "\n") . T.splitOn ";" . T.decodeUtf8) (drop version migrations)
|
||||
S.execute_ conn "DELETE FROM schema_version;"
|
||||
S.execute conn "INSERT INTO schema_version (version) VALUES (?);" (S.Only $ length migrations)
|
||||
return conn
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue