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

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

MySQL数据库

开发平台：

Visual C++

ndb_config.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. /**
  18.  * ndb_config --nodes --query=nodeid --type=ndbd --host=local1
  19.  */
  21. #include <ndb_global.h>
  22. #include <my_sys.h>
  23. #include <my_getopt.h>
  24. #include <mysql_version.h>
  26. #include <NdbOut.hpp>
  27. #include <mgmapi.h>
  28. #include <mgmapi_configuration.hpp>
  29. #include <ConfigInfo.hpp>
  31. static int g_verbose = 0;
  32. static int try_reconnect = 3;
  34. static int g_nodes = 1;
  35. static const char * g_connectstring = 0;
  36. static const char * g_query = 0;
  38. static int g_nodeid = 0;
  39. static const char * g_type = 0;
  40. static const char * g_host = 0;
  41. static const char * g_field_delimiter=",";
  42. static const char * g_row_delimiter=" ";
  43. static const char * g_config_file = 0;
  45. int g_print_full_config, opt_ndb_shm;
  46. my_bool opt_core;
  48. typedef ndb_mgm_configuration_iterator Iter;
  50. static void ndb_std_print_version()
  51. {
  52.   printf("MySQL distrib %s, for %s (%s)n",
  53.  MYSQL_SERVER_VERSION,SYSTEM_TYPE,MACHINE_TYPE);
  54. }
  56. static struct my_option my_long_options[] =
  57. {
  58.   { "usage", '?', "Display this help and exit.", 
  59.     0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0 }, 
  60.   { "help", '?', "Display this help and exit.", 
  61.     0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0 }, 
  62.   { "version", 'V', "Output version information and exit.", 0, 0, 0, 
  63.     GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0 }, 
  64.   { "ndb-connectstring", 256,
  65.     "Set connect string for connecting to ndb_mgmd. " 
  66.     "Syntax: "[nodeid=<id>;][host=]<hostname>[:<port>]". " 
  67.     "Overides specifying entries in NDB_CONNECTSTRING and Ndb.cfg", 
  68.     (gptr*) &g_connectstring, (gptr*) &g_connectstring, 
  69.     0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0 },
  70.   { "nodes", 256, "Print nodes",
  71.     (gptr*) &g_nodes, (gptr*) &g_nodes,
  72.     0, GET_BOOL, NO_ARG, 1, 0, 0, 0, 0, 0},
  73.   { "query", 'q', "Query option(s)",
  74.     (gptr*) &g_query, (gptr*) &g_query,
  75.     0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  76.   { "host", 257, "Host",
  77.     (gptr*) &g_host, (gptr*) &g_host,
  78.     0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  79.   { "type", 258, "Type of node/connection",
  80.     (gptr*) &g_type, (gptr*) &g_type,
  81.     0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  82.   { "id", 258, "Nodeid",
  83.     (gptr*) &g_nodeid, (gptr*) &g_nodeid,
  84.     0, GET_INT, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  85.   { "nodeid", 258, "Nodeid",
  86.     (gptr*) &g_nodeid, (gptr*) &g_nodeid,
  87.     0, GET_INT, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  88.   { "fields", 'f', "Field separator",
  89.     (gptr*) &g_field_delimiter, (gptr*) &g_field_delimiter,
  90.     0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  91.   { "rows", 'r', "Row separator",
  92.     (gptr*) &g_row_delimiter, (gptr*) &g_row_delimiter,
  93.     0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  94.   { "config-file", 256, "Path to config.ini",
  95.     (gptr*) &g_config_file, (gptr*) &g_config_file,
  96.     0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  97.   { 0, 0, 0, 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0}
  98. };
  100. static void usage()
  101. {
  102.   char desc[] = 
  103.     "This program will retreive config options for a ndb clustern";
  104.   ndb_std_print_version();
  105.   my_print_help(my_long_options);
  106.   my_print_variables(my_long_options);
  107. }
  108. static my_bool
  109. ndb_std_get_one_option(int optid,
  110.        const struct my_option *opt __attribute__((unused)),
  111.        char *argument)
  112. {
  113.   switch (optid) {
  114.   case 'V':
  115.     ndb_std_print_version();
  116.     exit(0);
  117.   case '?':
  118.     usage();
  119.     exit(0);
  120.   }
  121.   return 0;
  122. }
  124. /**
  125.  * Match/Apply framework
  126.  */
  127. struct Match
  128. {
  129.   int m_key;
  130.   BaseString m_value;
  131.   virtual int eval(const Iter&);
  132. };
  134. struct Apply
  135. {
  136.   Apply() {}
  137.   Apply(int val) { m_key = val;}
  138.   int m_key;
  139.   virtual int apply(const Iter&);
  140. };
  142. struct NodeTypeApply : public Apply
  143. {
  144.   virtual int apply(const Iter&);
  145. };
  147. static int parse_query(Vector<Apply*>&, int &argc, char**& argv);
  148. static int parse_where(Vector<Match*>&, int &argc, char**& argv);
  149. static int eval(const Iter&, const Vector<Match*>&);
  150. static int apply(const Iter&, const Vector<Apply*>&);
  151. static ndb_mgm_configuration* fetch_configuration();
  152. static ndb_mgm_configuration* load_configuration();
  154. int
  155. main(int argc, char** argv){
  156.   NDB_INIT(argv[0]);
  157.   const char *load_default_groups[]= { "mysql_cluster",0 };
  158.   load_defaults("my",load_default_groups,&argc,&argv);
  159.   int ho_error;
  160.   if ((ho_error=handle_options(&argc, &argv, my_long_options,
  161.        ndb_std_get_one_option)))
  162.     return -1;
  164.   ndb_mgm_configuration * conf = 0;
  166.   if (g_config_file)
  167.     conf = load_configuration();
  168.   else
  169.     conf = fetch_configuration();
  171.   if (conf == 0)
  172.   {
  173.     return -1;
  174.   }
  175.   
  176.   Vector<Apply*> select_list;
  177.   Vector<Match*> where_clause;
  179.   if(strcmp(g_row_delimiter, "\n") == 0)
  180.     g_row_delimiter = "n";
  181.   if(strcmp(g_field_delimiter, "\n") == 0)
  182.     g_field_delimiter = "n";
  184.   if(parse_query(select_list, argc, argv))
  185.   {
  186.     exit(0);
  187.   }
  189.   if(parse_where(where_clause, argc, argv))
  190.   {
  191.     exit(0);
  192.   }
  194.   Iter iter(* conf, CFG_SECTION_NODE);
  195.   bool prev= false;
  196.   iter.first();
  197.   for(iter.first(); iter.valid(); iter.next())
  198.   {
  199.     if(eval(iter, where_clause))
  200.     {
  201.       if(prev)
  202. printf("%s", g_row_delimiter);
  203.       prev= true;
  204.       apply(iter, select_list);
  205.     }
  206.   }
  207.   printf("n");
  208.   return 0;
  209. }
  211. static
  212. int
  213. parse_query(Vector<Apply*>& select, int &argc, char**& argv)
  214. {
  215.   if(g_query)
  216.   {
  217.     BaseString q(g_query);
  218.     Vector<BaseString> list;
  219.     q.split(list, ",");
  220.     for(unsigned i = 0; i<list.size(); i++)
  221.     {
  222.       const char * str= list[i].c_str();
  223.       if(strcasecmp(str, "id") == 0 || strcasecmp(str, "nodeid") == 0)
  224. select.push_back(new Apply(CFG_NODE_ID));
  225.       else if(strncasecmp(str, "host", 4) == 0)
  226. select.push_back(new Apply(CFG_NODE_HOST));
  227.       else if(strcasecmp(str, "type") == 0)
  228. select.push_back(new NodeTypeApply());
  229.       else if(g_nodes)
  230.       {
  231. bool found = false;
  232. for(int p = 0; p<ConfigInfo::m_NoOfParams; p++)
  233. {
  234.   if(0)ndbout_c("%s %s",
  235. ConfigInfo::m_ParamInfo[p]._section,
  236. ConfigInfo::m_ParamInfo[p]._fname);
  237.   if(strcmp(ConfigInfo::m_ParamInfo[p]._section, "DB") == 0 ||
  238.      strcmp(ConfigInfo::m_ParamInfo[p]._section, "API") == 0 ||
  239.      strcmp(ConfigInfo::m_ParamInfo[p]._section, "MGM") == 0)
  240.   {
  241.     if(strcasecmp(ConfigInfo::m_ParamInfo[p]._fname, str) == 0)
  242.     {
  243.       select.push_back(new Apply(ConfigInfo::m_ParamInfo[p]._paramId));
  244.       found = true;
  245.       break;
  246.     }
  247.   }
  248. }
  249. if(!found)
  250. {
  251.   fprintf(stderr, "Unknown query option: %sn", str);
  252.   return 1;
  253. }
  254.       }
  255.       else
  256.       {
  257. fprintf(stderr, "Unknown query option: %sn", str);
  258. return 1;
  259.       }
  260.     }
  261.   }
  262.   return 0;
  263. }
  265. static 
  266. int 
  267. parse_where(Vector<Match*>& where, int &argc, char**& argv)
  268. {
  269.   Match m;
  270.   if(g_host)
  271.   {
  272.     m.m_key = CFG_NODE_HOST;
  273.     m.m_value.assfmt("%s", g_host);
  274.     where.push_back(new Match(m));
  275.   }
  276.   
  277.   if(g_type)
  278.   {
  279.     m.m_key = CFG_TYPE_OF_SECTION;
  280.     m.m_value.assfmt("%d", ndb_mgm_match_node_type(g_type));
  281.     where.push_back(new Match(m));
  282.   }
  284.   if(g_nodeid)
  285.   {
  286.     m.m_key = CFG_NODE_ID;
  287.     m.m_value.assfmt("%d", g_nodeid);
  288.     where.push_back(new Match(m));
  289.   }
  290.   return 0;
  291. }
  293. template class Vector<Apply*>;
  294. template class Vector<Match*>;
  296. static 
  297. int
  298. eval(const Iter& iter, const Vector<Match*>& where)
  299. {
  300.   for(unsigned i = 0; i<where.size(); i++)
  301.   {
  302.     if(where[i]->eval(iter) == 0)
  303.       return 0;
  304.   }
  305.   
  306.   return 1;
  307. }
  309. static 
  310. int 
  311. apply(const Iter& iter, const Vector<Apply*>& list)
  312. {
  313.   for(unsigned i = 0; i<list.size(); i++)
  314.   {
  315.     list[i]->apply(iter);
  316.     if(i + 1 != list.size())
  317.       printf("%s", g_field_delimiter);
  318.   }
  319.   return 0;
  320. }
  322. int
  323. Match::eval(const Iter& iter)
  324. {
  325.   Uint32 val32;
  326.   Uint64 val64;
  327.   const char* valc;
  328.   if (iter.get(m_key, &val32) == 0)
  329.   {
  330.     if(atoi(m_value.c_str()) != (int)val32)
  331.       return 0;
  332.   } 
  333.   else if(iter.get(m_key, &val64) == 0)
  334.   {
  335.     if(strtoll(m_value.c_str(), (char **)NULL, 10) != (long long)val64)
  336.       return 0;
  337.   }
  338.   else if(iter.get(m_key, &valc) == 0)
  339.   {
  340.     if(strcmp(m_value.c_str(), valc) != 0)
  341.       return 0;
  342.   }
  343.   else
  344.   {
  345.     return 0;
  346.   }
  347.   return 1;
  348. }
  350. int
  351. Apply::apply(const Iter& iter)
  352. {
  353.   Uint32 val32;
  354.   Uint64 val64;
  355.   const char* valc;
  356.   if (iter.get(m_key, &val32) == 0)
  357.   {
  358.     printf("%u", val32);
  359.   } 
  360.   else if(iter.get(m_key, &val64) == 0)
  361.   {
  362.     printf("%llu", val64);
  363.   }
  364.   else if(iter.get(m_key, &valc) == 0)
  365.   {
  366.     printf("%s", valc);
  367.   }
  368.   return 0;
  369. }
  371. int
  372. NodeTypeApply::apply(const Iter& iter)
  373. {
  374.   Uint32 val32;
  375.   if (iter.get(CFG_TYPE_OF_SECTION, &val32) == 0)
  376.   {
  377.     printf("%s", ndb_mgm_get_node_type_alias_string((ndb_mgm_node_type)val32, 0));
  378.   } 
  379.   return 0;
  380. }
  382. ndb_mgm_configuration*
  383. fetch_configuration()
  384. {  
  385.   ndb_mgm_configuration* conf = 0;
  386.   NdbMgmHandle mgm = ndb_mgm_create_handle();
  387.   if(mgm == NULL) {
  388.     fprintf(stderr, "Cannot create handle to management server.n");
  389.     return 0;
  390.   }
  392.   ndb_mgm_set_error_stream(mgm, stderr);
  393.   
  394.   if (ndb_mgm_set_connectstring(mgm, g_connectstring))
  395.   {
  396.     fprintf(stderr, "* %5d: %sn", 
  397.     ndb_mgm_get_latest_error(mgm),
  398.     ndb_mgm_get_latest_error_msg(mgm));
  399.     fprintf(stderr, 
  400.     "*        %s", ndb_mgm_get_latest_error_desc(mgm));
  401.     goto noconnect;
  402.   }
  404.   if(ndb_mgm_connect(mgm, try_reconnect-1, 5, 1))
  405.   {
  406.     fprintf(stderr, "Connect failed");
  407.     fprintf(stderr, " code: %d, msg: %sn",
  408.     ndb_mgm_get_latest_error(mgm),
  409.     ndb_mgm_get_latest_error_msg(mgm));
  410.     goto noconnect;
  411.   }
  412.   else if(g_verbose)
  413.   {
  414.     fprintf(stderr, "Connected to %s:%dn", 
  415.     ndb_mgm_get_connected_host(mgm),
  416.     ndb_mgm_get_connected_port(mgm));
  417.   }
  418.   
  419.   conf = ndb_mgm_get_configuration(mgm, 0);
  420.   if(conf == 0)
  421.   {
  422.     fprintf(stderr, "Could not get configuration");
  423.     fprintf(stderr, "code: %d, msg: %sn",
  424.     ndb_mgm_get_latest_error(mgm),
  425.     ndb_mgm_get_latest_error_msg(mgm));
  426.   }
  427.   else if(g_verbose)
  428.   {
  429.     fprintf(stderr, "Fetched configurationn");
  430.   }
  432.   ndb_mgm_disconnect(mgm);
  433. noconnect:
  434.   ndb_mgm_destroy_handle(&mgm);
  435.   
  436.   return conf;
  437. }
  439. #include <Config.hpp>
  441. ndb_mgm_configuration*
  442. load_configuration()
  443. {  
  444.   InitConfigFileParser parser(stderr);
  445.   if (g_verbose)
  446.     fprintf(stderr, "Using config.ini : %s", g_config_file);
  447.   
  448.   Config* conf = parser.parseConfig(g_config_file);
  449.   if (conf)
  450.     return conf->m_configValues;
  452.   return 0;
  453. }