如何使用pymongo为MongoDB插入符合TTL索引要求的指定格式日期

首先明确前提：MongoDB的TTL索引仅对BSON日期类型的字段生效，和构造该日期的输入格式无关，只要最终存入数据库的是标准BSON日期对象，即可正常触发过期逻辑，无需和官方示例的字符串输入格式保持一致。

官方插入示例参考：

db.log_events.insert( {
"expireAt": new Date('July 22, 2013 14:00:00'),
"logEvent": 2,
"logMessage": "Success!"
} )

转换代码实现 #

pymongo会自动将Python原生的datetime.datetime对象转换为MongoDB兼容的BSON日期类型，你可以按如下方式将毫秒时间戳转换为符合要求的日期对象：

1. 依赖导入 #

from datetime import datetime, timezone
import pymongo

2. 毫秒时间戳转目标日期对象 #

# 输入：纪元以来毫秒数
millisSinceEpoch = 1690000000000

# 转换为UTC时区的datetime对象，避免时区偏差问题
expire_at = datetime.fromtimestamp(millisSinceEpoch / 1000, tz=timezone.utc)

3. 文档插入&索引创建 #

# 插入文档，效果和官方示例完全一致
db.log_events.insert_one({
    "expireAt": expire_at,
    "logEvent": 2,
    "logMessage": "Success!"
})

# 创建TTL索引，expireAfterSeconds设为0表示到达expireAt指定时间立即过期
db.log_events.create_index("expireAt", expireAfterSeconds=0)

注意事项 #

• 禁止将expireAt字段存为字符串或数字时间戳，否则TTL索引不会识别，无法触发过期。
• 建议始终传入带UTC时区的datetime对象，如果传入不带时区的本地时间，MongoDB会默认当做UTC时间处理，会导致过期时间和预期不符。
• 无需对齐官方示例的日期字符串输入格式，只要最终存储的是BSON日期类型，TTL功能即可正常生效。

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