src/Graylog/Types.hs

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