代码优化 - 去掉无效的queue

沈斌

2018-05-03 8fbd26ea16e81c29926188ec5a8e149c032d102f

 server.js

@@ -7,12 +7,138 @@
var mongoClient = require('mongodb').MongoClient;
var config = require('./config');
var method = require('./method');
//var queue = require('./queue');

mongoClient.connect(config.URL, function(err, db) {
    if (err) return;
var map = {
    "5a0000010001": 32,  //心跳命令行
    "5a0000010002": 102, //云地址写入WIFI模块
    "5a0000010003": 162, //传感器数据上传
    "5a0000010004": 32,  //云端时间
    "5a0000010006": 122, //配网数据
    "5a0000010007": 32,  //主机休眠前发送即将休眠的命令
    "5a000001000c": 20   //联云通知
};

global.configs = {};

function handleData(db, socket, value) {

    //1.心跳命令行
    if(value.startsWith('5a0000010001')) {
        var output = [ 0x6A, 0x00, 0x00, 0x01, 0x00, 0x01, 0xA1, 0x1A, 0xC7, 0x6B ];
        method.getAppStatus(db, value, function(doc) {
            var app_status = doc.app_status;
            var app_last_updated = doc.app_last_updated;
            if(app_status == 1 && app_last_updated != null && Date.now() - app_last_updated <= 30000) {
                output[7] = 0x1B;
            }
            socket.write(new Buffer(output));
        });
        return;
    }

    //3.传感器数据上传
    if(value.startsWith('5a0000010003')) {
        var output = [ 0x6A, 0x00, 0x00, 0x01, 0x00, 0x03, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x6B ];
        method.insertData(db, value, function(data, rank) {
            var fields = method.padLeft(rank.toString(16), 8).match(/.{2}/g);
            output[6] = method.toDec(fields[0]);
            output[7] = method.toDec(fields[1]);
            output[8] = method.toDec(fields[2]);
            output[9] = method.toDec(fields[3]);
            socket.write(new Buffer(output));
        });
        return;
    }

    //7.主机休眠前发送即将休眠的命令
    if(value.startsWith('5a0000010007')) {
        method.updateDeviceSleep(db, value, function(data) {});
        return;
    }

    //4.云端时间
    if(value.startsWith('5a0000010004')) {
        method.updateDeviceWakeup(db, value, function(data) {});
        var current_time = moment();
        var output = [
            0x6A, 0x00, 0x00, 0x01, 0x00, 0x04,  //包头(0-5)
            parseInt(current_time.format('s')),  //秒
            parseInt(current_time.format('m')),  //分
            parseInt(current_time.format('H')),  //小时
            parseInt(current_time.format('D')),  //日
            parseInt(current_time.format('M')),  //月
            parseInt(current_time.format('E')),  //星期
            parseInt(current_time.format('YY')), //年
            0x36, //校验和
            0x6B  //包尾
        ];
        socket.write(new Buffer(output));
        return;
    }

    //2.云地址写入WIFI模块
    if(value.startsWith('5a0000010002')) {
        method.insertDocument(db, value, function(data) {});
        socket.write(new Buffer(config.OUTPUT_2));
        return;
    }

    //6.配网数据
    if(value.startsWith('5a0000010006')) {
        method.registerDevice(db, value, function(data) {});
        socket.write(new Buffer(config.OUTPUT_6));
        return;
    }

    //12.联云通知
    if(value.startsWith('5a000001000c')) {
        socket.write(new Buffer(config.OUTPUT_6));
        return;
    }
}

function doWork(db, socket, data) {
    if(data == "") return;
    var length = map[data.slice(0, 12)];
    if(length > 0) {
        var value = data.slice(0, length);
        //console.log(moment().format('YYYY-MM-DD HH:mm:ss') + " => " + value);

        //queue.publishMessage(value);

        handleData(db, socket, value);

        doWork(db, socket, data.slice(length));
    }
}

mongoClient.connect(config.URL, {
        server: {
            poolSize: 10,
            keepAlive: 120,
            auto_reconnect: true,
            reconnectTries: 1000,
            reconnectInterval: 5000,
            socketOptions: {
                keepAlive: 300000,
                connectTimeoutMS: 30000
            }
        }
    }, function(err, db) {
    if (err) {
        console.log(err.message);
        return;
    }
    console.log('Connecting to Mongo DB at ' + config.URL);

    net.createServer(function(socket) {
    //初始化三级警报通讯方式
    method.initConfigs(db);

    // //监听三级警报通讯方式的变化
    // queue.listenToMQ("ex_data_config");

    net.createServer().on('connection', function(socket){
        console.log('CONNECTED: ' + socket.remoteAddress +':'+ socket.remotePort);

        socket.on('data', function(data) {
@@ -27,85 +153,26 @@
                return;
            }

            //1.心跳命令行
            if(value.startsWith('5a0000010001')) {
                var output = [ 0x6A, 0x00, 0x00, 0x01, 0x00, 0x01, 0xA1, 0x1A, 0xC7, 0x6B ];
                method.getAppStatus(db, value, function(doc) {
                    var app_status = doc.app_status;
                    var app_last_updated = doc.app_last_updated;
                    if(app_status == 1 && app_last_updated != null && Date.now() - app_last_updated <= 30000) {
                        output[7] = 0x1B;
                    }
                    socket.write(new Buffer(output));
                });
                return;
            }

            //3.传感器数据上传
            if(value.startsWith('5a0000010003')) {
                var output = [ 0x6A, 0x00, 0x00, 0x01, 0x00, 0x03, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x6B ];
                method.insertData(db, value, function(data, rank) {
                    method.insertDocument2(db, value, rank, function(data) {});
                    var fields = method.padLeft(rank.toString(16), 8).match(/.{2}/g);
                    output[6] = method.toDec(fields[0]);
                    output[7] = method.toDec(fields[1]);
                    output[8] = method.toDec(fields[2]);
                    output[9] = method.toDec(fields[3]);
                    socket.write(new Buffer(output));
                });
                return;
            }

            //7.主机休眠前发送即将休眠的命令
            if(value.startsWith('5a0000010007')) {
                method.updateDeviceSleep(db, value, function(data) {});
                return;
            }

            //4.云端时间
            if(value.startsWith('5a0000010004')) {
                method.updateDeviceWakeup(db, value, function(data) {});
                var current_time = moment();
                var output = [
                    0x6A, 0x00, 0x00, 0x01, 0x00, 0x04,  //包头(0-5)
                    parseInt(current_time.format('s')),  //秒
                    parseInt(current_time.format('m')),  //分
                    parseInt(current_time.format('H')),  //小时
                    parseInt(current_time.format('D')),  //日
                    parseInt(current_time.format('M')),  //月
                    parseInt(current_time.format('E')),  //星期
                    parseInt(current_time.format('YY')), //年
                    0x36, //校验和
                    0x6B  //包尾
                ];
                socket.write(new Buffer(output));
                return;
            }

            //2.云地址写入WIFI模块
            if(value.startsWith('5a0000010002')) {
                method.insertDocument(db, value, function(data) {});
                socket.write(new Buffer(config.OUTPUT_2));
                return;
            }

            //6.配网数据
            if(value.startsWith('5a0000010006')) {
                method.registerDevice(db, value, function(data) {});
                socket.write(new Buffer(config.OUTPUT_6));
                return;
            }

            //12.联云通知
            if(value.startsWith('5a000001000c')) {
                socket.write(new Buffer(config.OUTPUT_6));
                return;
            }
            doWork(db, socket, value);
        });

        socket.on('end', function(){

        });

        socket.on('error', function(error) {
            console.log(error);
            socket.end();
        });

        socket.on('timeout',function(){
            socket.end();
        });

        socket.on('close', function(data) {
            console.log('Closed socket: ' + socket.remoteAddress + ' ' + socket.remotePort);
            console.log('Closed socket: ' + socket.remoteAddress +' '+ socket.remotePort);
        });
    }).listen(config.PORT, config.HOST);
    

    console.log('TCP Server listening on ' + config.HOST + ':' + config.PORT);
});
});