mirror of
https://github.com/facebook/Haxl.git
synced 2024-12-23 16:53:02 +03:00
c4ca10b6ea
Summary: Add serialization and deserialization functions to non-risk datasources that don't require special treatment (like CacheClient, etc), for use with D2628780 Reviewed By: simonmar Differential Revision: D2645436 fbshipit-source-id: 2777dbfbb11528bc079e2e82e88ebbdc880a8914
45 lines
1.0 KiB
Haskell
45 lines
1.0 KiB
Haskell
{-# LANGUAGE OverloadedStrings #-}
|
|
{-# LANGUAGE GeneralizedNewtypeDeriving #-}
|
|
{-# LANGUAGE DeriveDataTypeable #-}
|
|
|
|
module TestTypes
|
|
( UserEnv
|
|
, Haxl
|
|
, lookupInput
|
|
, Id(..)
|
|
) where
|
|
|
|
import Data.Aeson
|
|
import Data.Binary (Binary)
|
|
import Data.Text (Text)
|
|
import qualified Data.Text as Text
|
|
import qualified Data.HashMap.Strict as HashMap
|
|
import Data.Hashable
|
|
import Data.Typeable
|
|
|
|
import Haxl.Core
|
|
|
|
type UserEnv = Object
|
|
type Haxl a = GenHaxl UserEnv a
|
|
|
|
lookupInput :: FromJSON a => Text -> Haxl a
|
|
lookupInput field = do
|
|
mb_val <- env (HashMap.lookup field . userEnv)
|
|
case mb_val of
|
|
Nothing ->
|
|
throw (NotFound (Text.concat ["field ", field, " was not found."]))
|
|
Just val ->
|
|
case fromJSON val of
|
|
Error str ->
|
|
throw (UnexpectedType (Text.concat
|
|
["field ", field, ": ", Text.pack str]))
|
|
Success a -> return a
|
|
|
|
|
|
newtype Id = Id Int
|
|
deriving (Eq, Ord, Binary, Enum, Num, Integral, Real, Hashable, Typeable,
|
|
ToJSON, FromJSON)
|
|
|
|
instance Show Id where
|
|
show (Id i) = show i
|