newbeely-nodejs/lib/utils/protocol.js

简单易用的轻量级nodejs服务框架. 框架简单到只有组件逻辑,不同的组件提供不同的服务,使用外部的配置文件(只有一个配置文件)方便的组合成一个完整的服务框架. 整个服务使用bearcat(网易开源的nodejs面向切面编程的轻量级框架(AOP))管理,极大的解耦组件间的耦合.(关于代码热更新后续开放).

/**
 * @filename protocol
 *
 * @module Utils
 *
 * @author Gandalfull <orientcountry@gmail.com>
 * @version 1
 * @time 2016-02-19 10:03
 */
var Zlib = require('zlib');
var Async = require('async');
var Logger = require('pomelo-logger').getLogger('newbeely', 'protocol');
var __ = require('underscore');

/**
 * 消息解析协议
 *
 * @class protocol
 * @constructor
 */
function protocol() {

    var encrypt = null;
    var rc4key = "";
}

/**
 * 解码协议
 *
 * @method decode
 * @for protocol
 * @param buf 消息提
 * @param protocols 协议类型 { base64 | zlib | googleprotocolbuf | cr4 }
 * @returns {*}
 */
protocol.prototype.decode = function (buf, protocols, cb) {
    var s = protocols.split('|');
    s = __.compact(s);
    if (s.length == 0) {
        cb(null, buf);
        return;
    }
    var recived_buf = buf;
    var _this = this;
    Async.eachSeries(s, function (item, next) {
        if (item == 'zlib') {
            Zlib.unzip(new Buffer(recived_buf, 'binary'), function (error, bytes) {
                if (error) {
                    Logger.error("解码压缩文件错误:" + error);
                }
                recived_buf = bytes;
                next(error);
            });
        } else if (item == "base64") {
            recived_buf = _this.encrypt.base64Decode(recived_buf.toString());
            next(null);
        } else if (item = "rc4") {
            recived_buf = _this.encrypt.rc4Decode(recived_buf, _this.rc4key);
            next(null);
        } else {
            next(null);
        }
    }, function (error) {
        if (error) {
            Logger.error("protocol.prototype.decode error:" + error);
            cb(error, "");
            return;
        }
        cb(null, recived_buf);
    });
};

/**
 * 编码协议
 *
 * @method encode
 * @for protocol
 * @param buf 消息提
 * @param protocols 协议类型 { base64 | zlib | googleprotocolbuf | cr4}
 * @returns {*}
 */
protocol.prototype.encode = function (buf, protocols, cb) {
    var bytes_buf = buf;
    var s = protocols.split('|');
    s = s.reverse();
    s = __.compact(s);
    if (s.length == 0) {
        cb(null, bytes_buf);
        return;
    }
    var _this = this;
    Async.eachSeries(s, function (item, next) {
        if (item == 'zlib') {
            Zlib.gzip(new Buffer(bytes_buf, "utf8").toString("binary"), function (error, bytes) {
                if (error) {
                    next(error);
                    return;
                }
                bytes_buf = bytes;
                next(null);
            });
        } else if (item == 'base64') {
            bytes_buf = _this.encrypt.base64Encode(bytes_buf.toString("binary"));
            next(null);
        } else if (item == 'rc4') {
            bytes_buf = _this.encrypt.rc4Encode(bytes_buf, _this.rc4key);
            next(null);
        } else {
            next(null);
        }
    }, function (error) {
        if (error) {
            Logger.error("protocol.prototype.decode error:" + error);
            cb(null, "");
            return;
        }
        cb(null, bytes_buf);
    });
};

module.exports = {
    id: "protocol",
    func: protocol,
    props: [
        {name: "encrypt", "ref": "encrypt"},
        {key: "rc4key", "value": "zplay-rc4-key"}
    ]
};