1 files changed, 67 insertions, 0 deletions

diff --git a/models/poolRequests.js b/models/poolRequests.js
new file mode 100644
index 000000000..0f488ef04
--- /dev/null
+++ b/models/poolRequests.js
@@ -0,0 +1,67 @@
+;(function () {
+  'use strict'
+
+  var mongoose = require('mongoose')
+
+  var logger = require('../helpers/logger')
+
+  // ---------------------------------------------------------------------------
+
+  var poolRequestsSchema = mongoose.Schema({
+    type: String,
+    id: String, // Special id to find duplicates (video created we want to remove...)
+    request: mongoose.Schema.Types.Mixed
+  })
+  var PoolRequestsDB = mongoose.model('poolRequests', poolRequestsSchema)
+
+  // ---------------------------------------------------------------------------
+
+  var PoolRequests = {
+    addRequest: addRequest,
+    list: list,
+    removeRequests: removeRequests
+  }
+
+  function addRequest (id, type, request) {
+    logger.debug('Add request to the pool requests.', { id: id, type: type, request: request })
+
+    PoolRequestsDB.findOne({ id: id }, function (err, entity) {
+      if (err) logger.error(err)
+
+      if (entity) {
+        if (entity.type === type) {
+          logger.error(new Error('Cannot insert two same requests.'))
+          return
+        }
+
+        // Remove the request of the other type
+        PoolRequestsDB.remove({ id: id }, function (err) {
+          if (err) logger.error(err)
+        })
+      } else {
+        PoolRequestsDB.create({ id: id, type: type, request: request }, function (err) {
+          if (err) logger.error(err)
+        })
+      }
+    })
+  }
+
+  function list (callback) {
+    PoolRequestsDB.find({}, { _id: 1, type: 1, request: 1 }, callback)
+  }
+
+  function removeRequests (ids) {
+    PoolRequestsDB.remove({ _id: { $in: ids } }, function (err) {
+      if (err) {
+        logger.error('Cannot remove requests from the pool requests database.', { error: err })
+        return
+      }
+
+      logger.info('Pool requests flushed.')
+    })
+  }
+
+  // ---------------------------------------------------------------------------
+
+  module.exports = PoolRequests
+})()