You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
70 lines
1.8 KiB
70 lines
1.8 KiB
3 years ago
|
'use strict'
|
||
|
|
||
|
const Ajv = require('ajv')
|
||
|
|
||
|
function ValidatorSelector () {
|
||
|
const validatorPool = new Map()
|
||
|
const cache = new Map()
|
||
|
cache.put = cache.set
|
||
|
|
||
|
return function buildCompilerFromPool (externalSchemas, options) {
|
||
|
const externals = JSON.stringify(externalSchemas)
|
||
|
const ajvConfig = JSON.stringify(options.customOptions)
|
||
|
|
||
|
const uniqueAjvKey = `${externals}${ajvConfig}`
|
||
|
if (validatorPool.has(uniqueAjvKey)) {
|
||
|
return validatorPool.get(uniqueAjvKey)
|
||
|
}
|
||
|
|
||
|
const compiler = ValidatorCompiler(externalSchemas, options, cache)
|
||
|
validatorPool.set(uniqueAjvKey, compiler)
|
||
|
|
||
|
return compiler
|
||
|
}
|
||
|
}
|
||
|
|
||
|
function ValidatorCompiler (externalSchemas, options, cache) {
|
||
|
// This instance of Ajv is private
|
||
|
// it should not be customized or used
|
||
|
const ajv = new Ajv(Object.assign({
|
||
|
coerceTypes: true,
|
||
|
useDefaults: true,
|
||
|
removeAdditional: true,
|
||
|
// Explicitly set allErrors to `false`.
|
||
|
// When set to `true`, a DoS attack is possible.
|
||
|
allErrors: false,
|
||
|
nullable: true
|
||
|
}, options.customOptions, { cache }))
|
||
|
|
||
|
if (options.plugins && options.plugins.length > 0) {
|
||
|
for (const plugin of options.plugins) {
|
||
|
if (Array.isArray(plugin)) {
|
||
|
plugin[0](ajv, plugin[1])
|
||
|
} else {
|
||
|
plugin(ajv)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
const sourceSchemas = Object.values(externalSchemas)
|
||
|
for (const extSchema of sourceSchemas) {
|
||
|
ajv.addSchema(extSchema)
|
||
|
}
|
||
|
|
||
|
return function ({ schema/*, method, url, httpPart */ }) {
|
||
|
// Ajv does not support compiling two schemas with the same
|
||
|
// id inside the same instance. Therefore if we have already
|
||
|
// compiled the schema with the given id, we just return it.
|
||
|
if (schema.$id) {
|
||
|
const stored = ajv.getSchema(schema.$id)
|
||
|
if (stored) {
|
||
|
return stored
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return ajv.compile(schema)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
module.exports = ValidatorSelector
|