這個標題本身就是一個命題,因為使用默認方式的情況下,一個 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 ----- // require mongoose.js 引用mongoose.js var M = require('mongoose'); // connect to database 連接數據庫 M.connect('mongodb://localhost/test'); // ... some other code ... // ----- flight.js ----- // require mongoose.js 引用mongoose.js var M = require('mongoose'); // connect to database 連接數據庫 M.connect('mongodb://localhost/test'); // ... some other code ... // ----- models.js ----- var user = require('./user'), flight = require('./flight'); // ----- index.js ----- var Models = require('./models');
且不說這種寫法一點都不 DRY,這種方式本身就是錯誤的。當我運行 index.js 時,會出現如下錯誤。
> node index.js > Connection error: { [Error: Trying to open unclosed connection.] state: 2 }
錯誤是:嘗試打開未關閉的連接。
所以我們應該在一個地方連接一次數據庫,然後其他需要連接數據庫的模塊通過這個模塊來和數據庫交互。就好像插線板,義無反顧地吼叫道:“牆上就一個插座,你們不要搶了!放著我來!你們。。。就可以了!”
具體方案
我們把連接數據庫的動作放到一個模塊裡,並且把連接暴露給整個應用中的其他模塊,然後其他需要連接數據庫的模塊引用這個連接即可。
// ----- database.js ----- var M = require('mongoose'); M.connect('mongodb://localhost/test'); // reference to the database connection 為這個連接創建一個引用 var db = M.connection; // expose to modules that require database.js 把這個引用暴露給引用 database 模塊的其他模塊 module.exports = db; // ----- user.js ----- flight.js 類似 ----- // ... some other code ... // 我們會在 models.js 中,把數據庫連接的引用作為參數傳進來 module.exports = function( db ){ if( db ){ // ... do things with the connection ... 如果連接了數據庫,就可以執行數據庫相關的操作了 } } // ----- models.js ----- // require database module, retrieve the reference to database connection 引用 databse 模塊,獲取數據庫連接的引用 var db = require('./database'); // 把數據庫連接的引用傳入需要連接數據庫的模塊,任務完成! var user = require('./user')( db ), flight = require('./flight')( db );
這就是讓一個 Node.js 應用的多個模塊共享數據庫連接的一種方法。是我在 StackOverflow 上面看到的。如果你有更好的方法,歡迎在評論中分享給大家!