path: root/src/net/bandwidth.hpp

/*
 *  The Mana World Server
 *  Copyright 2008 The Mana World Development Team
 *
 *  This file is part of The Mana World.
 *
 *  The Mana World is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  any later version.
 *
 *  The Mana World is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with The Mana World; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

#ifndef _TMWSERV_BANDWIDTH_H_
#define _TMWSERV_BANDWIDTH_H_

#include <map>

class NetComputer;

class BandwidthMonitor
{
public:
    BandwidthMonitor();
    void increaseInterServerOutput(int size);
    void increaseInterServerInput(int size);
    void increaseClientOutput(NetComputer *nc, int size);
    void increaseClientInput(NetComputer *nc, int size);
    int totalInterServerOut() const { return mAmountServerOutput; }
    int totalInterServerIn() const { return mAmountServerInput; }
    int totalClientOut() const { return mAmountClientOutput; }
    int totalClientIn() const { return mAmountClientInput; }

private:
    int mAmountServerOutput;
    int mAmountServerInput;
    int mAmountClientOutput;
    int mAmountClientInput;
    // map of client to output and input
    typedef std::map<NetComputer*, std::pair<int, int> > ClientBandwidth;
    ClientBandwidth mClientBandwidth;
};

extern BandwidthMonitor *gBandwidth;

#endif