module.exports = function(RED) { "use strict"; var reconnect = RED.settings.mysqlReconnectTime || 20000; var mysqldb = require('mysql2'); function MySQLNode(n) { RED.nodes.createNode(this,n); this.host = n.host; this.port = n.port; this.tz = n.tz || "local"; this.charset = (n.charset || "UTF8_GENERAL_CI").toUpperCase(); this.connected = false; this.connecting = false; this.dbname = n.db; this.setMaxListeners(0); var node = this; function checkVer() { node.pool.query("SELECT version();", [], function(err, rows, fields) { if (err) { node.error(err); node.status({fill:"red",shape:"ring",text:RED._("mysql.status.badping")}); doConnect(); } }); } function doConnect() { node.connecting = true; node.emit("state","connecting"); if (!node.pool) { node.pool = mysqldb.createPool({ host : node.host, port : node.port, user : node.credentials.user, password : node.credentials.password, database : node.dbname, timezone : node.tz, insecureAuth: true, multipleStatements: true, connectionLimit: RED.settings.mysqlConnectionLimit || 50, connectTimeout: 30000, charset: node.charset, decimalNumbers: true }); } // connection test node.pool.getConnection(function(err, connection) { node.connecting = false; if (err) { node.emit("state",err.code); node.error(err); node.tick = setTimeout(doConnect, reconnect); } else { node.connected = true; node.emit("state","connected"); if (!node.check) { node.check = setInterval(checkVer, 290000); } connection.release(); } }); } this.connect = function() { if (!this.connected && !this.connecting) { doConnect(); } } this.on('close', function(done) { if (this.tick) { clearTimeout(this.tick); } if (this.check) { clearInterval(this.check); } node.connected = false; // node.connection.release(); node.emit("state"," "); node.pool.end(function(err) { done(); }); }); } RED.nodes.registerType("MySQLdatabase",MySQLNode, { credentials: { user: {type: "text"}, password: {type: "password"} } }); function MysqlDBNodeIn(n) { RED.nodes.createNode(this,n); this.mydb = n.mydb; this.mydbConfig = RED.nodes.getNode(this.mydb); this.status({}); if (this.mydbConfig) { this.mydbConfig.connect(); var node = this; var busy = false; var status = {}; node.mydbConfig.on("state", function(info) { if (info === "connecting") { node.status({fill:"grey",shape:"ring",text:info}); } else if (info === "connected") { node.status({fill:"green",shape:"dot",text:info}); } else { node.status({fill:"red",shape:"ring",text:info}); } }); node.on("input", function(msg, send, done) { send = send || function() { node.send.apply(node,arguments) }; if (node.mydbConfig.connected) { if (typeof msg.topic === 'string') { //console.log("query:",msg.topic); node.mydbConfig.pool.getConnection(function (err, conn) { if (err) { if (conn) { conn.release() } status = { fill: "red", shape: "ring", text: RED._("mysql.status.error") + ": " + err.code }; node.status(status); node.error(err, msg); if (done) { done(); } return } var bind = []; if (Array.isArray(msg.payload)) { bind = msg.payload; } else if (typeof msg.payload === 'object' && msg.payload !== null) { bind = msg.payload; } conn.config.queryFormat = Array.isArray(msg.payload) ? null : customQueryFormat conn.query(msg.topic, bind, function (err, rows) { conn.release() if (err) { status = { fill: "red", shape: "ring", text: RED._("mysql.status.error") + ": " + err.code }; node.status(status); node.error(err, msg); } else { msg.payload = rows; send(msg); status = { fill: "green", shape: "dot", text: RED._("mysql.status.ok") }; node.status(status); } if (done) { done(); } }); }) } else { if (typeof msg.topic !== 'string') { node.error("msg.topic : "+RED._("mysql.errors.notstring")); done(); } } } else { node.error(RED._("mysql.errors.notconnected"),msg); status = {fill:"red",shape:"ring",text:RED._("mysql.status.notconnected")}; if (done) { done(); } } if (!busy) { busy = true; node.status(status); node.tout = setTimeout(function() { busy = false; node.status(status); },500); } }); node.on('close', function() { if (node.tout) { clearTimeout(node.tout); } node.mydbConfig.removeAllListeners(); node.status({}); }); } else { this.error(RED._("mysql.errors.notconfigured")); } } RED.nodes.registerType("mysql",MysqlDBNodeIn); } function customQueryFormat(query, values) { if (!values) { return query; } return query.replace(/\:(\w+)/g, function(txt, key) { if (values.hasOwnProperty(key)) { return this.escape(values[key]); } return txt; }.bind(this)); }