-- A concurrent TCP server that echoes everything that it receives.

import Control.Exception (finally)
import Control.Monad.IO.Class (liftIO)
import Streamly
import Streamly.Network.Socket
import qualified Network.Socket as Net
import qualified Streamly.Network.Inet.TCP as TCP
import qualified Streamly.Prelude as S

main :: IO ()
main = S.drain
    $ parallely $ S.mapM (useWith echo)
    $ serially $ S.unfold TCP.acceptOnPort 8091
    where
    echo sk =
          S.fold (writeChunks sk)
        $ S.unfold readChunksWithBufferOf (32768, sk)
    useWith f sk = finally (f sk) (liftIO (Net.close sk))