git.m6w6.name Git - awesomized/libmemcached/blob - libtest/wait.h

   1 /*  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
   2  *
   3  *  libtest
   4  *
   5  *  Copyright (C) 2011 Data Differential, http://datadifferential.com/
   6  *
   7  *  This library is free software; you can redistribute it and/or
   8  *  modify it under the terms of the GNU Lesser General Public
   9  *  License as published by the Free Software Foundation; either
  10  *  version 3 of the License, or (at your option) any later version.
  11  *
  12  *  This library is distributed in the hope that it will be useful,
  13  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  *  Lesser General Public License for more details.
  16  *
  17  *  You should have received a copy of the GNU Lesser General Public
  18  *  License along with this library; if not, write to the Free Software
  19  *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
  20  */
  21
  22
  23 #pragma once
  24
  25 #include <unistd.h>
  26 #include <string>
  27
  28 namespace libtest {
  29
  30 class Wait
  31 {
  32 public:
  33
  34   Wait(const std::string &filename, uint32_t timeout= 6) :
  35     _successful(false)
  36   {
  37     uint32_t waited;
  38     uint32_t this_wait;
  39     uint32_t retry;
  40
  41     for (waited= 0, retry= 1; ; retry++, waited+= this_wait)
  42     {
  43       if (access(filename.c_str(), R_OK) == 0)
  44       {
  45         _successful= true;
  46         break;
  47       }
  48       else if (waited >= timeout)
  49       {
  50         break;
  51       }
  52
  53       this_wait= retry * retry / 3 + 1;
  54       sleep(this_wait);
  55 #ifdef WIN32
  56       sleep(this_wait);
  57 #else
  58       struct timespec global_sleep_value= { this_wait, 0 };
  59       nanosleep(&global_sleep_value, NULL);
  60 #endif
  61     }
  62   }
  63
  64   bool successful() const
  65   {
  66     return _successful;
  67   }
  68
  69 private:
  70   bool _successful;
  71 };
  72
  73 } // namespace libtest