haskoin-store-0.40.3: src/Haskoin/Store/Cache.hs
{-# LANGUAGE ApplicativeDo #-}
{-# LANGUAGE DeriveAnyClass #-}
{-# LANGUAGE DeriveGeneric #-}
{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE LambdaCase #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE TemplateHaskell #-}
{-# LANGUAGE TupleSections #-}
module Haskoin.Store.Cache
( CacheConfig(..)
, CacheT
, CacheError(..)
, withCache
, connectRedis
, blockRefScore
, scoreBlockRef
, CacheWriter
, CacheWriterInbox
, cacheNewBlock
, cachePing
, cacheWriter
, isInCache
, evictFromCache
) where
import Control.DeepSeq (NFData)
import Control.Monad (forM, forM_, forever, unless, void,
when)
import Control.Monad.Logger (MonadLoggerIO, logDebugS, logErrorS,
logWarnS)
import Control.Monad.Reader (ReaderT (..), ask, asks)
import Control.Monad.Trans (lift)
import Control.Monad.Trans.Maybe (MaybeT (..), runMaybeT)
import Data.Bits (shift, (.&.), (.|.))
import Data.ByteString (ByteString)
import qualified Data.ByteString.Short as BSS
import Data.Default (def)
import Data.Either (rights)
import Data.HashMap.Strict (HashMap)
import qualified Data.HashMap.Strict as HashMap
import qualified Data.HashSet as HashSet
import qualified Data.IntMap.Strict as I
import Data.List (sort)
import qualified Data.Map.Strict as Map
import Data.Maybe (catMaybes, mapMaybe)
import Data.Serialize (Serialize, decode, encode)
import Data.String.Conversions (cs)
import Data.Text (Text)
import Data.Time.Clock.System (getSystemTime, systemSeconds)
import Data.Word (Word32, Word64)
import Database.Redis (Connection, Redis, RedisCtx, Reply,
checkedConnect, defaultConnectInfo,
hgetall, parseConnectInfo, zadd,
zrangeWithscores,
zrangebyscoreWithscoresLimit, zrem)
import qualified Database.Redis as Redis
import GHC.Generics (Generic)
import Haskoin (Address, BlockHash,
BlockHeader (..), BlockNode (..),
DerivPathI (..), KeyIndex,
OutPoint (..), Tx (..), TxHash,
TxIn (..), TxOut (..), XPubKey,
blockHashToHex, derivePubPath,
eitherToMaybe, headerHash,
pathToList, scriptToAddressBS,
txHash, xPubAddr,
xPubCompatWitnessAddr, xPubExport,
xPubWitnessAddr)
import Haskoin.Node (Chain, chainBlockMain,
chainGetAncestor, chainGetBest,
chainGetBlock)
import Haskoin.Store.Common
import Haskoin.Store.Data
import NQE (Inbox, Mailbox, receive, send)
import System.Random (randomIO)
import UnliftIO (Exception, MonadIO, MonadUnliftIO,
bracket, liftIO, throwIO)
runRedis :: MonadLoggerIO m => Redis (Either Reply a) -> CacheX m a
runRedis action =
asks cacheConn >>= \conn ->
liftIO (Redis.runRedis conn action) >>= \case
Right x -> return x
Left e -> do
$(logErrorS) "Cache" $ "Got error from Redis: " <> cs (show e)
throwIO (RedisError e)
data CacheConfig = CacheConfig
{ cacheConn :: !Connection
, cacheMin :: !Int
, cacheMax :: !Integer
, cacheChain :: !Chain
, cacheRefresh :: !Int
}
type CacheT = ReaderT (Maybe CacheConfig)
type CacheX = ReaderT CacheConfig
data CacheError = RedisError Reply
| RedisTxError !String
| LogicError !String
deriving (Show, Eq, Generic, NFData, Exception)
connectRedis :: MonadIO m => String -> m Connection
connectRedis redisurl = do
conninfo <-
if null redisurl
then return defaultConnectInfo
else case parseConnectInfo redisurl of
Left e -> error e
Right r -> return r
liftIO (checkedConnect conninfo)
instance (MonadUnliftIO m, MonadLoggerIO m, StoreReadBase m) =>
StoreReadBase (CacheT m) where
getNetwork = lift getNetwork
getBestBlock = lift getBestBlock
getBlocksAtHeight = lift . getBlocksAtHeight
getBlock = lift . getBlock
getTxData = lift . getTxData
getSpender = lift . getSpender
getBalance = lift . getBalance
getUnspent = lift . getUnspent
getMempool = lift getMempool
instance (MonadUnliftIO m , MonadLoggerIO m, StoreReadExtra m) =>
StoreReadExtra (CacheT m) where
getBalances = lift . getBalances
getAddressesTxs addrs = lift . getAddressesTxs addrs
getAddressTxs addr = lift . getAddressTxs addr
getAddressUnspents addr = lift . getAddressUnspents addr
getAddressesUnspents addrs = lift . getAddressesUnspents addrs
xPubBals xpub =
ask >>= \case
Nothing -> lift (xPubBals xpub)
Just cfg -> lift (runReaderT (getXPubBalances xpub) cfg)
xPubUnspents xpub limits =
ask >>= \case
Nothing -> lift (xPubUnspents xpub limits)
Just cfg -> lift (runReaderT (getXPubUnspents xpub limits) cfg)
xPubTxs xpub limits =
ask >>= \case
Nothing -> lift (xPubTxs xpub limits)
Just cfg -> lift (runReaderT (getXPubTxs xpub limits) cfg)
getMaxGap = lift getMaxGap
getInitialGap = lift getInitialGap
withCache :: StoreReadBase m => Maybe CacheConfig -> CacheT m a -> m a
withCache s f = runReaderT f s
balancesPfx :: ByteString
balancesPfx = "b"
txSetPfx :: ByteString
txSetPfx = "t"
utxoPfx :: ByteString
utxoPfx = "u"
getXPubTxs ::
(MonadUnliftIO m, MonadLoggerIO m, StoreReadExtra m)
=> XPubSpec -> Limits -> CacheX m [TxRef]
getXPubTxs xpub limits = do
xpubtxt <- xpubText xpub
$(logDebugS) "Cache" $ "Getting xpub txs for " <> xpubtxt
isXPubCached xpub >>= \case
True -> do
txs <- cacheGetXPubTxs xpub limits
$(logDebugS) "Cache" $
"Returning " <> cs (show (length txs)) <>
" transactions for cached xpub: " <>
xpubtxt
return txs
False -> do
$(logDebugS) "Cache" $ "Caching new xpub " <> xpubtxt
newXPubC xpub >>= \(t, bals) ->
if t
then do
$(logDebugS) "Cache" $
"Successfully cached xpub " <> xpubtxt
cacheGetXPubTxs xpub limits
else do
$(logDebugS) "Cache" $
"Using DB to return txs for xpub " <> xpubtxt
lift $ xPubBalsTxs bals limits
getXPubUnspents ::
(MonadUnliftIO m, MonadLoggerIO m, StoreReadExtra m)
=> XPubSpec -> Limits -> CacheX m [XPubUnspent]
getXPubUnspents xpub limits = do
xpubtxt <- xpubText xpub
$(logDebugS) "Cache" $ "Getting utxo for xpub " <> xpubtxt
isXPubCached xpub >>= \case
True -> do
bals <- cacheGetXPubBalances xpub
process bals
False ->
newXPubC xpub >>= \(t, bals) ->
if t
then do
$(logDebugS) "Cache" $
"Successfully cached xpub " <> xpubtxt
process bals
else do
$(logDebugS) "Cache" $
"Using DB to return utxo for xpub " <> xpubtxt
lift $ xPubBalsUnspents bals limits
where
process bals = do
ops <- map snd <$> cacheGetXPubUnspents xpub limits
uns <- catMaybes <$> lift (mapM getUnspent ops)
let addrmap =
Map.fromList $
map (\b -> (balanceAddress (xPubBal b), xPubBalPath b)) bals
addrutxo =
mapMaybe
(\u ->
either
(const Nothing)
(\a -> Just (a, u))
(scriptToAddressBS
(BSS.fromShort (unspentScript u))))
uns
xpubutxo =
mapMaybe
(\(a, u) -> (`XPubUnspent` u) <$> Map.lookup a addrmap)
addrutxo
xpubtxt <- xpubText xpub
$(logDebugS) "Cache" $
"Returning " <> cs (show (length xpubutxo)) <>
" utxos for cached xpub: " <>
xpubtxt
return xpubutxo
getXPubBalances ::
(MonadUnliftIO m, MonadLoggerIO m, StoreReadExtra m)
=> XPubSpec -> CacheX m [XPubBal]
getXPubBalances xpub = do
xpubtxt <- xpubText xpub
$(logDebugS) "Cache" $ "Getting balances for xpub " <> xpubtxt
isXPubCached xpub >>= \case
True -> do
bals <- cacheGetXPubBalances xpub
$(logDebugS) "Cache" $
"Returning " <> cs (show (length bals)) <> " balances for xpub " <>
xpubtxt
return bals
False -> do
$(logDebugS) "Cache" $ "Caching balances for xpub " <> xpubtxt
snd <$> newXPubC xpub
isInCache :: MonadLoggerIO m => XPubSpec -> CacheT m Bool
isInCache xpub =
ask >>= \case
Nothing -> return False
Just cfg -> runReaderT (isXPubCached xpub) cfg
isXPubCached :: MonadLoggerIO m => XPubSpec -> CacheX m Bool
isXPubCached = runRedis . redisIsXPubCached
redisIsXPubCached :: RedisCtx m f => XPubSpec -> m (f Bool)
redisIsXPubCached xpub = Redis.exists (balancesPfx <> encode xpub)
cacheGetXPubBalances :: MonadLoggerIO m => XPubSpec -> CacheX m [XPubBal]
cacheGetXPubBalances xpub = do
bals <- runRedis $ redisGetXPubBalances xpub
touchKeys [xpub]
return bals
cacheGetXPubTxs ::
(StoreReadBase m, MonadLoggerIO m)
=> XPubSpec
-> Limits
-> CacheX m [TxRef]
cacheGetXPubTxs xpub limits = do
score <-
case start limits of
Nothing -> return Nothing
Just (AtTx th) ->
lift (getTxData th) >>= \case
Just TxData {txDataBlock = b@BlockRef {}} ->
return (Just (blockRefScore b))
_ -> return Nothing
Just (AtBlock h) ->
return (Just (blockRefScore (BlockRef h maxBound)))
xs <-
runRedis $
getFromSortedSet
(txSetPfx <> encode xpub)
score
(offset limits)
(limit limits)
touchKeys [xpub]
return $ map (uncurry f) xs
where
f t s = TxRef {txRefHash = t, txRefBlock = scoreBlockRef s}
cacheGetXPubUnspents ::
(StoreReadBase m, MonadLoggerIO m)
=> XPubSpec
-> Limits
-> CacheX m [(BlockRef, OutPoint)]
cacheGetXPubUnspents xpub limits = do
score <-
case start limits of
Nothing -> return Nothing
Just (AtTx th) ->
lift (getTxData th) >>= \case
Just TxData {txDataBlock = b@BlockRef {}} ->
return (Just (blockRefScore b))
_ -> return Nothing
Just (AtBlock h) ->
return (Just (blockRefScore (BlockRef h maxBound)))
xs <-
runRedis $
getFromSortedSet
(utxoPfx <> encode xpub)
score
(offset limits)
(limit limits)
touchKeys [xpub]
return $ map (uncurry f) xs
where
f o s = (scoreBlockRef s, o)
redisGetXPubBalances :: (Functor f, RedisCtx m f) => XPubSpec -> m (f [XPubBal])
redisGetXPubBalances xpub =
fmap (sort . map (uncurry f)) <$> getAllFromMap (balancesPfx <> encode xpub)
where
f p b = XPubBal {xPubBalPath = p, xPubBal = b}
blockRefScore :: BlockRef -> Double
blockRefScore BlockRef {blockRefHeight = h, blockRefPos = p} =
fromIntegral (0x001fffffffffffff - (h' .|. p'))
where
h' = (fromIntegral h .&. 0x07ffffff) `shift` 26 :: Word64
p' = (fromIntegral p .&. 0x03ffffff) :: Word64
blockRefScore MemRef {memRefTime = t} = negate t'
where
t' = fromIntegral (t .&. 0x001fffffffffffff)
scoreBlockRef :: Double -> BlockRef
scoreBlockRef s
| s < 0 = MemRef {memRefTime = n}
| otherwise = BlockRef {blockRefHeight = h, blockRefPos = p}
where
n = truncate (abs s) :: Word64
m = 0x001fffffffffffff - n
h = fromIntegral (m `shift` (-26))
p = fromIntegral (m .&. 0x03ffffff)
getFromSortedSet ::
(Applicative f, RedisCtx m f, Serialize a)
=> ByteString
-> Maybe Double
-> Word32
-> Word32
-> m (f [(a, Double)])
getFromSortedSet key Nothing off 0 = do
xs <- zrangeWithscores key (fromIntegral off) (-1)
return $ do
ys <- map (\(x, s) -> (, s) <$> decode x) <$> xs
return (rights ys)
getFromSortedSet key Nothing off count = do
xs <-
zrangeWithscores
key
(fromIntegral off)
(fromIntegral off + fromIntegral count - 1)
return $ do
ys <- map (\(x, s) -> (, s) <$> decode x) <$> xs
return (rights ys)
getFromSortedSet key (Just score) off 0 = do
xs <- zrangebyscoreWithscoresLimit key score (1 / 0) (fromIntegral off) (-1)
return $ do
ys <- map (\(x, s) -> (, s) <$> decode x) <$> xs
return (rights ys)
getFromSortedSet key (Just score) off count = do
xs <-
zrangebyscoreWithscoresLimit
key
score
(1 / 0)
(fromIntegral off)
(fromIntegral count)
return $ do
ys <- map (\(x, s) -> (, s) <$> decode x) <$> xs
return (rights ys)
getAllFromMap ::
(Functor f, RedisCtx m f, Serialize k, Serialize v)
=> ByteString
-> m (f [(k, v)])
getAllFromMap n = do
fxs <- hgetall n
return $ do
xs <- fxs
return
[ (k, v)
| (k', v') <- xs
, let Right k = decode k'
, let Right v = decode v'
]
data CacheWriterMessage
= CacheNewBlock
| CachePing
type CacheWriterInbox = Inbox CacheWriterMessage
type CacheWriter = Mailbox CacheWriterMessage
data AddressXPub = AddressXPub
{ addressXPubSpec :: !XPubSpec
, addressXPubPath :: ![KeyIndex]
}
deriving (Show, Eq, Generic, NFData, Serialize)
mempoolSetKey :: ByteString
mempoolSetKey = "mempool"
addrPfx :: ByteString
addrPfx = "a"
bestBlockKey :: ByteString
bestBlockKey = "head"
maxKey :: ByteString
maxKey = "max"
xPubAddrFunction :: DeriveType -> XPubKey -> Address
xPubAddrFunction DeriveNormal = xPubAddr
xPubAddrFunction DeriveP2SH = xPubCompatWitnessAddr
xPubAddrFunction DeriveP2WPKH = xPubWitnessAddr
cacheWriter ::
(MonadUnliftIO m, MonadLoggerIO m, StoreReadExtra m)
=> CacheConfig -> CacheWriterInbox -> m ()
cacheWriter cfg inbox = runReaderT go cfg
where
go = do
newBlockC
forever $ do
x <- receive inbox
cacheWriterReact x
lockIt :: MonadLoggerIO m => CacheX m (Maybe Word32)
lockIt = do
rnd <- liftIO randomIO
go rnd >>= \case
Right Redis.Ok -> do
$(logDebugS) "Cache" $
"Acquired lock with value " <> cs (show rnd)
return (Just rnd)
Right Redis.Pong -> do
$(logErrorS) "Cache"
"Unexpected pong when acquiring lock"
return Nothing
Right (Redis.Status s) -> do
$(logErrorS) "Cache" $
"Unexpected status acquiring lock: " <> cs s
return Nothing
Left (Redis.Bulk Nothing) -> return Nothing
Left e -> do
$(logErrorS) "Cache"
"Error when trying to acquire lock"
throwIO (RedisError e)
where
go rnd = do
conn <- asks cacheConn
liftIO . Redis.runRedis conn $ do
let opts =
Redis.SetOpts
{ Redis.setSeconds = Just 300
, Redis.setMilliseconds = Nothing
, Redis.setCondition = Just Redis.Nx
}
Redis.setOpts "lock" (cs (show rnd)) opts
unlockIt :: MonadLoggerIO m => Maybe Word32 -> CacheX m ()
unlockIt Nothing = return ()
unlockIt (Just i) =
runRedis (Redis.get "lock") >>= \case
Nothing ->
$(logErrorS) "Cache" $
"Not releasing lock with value " <> cs (show i) <>
": not locked"
Just bs ->
if read (cs bs) == i
then do
void $ runRedis (Redis.del ["lock"])
$(logDebugS) "Cache" $
"Released lock with value " <>
cs (show i)
else
$(logErrorS) "Cache" $
"Could not release lock: value is not " <>
cs (show i)
withLock ::
(MonadLoggerIO m, MonadUnliftIO m)
=> CacheX m a
-> CacheX m (Maybe a)
withLock f =
bracket lockIt unlockIt $ \case
Just _ -> Just <$> f
Nothing -> return Nothing
pruneDB :: (MonadUnliftIO m, MonadLoggerIO m, StoreReadBase m)
=> CacheX m Integer
pruneDB = do
x <- asks cacheMax
s <- runRedis Redis.dbsize
if s > x then flush (s - x) else return 0
where
flush n =
case min 1000 (n `div` 64) of
0 -> return 0
x -> do
m <- withLock $ do
ks <- fmap (map fst) . runRedis $
getFromSortedSet maxKey Nothing 0 (fromIntegral x)
$(logDebugS) "Cache" $
"Pruning " <> cs (show (length ks)) <> " old xpubs"
delXPubKeys ks
case m of
Nothing -> return 0
Just y -> return y
touchKeys :: MonadLoggerIO m => [XPubSpec] -> CacheX m ()
touchKeys xpubs = do
now <- systemSeconds <$> liftIO getSystemTime
runRedis $ redisTouchKeys now xpubs
redisTouchKeys :: (Monad f, RedisCtx m f, Real a) => a -> [XPubSpec] -> m (f ())
redisTouchKeys _ [] = return $ return ()
redisTouchKeys now xpubs =
void <$> Redis.zadd maxKey (map ((realToFrac now, ) . encode) xpubs)
cacheWriterReact ::
(MonadUnliftIO m, MonadLoggerIO m, StoreReadExtra m)
=> CacheWriterMessage -> CacheX m ()
cacheWriterReact CacheNewBlock =
inSync >>= \s ->
when s $ do
newBlockC
syncMempoolC
cacheWriterReact CachePing =
inSync >>= \s ->
when s $ do
pruneDB
newBlockC
syncMempoolC
lenNotNull :: [XPubBal] -> Int
lenNotNull bals = length $ filter (not . nullBalance . xPubBal) bals
newXPubC ::
(MonadUnliftIO m, MonadLoggerIO m, StoreReadExtra m)
=> XPubSpec -> CacheX m (Bool, [XPubBal])
newXPubC xpub = do
bals <- lift $ xPubBals xpub
x <- asks cacheMin
t <- xpubText xpub
let n = lenNotNull bals
if x <= n
then inSync >>= \s -> if s then go bals else return (False, bals)
else do
$(logDebugS) "Cache" $ "Not caching xpub: " <> t
return (False, bals)
where
op XPubUnspent {xPubUnspent = u} = (unspentPoint u, unspentBlock u)
go bals = do
xpubtxt <- xpubText xpub
$(logDebugS) "Cache" $
"Caching xpub with " <> cs (show (length bals)) <>
" addresses (used: " <>
cs (show (lenNotNull bals)) <>
"): " <>
xpubtxt
utxo <- lift $ xPubUnspents xpub def
$(logDebugS) "Cache" $
"Caching xpub with " <> cs (show (length utxo)) <> " utxos: " <>
xpubtxt
xtxs <- lift $ xPubTxs xpub def
$(logDebugS) "Cache" $
"Caching xpub with " <> cs (show (length xtxs)) <> " txs: " <>
xpubtxt
now <- systemSeconds <$> liftIO getSystemTime
$(logDebugS) "Cache" $
"Running Redis pipeline to cache xpub: " <> xpubtxt <> ""
runRedis $ do
b <- redisTouchKeys now [xpub]
c <- redisAddXPubBalances xpub bals
d <- redisAddXPubUnspents xpub (map op utxo)
e <- redisAddXPubTxs xpub xtxs
return $ b >> c >> d >> e >> return ()
$(logDebugS) "Cache" $ "Done caching xpub: " <> xpubtxt
return (True, bals)
inSync :: (MonadUnliftIO m, MonadLoggerIO m, StoreReadExtra m)
=> CacheX m Bool
inSync =
lift getBestBlock >>= \case
Nothing -> return False
Just bb ->
asks cacheChain >>= \ch ->
chainGetBest ch >>= \cb ->
return $ headerHash (nodeHeader cb) == bb
newBlockC :: (MonadUnliftIO m, MonadLoggerIO m, StoreReadExtra m)
=> CacheX m ()
newBlockC =
lift getBestBlock >>= \m ->
forM_ m $ \bb -> void $ withLock $ f bb
where
f bb = cacheGetHead >>= go bb
go bb Nothing =
importBlockC bb
go bb (Just cb)
| cb == bb =
$(logDebugS) "Cache" "Cache in sync"
| otherwise =
sync bb cb
sync bb cb =
asks cacheChain >>= \ch ->
chainBlockMain bb ch >>= \case
False ->
$(logErrorS) "Cache" $
"New head not in main chain: " <> blockHashToHex bb
True ->
chainGetBlock cb ch >>= \case
Nothing ->
$(logErrorS) "Cache" $
"Cache head block node not found: " <>
blockHashToHex cb
Just cn ->
chainBlockMain cb ch >>= \case
False -> do
$(logDebugS) "Cache" $
"Reverting cache head not in main chain: " <>
blockHashToHex cb
removeHeadC >> f bb
True ->
chainGetBlock bb ch >>= \case
Just nn -> next bb nn cn
Nothing -> do
$(logErrorS) "Cache" $
"Cache head node not found: "
<> blockHashToHex bb
throwIO $
LogicError $
"Cache head node not found: "
<> cs (blockHashToHex bb)
next bb nn cn =
asks cacheChain >>= \ch ->
chainGetAncestor (nodeHeight cn + 1) nn ch >>= \case
Nothing ->
$(logWarnS) "Cache" $
"Ancestor not found at height "
<> cs (show (nodeHeight cn + 1))
<> " for block: "
<> blockHashToHex (headerHash (nodeHeader nn))
Just cn' ->
importBlockC (headerHash (nodeHeader cn')) >> f bb
importBlockC :: (MonadUnliftIO m, StoreReadExtra m, MonadLoggerIO m)
=> BlockHash -> CacheX m ()
importBlockC bh =
lift (getBlock bh) >>= \case
Just bd -> do
let ths = blockDataTxs bd
tds <- sortTxData . catMaybes <$> mapM (lift . getTxData) ths
$(logDebugS) "Cache" $
"Importing " <> cs (show (length tds)) <>
" transactions from block " <>
blockHashToHex bh
importMultiTxC tds
$(logDebugS) "Cache" $
"Done importing " <> cs (show (length tds)) <>
" transactions from block " <>
blockHashToHex bh
cacheSetHead bh
Nothing -> do
$(logErrorS) "Cache" $
"Could not get block: "
<> blockHashToHex bh
throwIO . LogicError . cs $
"Could not get block: "
<> blockHashToHex bh
removeHeadC :: (StoreReadExtra m, MonadUnliftIO m, MonadLoggerIO m)
=> CacheX m ()
removeHeadC =
void . runMaybeT $ do
bh <- MaybeT cacheGetHead
bd <- MaybeT (lift (getBlock bh))
lift $ do
tds <- sortTxData . catMaybes <$>
mapM (lift . getTxData) (blockDataTxs bd)
$(logDebugS) "Cache" $ "Reverting head: " <> blockHashToHex bh
importMultiTxC tds
$(logWarnS) "Cache" $
"Reverted block head "
<> blockHashToHex bh
<> " to parent "
<> blockHashToHex (prevBlock (blockDataHeader bd))
cacheSetHead (prevBlock (blockDataHeader bd))
importMultiTxC ::
(MonadUnliftIO m, StoreReadExtra m, MonadLoggerIO m)
=> [TxData] -> CacheX m ()
importMultiTxC txs = do
$(logDebugS) "Cache" $ "Processing " <> cs (show (length txs)) <> " txs"
$(logDebugS) "Cache" $
"Getting address information for "
<> cs (show (length alladdrs))
<> " addresses"
addrmap <- getaddrmap
let addrs = HashMap.keys addrmap
$(logDebugS) "Cache" $
"Getting balances for "
<> cs (show (HashMap.size addrmap))
<> " addresses"
balmap <- getbalances addrs
$(logDebugS) "Cache" $
"Getting unspent data for "
<> cs (show (length allops))
<> " outputs"
unspentmap <- getunspents
gap <- lift getMaxGap
now <- systemSeconds <$> liftIO getSystemTime
let xpubs = allxpubsls addrmap
forM_ (zip [(1 :: Int) ..] xpubs) $ \(i, xpub) -> do
xpubtxt <- xpubText xpub
$(logDebugS) "Cache" $
"Affected xpub "
<> cs (show i) <> "/" <> cs (show (length xpubs))
<> ": " <> xpubtxt
addrs' <- do
$(logDebugS) "Cache" $
"Getting xpub balances for "
<> cs (show (length xpubs)) <> " xpubs"
xmap <- getxbals xpubs
let addrmap' = faddrmap (HashMap.keysSet xmap) addrmap
$(logDebugS) "Cache" "Starting Redis import pipeline"
runRedis $ do
x <- redisImportMultiTx addrmap' unspentmap txs
y <- redisUpdateBalances addrmap' balmap
z <- redisTouchKeys now (HashMap.keys xmap)
return $ x >> y >> z >> return ()
$(logDebugS) "Cache" "Completed Redis pipeline"
return $ getNewAddrs gap xmap (HashMap.elems addrmap')
cacheAddAddresses addrs'
where
alladdrsls = HashSet.toList alladdrs
faddrmap xmap = HashMap.filter (\a -> addressXPubSpec a `elem` xmap)
getaddrmap =
HashMap.fromList . catMaybes . zipWith (\a -> fmap (a, )) alladdrsls <$>
cacheGetAddrsInfo alladdrsls
getunspents =
HashMap.fromList . catMaybes . zipWith (\p -> fmap (p, )) allops <$>
lift (mapM getUnspent allops)
getbalances addrs =
HashMap.fromList . zip addrs <$> mapM (lift . getDefaultBalance) addrs
getxbals xpubs = do
bals <- runRedis . fmap sequence . forM xpubs $ \xpub -> do
bs <- redisGetXPubBalances xpub
return $ (xpub, ) <$> bs
return $ HashMap.filter (not . null) (HashMap.fromList bals)
allops = map snd $ concatMap txInputs txs <> concatMap txOutputs txs
alladdrs =
HashSet.fromList . map fst $
concatMap txInputs txs <> concatMap txOutputs txs
allxpubsls addrmap = HashSet.toList (allxpubs addrmap)
allxpubs addrmap =
HashSet.fromList . map addressXPubSpec $ HashMap.elems addrmap
redisImportMultiTx ::
(Monad f, RedisCtx m f)
=> HashMap Address AddressXPub
-> HashMap OutPoint Unspent
-> [TxData]
-> m (f ())
redisImportMultiTx addrmap unspentmap txs = do
xs <- mapM importtxentries txs
return $ sequence_ xs
where
uns p i =
case HashMap.lookup p unspentmap of
Just u ->
redisAddXPubUnspents (addressXPubSpec i) [(p, unspentBlock u)]
Nothing -> redisRemXPubUnspents (addressXPubSpec i) [p]
addtx tx a p =
case HashMap.lookup a addrmap of
Just i -> do
let tr = TxRef { txRefHash = txHash (txData tx)
, txRefBlock = txDataBlock tx
}
x <- redisAddXPubTxs (addressXPubSpec i) [tr]
y <- uns p i
return $ x >> y >> return ()
Nothing -> return (pure ())
remtx tx a p =
case HashMap.lookup a addrmap of
Just i -> do
x <- redisRemXPubTxs (addressXPubSpec i) [txHash (txData tx)]
y <- uns p i
return $ x >> y >> return ()
Nothing -> return (pure ())
importtxentries tx =
if txDataDeleted tx
then do
x <- mapM (uncurry (remtx tx)) (txaddrops tx)
y <- redisRemFromMempool [txHash (txData tx)]
return $ sequence_ x >> void y
else do
a <- sequence <$> mapM (uncurry (addtx tx)) (txaddrops tx)
b <-
case txDataBlock tx of
b@MemRef {} ->
let tr = TxRef { txRefHash = txHash (txData tx)
, txRefBlock = b
}
in redisAddToMempool [tr]
_ -> redisRemFromMempool [txHash (txData tx)]
return $ a >> b >> return ()
txaddrops td = txInputs td <> txOutputs td
redisUpdateBalances ::
(Monad f, RedisCtx m f)
=> HashMap Address AddressXPub
-> HashMap Address Balance
-> m (f ())
redisUpdateBalances addrmap balmap =
fmap (fmap mconcat . sequence) . forM (HashMap.keys addrmap) $ \a ->
case (HashMap.lookup a addrmap, HashMap.lookup a balmap) of
(Just ainfo, Just bal) ->
redisAddXPubBalances (addressXPubSpec ainfo) [xpubbal ainfo bal]
_ -> return (pure ())
where
xpubbal ainfo bal =
XPubBal {xPubBalPath = addressXPubPath ainfo, xPubBal = bal}
cacheAddAddresses ::
(StoreReadExtra m, MonadUnliftIO m, MonadLoggerIO m)
=> [(Address, AddressXPub)]
-> CacheX m ()
cacheAddAddresses [] = $(logDebugS) "Cache" "No further addresses to add"
cacheAddAddresses addrs = do
$(logDebugS) "Cache" $
"Adding " <> cs (show (length addrs)) <> " new generated addresses"
$(logDebugS) "Cache" "Getting balances"
balmap <- HashMap.fromListWith (<>) <$> mapM (uncurry getbal) addrs
$(logDebugS) "Cache" "Getting unspent outputs"
utxomap <- HashMap.fromListWith (<>) <$> mapM (uncurry getutxo) addrs
$(logDebugS) "Cache" "Getting transactions"
txmap <- HashMap.fromListWith (<>) <$> mapM (uncurry gettxmap) addrs
$(logDebugS) "Cache" "Running Redis pipeline"
runRedis $ do
a <- forM (HashMap.toList balmap) (uncurry redisAddXPubBalances)
b <- forM (HashMap.toList utxomap) (uncurry redisAddXPubUnspents)
c <- forM (HashMap.toList txmap) (uncurry redisAddXPubTxs)
return $ sequence_ a >> sequence_ b >> sequence_ c
$(logDebugS) "Cache" "Completed Redis pipeline"
let xpubs = HashSet.toList
. HashSet.fromList
. map addressXPubSpec
$ Map.elems amap
$(logDebugS) "Cache" "Getting xpub balances"
xmap <- getbals xpubs
gap <- lift getMaxGap
let notnulls = getnotnull balmap
addrs' = getNewAddrs gap xmap notnulls
cacheAddAddresses addrs'
where
getbals xpubs = runRedis $ do
bs <- sequence <$> forM xpubs redisGetXPubBalances
return $
HashMap.filter (not . null) . HashMap.fromList . zip xpubs <$> bs
amap = Map.fromList addrs
getnotnull =
let f xpub =
map $ \bal ->
AddressXPub
{ addressXPubSpec = xpub
, addressXPubPath = xPubBalPath bal
}
g = filter (not . nullBalance . xPubBal)
in concatMap (uncurry f) . HashMap.toList . HashMap.map g
getbal a i =
let f b = ( addressXPubSpec i
, [XPubBal {xPubBal = b, xPubBalPath = addressXPubPath i}] )
in f <$> lift (getDefaultBalance a)
getutxo a i =
let f us = ( addressXPubSpec i
, map (\u -> (unspentPoint u, unspentBlock u)) us )
in f <$> lift (getAddressUnspents a def)
gettxmap a i =
let f ts = (addressXPubSpec i, ts)
in f <$> lift (getAddressTxs a def)
getNewAddrs :: KeyIndex
-> HashMap XPubSpec [XPubBal]
-> [AddressXPub]
-> [(Address, AddressXPub)]
getNewAddrs gap xpubs =
concatMap $ \a ->
case HashMap.lookup (addressXPubSpec a) xpubs of
Nothing -> []
Just bals -> addrsToAdd gap bals a
cacheCoolKey :: ByteString
cacheCoolKey = "cooldown"
isCool :: (MonadUnliftIO m, MonadLoggerIO m) => CacheX m Bool
isCool =
runRedis (Redis.get cacheCoolKey) >>= \case
Nothing -> return True
Just bs -> do
let t = read (cs bs)
cooldown <- toInteger . (* 500) <$> asks cacheRefresh
now <- microseconds
return (cooldown <= now - t)
startCooldown :: (MonadUnliftIO m, MonadLoggerIO m) => CacheX m ()
startCooldown = do
now <- microseconds
void $ runRedis (Redis.set cacheCoolKey (cs (show now)))
syncMempoolC :: (MonadUnliftIO m, MonadLoggerIO m, StoreReadExtra m)
=> CacheX m ()
syncMempoolC =
void . withLock $
isCool >>= \cool ->
when cool $ do
nodepool <- HashSet.fromList . map snd <$> lift getMempool
cachepool <- HashSet.fromList . map snd <$> cacheGetMempool
txs <- mapM getit . HashSet.toList $
mappend
(HashSet.difference nodepool cachepool)
(HashSet.difference cachepool nodepool)
unless (null txs) $ do
$(logDebugS) "Cache" "Importing mempool transactions"
importMultiTxC (rights txs)
startCooldown
where
in_sync bb =
asks cacheChain >>= \ch ->
chainGetBest ch >>= \cb ->
return $ headerHash (nodeHeader cb) == bb
getit th =
lift (getTxData th) >>= \case
Nothing -> return (Left th)
Just tx -> return (Right tx)
cacheGetMempool :: MonadLoggerIO m => CacheX m [(UnixTime, TxHash)]
cacheGetMempool = runRedis redisGetMempool
cacheGetHead :: MonadLoggerIO m => CacheX m (Maybe BlockHash)
cacheGetHead = runRedis redisGetHead
cacheSetHead :: (MonadLoggerIO m, StoreReadBase m) => BlockHash -> CacheX m ()
cacheSetHead bh = do
$(logDebugS) "Cache" $ "Cache head set to: " <> blockHashToHex bh
void $ runRedis (redisSetHead bh)
cacheGetAddrsInfo ::
MonadLoggerIO m => [Address] -> CacheX m [Maybe AddressXPub]
cacheGetAddrsInfo as = runRedis (redisGetAddrsInfo as)
redisAddToMempool :: (Applicative f, RedisCtx m f) => [TxRef] -> m (f Integer)
redisAddToMempool [] = return (pure 0)
redisAddToMempool btxs =
zadd mempoolSetKey $
map (\btx -> (blockRefScore (txRefBlock btx), encode (txRefHash btx)))
btxs
redisRemFromMempool ::
(Applicative f, RedisCtx m f) => [TxHash] -> m (f Integer)
redisRemFromMempool [] = return (pure 0)
redisRemFromMempool xs = zrem mempoolSetKey $ map encode xs
redisSetAddrInfo ::
(Functor f, RedisCtx m f) => Address -> AddressXPub -> m (f ())
redisSetAddrInfo a i = void <$> Redis.set (addrPfx <> encode a) (encode i)
evictFromCache ::
(MonadUnliftIO m, MonadLoggerIO m, StoreReadBase m)
=> [XPubSpec]
-> CacheT m ()
evictFromCache xpubs = ask >>= \case
Nothing -> return ()
Just cfg -> void (runReaderT (delXPubKeys xpubs) cfg)
delXPubKeys ::
(MonadUnliftIO m, MonadLoggerIO m, StoreReadBase m)
=> [XPubSpec]
-> CacheX m Integer
delXPubKeys [] = return 0
delXPubKeys xpubs = do
forM_ xpubs $ \x -> do
xtxt <- xpubText x
$(logDebugS) "Cache" $ "Deleting xpub: " <> xtxt
xbals <-
runRedis . fmap sequence . forM xpubs $ \xpub -> do
bs <- redisGetXPubBalances xpub
return $ (xpub, ) <$> bs
runRedis $ fmap sum . sequence <$> forM xbals (uncurry redisDelXPubKeys)
redisDelXPubKeys ::
(Monad f, RedisCtx m f) => XPubSpec -> [XPubBal] -> m (f Integer)
redisDelXPubKeys xpub bals = go (map (balanceAddress . xPubBal) bals)
where
go addrs = do
addrcount <-
case addrs of
[] -> return (pure 0)
_ -> Redis.del (map ((addrPfx <>) . encode) addrs)
txsetcount <- Redis.del [txSetPfx <> encode xpub]
utxocount <- Redis.del [utxoPfx <> encode xpub]
balcount <- Redis.del [balancesPfx <> encode xpub]
x <- Redis.zrem maxKey [encode xpub]
return $ do
_ <- x
addrs' <- addrcount
txset' <- txsetcount
utxo' <- utxocount
bal' <- balcount
return $ addrs' + txset' + utxo' + bal'
redisAddXPubTxs ::
(Applicative f, RedisCtx m f) => XPubSpec -> [TxRef] -> m (f Integer)
redisAddXPubTxs _ [] = return (pure 0)
redisAddXPubTxs xpub btxs =
zadd (txSetPfx <> encode xpub) $
map (\t -> (blockRefScore (txRefBlock t), encode (txRefHash t))) btxs
redisRemXPubTxs ::
(Applicative f, RedisCtx m f) => XPubSpec -> [TxHash] -> m (f Integer)
redisRemXPubTxs _ [] = return (pure 0)
redisRemXPubTxs xpub txhs = zrem (txSetPfx <> encode xpub) (map encode txhs)
redisAddXPubUnspents ::
(Applicative f, RedisCtx m f)
=> XPubSpec
-> [(OutPoint, BlockRef)]
-> m (f Integer)
redisAddXPubUnspents _ [] = return (pure 0)
redisAddXPubUnspents xpub utxo =
zadd (utxoPfx <> encode xpub) $
map (\(p, r) -> (blockRefScore r, encode p)) utxo
redisRemXPubUnspents ::
(Applicative f, RedisCtx m f) => XPubSpec -> [OutPoint] -> m (f Integer)
redisRemXPubUnspents _ [] = return (pure 0)
redisRemXPubUnspents xpub ops = zrem (utxoPfx <> encode xpub) (map encode ops)
redisAddXPubBalances ::
(Monad f, RedisCtx m f) => XPubSpec -> [XPubBal] -> m (f ())
redisAddXPubBalances _ [] = return (pure ())
redisAddXPubBalances xpub bals = do
xs <- mapM (uncurry (Redis.hset (balancesPfx <> encode xpub))) entries
ys <-
forM bals $ \b ->
redisSetAddrInfo
(balanceAddress (xPubBal b))
AddressXPub
{addressXPubSpec = xpub, addressXPubPath = xPubBalPath b}
return $ sequence_ xs >> sequence_ ys
where
entries = map (\b -> (encode (xPubBalPath b), encode (xPubBal b))) bals
redisSetHead :: RedisCtx m f => BlockHash -> m (f Redis.Status)
redisSetHead bh = Redis.set bestBlockKey (encode bh)
redisGetAddrsInfo ::
(Monad f, RedisCtx m f) => [Address] -> m (f [Maybe AddressXPub])
redisGetAddrsInfo [] = return (pure [])
redisGetAddrsInfo as = do
is <- mapM (\a -> Redis.get (addrPfx <> encode a)) as
return $ do
is' <- sequence is
return $ map (eitherToMaybe . decode =<<) is'
addrsToAdd :: KeyIndex -> [XPubBal] -> AddressXPub -> [(Address, AddressXPub)]
addrsToAdd gap xbals addrinfo
| null fbals = []
| otherwise = zipWith f addrs list
where
f a p = (a, AddressXPub {addressXPubSpec = xpub, addressXPubPath = p})
dchain = head (addressXPubPath addrinfo)
fbals = filter ((== dchain) . head . xPubBalPath) xbals
maxidx = maximum (map (head . tail . xPubBalPath) fbals)
xpub = addressXPubSpec addrinfo
aidx = (head . tail) (addressXPubPath addrinfo)
ixs =
if gap > maxidx - aidx
then [maxidx + 1 .. aidx + gap]
else []
paths = map (Deriv :/ dchain :/) ixs
keys = map (\p -> derivePubPath p (xPubSpecKey xpub)) paths
list = map pathToList paths
xpubf = xPubAddrFunction (xPubDeriveType xpub)
addrs = map xpubf keys
sortTxData :: [TxData] -> [TxData]
sortTxData tds =
let txm = Map.fromList (map (\d -> (txHash (txData d), d)) tds)
ths = map (txHash . snd) (sortTxs (map txData tds))
in mapMaybe (`Map.lookup` txm) ths
txInputs :: TxData -> [(Address, OutPoint)]
txInputs td =
let is = txIn (txData td)
ps = I.toAscList (txDataPrevs td)
as = map (scriptToAddressBS . prevScript . snd) ps
f (Right a) i = Just (a, prevOutput i)
f (Left _) _ = Nothing
in catMaybes (zipWith f as is)
txOutputs :: TxData -> [(Address, OutPoint)]
txOutputs td =
let ps =
zipWith
(\i _ ->
OutPoint
{outPointHash = txHash (txData td), outPointIndex = i})
[0 ..]
(txOut (txData td))
as = map (scriptToAddressBS . scriptOutput) (txOut (txData td))
f (Right a) p = Just (a, p)
f (Left _) _ = Nothing
in catMaybes (zipWith f as ps)
redisGetHead :: (Functor f, RedisCtx m f) => m (f (Maybe BlockHash))
redisGetHead = do
x <- Redis.get bestBlockKey
return $ (eitherToMaybe . decode =<<) <$> x
redisGetMempool :: (Applicative f, RedisCtx m f) => m (f [(UnixTime, TxHash)])
redisGetMempool = do
xs <- getFromSortedSet mempoolSetKey Nothing 0 0
return $ map (uncurry f) <$> xs
where
f t s = (memRefTime (scoreBlockRef s), t)
xpubText :: (MonadUnliftIO m, MonadLoggerIO m, StoreReadBase m)
=> XPubSpec -> CacheX m Text
xpubText xpub = do
net <- lift getNetwork
return . cs $ xPubExport net (xPubSpecKey xpub)
cacheNewBlock :: MonadIO m => CacheWriter -> m ()
cacheNewBlock = send CacheNewBlock
cachePing :: MonadIO m => CacheWriter -> m ()
cachePing = send CachePing