aboutsummaryrefslogblamecommitdiffhomepage
path: root/server/lib/request/request-scheduler.ts
blob: 3945ace200aa9bc1a5a0ec24bae5c9bff094c650 (plain) (tree)
1
2
3
4
5
6
7
8
9

                                      
                                                            
                                                                       
                                      

                                                                                
                                                 
 

                                       
                           




                                    
                                                                          
                  
           
 
                                        

                                             
 
                                 

   

                      

   

                           

   
                                                          

                                    

                                                     
















                                                                             
 
                                

   
                                                                                         
                                             
                                            
 







                         
                                                       


                 

                                                           
                                                       
        



                                                                                

                                 
                                              
                                                                                    

   





                                                                              
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) {
    // 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
}