create_tab.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 <ndb_global.h>
  14. #include <NdbOut.hpp>
  15. #include <NdbApi.hpp>
  16. #include <NDBT.hpp>
  17. #include <getarg.h>
  18. int main(int argc, const char** argv){
  19.   ndb_init();
  20.   int _temp = false;
  21.   int _help = 0;
  22.   int _all = 0;
  23.   int _print = 0;
  24.   const char* _connectstr = NULL;
  25.   struct getargs args[] = {
  26.     { "all", 'a', arg_flag, &_all, "Create/print all tables" },
  27.     { "print", 'p', arg_flag, &_print, "Print table(s) instead of creating it"},
  28.     { "temp", 't', arg_flag, &_temp, "Temporary table", "temp" },
  29.     { "connstr", 'c', arg_string, &_connectstr, "connect string", 
  30.       "How to connect to NDB"}, 
  31.     { "usage", '?', arg_flag, &_help, "Print help", "" }
  32.   };
  33.   int num_args = sizeof(args) / sizeof(args[0]);
  34.   int optind = 0;
  35.   char desc[] = 
  36.     "tabnamen"
  37.     "This program will create one table in Ndb.n"
  38.     "The tables may be selected from a fixed list of tablesn"
  39.     "defined in NDBT_Tables classn";
  40.   if(getarg(args, num_args, argc, argv, &optind) || _help){
  41.     arg_printusage(args, num_args, argv[0], desc);
  42.     return NDBT_ProgramExit(NDBT_WRONGARGS);
  43.   }
  44.   if(argv[optind] == NULL && !_all){
  45.     arg_printusage(args, num_args, argv[0], desc);
  46.     return NDBT_ProgramExit(NDBT_WRONGARGS);
  47.   }
  48.   int res = 0;
  49.   if(_print){
  50.     /**
  51.      * Print instead of creating
  52.      */
  53.     if(optind < argc){
  54.       for(int i = optind; i<argc; i++){
  55. NDBT_Tables::print(argv[i]);
  56.       }
  57.     } else {
  58.       NDBT_Tables::printAll();
  59.     }
  60.   } else {
  61.     /**
  62.      * Creating
  63.      */
  64.     
  65.     // Connect to Ndb
  66.     Ndb::setConnectString(_connectstr);
  67.     Ndb MyNdb( "TEST_DB" );
  68.     
  69.     if(MyNdb.init() != 0){
  70.       ERR(MyNdb.getNdbError());
  71.       return NDBT_ProgramExit(NDBT_FAILED);
  72.     }
  73.     
  74.     while(MyNdb.waitUntilReady() != 0)
  75.       ndbout << "Waiting for ndb to become ready..." << endl;
  76.     
  77.     if(_all){
  78.       res = NDBT_Tables::createAllTables(&MyNdb, _temp);
  79.     } else {
  80.       int tmp;
  81.       for(int i = optind; i<argc; i++){
  82. ndbout << "Trying to create " <<  argv[i] << endl;
  83. if((tmp = NDBT_Tables::createTable(&MyNdb, argv[i], _temp)) != 0)
  84.   res = tmp;
  85.       }
  86.     } 
  87.   }
  88.   
  89.   if(res != 0)
  90.     return NDBT_ProgramExit(NDBT_FAILED);
  91.   else
  92.     return NDBT_ProgramExit(NDBT_OK);
  93. }