一日一技:使用Pymongo实现更新并返回数据

有这样一个业务场景:

我有100篇故事,放在 MongoDB 里面。我做了一个 web 接口,每次请求返回一篇故事。希望能够实现:

每次请求返回的故事都不一样,在100次请求里面,每篇故事都需要返回。直到100篇故事全部返回完成。第101次与第1次相同,第102次与第2次相同……

为了实现这个目的,在 MongoDB 里面每一篇故事有3个字段:

{"title": "故事标题", "content": "故事正文", "count": "请求计数"}

如果只有一个人请求这个接口,并且他会等着上一次请求返回完成了再请求下一次,那么关键代码可以这样写:

  • 一开始,所有故事的 count字段都是0
import pymongo

handler = pymongo.MongoClient().story_project.story

story = list(handler.find().sort('count', 1).limit(1))[0]
_id = story['_id']
handler.update_one({'_id': _id}, {'$inc': {'count': 1}})

每次查询都取 count 最小的这一条。查询完成以后,就更新 count 字段。

但是,如果有很多人同时访问这个 web 接口,那么就会出现并发读写冲突的问题。当第一个人读取了一篇故事,还没有来得及更新 count 时,另一个人又访问接口,这样他们两人就会得到同一篇故事。

为了解决这个问题,就需要把查询和更新作为一个原子操作来执行。此时,就要用到 pymongo 的find_one_and_update方法。它根据特定条件查询一条记录,同时更新这条记录的 count 值。这两个行为是通过一条命令完成的,即使多个人同时请求也不会出现并发冲突的情况。

find_one_and_update的使用方法如下:

import pymongo
handler.find_one_and_update({},
                         {'$inc':{'count': 1}},
                         sort=[('count', 1)])

其中,第一个参数表示查询条件,跟find的第一个参数一致。第二个参数表示更新的内容,与update_one的第二个参数一致。sort表示排序方式,它的值是一个包含元组的列表,元组的第一个元素为被排序的字段名,第二个元素为1表示升序,为-1表示降序。

update_one_and_update还有其他参数,可以参考它的官方文档[1]

参考资料

[1]

官方文档: https://api.mongodb.com/python/current/api/pymongo/collection.html#pymongo.collection.Collection.find_one_and_update

原文发布于微信公众号 - 未闻Code(itskingname)

原文发表时间:2019-09-28

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

扫码关注云+社区

领取腾讯云代金券