module System.Log.FastLogger (
Logger
, mkLogger
, renewLogger
, loggerPutStr
, loggerPutBuilder
, loggerFlush
, LogStr(..)
, ToLogStr(..)
, loggerDate
, module System.Log.FastLogger.File
) where
import Blaze.ByteString.Builder
import Control.Applicative
import Control.Monad
import qualified Data.ByteString as BS
import Data.ByteString.Internal (ByteString(..), c2w)
import Data.List
import Data.Maybe
import Data.Typeable
import Foreign
import GHC.Base
import GHC.IO.Buffer
import qualified GHC.IO.BufferedIO as Buffered
import qualified GHC.IO.Device as RawIO
import GHC.IO.FD
import GHC.IO.Handle.Internals
import GHC.IO.Handle.Text
import GHC.IO.Handle.Types
import GHC.IORef
import GHC.Num
import GHC.Real
import System.IO
import System.Log.FastLogger.Date
import System.Log.FastLogger.File
import qualified Data.Text as TS
import qualified Data.Text.Encoding as TE
import qualified Data.Text.Lazy as TL
import qualified Data.ByteString as S
import qualified Data.ByteString.Lazy as L
data Logger = Logger {
loggerAutoFlush :: Bool
, loggerHandle :: Handle
, loggerDateRef :: DateRef
}
logBufSize :: Int
logBufSize = 4096
initHandle :: Handle -> IO ()
initHandle hdl = hSetBuffering hdl (BlockBuffering (Just logBufSize))
mkLogger :: Bool
-> Handle
-> IO Logger
mkLogger autoFlush hdl = do
initHandle hdl
Logger autoFlush hdl <$> dateInit
renewLogger :: Logger -> Handle -> IO Logger
renewLogger logger newhdl = do
let oldhdl = loggerHandle logger
hFlush oldhdl
hClose oldhdl
initHandle newhdl
return $ logger { loggerHandle = newhdl }
data LogStr = LS !String | LB !ByteString
class ToLogStr a where toLogStr :: a -> LogStr
instance ToLogStr [Char] where toLogStr = LS
instance ToLogStr ByteString where toLogStr = LB
instance ToLogStr L.ByteString where toLogStr = LB . S.concat . L.toChunks
instance ToLogStr TS.Text where toLogStr = LB . TE.encodeUtf8
instance ToLogStr TL.Text where toLogStr = LB . TE.encodeUtf8 . TL.toStrict
hPutLogStr :: Handle -> [LogStr] -> IO ()
hPutLogStr handle bss =
wantWritableHandle "hPutLogStr" handle $ \h_ -> bufsWrite h_ bss
bufsWrite :: Handle__ -> [LogStr] -> IO ()
bufsWrite h_@Handle__{..} bss = do
old_buf@Buffer{
bufRaw = old_raw
, bufR = w
, bufSize = size
} <- readIORef haByteBuffer
if size w > len then do
withRawBuffer old_raw $ \ptr ->
go (ptr `plusPtr` w) bss
writeIORef haByteBuffer old_buf{ bufR = w + len }
else do
old_buf' <- Buffered.flushWriteBuffer haDevice old_buf
writeIORef haByteBuffer old_buf'
if size > len then
bufsWrite h_ bss
else allocaBytes size $ \ptr -> do
go ptr bss
let Just fd = cast haDevice :: Maybe FD
_ <- RawIO.writeNonBlocking fd ptr size
return ()
where
len = foldl' (\x y -> x + getLength y) 0 bss
getLength (LB s) = BS.length s
getLength (LS s) = length s
go :: Ptr Word8 -> [LogStr] -> IO ()
go _ [] = return ()
go dst (LB b:bs) = do
dst' <- copy dst b
go dst' bs
go dst (LS s:ss) = do
dst' <- copy' dst s
go dst' ss
copy :: Ptr Word8 -> ByteString -> IO (Ptr Word8)
copy dst (PS ptr off len) = withForeignPtr ptr $ \s -> do
let src = s `plusPtr` off
_ <- memcpy dst src (fromIntegral len)
return (dst `plusPtr` len)
copy' :: Ptr Word8 -> String -> IO (Ptr Word8)
copy' dst [] = return dst
copy' dst (x:xs) = do
poke dst (c2w x)
copy' (dst `plusPtr` 1) xs
loggerPutStr :: Logger -> [LogStr] -> IO ()
loggerPutStr logger strs = do
hPutLogStr hdl strs
when autoFlush $ hFlush hdl
where
hdl = loggerHandle logger
autoFlush = loggerAutoFlush logger
loggerPutBuilder :: Logger -> Builder -> IO ()
loggerPutBuilder logger builder = do
loggerPutStr logger . return . LB . toByteString $ builder
when autoFlush $ hFlush hdl
where
hdl = loggerHandle logger
autoFlush = loggerAutoFlush logger
loggerFlush :: Logger -> IO ()
loggerFlush logger = hFlush $ loggerHandle logger
loggerDate :: Logger -> IO ZonedDate
loggerDate logger = getDate $ loggerDateRef logger