ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/gvpe/src/iom.C
(Generate patch)

Comparing gvpe/src/iom.C (file contents):
Revision 1.3 by pcg, Fri Mar 21 21:21:02 2003 UTC vs.
Revision 1.10 by pcg, Fri Apr 4 05:26:45 2003 UTC

… …
24#include <functional> 24#include <functional>
25 25
26#include "slog.h" 26#include "slog.h"
27#include "iom.h" 27#include "iom.h"
28 28
29tstamp NOW;
30bool iom_valid;
31io_manager iom;
32
29inline bool lowest_first (const time_watcher *a, const time_watcher *b) 33inline bool earliest_first (const time_watcher *a, const time_watcher *b)
30{ 34{
31 return a->at > b->at; 35 return a->at > b->at;
32} 36}
33 37
34tstamp NOW;
35
36io_manager iom;
37
38void time_watcher::set (tstamp when) 38void time_watcher::set (tstamp when)
39{ 39{
40 iom.unreg (this);
41 at = when; 40 at = when;
41
42 if (registered)
43 iom.reschedule_time_watchers ();
44 else
42 iom.reg (this); 45 iom.reg (this);
43} 46}
44 47
48void time_watcher::trigger ()
49{
50 call (*this);
51
52 if (registered)
53 iom.reschedule_time_watchers ();
54 else
55 iom.reg (this);
56}
57
58void time_watcher::start ()
59{
60 if (!registered)
61 iom.reg (this);
62}
63
64time_watcher::~time_watcher ()
65{
66 if (iom_valid)
67 iom.unreg (this);
68}
69
70io_watcher::~io_watcher ()
71{
72 if (iom_valid)
73 iom.unreg (this);
74}
75
45void io_manager::reg (int fd, short events, io_watcher *w) 76void io_manager::reg (io_watcher *w)
46{ 77{
47 pollfd pfd; 78 pollfd pfd;
48
49 pfd.fd = fd;
50 pfd.events = events;
51 79
52 pfs.push_back (pfd); 80 pfs.push_back (pfd);
53 iow.push_back (w); 81 iow.push_back (w);
82
83 w->p = &(*(pfs.end () - 1));
54} 84}
55 85
56void io_manager::unreg (io_watcher *w) 86void io_manager::unreg (io_watcher *w)
57{ 87{
58 unsigned int sz = iow.size (); 88 if (w->p)
59 unsigned int i = find (iow.begin (), iow.end (), w) - iow.begin ();
60
61 if (i != sz)
62 { 89 {
90 unsigned int sz = iow.size ();
91 unsigned int i = find (iow.begin (), iow.end (), w) - iow.begin ();
92
93 assert (i != sz);
94
63 if (sz == 1) 95 if (sz == 1)
64 { 96 {
65 pfs.clear (); 97 pfs.clear ();
66 iow.clear (); 98 iow.clear ();
67 } 99 }
… …
73 else 105 else
74 { 106 {
75 iow[i] = iow[sz - 1]; iow.pop_back (); 107 iow[i] = iow[sz - 1]; iow.pop_back ();
76 pfs[i] = pfs[sz - 1]; pfs.pop_back (); 108 pfs[i] = pfs[sz - 1]; pfs.pop_back ();
77 } 109 }
110
111 w->p = 0;
78 } 112 }
79} 113}
80 114
115void io_manager::reschedule_time_watchers ()
116{
117 make_heap (tw.begin (), tw.end (), earliest_first);
118}
119
81void io_manager::reg (time_watcher *w) 120void io_manager::reg (time_watcher *w)
82{ 121{
122 w->registered = true;
123
83 tw.push_back (w); 124 tw.push_back (w);
84 push_heap (tw.begin (), tw.end (), lowest_first); 125 push_heap (tw.begin (), tw.end (), earliest_first);
85} 126}
86 127
87void io_manager::unreg (time_watcher *w) 128void io_manager::unreg (time_watcher *w)
88{ 129{
89 unsigned int sz = tw.size (); 130 if (w->registered)
90 unsigned int i = find (tw.begin (), tw.end (), w) - tw.begin ();
91
92 if (i != sz)
93 { 131 {
94 if (sz == 1) 132 unsigned int sz = tw.size ();
95 tw.clear (); 133 unsigned int i = find (tw.begin (), tw.end (), w) - tw.begin ();
96 else 134
135 assert (i != sz);
97 { 136
98 if (i != sz - 1) 137 if (i != sz - 1)
99 tw[i] = tw[sz - 1]; 138 tw[i] = tw[sz - 1];
100 139
140 tw.pop_back ();
141 reschedule_time_watchers ();
142
143 w->registered = false;
144 }
145}
146
147inline void set_now (void)
148{
149 struct timeval tv;
150
151 gettimeofday (&tv, 0);
152
153 NOW = (tstamp)tv.tv_sec + (tstamp)tv.tv_usec / 1000000;
154}
155
156void io_manager::loop ()
157{
158 set_now ();
159
160 for (;;)
161 {
162 while (tw[0]->at <= NOW)
163 {
164 // remove the first watcher
165 time_watcher *w = tw[0];
166
167 pop_heap (tw.begin (), tw.end (), earliest_first);
101 tw.pop_back (); 168 tw.pop_back ();
102 make_heap (tw.begin (), tw.end (), lowest_first); 169
170 w->registered = false;
171
172 // call it
173 w->call (*w);
174
175 // re-add it if necessary
176 if (w->at >= 0 && !w->registered)
177 reg (w);
178 }
179
180 int timeout = (int) ((tw[0]->at - NOW) * 1000);
181
182 int fds = poll (&pfs[0], pfs.size (), timeout);
183
184 set_now ();
185
186 vector<io_watcher *>::iterator w;
187 vector<pollfd>::iterator p;
188
189 for (w = iow.begin (), p = pfs.begin ();
190 fds > 0 && w < iow.end ();
191 ++w, ++p)
192 if (p->revents)
193 {
194 --fds;
195 (*w)->call (**w, p->revents);
103 } 196 }
104 } 197 }
105} 198}
106 199
107inline void set_now (void) 200void io_manager::idle_cb (time_watcher &w)
108{ 201{
109 struct timeval tv; 202 w.at = NOW + 86400; // wake up every day, for no good reason
110
111 gettimeofday (&tv, 0);
112
113 NOW = (tstamp)tv.tv_sec + (tstamp)tv.tv_usec / 1000000;
114} 203}
115 204
116void io_manager::loop () 205io_manager::io_manager ()
117{ 206{
207 iom_valid = true;
208
118 set_now (); 209 set_now ();
119 210 idle = new time_watcher (this, &io_manager::idle_cb);
120 while (!(iow.empty () && tw.empty ())) 211 idle->start (0);
121 {
122 int timeout = tw.empty ()
123 ? 3600 * 1000 // wake up at least every hour
124 : (int) ((tw[0]->at - NOW) * 1000);
125
126 printf ("s%d t%d #%d\n", pfs.size (), timeout, tw.size ());
127
128 if (timeout >= 0)
129 {
130 int fds = poll (&pfs[0], pfs.size (), timeout);
131
132 set_now ();
133
134 for (unsigned int i = iow.size (); fds && i--; )
135 if (pfs[i].revents)
136 {
137 --fds;
138 iow[i]->call (pfs[i].revents);
139 }
140 }
141
142 while (!tw.empty () && tw[0]->at <= NOW)
143 {
144 pop_heap (tw.begin (), tw.end (), lowest_first);
145 (*(tw.end () - 1))->trigger ();
146 push_heap (tw.begin (), tw.end (), lowest_first);
147 }
148 }
149}
150
151io_manager::io_manager ()
152{
153 set_now ();
154} 212}
155 213
156io_manager::~io_manager () 214io_manager::~io_manager ()
157{ 215{
158 // 216 iom_valid = false;
159} 217}
160 218

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines