git.iko.soy/streamly
on 8ecf96a help sign in
streamly/examples/FileSinkServer.hs
36 lines 998 B history raw
1-- A concurrent TCP server that:
2--
3-- * receives connections from clients
4-- * splits the incoming data into lines
5-- * lines from concurrent connections are merged into a single srteam
6-- * writes the line stream to an output file
8import System.Environment (getArgs)
10import Streamly
11import Streamly.String
12import qualified Streamly.FileSystem.Handle as FH
13import qualified Streamly.Fold as FL
14import qualified Streamly.Mem.Array as A
15import qualified Streamly.Network.Socket as NS
16import qualified Streamly.Network.Server as NS
17import qualified Streamly.Prelude as S
19import System.IO (withFile, IOMode(..))
21main :: IO ()
22main = do
23 file <- fmap head getArgs
24 withFile file AppendMode
25 (\src -> FH.write src
26 $ encodeChar8Unchecked
27 $ S.concatMap A.read
28 $ S.concatMapBy parallel (flip NS.withSocketS recv)
29 $ NS.connectionsOnAllAddrs 8090)
31 where
33 recv =
34 FL.splitSuffixBy' (== '\n') A.toArray
35 . decodeChar8
36 . NS.read