]> git.stg.codes - stg.git/blob - include/stg/user_traff.h
Fix passing params to OnDisconnect in rscriptd.
[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     DIR_TRAFF(const DIR_TRAFF & ts) : traff(ts.traff) {}
49     DIR_TRAFF & operator=(const DIR_TRAFF & ts) { traff = ts.traff; return *this; }
50     const uint64_t & operator[](IndexType idx) const { return traff[idx]; }
51     uint64_t & operator[](IndexType idx) { return traff[idx]; }
52     IndexType size() const { return traff.size(); }
53
54     void Reset()
55     {
56     for (IndexType i = 0; i < traff.size(); ++i)
57         traff[i] = 0;
58     }
59
60 private:
61     ContainerType traff;
62 };
63
64 //-----------------------------------------------------------------------------
65 inline
66 std::ostream & operator<<(std::ostream & o, const DIR_TRAFF & traff)
67 {
68 bool first = true;
69 for (DIR_TRAFF::IndexType i = 0; i < traff.size(); ++i)
70     {
71     if (first)
72         first = false;
73     else
74         o << ",";
75     o << traff[i];
76     }
77 return o;
78 }
79
80 class DIR_TRAFF_RES
81 {
82 public:
83     typedef RESETABLE<uint64_t> ValueType;
84     typedef std::vector<ValueType> ContainerType;
85     typedef ContainerType::size_type IndexType;
86
87     DIR_TRAFF_RES() : traff(DIR_NUM) {}
88     DIR_TRAFF_RES(const DIR_TRAFF & ts)
89         : traff(ts.size())
90     {
91     for (IndexType i = 0; i < ts.size(); ++i)
92         traff[i] = ts[i];
93     }
94     const ValueType & operator[](IndexType idx) const { return traff[idx]; }
95     ValueType & operator[](IndexType idx) { return traff[idx]; }
96     DIR_TRAFF GetData() const
97     {
98     DIR_TRAFF res;
99     for (IndexType i = 0; i < traff.size(); ++i)
100         if (!traff[i].empty())
101             res[i] = traff[i].data();
102     return res;
103     }
104
105 private:
106     ContainerType traff;
107 };
108
109 #endif