MongoDB
 sql >> Base de Dados >  >> NoSQL >> MongoDB

Conexão de banco de dados dinâmica para mongodb ou mongoose de nodejs


Isso é para ajudar outras pessoas que podem se encontrar em situação semelhante à minha. Espero que possa ser padronizado. Eu não acho que deveríamos ter que reinventar a roda toda vez que alguém precisa fazer um aplicativo multi-tenant.

Este exemplo descreve uma estrutura multi-tenant com cada cliente tendo seu próprio banco de dados. Como eu disse, pode haver uma maneira melhor de fazer isso, mas porque eu não consegui ajuda, esta foi a minha solução.

Então, aqui estão os objetivos desta solução:
  • cada cliente é identificado por subdomínio, por exemplo, client1.application.com,
  • o aplicativo verifica se o subdomínio é válido,
  • o aplicativo pesquisa e obtém informações de conexão (url do banco de dados, credenciais, etc.) do banco de dados mestre,
  • aplicativo se conecta ao banco de dados do cliente (praticamente passa para o cliente),
  • o aplicativo toma medidas para garantir a integridade e o gerenciamento de recursos (por exemplo, usar a mesma conexão de banco de dados para membros do mesmo cliente, em vez de fazer uma nova conexão).

Aqui está o código


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

Criei dois middlewares:
  • clientListener - para identificar o cliente que está se conectando,
  • setclientdb - obtém os detalhes do cliente do banco de dados mestre, depois que o cliente é identificado, e estabelece a conexão com o banco de dados do cliente.

middleware clientListener


Eu verifico quem é o cliente verificando o subdomínio do objeto de solicitação. Eu faço várias verificações para ter certeza de que o cliente é válido (eu sei que o código está confuso e pode ser mais limpo). Depois de garantir que o cliente é válido, armazeno as informações do cliente na sessão. Também verifico se as informações do cliente já estão armazenadas na sessão, não há necessidade de consultar o banco de dados novamente. Só precisamos garantir que o subdomínio da solicitação corresponda ao que já está armazenado na sessão.
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 middleware:


Eu verifico tudo novamente, certificando-me de que o cliente é válido. Em seguida, a conexão com o banco de dados do cliente com as informações recuperadas da sessão é aberta.

Eu também me certifico de armazenar todas as conexões ativas em um objeto global, para evitar novas conexões com o banco de dados a cada solicitação (não queremos sobrecarregar cada servidor mongodb de clientes com conexões).
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;

Por último, mas não menos importante


Como estou usando uma combinação de mongoose e mongo nativo, temos que compilar nossos modelos em tempo de execução. Por favor veja abaixo

Adicione isso ao seu 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();
});

Explicação:

Como eu disse anteriormente, criei um objeto global para armazenar o objeto de conexão de banco de dados ativo:global.App.activdb

Então eu uso este objeto de conexão para criar (compilar) o modelo mangusto, depois de armazená-lo na propriedade db do objeto req:req.db . Eu faço isso para que eu possa acessar meus modelos no meu controlador como este, por exemplo.

Exemplo do meu controlador de usuários:
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!');
    });

};

Eu vou voltar e limpar isso eventualmente. Se alguém quiser me ajudar, que bom.