bloomfilter-redis-0.1.0.0: test/Common.hs

{-# LANGUAGE CPP #-}
{-# LANGUAGE Trustworthy #-}

module Common where

#if !MIN_VERSION_base(4,8,0)
import Prelude hiding (mapM, mapM_)
import Data.Traversable (Traversable(..))
#endif
import Data.Hashable
import Database.Redis
import qualified Data.ByteString.Char8 as BS

import Data.RedisBloom
import Data.RedisBloom.Hash

test_key :: Key
test_key = BS.pack "tk"

test_bf :: Bloom Int
test_bf = Bloom test_key sz hb
    where
      hc = HashCount 16
      hb = hashFamilyFNV1a hc :: HashFamily Int
      kb = (*8192)
      sz = Capacity $ kb 32

createAddQuery :: Hashable a => Bloom a -> a -> Redis Bool
createAddQuery b x = createBF b >> addBF b x >> queryBF b x
createAddL :: (Hashable a, Traversable t) => Bloom a -> t a -> Redis ()
createAddL b l = createBF b >> mapM_ (addBF b) l
queryL :: (Hashable a, Traversable t) => Bloom a -> t a -> Redis (t Bool)
queryL b = mapM (queryBF b)
createAddQueryL :: (Hashable a, Traversable t) => Bloom a -> t a -> Redis (t Bool)
createAddQueryL b l = createAddL b l >> queryL b l