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

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. /*
  14.  * list_tables
  15.  *
  16.  * List objects(tables, triggers, etc.) in NDB Cluster
  17.  *
  18.  */
  19. #include <ndb_global.h>
  20. #include <ndb_opts.h>
  21. #include <NdbApi.hpp>
  22. #include <NDBT.hpp>
  23. static Ndb_cluster_connection *ndb_cluster_connection= 0;
  24. static Ndb* ndb = 0;
  25. static NdbDictionary::Dictionary* dic = 0;
  26. static int _unqualified = 0;
  27. static void
  28. fatal(char const* fmt, ...)
  29. {
  30.     va_list ap;
  31.     char buf[500];
  32.     va_start(ap, fmt);
  33.     BaseString::vsnprintf(buf, sizeof(buf), fmt, ap);
  34.     va_end(ap);
  35.     ndbout << buf;
  36.     if (ndb)
  37.       ndbout << " - " << ndb->getNdbError();
  38.     ndbout << endl;
  39.     NDBT_ProgramExit(NDBT_FAILED);
  40.     exit(1);
  41. }
  42. static void
  43. fatal_dict(char const* fmt, ...)
  44. {
  45.     va_list ap;
  46.     char buf[500];
  47.     va_start(ap, fmt);
  48.     BaseString::vsnprintf(buf, sizeof(buf), fmt, ap);
  49.     va_end(ap);
  50.     ndbout << buf;
  51.     if (dic)
  52.       ndbout << " - " << dic->getNdbError();
  53.     ndbout << endl;
  54.     NDBT_ProgramExit(NDBT_FAILED);
  55.     exit(1);
  56. }
  57. static void
  58. list(const char * tabname, 
  59.      NdbDictionary::Object::Type type)
  60. {
  61.     NdbDictionary::Dictionary::List list;
  62.     if (tabname == 0) {
  63. if (dic->listObjects(list, type) == -1)
  64.     fatal_dict("listObjects");
  65.     } else {
  66. if (dic->listIndexes(list, tabname) == -1)
  67.     fatal_dict("listIndexes");
  68.     }
  69.     if (ndb->usingFullyQualifiedNames())
  70.        ndbout_c("%-5s %-20s %-8s %-7s %-12s %-8s %s", "id", "type", "state", "logging", "database", "schema", "name");
  71.      else
  72.        ndbout_c("%-5s %-20s %-8s %-7s %s", "id", "type", "state", "logging", "name");
  73.     for (unsigned i = 0; i < list.count; i++) {
  74. NdbDictionary::Dictionary::List::Element& elt = list.elements[i];
  75.         char type[100];
  76. bool isTable = false;
  77.         switch (elt.type) {
  78.         case NdbDictionary::Object::SystemTable:
  79.             strcpy(type, "SystemTable");
  80.     isTable = true;
  81.             break;
  82.         case NdbDictionary::Object::UserTable:
  83.             strcpy(type, "UserTable");
  84.     isTable = true;
  85.             break;
  86.         case NdbDictionary::Object::UniqueHashIndex:
  87.             strcpy(type, "UniqueHashIndex");
  88.     isTable = true;
  89.             break;
  90.         case NdbDictionary::Object::OrderedIndex:
  91.             strcpy(type, "OrderedIndex");
  92.     isTable = true;
  93.             break;
  94.         case NdbDictionary::Object::HashIndexTrigger:
  95.             strcpy(type, "HashIndexTrigger");
  96.             break;
  97.         case NdbDictionary::Object::IndexTrigger:
  98.             strcpy(type, "IndexTrigger");
  99.             break;
  100.         case NdbDictionary::Object::SubscriptionTrigger:
  101.             strcpy(type, "SubscriptionTrigger");
  102.             break;
  103.         case NdbDictionary::Object::ReadOnlyConstraint:
  104.             strcpy(type, "ReadOnlyConstraint");
  105.             break;
  106.         default:
  107.             sprintf(type, "%d", (int)elt.type);
  108.             break;
  109.         }
  110.         char state[100];
  111.         switch (elt.state) {
  112.         case NdbDictionary::Object::StateOffline:
  113.             strcpy(state, "Offline");
  114.             break;
  115.         case NdbDictionary::Object::StateBuilding:
  116.             strcpy(state, "Building");
  117.             break;
  118.         case NdbDictionary::Object::StateDropping:
  119.             strcpy(state, "Dropping");
  120.             break;
  121.         case NdbDictionary::Object::StateOnline:
  122.             strcpy(state, "Online");
  123.             break;
  124.         case NdbDictionary::Object::StateBackup:
  125.     strcpy(state, "Backup");
  126.             break;
  127.         case NdbDictionary::Object::StateBroken:
  128.             strcpy(state, "Broken");
  129.             break;
  130.         default:
  131.             sprintf(state, "%d", (int)elt.state);
  132.             break;
  133.         }
  134.         char store[100];
  135. if (! isTable)
  136.     strcpy(store, "-");
  137. else {
  138.     switch (elt.store) {
  139.     case NdbDictionary::Object::StoreTemporary:
  140. strcpy(store, "No");
  141. break;
  142.     case NdbDictionary::Object::StorePermanent:
  143. strcpy(store, "Yes");
  144. break;
  145.     default:
  146. sprintf(state, "%d", (int)elt.store);
  147. break;
  148.     }
  149. }
  150. if (ndb->usingFullyQualifiedNames())
  151.   ndbout_c("%-5d %-20s %-8s %-7s %-12s %-8s %s", elt.id, type, state, store, (elt.database)?elt.database:"", (elt.schema)?elt.schema:"", elt.name);
  152.        else
  153.  ndbout_c("%-5d %-20s %-8s %-7s %s", elt.id, type, state, store, elt.name);
  154.     }
  155. }
  156. NDB_STD_OPTS_VARS;
  157. static const char* _dbname = "TEST_DB";
  158. static int _loops;
  159. static int _type;
  160. static struct my_option my_long_options[] =
  161. {
  162.   NDB_STD_OPTS("ndb_show_tables"),
  163.   { "database", 'd', "Name of database table is in",
  164.     (gptr*) &_dbname, (gptr*) &_dbname, 0,
  165.     GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0 },
  166.   { "loops", 'l', "loops",
  167.     (gptr*) &_loops, (gptr*) &_loops, 0,
  168.     GET_INT, REQUIRED_ARG, 1, 0, 0, 0, 0, 0 }, 
  169.   { "type", 't', "type",
  170.     (gptr*) &_type, (gptr*) &_type, 0,
  171.     GET_INT, REQUIRED_ARG, 0, 0, 0, 0, 0, 0 }, 
  172.   { "unqualified", 'u', "Use unqualified table names",
  173.     (gptr*) &_unqualified, (gptr*) &_unqualified, 0,
  174.     GET_BOOL, NO_ARG, 0, 0, 0, 0, 0, 0 }, 
  175.   { 0, 0, 0, 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0}
  176. };
  177. static void usage()
  178. {
  179.   char desc[] = 
  180.     "tabnamen"
  181.     "This program list all system objects in  NDB Cluster.n"
  182.     "Type of objects to display can be limited with -t optionn"
  183.     " ex: ndb_show_tables -t 2 would show all UserTablesn"
  184.     "To show all indexes for a table write table name as final argumentn"
  185.     "  ex: ndb_show_tables T1n";
  186.   ndb_std_print_version();
  187.   my_print_help(my_long_options);
  188.   my_print_variables(my_long_options);
  189. }
  190. static my_bool
  191. get_one_option(int optid, const struct my_option *opt __attribute__((unused)),
  192.        char *argument)
  193. {
  194.   return ndb_std_get_one_option(optid, opt, argument ? argument :
  195. "d:t:O,/tmp/ndb_show_tables.trace");
  196. }
  197. int main(int argc, char** argv){
  198.   NDB_INIT(argv[0]);
  199.   const char* _tabname;
  200.   const char *load_default_groups[]= { "mysql_cluster",0 };
  201.   load_defaults("my",load_default_groups,&argc,&argv);
  202.   int ho_error;
  203.   if ((ho_error=handle_options(&argc, &argv, my_long_options, get_one_option)))
  204.     return NDBT_ProgramExit(NDBT_WRONGARGS);
  205.   _tabname = argv[0];
  206.   ndb_cluster_connection = new Ndb_cluster_connection(opt_connect_str);
  207.   if (ndb_cluster_connection->connect(12,5,1))
  208.     fatal("unable to connect");
  209.   ndb = new Ndb(ndb_cluster_connection, _dbname);
  210.   if (ndb->init() != 0)
  211.     fatal("init");
  212.   if (ndb->waitUntilReady(30) < 0)
  213.     fatal("waitUntilReady");
  214.   dic = ndb->getDictionary();
  215.   for (int i = 0; _loops == 0 || i < _loops; i++) {
  216.     list(_tabname, (NdbDictionary::Object::Type)_type);
  217.   }
  218.   return NDBT_ProgramExit(NDBT_OK);
  219. }
  220. // vim: set sw=4: