|
@@ -1,35 +1,58 @@
|
|
|
-{-# LANGUAGE QuasiQuotes, TemplateHaskell, TypeFamilies, MultiParamTypeClasses, OverloadedStrings #-}
|
|
|
-{-# LANGUAGE GADTs #-}
|
|
|
-{-# LANGUAGE RecordWildCards #-}
|
|
|
-{-# LANGUAGE EmptyDataDecls #-}
|
|
|
-{-# LANGUAGE RankNTypes #-}
|
|
|
-import Yesod hiding (Field)
|
|
|
-import System.Environment (getArgs)
|
|
|
-import qualified Network.Wai.Handler.Warp as Warp
|
|
|
-import Data.Text (Text)
|
|
|
-import Data.Conduit.Pool (Pool)
|
|
|
-import qualified Database.Persist.MySQL as My
|
|
|
-import qualified Database.Persist.MongoDB as Mongo
|
|
|
-import qualified Database.MongoDB as Mongo
|
|
|
-import Database.MongoDB ((=:), Field((:=)))
|
|
|
-import qualified System.Random.MWC as R
|
|
|
-import Control.Monad.Primitive (PrimState)
|
|
|
-import Control.Monad (replicateM)
|
|
|
-import Data.Conduit.Network (bindPort)
|
|
|
-import System.Posix.Process (forkProcess)
|
|
|
-import Control.Monad (replicateM_)
|
|
|
-import Network (PortID (PortNumber))
|
|
|
-import Data.Int (Int64)
|
|
|
-
|
|
|
-mkPersist sqlSettings [persistLowerCase|
|
|
|
+{-# LANGUAGE EmptyDataDecls #-}
|
|
|
+{-# LANGUAGE FlexibleContexts #-}
|
|
|
+{-# LANGUAGE FlexibleInstances #-}
|
|
|
+{-# LANGUAGE GADTs #-}
|
|
|
+{-# LANGUAGE GeneralizedNewtypeDeriving #-}
|
|
|
+{-# LANGUAGE MultiParamTypeClasses #-}
|
|
|
+{-# LANGUAGE OverloadedStrings #-}
|
|
|
+{-# LANGUAGE QuasiQuotes #-}
|
|
|
+{-# LANGUAGE RankNTypes #-}
|
|
|
+{-# LANGUAGE RecordWildCards #-}
|
|
|
+{-# LANGUAGE TemplateHaskell #-}
|
|
|
+{-# LANGUAGE TypeFamilies #-}
|
|
|
+{-# LANGUAGE ViewPatterns #-}
|
|
|
+{-# OPTIONS_GHC -fno-warn-orphans #-}
|
|
|
+module Main (main, resourcesApp, Widget, WorldId) where
|
|
|
+import Blaze.ByteString.Builder
|
|
|
+import Control.Concurrent (runInUnboundThread)
|
|
|
+import Control.Monad (replicateM)
|
|
|
+import Control.Monad.Logger (runNoLoggingT)
|
|
|
+import Control.Monad.Primitive (PrimState)
|
|
|
+import Control.Monad.Reader (ReaderT)
|
|
|
+import Control.Monad.Trans.Resource (InternalState)
|
|
|
+import Data.Aeson (encode)
|
|
|
+import qualified Data.ByteString.Lazy as L
|
|
|
+import Data.Conduit.Pool (Pool, createPool)
|
|
|
+import Data.Int (Int64)
|
|
|
+import Data.IORef (newIORef)
|
|
|
+import Data.Pool (withResource)
|
|
|
+import Data.Text (Text)
|
|
|
+import Database.MongoDB (Field ((:=)), (=:))
|
|
|
+import qualified Database.MongoDB as Mongo
|
|
|
+import Database.Persist (Key, PersistEntity,
|
|
|
+ PersistEntityBackend,
|
|
|
+ PersistStore, get)
|
|
|
+import qualified Database.Persist.MySQL as My
|
|
|
+import Database.Persist.TH (mkPersist, mpsGeneric,
|
|
|
+ persistLowerCase, sqlSettings)
|
|
|
+import Network (PortID (PortNumber))
|
|
|
+import Network.HTTP.Types
|
|
|
+import Network.Wai
|
|
|
+import qualified Network.Wai.Handler.Warp as Warp
|
|
|
+import System.Environment (getArgs)
|
|
|
+import System.IO.Unsafe (unsafePerformIO)
|
|
|
+import qualified System.Random.MWC as R
|
|
|
+import Yesod.Core
|
|
|
+
|
|
|
+mkPersist sqlSettings { mpsGeneric = True } [persistLowerCase|
|
|
|
World sql=World
|
|
|
randomNumber Int sql=randomNumber
|
|
|
|]
|
|
|
|
|
|
data App = App
|
|
|
- { appGen :: R.Gen (PrimState IO)
|
|
|
- , mySqlPool :: Pool My.Connection
|
|
|
- , mongoDBPool :: Pool Mongo.Connection
|
|
|
+ { appGen :: !(R.Gen (PrimState IO))
|
|
|
+ , mySqlPool :: !(Pool My.SqlBackend)
|
|
|
+ , mongoDBPool :: !(Pool Mongo.Pipe)
|
|
|
}
|
|
|
|
|
|
-- | Not actually using the non-raw mongoDB.
|
|
@@ -40,27 +63,41 @@ mkYesod "App" [parseRoutes|
|
|
|
/db DbR GET
|
|
|
/dbs/#Int DbsR GET
|
|
|
|
|
|
-/mongo/db MongoDbR GET
|
|
|
-/mongo/dbs/#Int MongoDbsR GET
|
|
|
-
|
|
|
/mongo/raw/db MongoRawDbR GET
|
|
|
/mongo/raw/dbs/#Int MongoRawDbsR GET
|
|
|
|]
|
|
|
|
|
|
+fakeInternalState :: InternalState
|
|
|
+fakeInternalState = unsafePerformIO $ newIORef $ error "fakeInternalState forced"
|
|
|
+{-# NOINLINE fakeInternalState #-}
|
|
|
+
|
|
|
instance Yesod App where
|
|
|
makeSessionBackend _ = return Nothing
|
|
|
+ {-# INLINE makeSessionBackend #-}
|
|
|
shouldLog _ _ _ = False
|
|
|
+ {-# INLINE shouldLog #-}
|
|
|
yesodMiddleware = id
|
|
|
-
|
|
|
-getJsonR :: Handler Value
|
|
|
-getJsonR = return $ object ["message" .= ("Hello, World!" :: Text)]
|
|
|
+ {-# INLINE yesodMiddleware #-}
|
|
|
+ cleanPath _ = Right
|
|
|
+ {-# INLINE cleanPath #-}
|
|
|
+ yesodWithInternalState _ _ = ($ fakeInternalState)
|
|
|
+ {-# INLINE yesodWithInternalState #-}
|
|
|
+ maximumContentLength _ _ = Nothing
|
|
|
+ {-# INLINE maximumContentLength #-}
|
|
|
+
|
|
|
+getJsonR :: Handler ()
|
|
|
+getJsonR = sendWaiResponse
|
|
|
+ $ responseBuilder
|
|
|
+ status200
|
|
|
+ [("Content-Type", typeJson)]
|
|
|
+ $ copyByteString
|
|
|
+ $ L.toStrict
|
|
|
+ $ encode
|
|
|
+ $ object ["message" .= ("Hello, World!" :: Text)]
|
|
|
|
|
|
|
|
|
getDbR :: Handler Value
|
|
|
-getDbR = getDb (intQuery runMySQL )
|
|
|
-
|
|
|
-getMongoDbR :: Handler Value
|
|
|
-getMongoDbR = getDb (intQuery runMongoDB )
|
|
|
+getDbR = getDb (intQuery runMySQL My.toSqlKey)
|
|
|
|
|
|
getMongoRawDbR :: Handler Value
|
|
|
getMongoRawDbR = getDb rawMongoIntQuery
|
|
@@ -68,10 +105,7 @@ getMongoRawDbR = getDb rawMongoIntQuery
|
|
|
getDbsR :: Int -> Handler Value
|
|
|
getDbsR cnt = do
|
|
|
App {..} <- getYesod
|
|
|
- multiRandomHandler (intQuery runMySQL) cnt
|
|
|
-
|
|
|
-getMongoDbsR :: Int -> Handler Value
|
|
|
-getMongoDbsR cnt = multiRandomHandler (intQuery runMongoDB) cnt
|
|
|
+ multiRandomHandler (intQuery runMySQL My.toSqlKey) cnt
|
|
|
|
|
|
getMongoRawDbsR :: Int -> Handler Value
|
|
|
getMongoRawDbsR cnt = multiRandomHandler rawMongoIntQuery cnt
|
|
@@ -84,26 +118,35 @@ getDb :: (Int64 -> Handler Value) -> Handler Value
|
|
|
getDb query = do
|
|
|
app <- getYesod
|
|
|
i <- liftIO (randomNumber (appGen app))
|
|
|
- query i
|
|
|
+ value <- query i
|
|
|
+ sendWaiResponse
|
|
|
+ $ responseBuilder
|
|
|
+ status200
|
|
|
+ [("Content-Type", typeJson)]
|
|
|
+ $ copyByteString
|
|
|
+ $ L.toStrict
|
|
|
+ $ encode value
|
|
|
|
|
|
|
|
|
runMongoDB :: Mongo.Action Handler b -> Handler b
|
|
|
runMongoDB f = do
|
|
|
App {..} <- getYesod
|
|
|
- Mongo.runMongoDBPoolDef f mongoDBPool
|
|
|
+ withResource mongoDBPool $ \pipe ->
|
|
|
+ Mongo.access pipe Mongo.ReadStaleOk "hello_world" f
|
|
|
|
|
|
runMySQL :: My.SqlPersistT Handler b -> Handler b
|
|
|
runMySQL f = do
|
|
|
App {..} <- getYesod
|
|
|
My.runSqlPool f mySqlPool
|
|
|
|
|
|
-intQuery :: forall (m :: * -> *) (m1 :: * -> *) val backend.
|
|
|
- (Monad m, PersistEntity val, PersistStore m1,
|
|
|
- PersistEntityBackend val ~ PersistMonadBackend m1) =>
|
|
|
- (m1 (Maybe val) -> m (Maybe (WorldGeneric backend)))
|
|
|
+intQuery :: (MonadIO m, PersistEntity val, PersistStore backend
|
|
|
+ , backend ~ PersistEntityBackend val
|
|
|
+ ) =>
|
|
|
+ (ReaderT backend m (Maybe val) -> m (Maybe (WorldGeneric backend)))
|
|
|
+ -> (Int64 -> Key val)
|
|
|
-> Int64 -> m Value
|
|
|
-intQuery db i = do
|
|
|
- Just x <- db $ get (Key $ PersistInt64 i)
|
|
|
+intQuery db toKey i = do
|
|
|
+ Just x <- db $ get $ toKey i
|
|
|
return $ jsonResult (worldRandomNumber x)
|
|
|
where
|
|
|
jsonResult :: Int -> Value
|
|
@@ -111,7 +154,7 @@ intQuery db i = do
|
|
|
|
|
|
rawMongoIntQuery :: Mongo.Val v => v -> Handler Value
|
|
|
rawMongoIntQuery i = do
|
|
|
- Just x <- runMongoDB $ Mongo.findOne (Mongo.select ["id" =: i] "world")
|
|
|
+ Just x <- runMongoDB $ Mongo.findOne (Mongo.select ["id" =: i] "World")
|
|
|
return $ documentToJson x
|
|
|
|
|
|
multiRandomHandler :: ToJSON a
|
|
@@ -141,30 +184,29 @@ instance ToJSON Mongo.Value where
|
|
|
|
|
|
main :: IO ()
|
|
|
main = R.withSystemRandom $ \gen -> do
|
|
|
- socket <- bindPort 8000 "*"
|
|
|
[cores, host] <- getArgs
|
|
|
- myPool <- My.createMySQLPool My.defaultConnectInfo
|
|
|
+ myPool <- runNoLoggingT $ My.createMySQLPool My.defaultConnectInfo
|
|
|
{ My.connectUser = "benchmarkdbuser"
|
|
|
, My.connectPassword = "benchmarkdbpass"
|
|
|
, My.connectDatabase = "hello_world"
|
|
|
, My.connectHost = host
|
|
|
} 1000
|
|
|
|
|
|
- mongoPool <- Mongo.createMongoDBPool "hello_world" host (PortNumber 27017)
|
|
|
- (Just (Mongo.MongoAuth "benchmarkdbuser" "benchmarkdbpass"))
|
|
|
- 1 -- what is the optimal stripe count? 1 is said to be a good default
|
|
|
- 1000
|
|
|
- 3 -- 3 second timeout
|
|
|
+ mongoPool <- createPool
|
|
|
+ (Mongo.connect $ Mongo.Host host $ PortNumber 27017)
|
|
|
+ Mongo.close
|
|
|
+ (read cores) -- what is the optimal stripe count? 1 is said to be a good default
|
|
|
+ 3 -- 3 second timeout
|
|
|
+ 1000
|
|
|
|
|
|
app <- toWaiAppPlain App
|
|
|
{ appGen = gen
|
|
|
, mySqlPool = myPool
|
|
|
, mongoDBPool = mongoPool
|
|
|
}
|
|
|
- let run = Warp.runSettingsSocket Warp.defaultSettings
|
|
|
- { Warp.settingsPort = 8000
|
|
|
- , Warp.settingsHost = "*"
|
|
|
- , Warp.settingsOnException = const $ return ()
|
|
|
- } socket app
|
|
|
- replicateM_ (read cores - 1) $ forkProcess run
|
|
|
- run
|
|
|
+ runInUnboundThread $ Warp.runSettings
|
|
|
+ ( Warp.setPort 8000
|
|
|
+ $ Warp.setHost "*"
|
|
|
+ $ Warp.setOnException (\_ _ -> return ())
|
|
|
+ Warp.defaultSettings
|
|
|
+ ) app
|