RethinkBenchmark.hs

{-# LANGUAGE OverloadedStrings #-}

module Main where


import           Control.Monad

import           Data.Text (Text)
import           Data.Aeson.Types (Value)

import qualified Database.RethinkDB as R

import           Criterion.Main


createRethinkHandle :: IO R.RethinkDBHandle
createRethinkHandle = do
    handle <- R.connect "localhost" 28015 Nothing
    return $ R.use handle (R.db "test")


rawLookup :: R.RethinkDBHandle -> IO ()
rawLookup handle = do
    void (R.run handle $ R.get key table :: IO (Maybe Value))
  where
    table = R.table "snapshots"
    key   = R.expr ("0ovUatO5oD@5" :: Text)


main :: IO ()
main = do
    handle <- createRethinkHandle
    defaultMain [ bench "lookup" $ nfIO $ rawLookup handle ]