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 #include <signal.h>
  28
  29 #include <libtest/dream.h>
  30
  31 namespace libtest {
  32
  33 class Wait
  34 {
  35 public:
  36
  37   Wait(const std::string &filename, uint32_t timeout= 6) :
  38     _successful(false)
  39   {
  40     uint32_t waited;
  41     uint32_t this_wait;
  42     uint32_t retry;
  43
  44     if (filename.empty())
  45     {
  46       _successful= false;
  47       return;
  48     }
  49
  50     for (waited= 0, retry= 1; ; retry++, waited+= this_wait)
  51     {
  52       if (access(filename.c_str(), R_OK) == 0)
  53       {
  54         _successful= true;
  55         break;
  56       }
  57       else if (waited >= timeout)
  58       {
  59         break;
  60       }
  61
  62       this_wait= retry * retry / 3 + 1;
  63       libtest::dream(this_wait, 0);
  64     }
  65   }
  66
  67   Wait(const pid_t &_pid_arg, uint32_t timeout= 6) :
  68     _successful(false)
  69   {
  70     uint32_t waited;
  71     uint32_t this_wait;
  72     uint32_t retry;
  73
  74     for (waited= 0, retry= 1; ; retry++, waited+= this_wait)
  75     {
  76       if (kill(_pid_arg, 0) == 0)
  77       {
  78         _successful= true;
  79         break;
  80       }
  81       else if (waited >= timeout)
  82       {
  83         break;
  84       }
  85
  86       this_wait= retry * retry / 3 + 1;
  87       libtest::dream(this_wait, 0);
  88     }
  89   }
  90
  91   bool successful() const
  92   {
  93     return _successful;
  94   }
  95
  96 private:
  97   bool _successful;
  98 };
  99
 100 } // namespace libtest