javascript - 从 nodejs 到 mongodb 或 mongoose 的动态数据库连接

Question

我正在尝试创建一个 Multi-Tenancy 应用 (saas)，其中每个客户端都有自己的数据库。

我的情况是:

我创建了一个中间件，它可以根据子域确定客户端是谁，然后从通用数据库中检索客户端的数据库连接信息。我不知道如何为此客户端建立连接对象以便能够在我的 Controller 中使用。我应该在中间件还是 Controller 中执行此操作？如果它在模型中，我如何传递连接字符串和参数(我可以使用 session ，但我不知道如何从模型中访问 session )。

我该怎么做？

1. 组织:我在哪里为客户端动态创建数据库连接？
2. 将连接参数注入(inject)/传递给 Controller ​​或模型(进行连接定义的地方)
3. 建立动态连接后，我如何为该客户端全局访问它？

这是我的中间件的一个例子，我想创建一个我想动态的 Mongoose 连接(传递客户端的连接信息):

function clientlistener() {
    return function (req, res, next) {
       console.dir('look at my sub domain  ' + req.subdomains[0]);
       // console.log(req.session.Client.name);

    if (req.session.Client && req.session.Client.name === req.subdomains[0]) {
          var options = session.Client.options;
          var url = session.Client.url
          var conn = mongoose.createConnection(url, options);
          next();
       }
    }
}

如何从 Controller 内部访问此连接对象？还是来自模型？

谢谢。

Answer 1

这是为了帮助可能与我处于类似情况的其他人。我希望它可以标准化。我不认为每次有人需要制作 Multi-Tenancy 应用程序时我们都必须重新发明轮子。

这个例子描述了一个 Multi-Tenancy 结构，每个客户端都有自己的数据库。 就像我说的那样，可能有更好的方法可以做到这一点，但是因为我自己没有得到帮助，所以这是我的解决方案。

以下是此解决方案的目标:

• 每个客户端都由子域标识，例如 client1.application.com，
• 应用程序检查子域是否有效，
• 应用程序从主数据库中查找并获取连接信息(数据库 url、凭据等)，
• 应用程序连接到客户端数据库(几乎移交给客户端)，
• 应用程序采取措施确保完整性和资源管理(例如，为同一客户端的成员使用同一数据库连接，而不是建立新连接)。

这里是代码

在你的 app.js 文件中

app.use(clientListener()); // checks and identify valid clients
app.use(setclientdb());// sets db for valid clients

我创建了两个中间件:

• clientListener - 识别正在连接的客户端，
• setclientdb - 识别客户端后，从主数据库获取客户端详细信息，然后建立与客户端数据库的连接。

clientListener 中间件

我通过检查请求对象的子域来检查客户是谁。我做了一堆检查以确保客户端是有效的(我知道代码很乱，并且可以变得更干净)。确保客户端有效后，我将客户端信息存储在 session 中。我还检查了如果客户信息已经存储在 session 中，则无需再次查询数据库。我们只需要确保请求子域与 session 中已经存储的子域匹配。

var Clients = require('../models/clients');
var basedomain = dbConfig.baseDomain;
var allowedSubs = {'admin':true, 'www':true };
allowedSubs[basedomain] = true;
function clientlistener() {
return function(req, res, next) {
    //console.dir('look at my sub domain  ' + req.subdomains[0]);
    // console.log(req.session.Client.name);

    if( req.subdomains[0] in allowedSubs ||  typeof req.subdomains[0] === 'undefined' || req.session.Client && req.session.Client.name === req.subdomains[0] ){
        //console.dir('look at the sub domain  ' + req.subdomains[0]);
        //console.dir('testing Session ' + req.session.Client);
        console.log('did not search database for '+ req.subdomains[0]);
        //console.log(JSON.stringify(req.session.Client, null, 4));
        next();
    }
    else{

        Clients.findOne({subdomain: req.subdomains[0]}, function (err, client) {
            if(!err){
                if(!client){
                    //res.send(client);
                    res.send(403, 'Sorry! you cant see that.');
                }
                else{
                    console.log('searched database for '+ req.subdomains[0]);
                    //console.log(JSON.stringify(client, null, 4));
                    //console.log(client);
                   // req.session.tester = "moyo cow";
                    req.session.Client = client;
                    return next();

                }
            }
            else{
                console.log(err);
                return next(err)
            }

        });
    }

   }
 }

