MongoDB的upsert状态判断和pymongo使用方法

在mongo中，有一个命令非常的方便，就是upsert，顾名思义就是update+insert的作用

根据条件判断有无记录，有的话就更新记录，没有的话就插入一条记录

upsert的使用方法：

MongoDB shell version: 2.4.9
connecting to: test
> use date
switched to db date
> db.user.insert({"name":"user1", "age":12, "sex":"male"})
> db.user.insert({"name":"user2", "age":13, "sex":"male"})
> db.user.insert({"name":"user3", "age":14, "sex":"male"})
> db.user.find()
{ "_id" : ObjectId("54c75876662ee1617463f0a3"), "name" : "user1", "age" : 12, "sex" : "male" }
{ "_id" : ObjectId("54c75880662ee1617463f0a4"), "name" : "user2", "age" : 13, "sex" : "male" }
{ "_id" : ObjectId("54c7588d662ee1617463f0a5"), "name" : "user3", "age" : 14, "sex" : "male" }
> db.user.update({"name":"user1"}, {"$set":{"age":21}}, {"upsert":"true"})
> db.user.find()
{ "_id" : ObjectId("54c75876662ee1617463f0a3"), "name" : "user1", "age" : 21, "sex" : "male" }
{ "_id" : ObjectId("54c75880662ee1617463f0a4"), "name" : "user2", "age" : 13, "sex" : "male" }
{ "_id" : ObjectId("54c7588d662ee1617463f0a5"), "name" : "user3", "age" : 14, "sex" : "male" }
> db.user.update({"name":"user1"}, {"$set":{"age":21}, "$setOnInsert":{"sex":"female"}}, {"upsert":"true"})
> db.user.find()
{ "_id" : ObjectId("54c75876662ee1617463f0a3"), "name" : "user1", "age" : 21, "sex" : "male" }
{ "_id" : ObjectId("54c75880662ee1617463f0a4"), "name" : "user2", "age" : 13, "sex" : "male" }
{ "_id" : ObjectId("54c7588d662ee1617463f0a5"), "name" : "user3", "age" : 14, "sex" : "male" }
> db.user.update({"name":"user4"}, {"$set":{"age":21}, "$setOnInsert":{"sex":"female"}}, {"upsert":"true"})
> db.user.find()
{ "_id" : ObjectId("54c75876662ee1617463f0a3"), "name" : "user1", "age" : 21, "sex" : "male" }
{ "_id" : ObjectId("54c75880662ee1617463f0a4"), "name" : "user2", "age" : 13, "sex" : "male" }
{ "_id" : ObjectId("54c7588d662ee1617463f0a5"), "name" : "user3", "age" : 14, "sex" : "male" }
{ "_id" : ObjectId("54c75a07478a805237d087c3"), "name" : "user4", "age" : 21, "sex" : "female" }
> 

上面的代码可以看出，$set搭配$setOnInsert使用就可以实现更新和插入时的一些事件了。

比如实现更新数据，如果一些需要更新的，就用$set设置，如果有些如创建日期这种字段，那么使用$setOnInsert更新

相当方便，如果想知道刚才执行的upsert是更新还是插入，使用如下的方法即可（如下代码在上面的代码数据基础之上）

> db.user.update({"name":"user4"}, {"$set":{"age":21}, "$setOnInsert":{"sex":"female"}}, {"upsert":"true"})
> db.runCommand('getlasterror')
{
    "connectionId" : 35,
    "updatedExisting" : true,
    "n" : 1,
    "syncMillis" : 0,
    "writtenTo" : null,
    "err" : null,
    "ok" : 1
}
> db.user.update({"name":"user5"}, {"$set":{"age":21}, "$setOnInsert":{"sex":"female"}}, {"upsert":"true"})
> db.runCommand('getlasterror')
{
    "connectionId" : 35,
    "updatedExisting" : false,
    "upserted" : ObjectId("54c75caa478a805237d087c4"),
    "n" : 1,
    "syncMillis" : 0,
    "writtenTo" : null,
    "err" : null,
    "ok" : 1
}
> 

上面的updatedExisting 就可以反馈刚才的状态，true为更新操作，false为插入操作

当upsert为插入操作还可以返回ObjectId呢，是不是很方便~

其实上面的操作都是在mongo中直接操作的，很多时候更新数据都是使用脚本，以python为例

代码还是稍微有点不同的

pymongo：

#!/bin/env python
#encoding:utf-8
import os
import pymongo

DATA_DIR = 'data'

# 连接数据库
conn = pymongo.Connection('127.0.0.1', 27017)
#conn = pymongo.Connection('115.28.55.217', 27017)
db = conn.haha


db.users.remove()

# Insert & Save
db.users.insert({'name': 'user1', 'age': 16, 'index': 1})
db.users.insert({'name': 'user2', 'age': 17, 'index': 2})
db.users.insert({'name': 'user3', 'age': 18, 'index': 3})
db.users.save({'name': 'user4', 'age': 19, 'index': 4})

# Update
db.users.update({'name':'user1'}, {'$set': {'age': 20}})
json = {
        'age':20
}
db.users.update({'name':'user2'}, {'$set': json})

# Upsert
db.users.update({'name':'user5'}, {'$set': {'age': 22}, '$setOnInsert': {'index':5}}, upsert=True)

json = {
        'age':36
}
name = 'user5'
db.users.update({'name':name}, {'$set': json, '$setOnInsert': {'index':5}}, upsert=True)

因为python中不认识true，所以使用True，并且为了代码的可读性，建议使用upsert=True

当然，pymongo中的返回信息也是有区别的。

并不是db.runCommand('getlasterror')而是db.command('getlasterror')

那么使用如下代码就可以获取刚才update的类型了

status = db.command('getlasterror')['updatedExisting']
print status

记录完成