From 45b4169bf7edd0a7e5585df7713a4464670c7048 Mon Sep 17 00:00:00 2001
From: Luke Lau <luke_lau@icloud.com>
Date: Fri, 28 Aug 2020 17:07:49 +0100
Subject: [PATCH] Update request map in listenServer

---
 src/Language/Haskell/LSP/Test.hs | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/Language/Haskell/LSP/Test.hs b/src/Language/Haskell/LSP/Test.hs
index 8dd252c..80ea0e3 100644
--- a/src/Language/Haskell/LSP/Test.hs
+++ b/src/Language/Haskell/LSP/Test.hs
@@ -229,10 +229,10 @@ runSessionWithHandles' serverProc serverIn serverOut config' caps rootDir sessio
   listenServer serverOut context = do
     msgBytes <- getNextMessage serverOut
 
-    reqMap <- readMVar $ requestMap context
-
-    let msg = fst $ decodeFromServerMsg reqMap msgBytes
+    msg <- modifyMVar (requestMap context) $ \reqMap -> do
+      let (msg, newReqMap) = decodeFromServerMsg reqMap msgBytes
       writeChan (messageChan context) (ServerMessage msg)
+      pure (newReqMap, msg)
 
     case msg of
       (FromServerRsp SShutdown _) -> return ()
-- 
2.30.2