add MsgGetSizes / MsgReplyGetSizes / MsgGetMeasures / MsgReplyGetMeas…

…ures handling to codecLocalTxMonitorId
IntersectMBO · Jan 23, 2025 · 13e39ad · 13e39ad
1 parent b7dae6c
commit 13e39ad
Showing 1 changed file with 13 additions and 9 deletions.
diff --git a/ouroboros-network-protocols/src/Ouroboros/Network/Protocol/LocalTxMonitor/Codec.hs b/ouroboros-network-protocols/src/Ouroboros/Network/Protocol/LocalTxMonitor/Codec.hs
@@ -216,15 +216,19 @@ codecLocalTxMonitorId =
               => Message ptcl st st' -> m (DecodeStep bytes failure m (SomeMessage st))
           res msg = return (DecodeDone (SomeMessage msg) Nothing)
       in return $ DecodePartial $ \bytes -> case (stok, bytes) of
-        (SingIdle,     Just (AnyMessage msg@MsgAcquire{}))      -> res msg
-        (SingIdle,     Just (AnyMessage msg@MsgDone{}))         -> res msg
-        (SingAcquired, Just (AnyMessage msg@MsgAwaitAcquire{})) -> res msg
-        (SingAcquired, Just (AnyMessage msg@MsgNextTx{}))       -> res msg
-        (SingAcquired, Just (AnyMessage msg@MsgHasTx{}))        -> res msg
-        (SingAcquired, Just (AnyMessage msg@MsgRelease{}))      -> res msg
-        (SingAcquiring,       Just (AnyMessage msg@MsgAcquired{}))    -> res msg
-        (SingBusy SingNextTx, Just (AnyMessage msg@MsgReplyNextTx{})) -> res msg
-        (SingBusy SingHasTx,  Just (AnyMessage msg@MsgReplyHasTx{}))  -> res msg
+        (SingIdle,                 Just (AnyMessage msg@MsgAcquire{}))          -> res msg
+        (SingIdle,                 Just (AnyMessage msg@MsgDone{}))             -> res msg
+        (SingAcquired,             Just (AnyMessage msg@MsgAwaitAcquire{}))     -> res msg
+        (SingAcquired,             Just (AnyMessage msg@MsgNextTx{}))           -> res msg
+        (SingAcquired,             Just (AnyMessage msg@MsgHasTx{}))            -> res msg
+        (SingAcquired,             Just (AnyMessage msg@MsgRelease{}))          -> res msg
+        (SingAcquired,             Just (AnyMessage msg@MsgGetSizes{}))         -> res msg
+        (SingAcquired,             Just (AnyMessage msg@MsgGetMeasures{}))      -> res msg
+        (SingAcquiring,            Just (AnyMessage msg@MsgAcquired{}))         -> res msg
+        (SingBusy SingNextTx,      Just (AnyMessage msg@MsgReplyNextTx{}))      -> res msg
+        (SingBusy SingHasTx,       Just (AnyMessage msg@MsgReplyHasTx{}))       -> res msg
+        (SingBusy SingGetSizes,    Just (AnyMessage msg@MsgReplyGetSizes{}))    -> res msg
+        (SingBusy SingGetMeasures, Just (AnyMessage msg@MsgReplyGetMeasures{})) -> res msg
 
         (SingDone, _) -> notActiveState stok