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

   1 /*  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
   2  *
   3  *  uTest
   4  *
   5  *  Copyright (C) 2011 Data Differential, http://datadifferential.com/
   6  *  All rights reserved.
   7  *
   8  *  Redistribution and use in source and binary forms, with or without
   9  *  modification, are permitted provided that the following conditions are
  10  *  met:
  11  *
  12  *      * Redistributions of source code must retain the above copyright
  13  *  notice, this list of conditions and the following disclaimer.
  14  *
  15  *      * Redistributions in binary form must reproduce the above
  16  *  copyright notice, this list of conditions and the following disclaimer
  17  *  in the documentation and/or other materials provided with the
  18  *  distribution.
  19  *
  20  *      * The names of its contributors may not be used to endorse or
  21  *  promote products derived from this software without specific prior
  22  *  written permission.
  23  *
  24  *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  25  *  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  26  *  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  27  *  A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  28  *  OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  29  *  SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  30  *  LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  31  *  DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  32  *  THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  33  *  (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  34  *  OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  35  *
  36  */
  37
  38 #pragma once
  39
  40 #include <unistd.h>
  41 #include <string>
  42
  43 namespace libtest {
  44
  45 class Wait
  46 {
  47 public:
  48
  49   Wait(const std::string &filename, uint32_t timeout= 6) :
  50     _successful(false)
  51   {
  52     uint32_t waited;
  53     uint32_t this_wait;
  54     uint32_t retry;
  55
  56     for (waited= 0, retry= 1; ; retry++, waited+= this_wait)
  57     {
  58       if (access(filename.c_str(), R_OK) == 0)
  59       {
  60         _successful= true;
  61         break;
  62       }
  63       else if (waited >= timeout)
  64       {
  65         break;
  66       }
  67
  68       this_wait= retry * retry / 3 + 1;
  69       sleep(this_wait);
  70 #ifdef WIN32
  71       sleep(this_wait);
  72 #else
  73       struct timespec global_sleep_value= { this_wait, 0 };
  74       nanosleep(&global_sleep_value, NULL);
  75 #endif
  76     }
  77   }
  78
  79   bool successful() const
  80   {
  81     return _successful;
  82   }
  83
  84 private:
  85   bool _successful;
  86 };
  87
  88 } // namespace libtest