@wmfs/tymly-rankings-plugin/lib/components/state-resources/refresh-ranking/index.js

Plugin which handles ranking of data for Tymly framework

const generateView = require('./../../services/rankings/generate-view-statement.js')
const generateStats = require('./../../services/rankings/generate-stats.js')
const { snakeCase } = require('lodash')

class RefreshRanking {
  init (resourceConfig, env) {
    this.logger = env.bootedServices.logger.child('stateResource:refreshRanking')
    this.rankings = env.blueprintComponents.rankings
    this.client = env.bootedServices.storage.client
    this.registry = env.bootedServices.registry
    this.storage = env.bootedServices.storage
    this.timestamp = env.bootedServices.timestamp
    this.bootedServices = env.bootedServices
  }

  async run (event, context) {
    const { schema, category } = event

    if (!schema && !category) {
      return context.sendTaskFailure({ error: 'noSchemaOrCategory', cause: 'No schema or category on input.' })
    }

    this.logger.info(`Begin refreshing ${schema} ${category}`)

    const key = schema + '_' + category
    const registry = this.registry.get(key)

    if (!registry) {
      return context.sendTaskFailure({ error: 'noRegistryFound', cause: `No registry found for: ${key}` })
    }

    const refreshModel = this.storage.models.wmfs_rankingRefreshStatus
    const refreshDoc = await refreshModel.upsert({ key, status: 'STARTED' }, {})

    const factors =
      Object.keys(this.rankings[key].factors)
        .filter(factor => Object.keys(registry).includes(factor))
        .reduce((obj, k) => {
          obj[k] = this.rankings[key].factors[k]
          return obj
        }, {})

    const statement = generateView({
      category: snakeCase(category),
      schema: snakeCase(schema),
      source: this.rankings[key].source,
      ranking: factors,
      registry: { value: registry }
    })

    await this.client.query(statement)

    const rankingModel = this.storage.models[`${schema}_${this.rankings[key].rankingModel}`]
    const riskScoreTrailModel = this.storage.models[`${schema}_${this.rankings[key].riskScoreTrailModel}`]
    const statsViewKey = `${snakeCase(schema)}.${snakeCase(this.rankings[key].rankingModel)}_stats`

    await generateStats({
      client: this.client,
      category,
      schema,
      pk: this.rankings[key].source.property,
      name: schema,
      rankingModel,
      riskScoreTrailModel,
      statsViewKey,
      getStats: this.bootedServices.rankings.getStats,
      registry: this.registry.registry[key],
      timestamp: this.timestamp,
      logger: this.logger
    })

    await refreshModel.upsert({ key, status: 'ENDED', id: refreshDoc.idProperties.id }, {})

    this.logger.info(`Finish refreshing ${schema} ${category}`)

    context.sendTaskSuccess()
  }
}

module.exports = RefreshRanking