index.js

const mongoose = require('mongoose')
const Q = require('q')

const CachePage = require('./models/CachePage')
const CachePageList = require('./models/CachePageList')
const CacheTagList = require('./models/CacheTagList')
const CacheProfile = require('./models/CacheProfile')
const CacheGroup = require('./models/CacheGroup')
const CacheGroupList = require('./models/CacheGroupList')

const collections = {
  'page': CachePage,
  'pages': CachePageList,
  'tags': CacheTagList,
  'user' : CacheProfile,
  'group' : CacheGroup,
  'groups' : CacheGroupList
}

let isDebug
let isDisabled

function start({
  mongoDsn,
  debug,
  disabled
}) {

  isDebug = debug
  isDisabled = disabled

  //mongoose.set('debug', debug)

  const options = {
    autoIndex: false, // Don't build indexes
    useNewUrlParser: true,
    poolSize: 20, // Maintain up to 10 socket connection
    bufferMaxEntries: 0,
    connectTimeoutMS: 200000, // Give up initial connection after 10 seconds
    socketTimeoutMS: 300000, // Close sockets after 45 seconds of inactivity
    family: 4,
    useFindAndModify: false,
    useCreateIndex: true,
    useUnifiedTopology: true,
  }

  setTimeout(function () {
    const mongoDB = mongoose.connect(mongoDsn, options)
      //.then(() => console.log('connected to mongo'))
      .catch((err) => console.log('error connecting to mongo\n\r' + err))
  })
}

function exists(type, websiteId, params) {
  if (!collections[type]) throw new Error(`[gc cache] model for ${type} not found`)
  return collections[type].exists(websiteId, params)
}

function update(type, websiteId, params, content) {
  if (!collections[type]) throw new Error(`[gc cache] model for ${type} not found`)
  return collections[type].updateOrCreate(websiteId, params, content)
}

function remove(type) {
  if (!collections[type]) throw new Error(`[gc cache] model for ${type} not found`)
  return collections[type].deleteAll().then(
    result => console.log(`[cache ${type}] ${result.deletedCount} results removed`),
    err => console.log(`[cache ${type}] error`, err),
  )
}

function clear(cb) {
  return Promise.all(
    Object.keys(collections).map(remove)
  ).then(cb)
}

function get(type, websiteId, params, res, generate) {
  exists(type, websiteId, params).then(
    cached => {
      if (isDisabled) {
        generate().then(content => res.send(content))
      }
      else if (cached && cached.valid) {
        if (isDebug) console.log(`[cache ${type}] found ${JSON.stringify(params)}`)
        res.append('Last-Modified', (new Date(cached.updated_at)).toUTCString())
        res.send(cached.content)
      }
      else {
        if (isDebug) console.log(`[cache ${type}] generate ${JSON.stringify(params)}`)
        generate().then(content => {
          if (content.indexOf('data-ignore-caching') === -1) {
            update(type, websiteId, params, content).catch(err => console.error('[gc cache]', err))
          }
          res.send(content)
        })
      }
    },
    err => console.error('[gc cache]', err)
  )
}

function exit() {
  mongoose.disconnect()
}

module.exports = { start, exists, update, get, remove, clear, exit }