rest-api-node-express-mongoose/db.js at main · chuksys/rest-api-node-express-mongoose · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
const mongoose = require('mongoose')

//setup mongoDB
function connect() {
    const keys = require('./config/keys')
    return new Promise((resolve, reject) => {
        if(process.env.NODE_ENV === 'test') {
            mongoose
                .connect(keys.mongoTestURI, {useNewUrlParser: true, useUnifiedTopology: true})
                .then((res, err) => {
                    if (err) return reject(err)
                        resolve()
                        //console.log('MongoDB connected')

                })

        } else {
            mongoose
                .connect(keys.mongoDevURI, {useNewUrlParser: true, useUnifiedTopology: true})
                .then((res, err) => {
                    if (err) return reject(err)
                    resolve()
                    console.log('MongoDB connected')

                })
                .catch(err => console.log(`Error: ${err}`))
        }
    })
}

function close() {
    return mongoose.disconnect()
}

module.exports = {
    connect: connect,
    close: close
}