Skip to content

V
vernemq
Commit 74b7fd7f by zCaesar
Options

can publish onGroup

parent 14c11a3b
Showing with 33 additions and 21 deletions
authhook/auth_on_publish.js
var config = require('config'); var config = require('config');
// https://github.com/isaacs/node-lru-cache // https://github.com/isaacs/node-lru-cache
var LRU = require("lru-cache") , var LRU = require("lru-cache"),
cache = LRU({ cache = LRU({
max: 500, max: 500,
maxAge: 1000 * 60 * 5 maxAge: 1000 * 60 * 5
}); });
// for testing // for testing
function authCheck(client_id, username, topic, callback) { function authCheck(client_id, username, topic, callback) {
callback(true); callback(true);
} }
module.exports = function(options={}) { module.exports = function (options = {}) {
return function(req, res, next) { return function (req, res, next) {
if (req.header('vernemq-hook') == 'auth_on_publish' && req && req.body) { if (req.header('vernemq-hook') == 'auth_on_publish' && req && req.body) {
if (options.debug) { if (options.debug) {
...@@ -23,24 +23,36 @@ module.exports = function(options={}) { ...@@ -23,24 +23,36 @@ module.exports = function(options={}) {
var authstatus; var authstatus;
var topic = req.body.topic; var topic = req.body.topic;
var cachekey = 'pub:'+req.body.client_id +':'+ req.body.username+':'+topic; var cachekey = 'pub:' + req.body.client_id + ':' + req.body.username + ':' + topic;
var authstatus = cache.get(cachekey); var authstatus = cache.get(cachekey);
if (typeof(authstatus)=='undefined') { // Set Response
// cache missed var getGroupID = require('./utils/getGroupID').getGroupID
authCheck(req.body.client_id, req.body.username, topic, function(result) { getGroupID(req.body.username, req.body.client_id, (_id) => { // get groupID
cache.set(cachekey, result); // console.log(_id)
res.send({ var _ftopic = require('./utils/getTopic').getTopic(topic, _id) // get topic where concat with groupID
"result": result?"ok":"no" var response = {
'result': 'ok',
'modifiers': {
'topic': _ftopic,
'qos': 0,
'retain': false
}
}
console.log(response)
if (typeof (authstatus) == 'undefined') {
// cache missed
authCheck(req.body.client_id, req.body.username, topic, function (result) {
cache.set(cachekey, result);
res.send(response);
}); });
}); }
} else {
else { res.send(response);
res.send({ }
"result": authstatus?"ok":"no" next();
}); })
}
next();
} }
else { else {
res.send({ res.send({
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment