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

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 <ndb_global.h>
  14. #include <NdbMain.h>
  15. #include <OutputStream.hpp>
  16. #include <NdbOut.hpp>
  17. #include <NdbSleep.h>
  18. #include <getarg.h>
  19. #include <NdbRestarter.hpp>
  20. #include <NDBT.hpp>
  21. int main(int argc, const char** argv){
  22.   ndb_init();
  23.   const char* _hostName = NULL;
  24.   int _initial = 0;
  25.   int _help = 0;
  26.   int _wait = 1;
  27.   struct getargs args[] = {
  28.     { "initial", 'i', arg_flag, &_initial, "Do initial restart"},
  29.     { "wait", '', arg_negative_flag, &_wait, "Wait until restarted(default=true)"},
  30.     { "usage", '?', arg_flag, &_help, "Print help", "" }
  31.     
  32.   };
  33.   int num_args = sizeof(args) / sizeof(args[0]);
  34.   int optind = 0;
  35.   char desc[] = 
  36.     "hostname:portn"
  37.     "This program will connect to the mgmsrv of a NDB clustern"
  38.     " and restart the cluster. n";
  39.   if(getarg(args, num_args, argc, argv, &optind) || _help) {
  40.     arg_printusage(args, num_args, argv[0], desc);
  41.     return NDBT_ProgramExit(NDBT_WRONGARGS);
  42.   }
  43.   _hostName = argv[optind];
  44.   
  45.   NdbRestarter restarter(_hostName);
  46.   setOutputLevel(1); // Show only g_err
  47.   int result = NDBT_OK;
  48.   if (_initial){
  49.     ndbout << "Restarting cluster with initial restart" << endl;
  50.     if (restarter.restartAll(true, false, false) != 0)
  51.       result = NDBT_FAILED;
  52.   } else {
  53.     ndbout << "Restarting cluster " << endl;
  54.     if (restarter.restartAll() != 0)
  55.       result = NDBT_FAILED;
  56.   }
  57.   if (result == NDBT_FAILED){
  58.     g_err << "Failed to restart cluster" << endl;
  59.     return NDBT_ProgramExit(NDBT_FAILED);
  60.   }
  61.   
  62.   if (_wait == 1){
  63.     ndbout << "Waiting for cluster to start" << endl;
  64.     if ( restarter.waitClusterStarted(120) != 0){
  65.       ndbout << "Failed waiting for restart of cluster" << endl;
  66.       result = NDBT_FAILED;
  67.     }
  68.   }
  69.   ndbout << "Cluster restarted" << endl;
  70.   return NDBT_ProgramExit(result);
  71. }