Fixed windows build of network-mux

IntersectMBO · Mar 25, 2020 · 0e38adb · 0e38adb
1 parent 1957446
commit 0e38adb
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/network-mux/demo/mux-demo.hs b/network-mux/demo/mux-demo.hs
@@ -25,6 +25,7 @@ import Test.Mux.ReqResp
 import System.Win32
 import System.Win32.NamedPipes
 import qualified System.Win32.Async as Win32.Async
+import System.IOManager
 
 import System.IO
 import System.Exit
@@ -71,7 +72,7 @@ defaultProtocolLimits =
 -- | Server accept loop.
 --
 echoServer :: IO ()
-echoServer = Win32.Async.withIOManager $ \iocp -> do
+echoServer = withIOManager $ \ioManager -> do
     hpipe <- createNamedPipe pipeName
                              (pIPE_ACCESS_DUPLEX .|. fILE_FLAG_OVERLAPPED)
                              (pIPE_TYPE_BYTE .|. pIPE_READMODE_BYTE)
@@ -80,7 +81,7 @@ echoServer = Win32.Async.withIOManager $ \iocp -> do
                              1024
                              0
                              Nothing
-    Win32.Async.associateWithIOCompletionPort (Left hpipe) iocp
+    associateWithIOManager ioManager (Left hpipe)
     Win32.Async.connectNamedPipe hpipe
     _ <- forkIO $ do
            serverLoop hpipe
@@ -122,15 +123,15 @@ serverLoop h = do
 
 
 client :: Int -> String -> IO ()
-client n msg = Win32.Async.withIOManager $ \iocp -> do
+client n msg = withIOManager $ \ioManager -> do
     hpipe <- createFile pipeName
                         (gENERIC_READ .|. gENERIC_WRITE)
                         fILE_SHARE_NONE
                         Nothing
                         oPEN_EXISTING
                         fILE_FLAG_OVERLAPPED
                         Nothing
-    Win32.Async.associateWithIOCompletionPort (Left hpipe) iocp
+    associateWithIOManager ioManager (Left hpipe)
     let pipeChannel = Mx.pipeChannelFromNamedPipe hpipe
         bearer = Mx.pipeAsMuxBearer nullTracer pipeChannel
     Mx.muxStart

diff --git a/network-mux/test/Test/Mux.hs b/network-mux/test/Test/Mux.hs
@@ -50,6 +50,7 @@ import           Control.Tracer (Tracer (..), contramap, nullTracer)
 import qualified System.Win32.NamedPipes as Win32.NamedPipes
 import qualified System.Win32.File       as Win32.File
 import qualified System.Win32.Async      as Win32.Async
+import           System.IOManager
 #else
 import           System.IO (hClose)
 import           System.Process (createPipe)
@@ -460,7 +461,7 @@ runWithQueues initApp respApp = do
 runWithPipe :: RunMuxApplications
 runWithPipe initApp respApp =
 #if defined(mingw32_HOST_OS)
-    Win32.Async.withIOManager $ \iocp -> do
+    withIOManager $ \ioManager -> do
       let pipeName = "\\\\.\\pipe\\mux-test-pipe"
       bracket
         (Win32.NamedPipes.createNamedPipe
@@ -484,8 +485,8 @@ runWithPipe initApp respApp =
                    Nothing)
                 Win32.File.closeHandle
           $ \hCli -> do
-             Win32.Async.associateWithIOCompletionPort (Left hSrv) iocp
-             Win32.Async.associateWithIOCompletionPort (Left hCli) iocp
+             associateWithIOManager ioManager (Left hSrv)
+             associateWithIOManager ioManager (Left hCli)
 
              let clientChannel = pipeChannelFromNamedPipe hCli
                  serverChannel = pipeChannelFromNamedPipe hSrv