diff options
author | Julien Dessaux | 2023-07-18 00:02:33 +0200 |
---|---|---|
committer | Julien Dessaux | 2023-07-18 00:02:33 +0200 |
commit | 4af96da5670566c111fa5c7dac9572eaac021a4b (patch) | |
tree | 74ae00fb63e087437bc3a62152d4b7fecf8c039e /haskell/src/SpaceTraders/Database/Systems.hs | |
parent | [haskell] Trigger a program restart when the api client encounters a server r... (diff) | |
download | spacetraders-4af96da5670566c111fa5c7dac9572eaac021a4b.tar.gz spacetraders-4af96da5670566c111fa5c7dac9572eaac021a4b.tar.bz2 spacetraders-4af96da5670566c111fa5c7dac9572eaac021a4b.zip |
[haskell] Refactored everything with a ReaderT pattern
Diffstat (limited to 'haskell/src/SpaceTraders/Database/Systems.hs')
-rw-r--r-- | haskell/src/SpaceTraders/Database/Systems.hs | 22 |
1 files changed, 13 insertions, 9 deletions
diff --git a/haskell/src/SpaceTraders/Database/Systems.hs b/haskell/src/SpaceTraders/Database/Systems.hs index 5312fd0..f71508e 100644 --- a/haskell/src/SpaceTraders/Database/Systems.hs +++ b/haskell/src/SpaceTraders/Database/Systems.hs @@ -5,19 +5,23 @@ module SpaceTraders.Database.Systems , getSystems ) where +import Control.Monad.Reader import Data.Aeson import Data.Maybe import qualified Database.SQLite.Simple as S -import Data.Text.Encoding (encodeUtf8Builder) -import Data.ByteString.Builder(toLazyByteString) - +import SpaceTraders +import SpaceTraders.Database.Utils import SpaceTraders.Model.System -addSystems :: S.Connection -> [System] -> IO () -addSystems conn systems = S.withTransaction conn $ S.executeMany conn "INSERT INTO systems(data) VALUES (json(?));" $ S.Only <$> map encode systems +addSystems :: (HasDatabaseConn env, MonadIO m, MonadReader env m) => [System] -> m () +addSystems systems = do + env <- ask + let conn = getConn env + liftIO $ S.withTransaction conn $ S.executeMany conn "INSERT INTO systems(data) VALUES (json(?));" $ S.Only <$> map encode systems -getSystems :: S.Connection -> IO [System] -getSystems conn = do - ret <- S.query_ conn "SELECT data from systems;" - return . catMaybes $ map (decode . toLazyByteString . encodeUtf8Builder . head) ret +getSystems :: (HasDatabaseConn env, MonadIO m, MonadReader env m) => m [System] +getSystems = do + env <- ask + ret <- liftIO $ S.query_ (getConn env) "SELECT data from systems;" + return . catMaybes $ map (decodeText . head) ret |