首页 > 代码库 > 在 mongodb 终端环境下写多行 javascript 代码、函数
在 mongodb 终端环境下写多行 javascript 代码、函数
工作中碰到一个问题,需要把某个 collection 中的某些符合条件的数据取出来,逐行处理其中某些字段。mongodb 终端下支持直接写 js 代码、函数,也可以运行 js 文件。
1 首先需要设置 mongo 终端的代码编辑器,不设置的话只能输入一行代码文件,无法处理大段 js 逻辑
进入 mongo 终端后,输入
f={}
回车后,继续输入:
edit f
未设置过相关系统变量的会收到如下提示:
please define EDITOR as a JavaScript string or as an environment variable
我们可以输入以下语句,设置 EDITOR:
EDITOR="/usr/bin/vim"
这样就临时设置了 vim 编辑器作为后续的 js 代码编辑器。此为临时方法,关闭mongo 后再次使用就必须再次设置。可以试着将 EDITOR="/usr/bin/vim" 加入到系统环境变量中,我暂时没尝试。。。
2 写 js 代码:设置好编辑器后,终端输入:
tempFunc = {}
回车后继续输入:
edit tempFunc
就会进入到刚刚设置的编辑器环境,我这里是 vim 环境。注意:vim 中输入的所有内容,会被替换成 “tempFunc = “ 中等号的右部分。所以如果不是只想编辑一行代码的话,最好是在 vim 中编辑成一个函数,确保替换后会变成 tempFunc = function(){…} 的形式。我在 vim 中编辑的完整内容如下:
注意:
一定要仔细检查语法,如果不小心打错了字,调用时候是没有效果的,也没有办法再次编辑,只能从头再编辑一遍!!!
编辑好以后 wq 退出 vim,到此我们就完整定义了一个 tempFunc 函数。
在终端中输入函数调用:
tempFunc()
mongo 会把 entities 表中所有 grade<2 的数据取出来,然后对 grade +1,再写回数据库。
注:
1:这里 update语句 写的有一个问题:原来的 grade 如果是 int32 类型,写入后会变成 double 类型。包括使用 inc、dec 之类的操作,都会把 int32 类型的数据覆写成 double 类型。用 NumberInt 函数就可以解决
可以把 update 语句改成这样:
db.entities.update({_id:rowData["_id"]}, {$set:{grade:NumberInt(newGrade)}}, {multi:false});
2:尽量定义成一个函数体,方便外界整体调用。如果不想定义成一个函数,可以一开始的时候直接 输入 edit{}, 然后再 vim 中删掉默认保留的{},就可以写一行行的 js 代码了,最后退出 vim 的时候会压缩成一行代码,类似于这种,然后按回车就直接执行了:
var cursor = db.entities.find(); while(cursor.hasNext()){ var row = cursor.next(); print(row["grade"]); }
<style></style>
在 mongodb 终端环境下写多行 javascript 代码、函数