git.iko.soy/streamly
on master help sign in
update client/server examples
68690bbHarendra Kumar
examples/FileSinkServer.hs+24−9
compare with master
unifiedsplit
@@ -1,4 +1,14 @@
1−-- A concurrent TCP server that prints everything it receives to stdout.
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
7+
8+import Control.Monad.IO.Class (liftIO)
9+import Data.Char (ord, chr)
10+import System.IO (openFile, IOMode(..), hClose, hPutStrLn)
11+import System.Environment (getArgs)
2
12
3import Streamly
13import Streamly
4import qualified Streamly.Fold as FL
14import qualified Streamly.Fold as FL
@@ -8,15 +18,20 @@ import qualified Network.Socket as NS
8import qualified Streamly.Prelude as S
18import qualified Streamly.Prelude as S
9
19
10main :: IO ()
20main :: IO ()
11−main = S.drain
21+main = do
22+ h <- openOutput
23+ S.mapM_ (hPutLines h)
12− $ parallely $ S.mapM (withSocket recv)
24+ $ S.concatMapBy parallel (withSocket recv)
13− $ serially $ fmap fst (NS.recvConnectionsOn 8090)
25+ $ fmap fst (NS.recvConnectionsOn 8090)
26+ hClose h
14
27
15 where
28 where
16
29
30+ openOutput = fmap head getArgs >>= \file -> openFile file AppendMode
31+
17− withSocket f sk = f sk >> NS.close sk
32+ withSocket f sk = S.finally (liftIO (NS.close sk)) (f sk)
33+ hPutLines h = hPutStrLn h . map (chr . fromIntegral) . A.toList
34+ ord' = (fromIntegral . ord)
18 recv =
35 recv =
19− S.mapM_ print
36+ FL.splitSuffixBy (== ord' '\n') (A.toArrayN 1024)
20− . FL.sessionsOf 3 FL.sum
37+ . NS.read
21− . S.map A.length
22− . NS.readArrays
examples/FromFileClient.hs+15−5
@@ -1,19 +1,29 @@
1−-- A TCP client that reads from stdin and sends to a server
1+-- A TCP client that does the following:
2+-- * Reads multiple filenames passed on the command line
3+-- * Opens as many concurrent connections to the server
4+-- * Sends all the files concurrently to the server
2
5
3import Network.Socket
6import Network.Socket
4−import System.IO (stdin)
7+import System.IO (openFile, IOMode(..), hClose)
8+import System.Environment (getArgs)
5
9
10+import Streamly
6import qualified Streamly.Prelude as S
11import qualified Streamly.Prelude as S
7import qualified Streamly.FileSystem.File as File
12import qualified Streamly.FileSystem.File as File
8import qualified Streamly.Network.Socket as NS
13import qualified Streamly.Network.Socket as NS
9
14
10main :: IO ()
15main :: IO ()
11−main = do
16+main = getArgs >>= S.drain . parallely . S.mapM sendFile . S.fromList
12− openConnection >>= \sk ->
13− S.drain $ NS.writeArrays sk $ File.readArrays stdin
14
17
15 where
18 where
16
19
20+ sendFile file = do
21+ h <- openFile file ReadMode
22+ openConnection >>= \sk -> do
23+ S.drain $ NS.writeArrays sk $ File.readArrays h
24+ close sk
25+ hClose h
26+
17 openConnection = do
27 openConnection = do
18 sock <- socket AF_INET Stream defaultProtocol
28 sock <- socket AF_INET Stream defaultProtocol
19 connect sock $ SockAddrInet 8090 (tupleToHostAddress (127,0,0,1))
29 connect sock $ SockAddrInet 8090 (tupleToHostAddress (127,0,0,1))