98 lines
2.5 KiB
JavaScript
98 lines
2.5 KiB
JavaScript
'use strict'
|
|
|
|
const constants = require('../initializers/constants')
|
|
const BaseRequestScheduler = require('./base-request-scheduler')
|
|
const db = require('../initializers/database')
|
|
const logger = require('../helpers/logger')
|
|
|
|
module.exports = class RequestScheduler extends BaseRequestScheduler {
|
|
constructor () {
|
|
super()
|
|
|
|
// We limit the size of the requests
|
|
this.limitPods = constants.REQUESTS_LIMIT_PODS
|
|
this.limitPerPod = constants.REQUESTS_LIMIT_PER_POD
|
|
|
|
this.description = 'requests'
|
|
}
|
|
|
|
getRequestModel () {
|
|
return db.Request
|
|
}
|
|
|
|
getRequestToPodModel () {
|
|
return db.RequestToPod
|
|
}
|
|
|
|
buildRequestObjects (requests) {
|
|
const requestsToMakeGrouped = {}
|
|
|
|
Object.keys(requests).forEach(toPodId => {
|
|
requests[toPodId].forEach(data => {
|
|
const request = data.request
|
|
const pod = data.pod
|
|
const hashKey = toPodId + request.endpoint
|
|
|
|
if (!requestsToMakeGrouped[hashKey]) {
|
|
requestsToMakeGrouped[hashKey] = {
|
|
toPod: pod,
|
|
endpoint: request.endpoint,
|
|
ids: [], // request ids, to delete them from the DB in the future
|
|
datas: [] // requests data,
|
|
}
|
|
}
|
|
|
|
requestsToMakeGrouped[hashKey].ids.push(request.id)
|
|
requestsToMakeGrouped[hashKey].datas.push(request.request)
|
|
})
|
|
})
|
|
|
|
return requestsToMakeGrouped
|
|
}
|
|
|
|
// { type, endpoint, data, toIds, transaction }
|
|
createRequest (options, callback) {
|
|
const type = options.type
|
|
const endpoint = options.endpoint
|
|
const data = options.data
|
|
const toIds = options.toIds
|
|
const transaction = options.transaction
|
|
|
|
const pods = []
|
|
|
|
// If there are no destination pods abort
|
|
if (toIds.length === 0) return callback(null)
|
|
|
|
toIds.forEach(toPod => {
|
|
pods.push(db.Pod.build({ id: toPod }))
|
|
})
|
|
|
|
const createQuery = {
|
|
endpoint,
|
|
request: {
|
|
type: type,
|
|
data: data
|
|
}
|
|
}
|
|
|
|
const dbRequestOptions = {
|
|
transaction
|
|
}
|
|
|
|
return db.Request.create(createQuery, dbRequestOptions).asCallback((err, request) => {
|
|
if (err) return callback(err)
|
|
|
|
return request.setPods(pods, dbRequestOptions).asCallback(callback)
|
|
})
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
|
|
afterRequestsHook () {
|
|
// Flush requests with no pod
|
|
this.getRequestModel().removeWithEmptyTo(err => {
|
|
if (err) logger.error('Error when removing requests with no pods.', { error: err })
|
|
})
|
|
}
|
|
}
|