Add and implement getTx

IntersectMBO · Jun 26, 2019 · 706b772 · 706b772
1 parent 20cac90
commit 706b772
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/ouroboros-consensus/src/Ouroboros/Consensus/Mempool/API.hs b/ouroboros-consensus/src/Ouroboros/Consensus/Mempool/API.hs
@@ -107,6 +107,9 @@ data Mempool m blk idx = Mempool {
       -- transaction/.
     , getTxs :: STM m (Seq (GenTx blk, idx))
 
+      -- |
+    , getTx :: idx -> STM m (Maybe (GenTx blk))
+
       -- |
     , zeroIdx :: idx
     }
diff --git a/ouroboros-consensus/src/Ouroboros/Consensus/Mempool/Impl.hs b/ouroboros-consensus/src/Ouroboros/Consensus/Mempool/Impl.hs
@@ -21,7 +21,7 @@ import           Ouroboros.Consensus.Ledger.Abstract
 import           Ouroboros.Consensus.Ledger.Extended
 import           Ouroboros.Consensus.Mempool.API
 import           Ouroboros.Consensus.Mempool.TxSeq (TicketNo, TxSeq (..),
-                     appendTx, fromTxSeq, zeroTicketNo)
+                     appendTx, fromTxSeq, lookupByTicketNo, zeroTicketNo)
 import qualified Ouroboros.Consensus.Mempool.TxSeq as TxSeq
 import           Ouroboros.Consensus.Util (repeatedly)
 
@@ -38,6 +38,7 @@ openMempool chainDB cfg = do
     return Mempool {
         addTxs  = implAddTxs env
       , getTxs  = implGetTxs env
+      , getTx   = implGetTx  env
       , zeroIdx = zeroTicketNo
       }
 
@@ -109,6 +110,13 @@ implGetTxs mpEnv@MempoolEnv{mpEnvStateVar} = do
                                }
     pure $ fromTxSeq vrValid
 
+implGetTx :: (MonadSTM m, ApplyTx blk)
+          => MempoolEnv m blk hdr
+          -> TicketNo
+          -> STM m (Maybe (GenTx blk))
+implGetTx MempoolEnv{mpEnvStateVar} tn =
+  (`lookupByTicketNo` tn) . isTxs <$> readTVar mpEnvStateVar
+
 {-------------------------------------------------------------------------------
   Validation
 -------------------------------------------------------------------------------}