pomelo-upgrade/lib/common/service/connectionService.js

网易Pomelo升级版本

'use strict';
/**
 * connection statistics service
 * record connection, login count and list
 */
const Service = function(app)
{
    this.serverId = app.getServerId();
    this.connCount = 0;
    this.loginedCount = 0;
    this.logined = {};
};

module.exports = Service;

const pro = Service.prototype;

/**
 * Add logined user.
 *
 * @param uid {string} user id
 * @param info {object} record for logined user
 */
pro.addLoginedUser = function(uid, info)
{
    if (!this.logined[uid])
    {
        this.loginedCount++;
    }
    info.uid = uid;
    this.logined[uid] = info;
};

/**
 * Update user info.
 * @param uid {string} user id
 * @param info {object} info for update.
 */
pro.updateUserInfo = function(uid, info)
{
    const user = this.logined[uid];
    if (!user)
    {
        return;
    }

    for (const p in info)
    {
        if (info.hasOwnProperty(p) && typeof info[p] !== 'function')
        {
            user[p] = info[p];
        }
    }
};

/**
 * Increase connection count
 */
pro.increaseConnectionCount = function()
{
    this.connCount++;
};

/**
 * remove logined user
 *
 * @param uid {string} user id
 */
pro.removeLoginedUser = function(uid)
{
    if (this.logined[uid])
    {
        this.loginedCount--;
    }
    delete this.logined[uid];
};

/**
 * Decrease connection count
 *
 * @param uid {string} uid
 */
pro.decreaseConnectionCount = function(uid)
{
    if (this.connCount)
    {
        this.connCount--;
    }
    if (uid)
    {
        this.removeLoginedUser(uid);
    }
};

/**
 * Get statistics info
 *
 * @return {object} statistics info
 */
pro.getStatisticsInfo = function()
{
    const list = [];
    for (const uid in this.logined)
    {
        list.push(this.logined[uid]);
    }

    return {
        serverId       : this.serverId,
        totalConnCount : this.connCount,
        loginedCount   : this.loginedCount,
        loginedList    : list
    };
};