详解Node.js模块间共享数据库连接的方法
这个标题本身就是一个命题,因为使用默认方式的情况下,一个Node.js应用里的各个模块都是共享的同一个数据库连接。但是如果姿势不对,可能会很丑陋,甚至可能会出错。
你可以忽略下面这部分,直接切入正题。
背景
最近在做专业课程设计,题目是“机票预订管理系统”。需求比较简单,就试着拿最近在学的Node.js来做了。本来还在调研用何种Node.js框架比较合适,看了几个框架之后发现这是杀鸡用牛刀,有看文档查资料的时间还不如直接动手写了。最后写完我会把代码放到Github上,欢迎大家批评指正。
数据库方面,以为我比较熟悉和喜欢JSON(SQL没学好就承认呗-_-#),所以就选择了MongoDB。Node+Mongo是近几年越来越热门的后端组合,网上有很多关于如何一起使用的资料。但为了节约时间(课程设计也就一个多星期),把精力多集中在系统和逻辑上,我用了Mongoose这个专门用于MongoDB数据建模的Node.js扩展,用它来大大减少操作数据库的代码。
正题
我建立了两个数据模型(Model),一个是用户(User),一个是航班(Flight),分别封装到了user.js,flight.js这两个模块(Module)里面。Model专门负责和数据库交互,用户和航班这两个模块都需要连接数据库,一开始我的代码是这样的:
//-----user.js----- //requiremongoose.js引用mongoose.js varM=require('mongoose'); //connecttodatabase连接数据库 M.connect('mongodb://localhost/test'); //...someothercode... //-----flight.js----- //requiremongoose.js引用mongoose.js varM=require('mongoose'); //connecttodatabase连接数据库 M.connect('mongodb://localhost/test'); //...someothercode... //-----models.js----- varuser=require('./user'), flight=require('./flight'); //-----index.js----- varModels=require('./models');
且不说这种写法一点都不DRY,这种方式本身就是错误的。当我运行index.js时,会出现如下错误。
>nodeindex.js >Connectionerror:{[Error:Tryingtoopenunclosedconnection.]state:2}
错误是:尝试打开未关闭的连接。
所以我们应该在一个地方连接一次数据库,然后其他需要连接数据库的模块通过这个模块来和数据库交互。就好像插线板,义无反顾地吼叫道:“墙上就一个插座,你们不要抢了!放着我来!你们。。。就可以了!”
具体方案
我们把连接数据库的动作放到一个模块里,并且把连接暴露给整个应用中的其他模块,然后其他需要连接数据库的模块引用这个连接即可。
//-----database.js----- varM=require('mongoose'); M.connect('mongodb://localhost/test'); //referencetothedatabaseconnection为这个连接创建一个引用 vardb=M.connection; //exposetomodulesthatrequiredatabase.js把这个引用暴露给引用database模块的其他模块 module.exports=db; //-----user.js-----flight.js类似----- //...someothercode... //我们会在models.js中,把数据库连接的引用作为参数传进来 module.exports=function(db){ if(db){ //...dothingswiththeconnection...如果连接了数据库,就可以执行数据库相关的操作了 } } //-----models.js----- //requiredatabasemodule,retrievethereferencetodatabaseconnection引用databse模块,获取数据库连接的引用 vardb=require('./database'); //把数据库连接的引用传入需要连接数据库的模块,任务完成! varuser=require('./user')(db), flight=require('./flight')(db);
这就是让一个Node.js应用的多个模块共享数据库连接的一种方法。是我在StackOverflow上面看到的。如果你有更好的方法,欢迎在评论中分享给大家!