104 lines
2.6 KiB
TypeScript
104 lines
2.6 KiB
TypeScript
const db = require('../../initializers/database')
|
|
import { BaseRequestScheduler } from './base-request-scheduler'
|
|
import { logger } from '../../helpers'
|
|
import {
|
|
REQUESTS_LIMIT_PODS,
|
|
REQUESTS_LIMIT_PER_POD
|
|
} from '../../initializers'
|
|
|
|
class RequestScheduler extends BaseRequestScheduler {
|
|
constructor () {
|
|
super()
|
|
|
|
// We limit the size of the requests
|
|
this.limitPods = REQUESTS_LIMIT_PODS
|
|
this.limitPerPod = 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 })
|
|
})
|
|
}
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
|
|
export {
|
|
RequestScheduler
|
|
}
|