{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE CPP #-}
module Network.Run.TCP (
runTCPClient
, runTCPServer
) where
import Control.Concurrent (forkFinally)
import qualified Control.Exception as E
import Control.Monad (forever, void)
import Network.Socket
import Network.Run.Core
runTCPClient :: String -> String -> (Socket -> IO a) -> IO a
runTCPClient host port client = withSocketsDo $ do
addr <- resolve Stream (Just host) port False
#if MIN_VERSION_network(3,1,1)
E.bracket (open addr) (\sock -> gracefulClose sock 5000) client
#else
E.bracket (open addr) close client
#endif
where
open addr = do
sock <- openSocket addr
connect sock $ addrAddress addr
return sock
runTCPServer :: Maybe HostName -> ServiceName -> (Socket -> IO a) -> IO a
runTCPServer mhost port server = withSocketsDo $ do
addr <- resolve Stream mhost port True
E.bracket (open addr) close loop
where
open addr = do
sock <- openServerSocket addr
listen sock 1024
return sock
loop sock = forever $ do
(conn, _peer) <- accept sock
#if MIN_VERSION_network(3,1,1)
void $ forkFinally (server conn) (const $ gracefulClose conn 5000)
#else
void $ forkFinally (server conn) (const $ close conn)
#endif