diff options
Diffstat (limited to 'haskell/src/SpaceTraders/Database.hs')
-rw-r--r-- | haskell/src/SpaceTraders/Database.hs | 18 |
1 files changed, 9 insertions, 9 deletions
diff --git a/haskell/src/SpaceTraders/Database.hs b/haskell/src/SpaceTraders/Database.hs index 66ff893..0166da9 100644 --- a/haskell/src/SpaceTraders/Database.hs +++ b/haskell/src/SpaceTraders/Database.hs @@ -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 |