详解Node.js模块间共享数据库连接的方法

在Node.js项目中，数据库连接通常是需要共享的。不同的模块可能需要访问同一个数据库，因此需要实现数据库连接的共享。本文将详细介绍如何实现模块间共享数据库连接的方法。本文的代码将基于MongoDB数据库进行演示。

初始化数据库连接

首先，我们需要在项目的入口文件中初始化数据库连接，并将连接实例保存到全局对象中。可以使用Mongoose库来创建数据库连接。以下是一个示例代码：

const mongoose = require('mongoose');

const url = 'mongodb://localhost:27017/mydatabase'; //数据库连接字符串
mongoose.connect(url, {useNewUrlParser: true, useUnifiedTopology: true}) //连接数据库
    .then(() =>{
        console.log('Database connected successfully!');
    })
    .catch(err => {
        console.log('Error connecting to database!');
        console.error(err);
    });

global.db = mongoose.connection; //将数据库连接对象保存到全局对象中

连接成功后，将连接实例保存到global对象上，方便在不同的模块中共享。

在模块中使用数据库连接

在需要使用数据库连接的模块中，可以通过引用全局对象来访问连接实例。以下是一个示例代码：

const mongoose = require('mongoose');

const mySchema = new mongoose.Schema({
    name: String,
    age: Number
});

const MyModel = mongoose.model('MyModel', mySchema);

exports.findAll = async () => {
    try {
        const results = await MyModel.find();
        return results;
    } catch (err) {
        console.error(err);
        throw err;
    }
}

上面这个模块暴露了一个findAll方法，用于查询数据库中的全部数据。在方法中，我们通过mongoose库创建一个数据模型MyModel，并通过该模型进行数据查询。注意，我们在模块中引用mongoose库，但并没有进行数据库连接操作。这是因为连接实例已经在入口文件中初始化，并保存到global对象中。

示例 1: 实现 REST API

在Node.js中，REST API是一种常用的API设计风格。在实现REST API时，通常需要对数据库进行增删改查操作。以下是一个示例代码：

const express = require('express');

const app = express();
const port = 3000;

app.get('/api/mymodel', async (req, res) => {
    try {
        const results = await global.db.model('MyModel').find();
        res.send(results);
    } catch (err) {
        console.error(err);
        res.status(500).send('Internal server error');
    }
});

app.post('/api/mymodel', async (req, res) => {
    const { name, age } = req.body;
    try {
        const newModel = new global.db.model('MyModel')({ name, age });
        await newModel.save();
        res.send('Saved successfully');
    } catch (err) {
        console.error(err);
        res.status(500).send('Internal server error');
    }
});

app.listen(port, () => {
    console.log(`Server listening on port ${port}`);
});

上述示例代码实现了一个简单的REST API，通过访问/api/mymodel接口获取全部数据，通过访问/api/mymodel接口创建新的数据。在示例代码中，我们使用global.db.model方法来获取数据模型，并对模型进行CRUD操作。由于数据库连接实例已经保存在全局对象中，可以在不同的模块中共享。

示例 2: 使用工厂方法

在某些情况下，我们希望在模块中创建数据模型，而不是在入口文件中创建。这时可以使用工厂方法来创建数据模型并将数据库连接实例作为参数传递。以下是一个示例代码：

const mongoose = require('mongoose');

const createModel = (db) => {
    const mySchema = new mongoose.Schema({
        name: String,
        age: Number
    });

    return db.model('MyModel', mySchema);
}

module.exports = createModel;

这个模块可以接收一个数据库连接实例作为参数，并返回一个数据模型。在上述示例代码中，我们使用db.model方法来创建数据模型。接下来可以在其他模块中使用createModel方法来创建数据模型。

const db = global.db;
const createModel = require('./createModel');

const MyModel = createModel(db);

exports.findAll = async () => {
    try {
        const results = await MyModel.find();
        return results;
    } catch (err) {
        console.error(err);
        throw err;
    }
}

在上述示例代码中，我们将数据库连接实例db作为参数传入createModel方法，并返回数据模型MyModel。与前面的示例类似，我们在方法中使用MyModel进行数据查询。

总的来说，基于Node.js的模块化设计，通过将数据库连接实例保存到全局对象中，可以方便地实现模块间的数据库连接共享，避免重复连接数据库的开销。