开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 数据库系统 > MySQL数据库 > 查看源码
testSRBank.cpp
资源名称：mysql-4.1.16-win-src.zip　[点击查看]
上传用户：romrleung
上传日期：2022-05-23
资源大小：18897k
文件大小：6k
源码类别：

MySQL数据库

开发平台：

Visual C++

testSRBank.cpp：源码内容
  1. /* Copyright (C) 2003 MySQL AB
  3.    This program is free software; you can redistribute it and/or modify
  4.    it under the terms of the GNU General Public License as published by
  5.    the Free Software Foundation; either version 2 of the License, or
  6.    (at your option) any later version.
  8.    This program is distributed in the hope that it will be useful,
  9.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  10.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11.    GNU General Public License for more details.
  13.    You should have received a copy of the GNU General Public License
  14.    along with this program; if not, write to the Free Software
  15.    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  17. #include <NDBT.hpp>
  18. #include <NDBT_Test.hpp>
  19. #include <HugoTransactions.hpp>
  20. #include <UtilTransactions.hpp>
  21. #include <NdbBackup.hpp>
  23. #include "bank/Bank.hpp"
  25. int runCreateBank(NDBT_Context* ctx, NDBT_Step* step){
  26.   Bank bank;
  27.   int overWriteExisting = true;
  28.   if (bank.createAndLoadBank(overWriteExisting, 10) != NDBT_OK)
  29.     return NDBT_FAILED;
  30.   return NDBT_OK;
  31. }
  33. /**
  34.  *
  35.  * SR 0 - normal
  36.  * SR 1 - shutdown in progress
  37.  * SR 2 - restart in progress
  38.  */
  39. int runBankTimer(NDBT_Context* ctx, NDBT_Step* step){
  40.   int wait = 5; // Max seconds between each "day"
  41.   int yield = 1; // Loops before bank returns 
  43.   ctx->incProperty("ThreadCount");
  44.   while (!ctx->isTestStopped()) 
  45.   {
  46.     Bank bank;
  47.     while(!ctx->isTestStopped() && ctx->getProperty("SR") <= 1)
  48.       if(bank.performIncreaseTime(wait, yield) == NDBT_FAILED)
  49. break;
  51.     ndbout_c("runBankTimer is stopped");
  52.     ctx->incProperty("ThreadStopped");
  53.     if(ctx->getPropertyWait("SR", (Uint32)0))
  54.       break;
  55.   }
  56.   return NDBT_OK;
  57. }
  59. int runBankTransactions(NDBT_Context* ctx, NDBT_Step* step){
  60.   int wait = 0; // Max ms between each transaction
  61.   int yield = 1; // Loops before bank returns 
  63.   ctx->incProperty("ThreadCount");
  64.   while (!ctx->isTestStopped()) 
  65.   {
  66.     Bank bank;
  67.     while(!ctx->isTestStopped() && ctx->getProperty("SR") <= 1)
  68.       if(bank.performTransactions(0, 1) == NDBT_FAILED)
  69. break;
  70.     
  71.     ndbout_c("runBankTransactions is stopped");
  72.     ctx->incProperty("ThreadStopped");
  73.     if(ctx->getPropertyWait("SR", (Uint32)0))
  74.       break;
  75.   }
  76.   return NDBT_OK;
  77. }
  79. int runBankGL(NDBT_Context* ctx, NDBT_Step* step){
  80.   int yield = 1; // Loops before bank returns 
  81.   int result = NDBT_OK;
  83.   ctx->incProperty("ThreadCount");
  84.   while (ctx->isTestStopped() == false) 
  85.   {
  86.     Bank bank;
  87.     while(!ctx->isTestStopped() && ctx->getProperty("SR") <= 1)
  88.       if (bank.performMakeGLs(yield) != NDBT_OK)
  89.       {
  90. if(ctx->getProperty("SR") != 0)
  91.   break;
  92. ndbout << "bank.performMakeGLs FAILED" << endl;
  93. return NDBT_FAILED;
  94.       }
  95.     
  96.     ndbout_c("runBankGL is stopped");
  97.     ctx->incProperty("ThreadStopped");
  98.     if(ctx->getPropertyWait("SR", (Uint32)0))
  99.       break;
  100.   }
  101.   return NDBT_OK;
  102. }
  104. int runBankSum(NDBT_Context* ctx, NDBT_Step* step){
  105.   Bank bank;
  106.   int wait = 2000; // Max ms between each sum of accounts
  107.   int yield = 1; // Loops before bank returns 
  108.   int result = NDBT_OK;
  110.   while (ctx->isTestStopped() == false) {
  111.     if (bank.performSumAccounts(wait, yield) != NDBT_OK){
  112.       ndbout << "bank.performSumAccounts FAILED" << endl;
  113.       result = NDBT_FAILED;
  114.     }
  115.   }
  116.   return result ;
  117. }
  119. #define CHECK(b) if (!(b)) { 
  120.   g_err << "ERR: "<< step->getName() 
  121.          << " failed on line " << __LINE__ << endl; 
  122.   result = NDBT_FAILED; 
  123.   continue; } 
  125. int runSR(NDBT_Context* ctx, NDBT_Step* step)
  126. {
  127.   int result = NDBT_OK;
  128.   int runtime = ctx->getNumLoops();
  129.   int sleeptime = ctx->getNumRecords();
  130.   NdbRestarter restarter;
  131.   bool abort = true;
  132.   int timeout = 180;
  134.   Uint32 now;
  135.   const Uint32 stop = time(0)+ runtime;
  136.   while(!ctx->isTestStopped() && ((now= time(0)) < stop) && result == NDBT_OK)
  137.   {
  138.     ndbout << " -- Sleep " << sleeptime << "s " << endl;
  139.     NdbSleep_SecSleep(sleeptime);
  140.     ndbout << " -- Shutting down " << endl;
  141.     ctx->setProperty("SR", 1);
  142.     CHECK(restarter.restartAll(false, true, abort) == 0);
  143.     ctx->setProperty("SR", 2);
  144.     CHECK(restarter.waitClusterNoStart(timeout) == 0);
  146.     Uint32 cnt = ctx->getProperty("ThreadCount");
  147.     Uint32 curr= ctx->getProperty("ThreadStopped");
  148.     while(curr != cnt)
  149.     {
  150.       ndbout_c("%d %d", curr, cnt);
  151.       NdbSleep_MilliSleep(100);
  152.       curr= ctx->getProperty("ThreadStopped");
  153.     }
  155.     ctx->setProperty("ThreadStopped", (Uint32)0);
  156.     CHECK(restarter.startAll() == 0);
  157.     CHECK(restarter.waitClusterStarted(timeout) == 0);
  158.     
  159.     ndbout << " -- Validating starts " << endl;
  160.     {
  161.       int wait = 0;
  162.       int yield = 1;
  163.       Bank bank;
  164.       if (bank.performSumAccounts(wait, yield) != 0)
  165.       {
  166. ndbout << "bank.performSumAccounts FAILED" << endl;
  167. return NDBT_FAILED;
  168.       }
  170.       if (bank.performValidateAllGLs() != 0)
  171.       {
  172. ndbout << "bank.performValidateAllGLs FAILED" << endl;
  173. return NDBT_FAILED;
  174.       }
  175.     }
  177.     ndbout << " -- Validating complete " << endl;
  178.     ctx->setProperty("SR", (Uint32)0);
  179.     ctx->broadcast();
  180.   }
  181.   ctx->stopTest();
  182.   return NDBT_OK;
  183. }
  185. int runDropBank(NDBT_Context* ctx, NDBT_Step* step){
  186.   Bank bank;
  187.   if (bank.dropBank() != NDBT_OK)
  188.     return NDBT_FAILED;
  189.   return NDBT_OK;
  190. }
  193. NDBT_TESTSUITE(testSRBank);
  194. TESTCASE("Graceful", 
  195.  " Test that a consistent bank is restored after graceful shutdownn"
  196.  "1.  Create bankn"
  197.  "2.  Start bank and let it runn"
  198.  "3.  Restart ndb and verify consistencyn"
  199.  "4.  Drop bankn")
  200. {
  201.   INITIALIZER(runCreateBank);
  202.   STEP(runBankTimer);
  203.   STEP(runBankTransactions);
  204.   STEP(runBankTransactions);
  205.   STEP(runBankTransactions);
  206.   STEP(runBankTransactions);
  207.   STEP(runBankTransactions);
  208.   STEP(runBankTransactions);
  209.   STEP(runBankTransactions);
  210.   STEP(runBankTransactions);
  211.   STEP(runBankTransactions);
  212.   STEP(runBankTransactions);
  213.   STEP(runBankGL);
  214.   STEP(runSR);
  215. }
  216. TESTCASE("Abort", 
  217.  " Test that a consistent bank is restored after graceful shutdownn"
  218.  "1.  Create bankn"
  219.  "2.  Start bank and let it runn"
  220.  "3.  Restart ndb and verify consistencyn"
  221.  "4.  Drop bankn")
  222. {
  223.   INITIALIZER(runCreateBank);
  224.   STEP(runBankTimer);
  225.   STEP(runBankTransactions);
  226.   STEP(runBankTransactions);
  227.   STEP(runBankTransactions);
  228.   STEP(runBankTransactions);
  229.   STEP(runBankTransactions);
  230.   STEP(runBankTransactions);
  231.   STEP(runBankTransactions);
  232.   STEP(runBankTransactions);
  233.   STEP(runBankTransactions);
  234.   STEP(runBankTransactions);
  235.   STEP(runBankGL);
  236.   STEP(runSR);
  237.   FINALIZER(runDropBank);
  238. }
  239. NDBT_TESTSUITE_END(testSRBank);
  241. int main(int argc, const char** argv){
  242.   ndb_init();
  243.   return testSRBank.execute(argc, argv);
  244. }