jonathan/hostr
jonathan/hostr
1
0
Fork 0
Code Issues Pull requests 4 Projects Releases Packages 1 Wiki Activity Actions

Refactor mongodb connection middleware

Browse source
This commit is contained in:
Jonathan Cremin 2015-08-22 13:35:06 +01:00
parent 51968374dd
commit 0fea2d7158
4 changed files with 45 additions and 65 deletions
Download patch file Download diff file Expand all files Collapse all files

23
api/app.js
View file

@ -1,5 +1,6 @@
import koa from 'koa';
import route from 'koa-route';
import stats from 'koa-statsd';
import websockify from 'koa-websocket';
import logger from 'koa-logger';
import compress from 'koa-compress';
@ -10,12 +11,11 @@ import redis from 'redis-url';
import coRedis from 'co-redis';
import raven from 'raven';
import auth from './lib/auth';
import mongoConnect from '../config/mongo';
import mongo from '../lib/mongo';
import * as user from './routes/user';
import * as file from './routes/file';
import debugname from 'debug';
const debug = debugname('hostr-api');
import stats from 'koa-statsd';
import StatsD from 'statsy';
if (process.env.SENTRY_DSN) {
@ -62,26 +62,9 @@ co(function*() {
console.error(err);
});
let mongoConnecting = false;
const mongoDeferred = {};
mongoDeferred.promise = new Promise(function(resolve, reject) {
mongoDeferred.resolve = resolve;
mongoDeferred.reject = reject;
});
function* getMongo() {
if (!mongoConnecting) {
mongoConnecting = true;
const db = yield mongoConnect();
mongoDeferred.resolve(db);
return db;
} else {
return mongoDeferred.promise;
}
}
app.use(mongo());
function* setupConnections(next){
this.db = yield getMongo();
this.redis = coRedisConn;
yield next;
}