解决Data.Time与PostgreSQL间UTCTime时间精度不一致问题

针对你遇到的UTCTime精度不一致（预期9位小数、实际返回6位）问题，有两种直接的解决思路：

一、调整数据库存储的时间精度 #

大部分数据库（比如PostgreSQL）默认的时间类型精度是微秒级（6位小数），而Haskell的UTCTime支持纳秒级（9位小数）精度。要让数据库存储完整的纳秒精度，需要在实体定义时显式指定精度：

1. 在实体的PersistEntity实例中，给时间字段指定SqlTimestamp的精度参数：

import Database.Persist.Sql (SqlType(..))

data TestEntity = TestEntity
  { testEntityTime :: UTCTime
  , testEntityMaybeTime :: Maybe UTCTime
  } deriving (Eq, Show)

instance PersistEntity TestEntity where
  persistEntityDef _ = EntityDef
    { entityName = "test_entity"
    , entityAttrs = []
    , entityFields =
        [ FieldDef
            { fieldName = "time"
            , fieldSqlType = SqlTimestamp True 9  -- 第二个参数指定9位精度（纳秒）
            , fieldAttrs = []
            , fieldStrict = True
            , fieldNullable = False
            }
        , FieldDef
            { fieldName = "maybe_time"
            , fieldSqlType = SqlTimestamp True 9
            , fieldAttrs = []
            , fieldStrict = True
            , fieldNullable = True
            }
        ]
    -- ... 其他必要的实体定义项
    }

2. 执行数据库迁移，确保表字段类型被更新为支持纳秒精度的类型（比如PostgreSQL的timestamp with time zone(9)）。自动迁移工具会根据你指定的SqlType生成对应的DDL。

二、修改测试中的相等性判断逻辑 #

如果不需要数据库存储纳秒级精度，或者无法修改数据库结构，可以在测试时截断时间的精度后再比较：

1. 编写一个截断UTCTime到指定小数位数的辅助函数：

import Data.Time.Clock (UTCTime(..), diffTimeToPicoseconds, picosecondsToDiffTime)

-- 将UTCTime截断到指定的小数位数（例如6位对应微秒，9位对应纳秒）
truncateUTCTime :: Int -> UTCTime -> UTCTime
truncateUTCTime precision (UTCTime day dt) =
  let totalPicos = diffTimeToPicoseconds dt
      -- DiffTime的精度是皮秒（10^-12秒），计算需要保留的位数
      divisor = 10 ^ (12 - precision)
      truncatedPicos = totalPicos `div` divisor * divisor
  in UTCTime day (picosecondsToDiffTime truncatedPicos)

2. 在测试中使用这个函数统一截断预期值和实际值后再比较：

import Test.HUnit (assertEqual)

testTimePrecision :: IO ()
testTimePrecision = do
  let expected = read "2023-05-06 21:52:13.441270819 UTC" :: UTCTime
  actualEntity <- fetchTestEntity  -- 从数据库查询实体
  -- 截断到6位精度（与数据库存储一致）后比较
  assertEqual "时间值匹配" 
    (truncateUTCTime 6 expected) 
    (truncateUTCTime 6 $ testEntityTime actualEntity)
  -- 处理Maybe UTCTime的情况
  case (testEntityMaybeTime actualEntity, Just expected) of
    (Just actual, Just expVal) -> 
      assertEqual "可选时间值匹配" (truncateUTCTime 6 expVal) (truncateUTCTime 6 actual)
    _ -> return ()

额外注意事项 #

• 确认数据库驱动是否支持纳秒级时间读取：比如postgresql-simple的UTCTime实例已经支持读取纳秒精度的时间，只要数据库字段定义正确。
• 如果使用自动迁移，修改SqlType后需要重新生成迁移脚本，确保表结构更新生效。

内容的提问来源于stack exchange，提问作者GTF