git.m6w6.name Git - m6w6/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     if (filename.empty())
  42     {
  43       _successful= false;
  44       return;
  45     }
  46
  47     for (waited= 0, retry= 1; ; retry++, waited+= this_wait)
  48     {
  49       if (access(filename.c_str(), R_OK) == 0)
  50       {
  51         _successful= true;
  52         break;
  53       }
  54       else if (waited >= timeout)
  55       {
  56         break;
  57       }
  58
  59       this_wait= retry * retry / 3 + 1;
  60       sleep(this_wait);
  61 #ifdef WIN32
  62       sleep(this_wait);
  63 #else
  64       struct timespec global_sleep_value= { this_wait, 0 };
  65       nanosleep(&global_sleep_value, NULL);
  66 #endif
  67     }
  68   }
  69
  70   bool successful() const
  71   {
  72     return _successful;
  73   }
  74
  75 private:
  76   bool _successful;
  77 };
  78
  79 } // namespace libtest