56 lines
1.4 KiB
Haskell
56 lines
1.4 KiB
Haskell
module TestTypes (
|
|
IncPK(..)
|
|
, CompoundPK(..)
|
|
, incFromList
|
|
, compoundFromList
|
|
) where
|
|
|
|
import qualified Data.Aeson as JSON
|
|
import Data.Aeson ((.:))
|
|
import Data.Maybe (fromJust)
|
|
import Control.Applicative ((<$>), (<*>))
|
|
import Control.Monad (mzero)
|
|
|
|
import Database.HDBC (SqlValue, fromSql)
|
|
|
|
data IncPK = IncPK {
|
|
incId :: Int
|
|
, incNullableStr :: Maybe String
|
|
, incStr :: String
|
|
, incInsert :: String
|
|
} deriving (Eq, Show)
|
|
|
|
instance JSON.FromJSON IncPK where
|
|
parseJSON (JSON.Object r) = IncPK <$>
|
|
r .: "id" <*>
|
|
r .: "nullable_string" <*>
|
|
r .: "non_nullable_string" <*>
|
|
r .: "inserted_at"
|
|
parseJSON _ = mzero
|
|
|
|
incFromList :: [(String, SqlValue)] -> IncPK
|
|
incFromList row = IncPK
|
|
(fromSql . fromJust $ lookup "id" row)
|
|
(fromSql . fromJust $ lookup "nullable_string" row)
|
|
(fromSql . fromJust $ lookup "non_nullable_string" row)
|
|
(fromSql . fromJust $ lookup "inserted_at" row)
|
|
|
|
data CompoundPK = CompoundPK {
|
|
compoundK1 :: Int
|
|
, compoundK2 :: Int
|
|
, compoundExtra :: Maybe Int
|
|
}
|
|
|
|
instance JSON.FromJSON CompoundPK where
|
|
parseJSON (JSON.Object r) = CompoundPK <$>
|
|
r .: "k1" <*>
|
|
r .: "k2" <*>
|
|
r .: "extra"
|
|
parseJSON _ = mzero
|
|
|
|
compoundFromList :: [(String, SqlValue)] -> CompoundPK
|
|
compoundFromList row = CompoundPK
|
|
(fromSql . fromJust $ lookup "k1" row)
|
|
(fromSql . fromJust $ lookup "k2" row)
|
|
(fromSql . fromJust $ lookup "extra" row)
|