> module Data.DateTime where
> import Data.Fixed (div')
> import Data.Function (on)
> import Data.Maybe (fromJust)
> import Data.Time.Calendar hiding (fromGregorian, toGregorian)
> import Data.Time.Clock hiding (getCurrentTime)
> import Data.Time.Format
> import Data.Time.LocalTime
>
> import Numeric (fromRat)
> import System.Locale
> import System.Time hiding (toClockTime)
> import Test.QuickCheck
> import qualified Data.Time.Calendar as Calendar
> import qualified Data.Time.Clock as Clock
Define a local synonym for UTCTime just to get some insulation from
the craziness of the Haskell standard library date and time functions.
> type DateTime = UTCTime
> instance Arbitrary UTCTime where
> arbitrary = do
> offset <- choose (0, 20000) :: Gen Float
> return . fromMJD' $ offset + fromRational startOfTimeMJD
So that we can use our DateTime class with HDBC.
>
>
>
Defined here so that users don't need to know about Data.Time.Clock.
> getCurrentTime :: IO DateTime
> getCurrentTime = Clock.getCurrentTime
Conversion back and forth between DateTime and MJD.
> toMJD :: DateTime -> Rational
> toMJD = getModJulianDate . toUniversalTime
> toMJD' :: RealFloat a => DateTime -> a
> toMJD' = fromRat . toMJD
> fromMJD :: Rational -> DateTime
> fromMJD = fromUniversalTime . ModJulianDate
> fromMJD' :: RealFloat a => a -> DateTime
> fromMJD' = fromMJD . realToFrac
> invariant f x = f x == x
> prop_MJD = invariant $ fromMJD . toMJD
> prop_MJD' = invariant $ fromMJD' . toMJD'
Because UTCTime is opaque, we need to convert to UniversalTime in
order to do anything with it, but these functions are mainly of
interest internally.
> toUniversalTime :: DateTime -> UniversalTime
> toUniversalTime = localTimeToUT1 0 . utcToLocalTime utc
> fromUniversalTime :: UniversalTime -> DateTime
> fromUniversalTime = localTimeToUTC utc . ut1ToLocalTime 0
> prop_Universal = invariant $ fromUniversalTime . toUniversalTime
Take apart a DateTime into pieces and parts.
> toGregorian' :: DateTime -> (Integer, Int, Int)
> toGregorian' dt = (y, m, d)
> where
> (y, m, d, _, _, _) = toGregorian dt
> toGregorian :: DateTime -> (Integer, Int, Int, Int, Int, Int)
> toGregorian dt = (year, month, day', hours, minutes, seconds `div'` 1)
> where
> LocalTime day tod = utcToLocalTime utc dt
> (year, month, day') = Calendar.toGregorian day
> TimeOfDay hours minutes seconds = tod
Combine pieces and parts to produce a DateTime.
> fromGregorian' :: Integer -> Int -> Int -> DateTime
> fromGregorian' y m d = fromGregorian y m d 0 0 0
> fromGregorian :: Integer -> Int -> Int -> Int -> Int -> Int -> DateTime
> fromGregorian year month day hours minutes seconds =
> UTCTime day' (secondsToDiffTime . fromIntegral $ seconds')
> where
> day' = Calendar.fromGregorian year month day
> seconds' = 3600 * hours + 60 * minutes + seconds
Getting closer to the machine: Not all the functionality of
System.Time is available in Data.Time, and the only way we can convert
back and forth is to go through seconds.
> toSeconds :: DateTime -> Integer
> toSeconds dt = floor $
> 86400.0 * fromRational (toMJD dt startOfTimeMJD)
> fromSeconds :: Integer -> DateTime
> fromSeconds s = fromMJD $
> fromIntegral s / 86400 + startOfTimeMJD
> toClockTime :: DateTime -> ClockTime
> toClockTime dt = TOD (toSeconds dt) 0
> fromClockTime :: ClockTime -> DateTime
> fromClockTime (TOD s _) = fromSeconds s
> startOfTime :: DateTime
> startOfTime = fromGregorian' 1970 1 1
> prop_StartOfTime _ = toSeconds startOfTime == 0
> startOfTimeMJD :: Rational
> startOfTimeMJD = toMJD startOfTime
Formatting and parsing, with special attention to the format used by
ODBC and MySQL.
> toSqlString :: DateTime -> String
> toSqlString = formatDateTime sqlFormat
> fromSqlString :: String -> Maybe DateTime
> fromSqlString = parseDateTime sqlFormat
> prop_SqlString dt = (fromJust . fromSqlString . toSqlString $ dt') == dt'
> where
> Just dt' = fromSqlString . toSqlString $ dt
> prop_SqlStartOfTime _ = toSqlString startOfTime == "1970-01-01 00:00:00"
> formatDateTime :: String -> DateTime -> String
> formatDateTime = formatTime defaultTimeLocale
> parseDateTime :: String -> String -> Maybe DateTime
> parseDateTime = parseTime defaultTimeLocale
> sqlFormat = iso8601DateFormat (Just "%T")
Simple arithmetic.
> addMinutes' :: Int -> DateTime -> DateTime
> addMinutes' = addMinutes . fromIntegral
> addMinutes :: Integer -> DateTime -> DateTime
> addMinutes m = addSeconds (60 * m)
> diffMinutes' :: DateTime -> DateTime -> Int
> diffMinutes' x = fromIntegral . diffMinutes x
> diffMinutes :: DateTime -> DateTime -> Integer
> diffMinutes x = (`div` 60) . diffSeconds x
> addSeconds :: Integer -> DateTime -> DateTime
> addSeconds s dt = fromSeconds $ toSeconds dt + s
> diffSeconds :: DateTime -> DateTime -> Integer
> diffSeconds = () `on` toSeconds