Alexandre Bodin 07e7cfc0bd Make lint stricter and fix the errors
Signed-off-by: Alexandre Bodin <bodin.alex@gmail.com>
2020-11-02 19:41:42 +01:00

166 lines
4.0 KiB
JavaScript

'use strict';
/**
* Module dependencies
*/
// Core
const util = require('util');
/* eslint-disable prefer-template */
// Public node modules.
const _ = require('lodash');
const Redis = require('ioredis');
const stackTrace = require('stack-trace');
/**
* Redis hook
*/
module.exports = function(strapi) {
const hook = {
/**
* Default options
*/
defaults: {
port: 6379,
host: 'localhost',
options: {
db: 0,
},
showFriendlyErrorStack: process.env.NODE_ENV !== 'production',
},
/**
* Initialize the hook
*/
initialize: () => {
if (
_.isEmpty(strapi.models) ||
!_.pickBy(strapi.config.connections, {
connector: 'strapi-hook-redis',
})
) {
return;
}
const connections = _.pickBy(strapi.config.connections, {
connector: 'strapi-hook-redis',
});
if (_.size(connections) === 0) {
return;
}
const done = _.after(_.size(connections), () => {
return;
});
// For each connection in the config register a new Knex connection.
_.forEach(connections, (connection, name) => {
// Apply defaults
_.defaults(connection.settings, strapi.config.hook.settings.redis);
const redis = new Redis(
_.defaultsDeep(
{
port: _.get(connection.settings, 'port'),
host: _.get(connection.settings, 'host'),
options: {
db: _.get(connection.options, 'database') || 0,
},
},
strapi.config.hook.settings.redis
)
);
redis.on('error', err => {
strapi.log.error(err);
process.exit(0);
});
// Utils function.
// Behavior: Try to retrieve data from Redis, if null
// execute callback and set the value in Redis for this serial key.
redis.cache = async ({ expired = 60 * 60, serial }, cb, type) => {
if (_.isEmpty(serial)) {
strapi.log.warn(
`Be careful, you're using cache() function of strapi-redis without serial`
);
const traces = stackTrace.get();
strapi.log.warn(
`> [${traces[1].getLineNumber()}] ${traces[1]
.getFileName()
.replace(strapi.config.appPath, '')}`
);
return await cb();
}
let cache = await redis.get(serial);
if (!cache) {
cache = await cb();
if (cache && _.get(connection, 'options.disabledCaching') !== true) {
switch (type) {
case 'json':
redis.set(serial, JSON.stringify(cache), 'ex', expired);
break;
case 'int':
default:
redis.set(serial, cache, 'ex', expired);
break;
}
}
}
switch (type) {
case 'int':
return parseInt(cache);
case 'float':
return _.toNumber(cache);
case 'json':
try {
return _.isObject(cache) ? cache : JSON.parse(cache);
} catch (e) {
return cache;
}
default:
return cache;
}
};
// Define as new connection.
strapi.connections[name] = redis;
// Expose global
if (_.get(connection, 'options.global') !== false) {
global[_.get(connection, 'options.globalName') || 'redis'] = redis;
}
if (_.get(connection, 'options.debug') === true) {
redis.monitor((err, monitor) => {
if (err) {
console.error(err);
}
// Entering monitoring mode.
monitor.on('monitor', (time, args) => {
console.log(time + ': ' + util.inspect(args));
});
});
}
redis.on('ready', () => {
done();
});
});
},
};
return hook;
};