thinknode/lib/Extension/Middleware/WebSocket.js

A fast, flexible and all-in-one web framework for node.js.

'use strict';

exports.__esModule = true;

var _promise = require('babel-runtime/core-js/promise');

var _promise2 = _interopRequireDefault(_promise);

var _classCallCheck2 = require('babel-runtime/helpers/classCallCheck');

var _classCallCheck3 = _interopRequireDefault(_classCallCheck2);

var _possibleConstructorReturn2 = require('babel-runtime/helpers/possibleConstructorReturn');

var _possibleConstructorReturn3 = _interopRequireDefault(_possibleConstructorReturn2);

var _inherits2 = require('babel-runtime/helpers/inherits');

var _inherits3 = _interopRequireDefault(_inherits2);

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

/**
 * WebSocket中间件
 * @author     richen
 * @copyright  Copyright (c) 2015 - <richenlin(at)gmail.com>
 * @license    MIT
 * @version    15/11/19
 */
var _class = function (_THINK$Middleware) {
    (0, _inherits3.default)(_class, _THINK$Middleware);

    function _class() {
        (0, _classCallCheck3.default)(this, _class);
        return (0, _possibleConstructorReturn3.default)(this, _THINK$Middleware.apply(this, arguments));
    }

    _class.prototype.init = function init(param) {
        this.param = param;
    };

    _class.prototype.run = function run(data) {
        if (THINK.config('use_websocket')) {
            try {
                var instance = new (THINK.adapter('WebSocket'))({ server: this.param.server, app: this.param.app });
                instance.run();
                THINK.log('WebSocket running at ws://' + (THINK.config('app_host') || '127.0.0.1') + ':' + THINK.config('app_port') + '/', 'THINK');
            } catch (e) {
                THINK.log('Initialize WebSocket error: ' + e.stack, 'ERROR');
            }
        }
        return _promise2.default.resolve(data);
    };

    return _class;
}(THINK.Middleware);

exports.default = _class;