module.exports = clientlistener;

setclientdb 中间件:

我再次检查所有内容，确保客户端有效。然后使用从 session 中检索到的信息打开与客户端数据库的连接。

我还确保将所有事件连接存储到一个全局对象中，以防止在每次请求时都与数据库建立新连接(我们不想让每个客户端的 mongodb 服务器的连接过载)。

var mongoose = require('mongoose');
//var dynamicConnection = require('../models/dynamicMongoose');
function setclientdb() {
    return function(req, res, next){
        //check if client has an existing db connection                                                               /*** Check if client db is connected and pooled *****/
    if(/*typeof global.App.clientdbconn === 'undefined' && */ typeof(req.session.Client) !== 'undefined' && global.App.clients[req.session.Client.name] !== req.subdomains[0])
    {
        //check if client session, matches current client if it matches, establish new connection for client
        if(req.session.Client && req.session.Client.name === req.subdomains[0] )
        {
            console.log('setting db for client ' + req.subdomains[0]+ ' and '+ req.session.Client.dbUrl);
            client = mongoose.createConnection(req.session.Client.dbUrl /*, dbconfigoptions*/);


            client.on('connected', function () {
                console.log('Mongoose default connection open to  ' + req.session.Client.name);
            });
            // When the connection is disconnected
            client.on('disconnected', function () {
                console.log('Mongoose '+ req.session.Client.name +' connection disconnected');
            });

            // If the Node process ends, close the Mongoose connection
            process.on('SIGINT', function() {
                client.close(function () {
                    console.log(req.session.Client.name +' connection disconnected through app termination');
                    process.exit(0);
                });
            });

            //If pool has not been created, create it and Add new connection to the pool and set it as active connection

            if(typeof(global.App.clients) === 'undefined' || typeof(global.App.clients[req.session.Client.name]) === 'undefined' && typeof(global.App.clientdbconn[req.session.Client.name]) === 'undefined')
            {
                clientname = req.session.Client.name;
                global.App.clients[clientname] = req.session.Client.name;// Store name of client in the global clients array
                activedb = global.App.clientdbconn[clientname] = client; //Store connection in the global connection array
                console.log('I am now in the list of active clients  ' + global.App.clients[clientname]);
            }
            global.App.activdb = activedb;
            console.log('client connection established, and saved ' + req.session.Client.name);
            next();
        }
        //if current client, does not match session client, then do not establish connection
        else
        {
            delete req.session.Client;
            client = false;
            next();
        }
    }
    else
    {
        if(typeof(req.session.Client) === 'undefined')
        {
           next();
        }
        //if client already has a connection make it active
        else{
            global.App.activdb = global.App.clientdbconn[req.session.Client.name];
            console.log('did not make new connection for ' + req.session.Client.name);
            return next();
        }

    }
    }
}

module.exports = setclientdb;

最后但并非最不重要的一点

由于我使用的是 mongoose 和原生 mongo 的组合，我们必须在运行时编译我们的模型。请看下文

将此添加到您的 app.js

// require your models directory
var models = require('./models');

// Create models using mongoose connection for use in controllers
app.use(function db(req, res, next) {
    req.db = {
        User: global.App.activdb.model('User', models.agency_user, 'users')
        //Post: global.App.activdb.model('Post', models.Post, 'posts')
    };
    return next();
});

解释:

就像我之前说的，我创建了一个全局对象来存储事件的数据库连接对象:global.App.activdb

然后我使用这个连接对象来创建(编译) Mongoose 模型，之后我将它存储在 req 对象的 db 属性中:req.db。我这样做是为了让我可以像这样在我的 Controller 中访问我的模型。

我的用户 Controller 示例:

exports.list = function (req, res) {
    req.db.User.find(function (err, users) {

        res.send("respond with a resource" + users + 'and connections  ' + JSON.stringify(global.App.clients, null, 4));
        console.log('Worker ' + cluster.worker.id + ' running!');
    });

};

我最终会回来清理这个。如果有人想帮助我，那就太好了。