105 lines
2.7 KiB
TypeScript
105 lines
2.7 KiB
TypeScript
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<RequestsGrouped> {
|
|
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) {
|
|
// TODO: check the setPods works
|
|
const podIds = []
|
|
|
|
// If there are no destination pods abort
|
|
if (toIds.length === 0) return undefined
|
|
|
|
toIds.forEach(toPod => {
|
|
podIds.push(toPod)
|
|
})
|
|
|
|
const createQuery = {
|
|
endpoint,
|
|
request: {
|
|
type: type,
|
|
data: data
|
|
}
|
|
}
|
|
|
|
const dbRequestOptions: Sequelize.CreateOptions = {
|
|
transaction
|
|
}
|
|
|
|
return db.Request.create(createQuery, dbRequestOptions)
|
|
.then(request => {
|
|
return request.setPods(podIds, dbRequestOptions)
|
|
})
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
|
|
afterRequestsHook () {
|
|
// Flush requests with no pod
|
|
this.getRequestModel().removeWithEmptyTo()
|
|
.catch(err => logger.error('Error when removing requests with no pods.', { error: err }))
|
|
}
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
|
|
export {
|
|
RequestScheduler
|
|
}
|