node-red-nodes/storage/mysql/68-mysql.js

module.exports = function(RED) {
    "use strict";
    var reconnect = RED.settings.mysqlReconnectTime || 20000;
    var mysqldb = require('mysql');

    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: 50,
                    timeout: 30000,
                    connectTimeout: 30000,
                    charset: node.charset
                });
            }

            // 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);
                        var bind = [];
                        if (Array.isArray(msg.payload)) {
                            bind = msg.payload;
                            node.mydbConfig.pool.on('acquire', function(connection) {
                                    connection.config.queryFormat = null;
                            });
                        }
                        else if (typeof msg.payload === 'object' && msg.payload !== null) {
                            bind = msg.payload;
                            node.mydbConfig.pool.on('acquire', function(connection) {
                                connection.config.queryFormat = function(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));
                                };
                            });
                        }
                        node.mydbConfig.pool.query(msg.topic, bind, function(err, rows) {
                            if (err) {
                                status = {fill:"red",shape:"ring",text:RED._("mysql.status.error")+": "+err.code};
                                node.status(status);
                                node.error(err,msg);
                            }
                            else {
                                // if (rows.constructor.name === "OkPacket") {
                                //     msg.payload = JSON.parse(JSON.stringify(rows));
                                // }
                                // else if (rows.constructor.name === "Array") {
                                //     if (rows[0] && rows[0].constructor.name === "RowDataPacket") {
                                //         msg.payload = rows.map(v => Object.assign({}, v));
                                //     }
                                //     else if (rows[0] && rows[0].constructor.name === "Array") {
                                //         if (rows[0][0] && rows[0][0].constructor.name === "RowDataPacket") {
                                //             msg.payload = rows.map(function(v) {
                                //                 if (!Array.isArray(v)) { return v; }
                                //                 v.map(w => Object.assign({}, w))
                                //             });
                                //         }
                                //         else { msg.payload = rows; }
                                //     }
                                //     else  { msg.payload = rows; }
                                // }
                                // else { msg.payload = rows; }
                                msg.payload = rows;
                                send(msg);
                                status = {fill:"green",shape:"dot",text:RED._("mysql.status.ok")};
                                node.status(status);
                            }
                            if (done) { done(); }
                            // if (node.mydbConfig.pool._freeConnections.indexOf(node.mydbConfig.connection) === -1) {
                            //     node.mydbConfig.connection.release();
                            // }
                        });
                    }
                    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);
}