restarts.cpp
上传用户:romrleung
上传日期:2022-05-23
资源大小:18897k
文件大小:3k
源码类别:

MySQL数据库

开发平台:

Visual C++

  1. /* Copyright (C) 2003 MySQL AB
  2.    This program is free software; you can redistribute it and/or modify
  3.    it under the terms of the GNU General Public License as published by
  4.    the Free Software Foundation; either version 2 of the License, or
  5.    (at your option) any later version.
  6.    This program is distributed in the hope that it will be useful,
  7.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  8.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  9.    GNU General Public License for more details.
  10.    You should have received a copy of the GNU General Public License
  11.    along with this program; if not, write to the Free Software
  12.    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  13. #include "mgmapi.h"
  14. #include <string.h>
  15. #include <NdbMain.h>
  16. #include <OutputStream.hpp>
  17. #include <NdbOut.hpp>
  18. #include <NdbSleep.h>
  19. #include <getarg.h>
  20. #include <NdbRestarts.hpp>
  21. #include <NDBT.hpp>
  22. int main(int argc, const char** argv){
  23.   ndb_init();
  24.   const char* _restartName = NULL;
  25.   int _loops = 1;
  26.   int _wait = -1;
  27.   int _maxwait = 120;
  28.   int _help = 0;
  29.   int _list = 0;
  30.   int _random = 0;
  31.   int _timeout = 0;
  32.   int _all = 0;
  33.   struct getargs args[] = {
  34.     { "seconds", 's', arg_integer, &_wait, 
  35.       "Seconds to wait between each restart(0=random)", "secs" },
  36.     { "max seconds", 'm', arg_integer, &_maxwait, 
  37.       "Max seconds to wait between each restart. Default is 120 seconds", "msecs" },
  38.     { "loops", 'l', arg_integer, &_loops, "Number of loops(0=forever)", "loops"},
  39.     { "timeout", 't', arg_integer, &_timeout, "Timeout waiting for nodes to start", "seconds"},
  40.     { "random", 'r', arg_flag, &_random, "Select restart case randomly", 
  41.       ""},
  42.     { "all", 'a', arg_flag, &_all, "Run all restarts", 
  43.       ""},
  44.     { "list-restarts", '', arg_flag, &_list, "List available restarts", ""},
  45.     { "usage", '?', arg_flag, &_help, "Print help", "" }
  46.     
  47.   };
  48.   int num_args = sizeof(args) / sizeof(args[0]);
  49.   int optind = 0;
  50.   char desc[] = 
  51.     "testnamen" 
  52.     "This program will perform node restart, n"
  53.     "multiple node restart or system-restart.n"
  54.     "Use --list-restarts to see whats availablen";
  55.   if(getarg(args, num_args, argc, argv, &optind) || _help) {
  56.     arg_printusage(args, num_args, argv[0], desc);
  57.     return NDBT_ProgramExit(NDBT_WRONGARGS);
  58.   }
  59.   if (_list){
  60.     NdbRestarts restarts;
  61.     restarts.listRestarts();
  62.     return NDBT_ProgramExit(NDBT_OK);
  63.   }
  64.   if ((argv[optind] == NULL) && (_random == 0)){
  65.     arg_printusage(args, num_args, argv[0], desc);
  66.     return NDBT_ProgramExit(NDBT_WRONGARGS);
  67.   }
  68.   _restartName = argv[optind];
  69.   
  70.   NdbRestarts restarts;
  71.   int res = NDBT_OK;
  72.   int l = 0;
  73.   while (_loops == 0 || l < _loops){
  74.     if (_all) {
  75.       // Execute all restarts, set loops to numRestarts
  76.       // so that ecvery restart is executed once
  77.       _loops = restarts.getNumRestarts();
  78.       res = restarts.executeRestart(l, _timeout);
  79.     } else if (_random) {
  80.       int num = rand() % restarts.getNumRestarts();
  81.       res = restarts.executeRestart(num, _timeout);
  82.     } else {
  83.       res = restarts.executeRestart(_restartName, _timeout);
  84.     }
  85.     if (res != NDBT_OK)
  86.       break;
  87.     if (_wait >= 0){
  88.       int seconds = _wait;
  89.       if(seconds==0) {
  90. // Create random value, default 120 secs
  91. seconds = (rand() % _maxwait) + 1; 
  92.       }
  93.       g_info << "Waiting for " << seconds << "(" << _maxwait 
  94.      << ") secs " << endl;
  95.       NdbSleep_SecSleep(seconds);
  96.     }
  97.     l++;
  98.   }
  99.   return NDBT_ProgramExit(res);
  100. }