被数据库延迟加载坑了一把

背景

目前做Flask开发用的数据库框架是peewee。peewee查询时默认是延迟加载,也就是说一个查询语句不会立刻将所有结果返回,而是等到使用这个查询结果时才真正从数据库中查询出来。在这样一个背景下,就踩了这个坑。

过程

按照我需要处理的业务逻辑,首先要记录用户使用记录,然后给用户重置数据,最后恢复用户的使用记录。

1.查询记录保存到变量
在这业务逻辑下我想到的就是首先将用户的使用记录查询出来

learn_status = sm.UserTask.select().where(sm.UserTask.learning_status.in_([1,2]), sm.UserTask.plan_id==plan_id, sm.UserTask.user_id==user_id)

2.然后给用户重置

此处省略过程

3.从变量生成列表,恢复记录

learn_finished_days = [item.date for item in learn_status]
sm.UserTask.update(learning_status=0, checkin_status=0).where(sm.UserTask.user_id==34113904, sm.UserTask.plan_id==2).execute()
sm.UserTaskTrain.update(is_finish=0).where(sm.UserTaskTrain.user_id==34113904, sm.UserTaskTrain.plan_id==2).execute()

当我恢复记录时就傻眼了,发现一个记录都没有了,仔细一想问题就出现在peewee的延迟加载上,在第一步中查询出记录,此时并没有真正将记录赋值给变量,延迟加载机制生效了,而在第二步中又删除了用户所有的记录,所以最后在第三步中使用变量写列表生成式时,就找不到相应的记录。

教训是惨痛的,最后是我手动修复了那些有问题的数据。如果给我一次重来的机会,避免这个问题: 在数据库查询之后立刻生成列表推导式。当后面删除真正的数据时也不会有影响。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章