MongoDB用作数据存储时的并发处理

背景

我负责的是一个游戏服务器
最近项目新增了激活码的需求,需要给每个购买的用户发放一个激活码,但是激活码并不是我们的游戏服务器生成的,而是用的接入平台那边给的指定的激活码,平台的激活码是唯一的,每一个激活码发放的时候都会发一个短信给用户,短信通知的功能很好实现,关键在于服务器要保证的是不能重复发放激活码。
由于激活码是平台那里提前给出的,我要先把激活码存入数据库,由服务器去发放。举个例子,平台给了500个激活码,我就要把这500个激活码存入数据库,如果有500个用户来买游戏,服务器就要把这500个激活码发到每个用户,每个激活码只能用一次,不能重复发放。
根据万物皆对象的程序猿定理,我创建了激活码对象,具体属性如下:
key 激活码(平台那里给的激活码)
status 激活码状态(主要区分激活码是否被发放,0未发放/1已发放/2已使用)
createTime 创建时间
useTime 使用时间
uid 用户的UID(用来记录这个激活码发给了哪个用户)
isSms 是否已发送短信(0未发送/1已发送)

说下几个关键属性的用处,key是激活码,因为要保证唯一性所以在key上建立唯一索引;status用来判断激活码使用状态,查询未发放的激活码时候就需要通过这个字段进行筛选;uid要跟着激活码状态的改变添加进去,status从0变成1的时候必须要把uid更新进去。在导入激活码的时候也给每个激活码设置status为0,至于isSms是否成功发送短信是另一套逻辑进行操作的,不是本篇重点。

由于Java中操作MongoDB的工具甚多,方法不尽相同,所以我用MongoDB命令去写下列操作,使用时请替换成各自工具的方法
正常来说,如果你要更新一条数据的某几个字段,就先要把他找出来,即在使用update之前要先find
① db.activation_code.find({"status":0}).limit(1)
这样我就找到了一个未被使用的激活码,然后下一步就是把这个激活码的uid和status更新,然后接着执行更新操作
② db.activation_code.update({"key":"qwer"},{$set:{"uid":"123213","status":1}})
整个流程看似没什么问题,但在实际应用的时候不会是单线程运行的,如果同时两个线程都在执行①,MongoDB中find和limit会找到符合条件的第一条结果,那么这两个线程得到的将会是同一个激活码,然后两个线程都对这个激活码进行②更新操作,那么肯定会有一个uid被覆盖掉,这种情况肯定是不安全的。

这里有两种解决方案,一种是使用乐观锁,另一种是先执行更新,再去查uid对应的key。

先说乐观锁,查询和①一样,只是在更新的时候带条件去更新
③ db.activation_code.update({"key":"qwer","status":0},{$set:{"uid":"123213","status":1}})
即在②的基础上加上状态为0的条件,更新的时候会找到key为qwer且状态为0的激活码,如果该激活码已经被使用status肯定不为0,更新条件就查不到这条记录。每次更新都会有返回值,通过返回值判断是否更新成功,如果没有更新成功,则继续执行①和③操作,直到更新成功为止。

另一种是先执行更新,再去查对应的key
④ db.activation_code.update({"status":0},{$set:{"uid":"123213","status":1}},{upsert:true})
⑤ db.activation_code.find({"uid":"123213"}).limit(1)
先是④更新,再去通过更新时的uid去找对应的激活码key,即⑤,然后把激活码key返回给用户。
更新的时候有条件,必须是status为0的才能执行更新,MongoDB的更新语句有隐藏属性multi,默认为false时只更新符合条件的第一个文档,为true时更新符合条件的所有文档,这里我们不设置multi即为默认false。
upsert属性是不存在就插入这条记录,应对激活码用完,没有新的激活码存入这种情况,不至于程序挂掉。实际上是不可能会发生这种情况的,但是如果真的发生了,如果真的发生了,我们还能去查哪些记录只有uid没有key,从而去补发激活码。

这里讲一下为什么要先写入
MongoDB的写入操作是原子操作,在MongoDB2.0及以前的版本,MongoDB会在写入的时候对所有数据库上锁,这一点和MySQL的行级锁不同,MongoDB2.0是全局锁,而且写入的优先级是大于读的,当时因为效率问题屡被吐槽,在MongoDB2.2的时候实现了数据库锁,只对该数据库上锁,不影响其他数据库,依然没有逃脱被吐槽的命运,直到3.0的版本MongoDB加入了表级锁,执行写操作的时候会对整张表上锁,不影响其他表的操作,性能有了大的改善,才逐渐体现出NoSQL的优点。

这两种方案,乐观锁依靠的是服务器去执行写入操作,每次写入都有迹可循。而先更新再查找依靠的是数据库本身的性能,在更新的时候带入条件,每次更新都会找符合条件的数据进行更新,MongoDB的表级锁在这时候就起到了关键作用。

参考文档

MongoDB文档http://doc.redisfans.com/string/set.html

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 214,951评论 6 497
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,606评论 3 389
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 160,601评论 0 350
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,478评论 1 288
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,565评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,587评论 1 293
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,590评论 3 414
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,337评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,785评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,096评论 2 330
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,273评论 1 344
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,935评论 5 339
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,578评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,199评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,440评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,163评论 2 366
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,133评论 2 352

推荐阅读更多精彩内容