]> gitweb.michael.orlitzky.com - dead/htsn-import.git/commitdiff
Use Generics.to_tuple in TSN.XML.Heartbeat.
authorMichael Orlitzky <michael@orlitzky.com>
Mon, 29 Dec 2014 21:34:59 +0000 (16:34 -0500)
committerMichael Orlitzky <michael@orlitzky.com>
Tue, 30 Dec 2014 00:42:58 +0000 (19:42 -0500)
src/TSN/XML/Heartbeat.hs

index 8fafba1c6c80d6435793a1345ab733329ebfc9a9..b44f9309379cf9e8fb8d6e51d9d8367c0084c36a 100644 (file)
@@ -1,3 +1,4 @@
+{-# LANGUAGE DeriveGeneric #-}
 {-# LANGUAGE ScopedTypeVariables #-}
 
 -- | Handle documents defined by Heartbeat.dtd.
@@ -12,6 +13,7 @@ where
 -- System imports.
 import Data.Time.Clock ( UTCTime )
 import Data.Tuple.Curry ( uncurryN )
+import qualified GHC.Generics as GHC ( Generic )
 import Test.Tasty ( TestTree, testGroup )
 import Test.Tasty.HUnit ( (@?=), testCase )
 import Text.XML.HXT.Core (
@@ -25,6 +27,7 @@ import Text.XML.HXT.Core (
   xpWrap )
 
 -- Local imports.
+import Generics ( Generic(..), to_tuple )
 import TSN.DbImport ( ImportResult(..) )
 import TSN.Picklers ( xp_time_stamp )
 import Xml ( pickle_unpickle, unpickleable )
@@ -39,11 +42,15 @@ dtd = "Heartbeat.dtd"
 --   Heartbeat message.
 --
 data Message =
-  Message {
-    xml_file_id :: Int,
-    heading :: String,
-    time_stamp :: UTCTime }
-  deriving (Eq, Show)
+  Message
+    Int     -- xml_file_id
+    String  -- heading
+    UTCTime -- time_stamp
+  deriving (Eq, GHC.Generic, Show)
+
+-- | For 'Generics.to_tuple'.
+--
+instance Generic Message
 
 
 -- | A (un)pickler that turns a Heartbeat XML file into a 'Message'
@@ -58,9 +65,6 @@ pickle_message =
              (xpElem "time_stamp" xp_time_stamp)
   where
     from_tuple = uncurryN Message
-    to_tuple m = (xml_file_id m,
-                  heading m,
-                  time_stamp m)
 
 
 -- | Verify (and report) the received heartbeat. We return