src/Graylog/Types.hs

   1 {-# LANGUAGE RecordWildCards #-}
   2
   3 module Graylog.Types
   4    ( Graylog
   5    , _graylogHost
   6    , _graylogPort
   7    , _graylogAddress
   8    , _graylogSocket
   9    , _graylogHostName
  10    , ChunkSize
  11
  12    , defaultChunkSize
  13    , openGraylog
  14    , closeGraylog
  15    ) where
  16
  17 import           Data.Text      (Text)
  18 import qualified Data.Text      as T
  19 import           Network.BSD
  20 import           Network.Socket
  21
  22 type ChunkSize = Word
  23
  24 data Graylog
  25    = Graylog
  26       { _graylogHost      :: String
  27       , _graylogPort      :: String
  28       , _graylogAddress   :: AddrInfo
  29       , _graylogSocket    :: Socket
  30       , _graylogHostName  :: Text
  31       , _graylogChunkSize :: ChunkSize
  32       }
  33
  34 defaultChunkSize :: ChunkSize
  35 defaultChunkSize = 8192
  36
  37 openGraylog :: HostName -> ServiceName -> ChunkSize -> IO (Either String Graylog)
  38 openGraylog host port chksize = do
  39    infos <- getAddrInfo Nothing (Just host) (Just port)
  40    case infos of
  41       []     -> return $ Left "No address info found."
  42       [info] -> do
  43          sock <- socket (addrFamily info) Datagram defaultProtocol
  44          connect sock (addrAddress info)
  45          hostname <- getHostName
  46          return $ Right $ Graylog host port info sock (T.pack hostname) chksize
  47       _      -> return $ Left "Too many address infos found."
  48
  49 closeGraylog :: Graylog -> IO ()
  50 closeGraylog Graylog{..} = close _graylogSocket
  51
  52