pomelo-upgrade/lib/pushSchedulers/direct.js

网易Pomelo升级版本

const utils = require('../util/utils');

const Service = function(app, opts)
{
    if (!(this instanceof Service))
    {
        return new Service(app, opts);
    }

    opts = opts || {};
    this.app = app;
};

module.exports = Service;

Service.prototype.schedule = function(reqId, route, msg, recvs, opts, cb)
{
    opts = opts || {};
    if (opts.type === 'broadcast')
    {
        doBroadcast(this, msg, opts.userOptions);
    }
    else
    {
        doBatchPush(this, msg, recvs);
    }

    if (cb)
    {
        process.nextTick(function()
        {
            utils.invokeCallback(cb);
        });
    }
};

const doBroadcast = function(self, msg, opts)
{
    const channelService = self.app.get('channelService');
    const sessionService = self.app.get('sessionService');

    if (opts.binded)
    {
        sessionService.forEachBindedSession(function(session)
        {
            if (channelService.broadcastFilter &&
									!channelService.broadcastFilter(session, msg, opts.filterParam))
            {
                return;
            }

            sessionService.sendMessageByUid(session.uid, msg);
        });
    }
    else
    {
        sessionService.forEachSession(function(session)
        {
            if (channelService.broadcastFilter &&
									!channelService.broadcastFilter(session, msg, opts.filterParam))
            {
                return;
            }

            sessionService.sendMessage(session.id, msg);
        });
    }
};

const doBatchPush = function(self, msg, recvs)
{
    const sessionService = self.app.get('sessionService');
    for (let i = 0, l = recvs.length; i < l; i++)
    {
        sessionService.sendMessage(recvs[i], msg);
    }
};