Node.js 多模塊共享數(shù)據(jù)庫連接

zhouzhou 發(fā)布于2019-06-26 16:48 / 2508人閱讀

摘要：本文介紹了在項(xiàng)目中不同模塊之間共享數(shù)據(jù)庫連接的方法。專門負(fù)責(zé)和數(shù)據(jù)庫交互，用戶和航班這兩個(gè)模塊都需要連接數(shù)據(jù)庫，一開始我的代碼是這樣的引用連接數(shù)據(jù)庫引用連接數(shù)據(jù)庫且不說這種寫法一點(diǎn)都不，這種方式本身就是錯(cuò)誤的。

本文介紹了在 Node.js 項(xiàng)目中不同模塊之間共享數(shù)據(jù)庫連接的方法。

這個(gè)標(biāo)題本身就是一個(gè)命題，因?yàn)槭褂媚J(rèn)方式的情況下，一個(gè) Node.js 應(yīng)用里的各個(gè)模塊都是共享的同一個(gè)數(shù)據(jù)庫連接。但是如果姿勢(shì)不對(duì)，可能會(huì)很丑陋，甚至可能會(huì)出錯(cuò)。

你可以忽略下面這部分，直接切入正題。

背景

最近在做專業(yè)課程設(shè)計(jì)，題目是“機(jī)票預(yù)訂管理系統(tǒng)”。需求比較簡(jiǎn)單，就試著拿最近在學(xué)的 Node.js 來做了。本來還在調(diào)研用何種 Node.js 框架比較合適，看了幾個(gè)框架之后發(fā)現(xiàn)這是殺雞用牛刀，有看文檔查資料的時(shí)間還不如直接動(dòng)手寫了。最后寫完我會(huì)把代碼放到 Github 上，歡迎大家批評(píng)指正。

數(shù)據(jù)庫方面，以為我比較熟悉和喜歡 JSON （SQL 沒學(xué)好就承認(rèn)唄-_-#），所以就選擇了 MongoDB。Node + Mongo 是近幾年越來越熱門的后端組合，網(wǎng)上有很多關(guān)于如何一起使用的資料。但為了節(jié)約時(shí)間（課程設(shè)計(jì)也就一個(gè)多星期），把精力多集中在系統(tǒng)和邏輯上，我用了 Mongoose 這個(gè)專門用于 MongoDB 數(shù)據(jù)建模的 Node.js 擴(kuò)展，用它來大大減少操作數(shù)據(jù)庫的代碼。

正題

我建立了兩個(gè)數(shù)據(jù)模型（Model），一個(gè)是用戶（User），一個(gè)是航班（Flight），分別封裝到了 user.js, flight.js 這兩個(gè)模塊（Module）里面。Model 專門負(fù)責(zé)和數(shù)據(jù)庫交互，用戶和航班這兩個(gè)模塊都需要連接數(shù)據(jù)庫，一開始我的代碼是這樣的：

// ----- user.js -----
// require mongoose.js 引用mongoose.js
var M = require("mongoose");
// connect to database 連接數(shù)據(jù)庫
M.connect("mongodb://localhost/test");
// ... some other code ...

// ----- flight.js -----
// require mongoose.js 引用mongoose.js
var M = require("mongoose");
// connect to database 連接數(shù)據(jù)庫
M.connect("mongodb://localhost/test");
// ... some other code ...

// ----- models.js -----
var user = require("./user"),
    flight = require("./flight");

// ----- index.js -----
var Models = require("./models");

且不說這種寫法一點(diǎn)都不 DRY，這種方式本身就是錯(cuò)誤的。當(dāng)我運(yùn)行 index.js 時(shí)，會(huì)出現(xiàn)如下錯(cuò)誤。

> node index.js
> Connection error: { [Error: Trying to open unclosed connection.] state: 2 }

錯(cuò)誤是：嘗試打開未關(guān)閉的連接。

所以我們應(yīng)該在一個(gè)地方連接一次數(shù)據(jù)庫，然后其他需要連接數(shù)據(jù)庫的模塊通過這個(gè)模塊來和數(shù)據(jù)庫交互。就好像插線板，義無反顧地吼叫道：“墻上就一個(gè)插座，你們不要搶了！放著我來！你們。。。就可以了！”

具體方案

我們把連接數(shù)據(jù)庫的動(dòng)作放到一個(gè)模塊里，并且把連接暴露給整個(gè)應(yīng)用中的其他模塊，然后其他需要連接數(shù)據(jù)庫的模塊引用這個(gè)連接即可。

// ----- database.js -----
var M = require("mongoose");
M.connect("mongodb://localhost/test");
// reference to the database connection 為這個(gè)連接創(chuàng)建一個(gè)引用
var db = M.connection;
// expose to modules that require database.js 把這個(gè)引用暴露給引用 database 模塊的其他模塊
module.exports = db;

// ----- user.js ----- flight.js 類似 -----
// ... some other code ...
// 我們會(huì)在 models.js 中，把數(shù)據(jù)庫連接的引用作為參數(shù)傳進(jìn)來
module.exports = function( db ){
    if( db ){
        // ... do things with the connection ... 如果連接了數(shù)據(jù)庫，就可以執(zhí)行數(shù)據(jù)庫相關(guān)的操作了
    }
}

// ----- models.js -----
// require database module, retrieve the reference to database connection 引用 databse 模塊，獲取數(shù)據(jù)庫連接的引用
var db = require("./database");
// 把數(shù)據(jù)庫連接的引用傳入需要連接數(shù)據(jù)庫的模塊，任務(wù)完成！
var user = require("./user")( db ),
    flight = require("./flight")( db );

這就是讓一個(gè) Node.js 應(yīng)用的多個(gè)模塊共享數(shù)據(jù)庫連接的一種方法。是我在 StackOverflow 上面看到的。如果你有更好的方法，歡迎在評(píng)論中分享給大家！