import * as Sequelize from 'sequelize' import { database as db } from '../../initializers/database' import { AbstractRequestScheduler } from './abstract-request-scheduler' import { logger } from '../../helpers' import { REQUESTS_LIMIT_PODS, REQUESTS_LIMIT_PER_POD } from '../../initializers' import { RequestsGrouped } from '../../models' import { RequestEndpoint } from '../../../shared' export type RequestSchedulerOptions = { type: string endpoint: RequestEndpoint data: Object toIds: number[] transaction: Sequelize.Transaction } class RequestScheduler extends AbstractRequestScheduler { 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 (requestsGrouped: RequestsGrouped) { const requestsToMakeGrouped = {} Object.keys(requestsGrouped).forEach(toPodId => { requestsGrouped[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 } createRequest ({ type, endpoint, data, toIds, transaction }: RequestSchedulerOptions) { // If there are no destination pods abort if (toIds.length === 0) return undefined const createQuery = { endpoint, request: { type: type, data: data } } const dbRequestOptions: Sequelize.CreateOptions = { transaction } return db.Request.create(createQuery, dbRequestOptions) .then(request => { return request.setPods(toIds, dbRequestOptions) }) } // --------------------------------------------------------------------------- afterRequestsHook () { // Flush requests with no pod this.getRequestModel().removeWithEmptyTo() .catch(err => logger.error('Error when removing requests with no pods.', err)) } } // --------------------------------------------------------------------------- export { RequestScheduler }