include/stg/user_stat.h

   1 /*
   2  *    This program is free software; you can redistribute it and/or modify
   3  *    it under the terms of the GNU General Public License as published by
   4  *    the Free Software Foundation; either version 2 of the License, or
   5  *    (at your option) any later version.
   6  *
   7  *    This program is distributed in the hope that it will be useful,
   8  *    but WITHOUT ANY WARRANTY; without even the implied warranty of
   9  *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  10  *    GNU General Public License for more details.
  11  *
  12  *    You should have received a copy of the GNU General Public License
  13  *    along with this program; if not, write to the Free Software
  14  *    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  15  */
  16
  17 /*
  18  *    Author : Boris Mikhailenko <stg34@stargazer.dp.ua>
  19  */
  20
  21  /*
  22  $Revision: 1.15 $
  23  $Date: 2010/03/11 14:42:05 $
  24  $Author: faust $
  25  */
  26
  27 #ifndef USER_STAT_H
  28 #define USER_STAT_H
  29
  30 #include <ctime>
  31 #include <map>
  32
  33 #include "os_int.h"
  34 #include "resetable.h"
  35 #include "user_traff.h"
  36 //-----------------------------------------------------------------------------
  37 struct IP_DIR_PAIR
  38 {
  39     #ifdef TRAFF_STAT_WITH_PORTS
  40     IP_DIR_PAIR(uint32_t _ip,
  41                 int _dir,
  42                 uint16_t _port)
  43         : ip(_ip),
  44           dir(_dir),
  45           port(_port)
  46     {}
  47     #else
  48     IP_DIR_PAIR(uint32_t _ip,
  49                 int _dir)
  50         : ip(_ip),
  51           dir(_dir)
  52     {}
  53     #endif
  54     //------------------------
  55     bool operator<(const IP_DIR_PAIR & idp) const
  56         {
  57         if (ip < idp.ip)
  58             return true;
  59
  60         if (ip > idp.ip)
  61             return false;
  62
  63         #ifdef TRAFF_STAT_WITH_PORTS
  64         if (port < idp.port)
  65             return true;
  66
  67         if (port > idp.port)
  68             return false;
  69         #endif
  70
  71         if (dir < idp.dir)
  72             return true;
  73
  74         return false;
  75         }
  76     //------------------------
  77     bool operator!=(const IP_DIR_PAIR & rvalue) const
  78         {
  79         if (ip != rvalue.ip)
  80             return true;
  81
  82         #ifdef TRAFF_STAT_WITH_PORTS
  83         if (port != rvalue.port)
  84             return true;
  85         #endif
  86
  87         if (dir != rvalue.dir)
  88             return true;
  89
  90         return false;
  91         }
  92     //------------------------
  93     uint32_t        ip;
  94     int             dir;
  95     #ifdef TRAFF_STAT_WITH_PORTS
  96     uint16_t        port;
  97     #endif
  98 };
  99 //-----------------------------------------------------------------------------
 100 struct STAT_NODE
 101 {
 102     STAT_NODE(uint64_t _up,
 103               uint64_t _down,
 104               double   _cash)
 105         : up(_up),
 106           down(_down),
 107           cash(_cash)
 108     {}
 109     uint64_t        up;
 110     uint64_t        down;
 111     double          cash;
 112 };
 113 //-----------------------------------------------------------------------------
 114 struct USER_STAT
 115 {
 116     //USER_STAT & operator= (const USER_STAT_RES & usr);
 117     USER_STAT()
 118         : up(),
 119           down(),
 120           cash(0),
 121           freeMb(0),
 122           lastCashAdd(0),
 123           lastCashAddTime(0),
 124           passiveTime(0),
 125           lastActivityTime(0)
 126     {};
 127
 128     DIR_TRAFF   up;
 129     DIR_TRAFF   down;
 130     double      cash;
 131     double      freeMb;
 132     double      lastCashAdd;
 133     time_t      lastCashAddTime;
 134     time_t      passiveTime;
 135     time_t      lastActivityTime;
 136 };
 137 //-----------------------------------------------------------------------------
 138 typedef std::map<IP_DIR_PAIR, STAT_NODE> TRAFF_STAT;
 139 //-----------------------------------------------------------------------------
 140 struct USER_STAT_RES
 141 {
 142     USER_STAT_RES()
 143         : cash(),
 144           freeMb(),
 145           lastCashAdd(),
 146           lastCashAddTime(),
 147           passiveTime(),
 148           lastActivityTime(),
 149           up(),
 150           down()
 151     {}
 152
 153     USER_STAT_RES & operator= (const USER_STAT & us)
 154     {
 155         cash             = us.cash;
 156         freeMb           = us.freeMb;
 157         lastCashAdd      = us.lastCashAdd;
 158         lastCashAddTime  = us.lastCashAddTime;
 159         passiveTime      = us.passiveTime;
 160         lastActivityTime = us.lastActivityTime;
 161         up = us.up;
 162         down = us.down;
 163         return * this;
 164     };
 165     operator USER_STAT() const
 166     {
 167         USER_STAT us;
 168         us.cash             = cash;
 169         us.freeMb           = freeMb;
 170         us.lastCashAdd      = lastCashAdd;
 171         us.lastCashAddTime  = lastCashAddTime;
 172         us.passiveTime      = passiveTime;
 173         us.lastActivityTime = lastActivityTime;
 174         us.up               = up;
 175         us.down             = down;
 176         return us;
 177     };
 178
 179     RESETABLE<double>      cash;
 180     RESETABLE<double>      freeMb;
 181     RESETABLE<double>      lastCashAdd;
 182     RESETABLE<time_t>      lastCashAddTime;
 183     RESETABLE<time_t>      passiveTime;
 184     RESETABLE<time_t>      lastActivityTime;
 185     RESETABLE<DIR_TRAFF>   up;
 186     RESETABLE<DIR_TRAFF>   down;
 187 };
 188 //-----------------------------------------------------------------------------
 189 #endif