Move ready-route retry to happen earlier in OnionRouter STM.

1 files changed, 6 insertions, 6 deletions

diff --git a/OnionRouter.hs b/OnionRouter.hs
index 2391e13c..e3c32fbe 100644
--- a/OnionRouter.hs
+++ b/OnionRouter.hs
@@ -388,19 +388,19 @@ hookQueries :: OnionRouter -> (tid -> Nonce8)
                            -> TransactionMethods d tid (OnionDestination RouteId) x
 hookQueries or t8 tmethods = TransactionMethods
     { dispatchRegister = \mvar od d -> {-# SCC "hookQ.dispatchRegister" #-} do --  :: MVar x -> d -> STM (tid, d)
-        (tid,d') <- dispatchRegister tmethods mvar od d
-        let Nonce8 w8 = t8 tid
-            ni = onionNodeInfo od
+        let ni = onionNodeInfo od
             rid@(RouteId ridn) = fromMaybe (routeId (nodeId ni)) $ onionRouteSpec od
-            od' = case od of
-                OnionDestination {} -> od { onionRouteSpec = Just rid }
-                OnionToOwner a b    -> OnionToOwner a b -- Type cast.
         wanted <- {-# SCC "hookQ.wanted" #-} (readTVar (pendingRoutes or IntMap.! ridn))
         mr <- {-# SCC "hookQ.mr_action" #-} (readArray (routeMap or) ridn)
         -- Block query until a route is ready.
         check $ fromMaybe False $ do
             RouteRecord{routeVersion=rv} <- {-# SCC "hookQ.mr" #-} mr
             return $ wanted <= rv
+        (tid,d') <- dispatchRegister tmethods mvar od d
+        let Nonce8 w8 = t8 tid
+            od' = case od of
+                OnionDestination {} -> od { onionRouteSpec = Just rid }
+                OnionToOwner a b    -> OnionToOwner a b -- Type cast.
         let pq = PendingQuery { pendingDestination = od'
                               , pendingVersion = maybe 0 routeVersion mr
                               }