PollTimeout.h

Go to the documentation of this file.

#ifndef HYPERTABLE_POLLTIMEOUT_H
#define HYPERTABLE_POLLTIMEOUT_H

// remove
#include <iostream>

#include <boost/thread/xtime.hpp>

#include <cassert>

namespace Hypertable {

  class PollTimeout {

  public:

    PollTimeout() : ts_ptr(0), duration_millis(-1) { return; }

    void set(boost::xtime &now, boost::xtime &expire) {
      assert((xtime_cmp(now , expire) <= 0));
      if (now.sec == expire.sec) {
        duration_ts.tv_sec = 0;
        duration_ts.tv_nsec = expire.nsec - now.nsec;
      }
      else {
        uint64_t nanos = expire.nsec + (1000000000LL - now.nsec);
        duration_ts.tv_sec = ((expire.sec - now.sec) - 1)
                              + (nanos / 1000000000LL);
        duration_ts.tv_nsec = nanos % 1000000000LL;
      }
      ts_ptr = &duration_ts;
      duration_millis = (int)((duration_ts.tv_sec * 1000)
                              + (duration_ts.tv_nsec / 1000000));
      assert(duration_millis >= 0);
    }

    void set_indefinite() {
      ts_ptr = 0;
      duration_millis = -1;
    }

    int get_millis() { return duration_millis; }

    struct timespec *get_timespec() { return ts_ptr; }

  private:

    struct timespec *ts_ptr;
    struct timespec duration_ts;
    int duration_millis;
  };

}

#endif // HYPERTABLE_POLLTIMEOUT_H