forked from mirror/misskey
42 lines
942 B
TypeScript
42 lines
942 B
TypeScript
import config from '../conf';
|
|
|
|
const u = config.mongodb.user ? encodeURIComponent(config.mongodb.user) : null;
|
|
const p = config.mongodb.pass ? encodeURIComponent(config.mongodb.pass) : null;
|
|
|
|
const uri = u && p
|
|
? `mongodb://${u}:${p}@${config.mongodb.host}:${config.mongodb.port}/${config.mongodb.db}`
|
|
: `mongodb://${config.mongodb.host}:${config.mongodb.port}/${config.mongodb.db}`;
|
|
|
|
/**
|
|
* monk
|
|
*/
|
|
import * as mongo from 'monk';
|
|
|
|
const db = mongo(uri);
|
|
|
|
export default db;
|
|
|
|
/**
|
|
* MongoDB native module (officialy)
|
|
*/
|
|
import * as mongodb from 'mongodb';
|
|
|
|
let mdb: mongodb.Db;
|
|
|
|
const nativeDbConn = async (): Promise<mongodb.Db> => {
|
|
if (mdb) return mdb;
|
|
|
|
const db = await ((): Promise<mongodb.Db> => new Promise((resolve, reject) => {
|
|
(mongodb as any).MongoClient.connect(uri, (e, client) => {
|
|
if (e) return reject(e);
|
|
resolve(client.db(config.mongodb.db));
|
|
});
|
|
}))();
|
|
|
|
mdb = db;
|
|
|
|
return db;
|
|
};
|
|
|
|
export { nativeDbConn };
|