如何在MySQL中伪造CURRENT_TIMESTAMP？支持按用户隔离吗？

针对你的需求，以下是几种可行的实现方式，均支持AWS RDS MySQL，且能满足「自动填充伪造时间、无需手动设置、按用户隔离」的要求：

1. 会话级修改MySQL时间（轻量原生方案） #

MySQL支持在会话级别临时覆盖系统时间，让CURRENT_TIMESTAMP、NOW()等函数返回指定的伪造时间，直到会话结束或重置。

实现步骤： #

• 在Python中，每个用户的请求建立数据库连接后，立即执行以下SQL语句（替换为用户对应的伪造时间）：

  SET timestamp = UNIX_TIMESTAMP('2024-05-15 14:30:00');
  

• 后续该连接执行的所有SQL中，CURRENT_TIMESTAMP默认值、ON UPDATE CURRENT_TIMESTAMP都会使用这个伪造时间。

按用户隔离： #

确保每个用户的请求使用独立的数据库会话（或在连接复用前重置时间）。如果用连接池，需在获取连接后设置对应用户的伪造时间，请求结束后可选重置为系统时间：

SET timestamp = DEFAULT;

优缺点： #

• 优点：无需修改表结构或业务代码，直接利用MySQL原生功能，实现成本低。
• 缺点：依赖数据库会话隔离，连接池复用需额外处理，避免时间污染后续请求。

2. 自定义函数+会话变量（数据库层适配） #

通过自定义函数替代CURRENT_TIMESTAMP作为字段默认值，结合会话变量存储用户的伪造时间，实现自动填充。

实现步骤： #

1. 创建自定义函数（需RDS实例的权限支持）：

   DELIMITER //
   CREATE FUNCTION GET_FAKE_TIMESTAMP() RETURNS TIMESTAMP
   BEGIN
       IF @fake_timestamp IS NOT NULL THEN
           RETURN @fake_timestamp;
       ELSE
           RETURN CURRENT_TIMESTAMP;
       END IF;
   END //
   DELIMITER ;
   

2. 修改表结构，将time_created、time_updated的默认值替换为函数：

   ALTER TABLE your_table 
       MODIFY COLUMN time_created TIMESTAMP DEFAULT GET_FAKE_TIMESTAMP(),
       MODIFY COLUMN time_updated TIMESTAMP DEFAULT GET_FAKE_TIMESTAMP() ON UPDATE GET_FAKE_TIMESTAMP();
   

3. 在Python中，每个用户请求开始时，设置会话变量：

   # 假设db是你的数据库连接对象
   fake_time = "2024-05-15 14:30:00"
   db.execute(f"SET @fake_timestamp = '{fake_time}'")
   

按用户隔离： #

同样依赖会话隔离，每个用户的连接设置独立的@fake_timestamp变量，连接复用前需重置：

SET @fake_timestamp = NULL;

优缺点： #

• 优点：一次修改表结构后，业务代码只需设置变量，无需关注时间填充逻辑。
• 缺点：需要修改表结构，创建函数需数据库权限，连接池复用仍需处理变量重置。

3. 应用层ORM事件监听（灵活按用户隔离） #

如果使用SQLAlchemy等ORM框架，可以通过事件监听在应用层自动替换时间字段值，完全脱离数据库层的依赖，天然支持按用户隔离。

实现步骤（以Flask+SQLAlchemy为例）： #

1. 在请求上下文绑定用户的伪造时间（比如Flask的g对象）：

   from flask import g, request
   
   @app.before_request
   def set_fake_time():
       # 从请求参数/用户会话中获取伪造时间，示例中直接赋值
       g.fake_timestamp = "2024-05-15 14:30:00"
   

2. 监听ORM的插入/更新事件，自动填充伪造时间：

   from sqlalchemy import event
   from your_app.models import BaseModel  # 你的基础模型类
   
   @event.listens_for(BaseModel, 'before_insert')
   def auto_set_created_time(mapper, connection, target):
       if hasattr(g, 'fake_timestamp'):
           target.time_created = g.fake_timestamp
           target.time_updated = g.fake_timestamp
   
   @event.listens_for(BaseModel, 'before_update')
   def auto_set_updated_time(mapper, connection, target):
       if hasattr(g, 'fake_timestamp'):
           target.time_updated = g.fake_timestamp
   

按用户隔离： #

通过请求上下文（如Flask的g、Django的request）绑定每个用户的伪造时间，不同用户的请求上下文独立，天然实现隔离。

优缺点： #

• 优点：无需修改数据库，完全在应用层控制，按用户隔离逻辑灵活，不依赖数据库连接会话。
• 缺点：如果使用原生SQL而非ORM，需手动处理时间字段的赋值逻辑。

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