]> git.stg.codes - stg.git/blob - include/stg/user_traff.h
Merge pull request #2 from bobr-kun/MySQL_custom_port
[stg.git] / include / stg / user_traff.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.7 $
23  $Date: 2010/10/07 19:48:52 $
24  $Author: faust $
25  */
26
27 #ifndef USER_TRAFF_H
28 #define USER_TRAFF_H
29
30 #include "resetable.h"
31 #include "const.h"
32 #include "os_int.h"
33
34 #include <iostream>
35 #include <vector>
36
37 enum TRAFF_DIRECTION {TRAFF_UPLOAD, TRAFF_DOWNLOAD};
38
39 class DIR_TRAFF
40 {
41     friend std::ostream & operator<< (std::ostream & o, const DIR_TRAFF & traff);
42
43 public:
44     typedef std::vector<uint64_t> ContainerType;
45     typedef ContainerType::size_type IndexType;
46
47     DIR_TRAFF() : traff(DIR_NUM) {}
48     const uint64_t & operator[](IndexType idx) const { return traff[idx]; }
49     uint64_t & operator[](IndexType idx) { return traff[idx]; }
50     IndexType size() const { return traff.size(); }
51
52     void Reset()
53     {
54     for (IndexType i = 0; i < traff.size(); ++i)
55         traff[i] = 0;
56     }
57
58 private:
59     ContainerType traff;
60 };
61
62 //-----------------------------------------------------------------------------
63 inline
64 std::ostream & operator<<(std::ostream & o, const DIR_TRAFF & traff)
65 {
66 bool first = true;
67 for (DIR_TRAFF::IndexType i = 0; i < traff.size(); ++i)
68     {
69     if (first)
70         first = false;
71     else
72         o << ",";
73     o << traff[i];
74     }
75 return o;
76 }
77
78 class DIR_TRAFF_RES
79 {
80 public:
81     typedef RESETABLE<uint64_t> ValueType;
82     typedef std::vector<ValueType> ContainerType;
83     typedef ContainerType::size_type IndexType;
84
85     DIR_TRAFF_RES() : traff(DIR_NUM) {}
86     explicit DIR_TRAFF_RES(const DIR_TRAFF & ts)
87         : traff(ts.size())
88     {
89     for (IndexType i = 0; i < ts.size(); ++i)
90         traff[i] = ts[i];
91     }
92     DIR_TRAFF_RES & operator=(const DIR_TRAFF & ts)
93     {
94         for (IndexType i = 0; i < ts.size(); ++i)
95             traff[i] = ts[i];
96         return *this;
97     }
98     const ValueType & operator[](IndexType idx) const { return traff[idx]; }
99     ValueType & operator[](IndexType idx) { return traff[idx]; }
100     DIR_TRAFF GetData() const
101     {
102     DIR_TRAFF res;
103     for (IndexType i = 0; i < traff.size(); ++i)
104         if (!traff[i].empty())
105             res[i] = traff[i].data();
106     return res;
107     }
108
109 private:
110     ContainerType traff;
111 };
112
113 #endif