--- gvpe/src/iom.C 2003/03/21 21:21:02 1.3 +++ gvpe/src/iom.C 2003/03/28 04:05:10 1.6 @@ -37,9 +37,28 @@ void time_watcher::set (tstamp when) { - iom.unreg (this); at = when; - iom.reg (this); + + if (registered) + iom.reschedule_time_watchers (); + else + iom.reg (this); +} + +void time_watcher::trigger () +{ + call (at); + + if (registered) + iom.reschedule_time_watchers (); + else + iom.reg (this); +} + +void time_watcher::start () +{ + if (!registered) + iom.reg (this); } void io_manager::reg (int fd, short events, io_watcher *w) @@ -53,7 +72,7 @@ iow.push_back (w); } -void io_manager::unreg (io_watcher *w) +void io_manager::unreg (const io_watcher *w) { unsigned int sz = iow.size (); unsigned int i = find (iow.begin (), iow.end (), w) - iow.begin (); @@ -78,29 +97,32 @@ } } +void io_manager::reschedule_time_watchers () +{ + make_heap (tw.begin (), tw.end (), lowest_first); +} + void io_manager::reg (time_watcher *w) { + if (w->registered) + slog (L_CRIT, "FATAL: io_manager::reg(time_watcher) called on already-registered watcher"); + tw.push_back (w); push_heap (tw.begin (), tw.end (), lowest_first); } -void io_manager::unreg (time_watcher *w) +void io_manager::unreg (const time_watcher *w) { unsigned int sz = tw.size (); unsigned int i = find (tw.begin (), tw.end (), w) - tw.begin (); if (i != sz) { - if (sz == 1) - tw.clear (); - else - { - if (i != sz - 1) - tw[i] = tw[sz - 1]; + if (i != sz - 1) + tw[i] = tw[sz - 1]; - tw.pop_back (); - make_heap (tw.begin (), tw.end (), lowest_first); - } + tw.pop_back (); + reschedule_time_watchers (); } } @@ -117,40 +139,48 @@ { set_now (); - while (!(iow.empty () && tw.empty ())) + for (;;) { - int timeout = tw.empty () - ? 3600 * 1000 // wake up at least every hour - : (int) ((tw[0]->at - NOW) * 1000); - - printf ("s%d t%d #%d\n", pfs.size (), timeout, tw.size ()); - - if (timeout >= 0) + while (tw[0]->at <= NOW) { - int fds = poll (&pfs[0], pfs.size (), timeout); + pop_heap (tw.begin (), tw.end (), lowest_first); + time_watcher *w = *(tw.end () - 1); + tw.pop_back (); - set_now (); + if (w->at >= 0) + { + w->call (w->at); - for (unsigned int i = iow.size (); fds && i--; ) - if (pfs[i].revents) - { - --fds; - iow[i]->call (pfs[i].revents); - } + if (!w->registered) + reg (w); + } } - while (!tw.empty () && tw[0]->at <= NOW) - { - pop_heap (tw.begin (), tw.end (), lowest_first); - (*(tw.end () - 1))->trigger (); - push_heap (tw.begin (), tw.end (), lowest_first); - } + int timeout = (int) ((tw[0]->at - NOW) * 1000); + + int fds = poll (&pfs[0], pfs.size (), timeout); + + set_now (); + + for (unsigned int i = iow.size (); fds > 0 && i--; ) + if (pfs[i].revents) + { + --fds; + iow[i]->call (pfs[i].revents); + } } } +void io_manager::idle_cb (tstamp &ts) +{ + ts = NOW + 86400; // wake up every day, for no good reason +} + io_manager::io_manager () { set_now (); + idle = new time_watcher (this, &io_manager::idle_cb); + idle->start (0); } io_manager::~io_manager ()