mysql_tableinfo
上传用户:romrleung
上传日期:2022-05-23
资源大小:18897k
文件大小:13k
源码类别:

MySQL数据库

开发平台:

Visual C++

  1. #!/usr/bin/perl -w
  2. use strict;
  3. use Getopt::Long;
  4. use DBI;
  5. =head1 NAME
  6. mysql_tableinfo - creates and populates information tables with 
  7. the output of SHOW DATABASES, SHOW TABLES (or SHOW TABLE STATUS), 
  8. SHOW COLUMNS and SHOW INDEX.
  9. This is version 1.1.
  10. =head1 SYNOPSIS
  11.  
  12.   mysql_tableinfo [OPTIONS] database_to_write [database_like_wild] [table_like_wild]
  13.   Do not backquote (``) database_to_write, 
  14.   and do not quote ('') database_like_wild or table_like_wild
  15.   Examples:
  16.   mysql_tableinfo info
  17.   mysql_tableinfo info this_db
  18.   mysql_tableinfo info %a% b%
  19.   mysql_tableinfo info --clear-only
  20.   mysql_tableinfo info --col --idx --table-status
  21. =cut
  22. # Documentation continued at end of file
  23. sub usage {
  24.     die @_,"nExecute 'perldoc $0' for documentationn";
  25. }
  26. my %opt = (
  27.     'user' => scalar getpwuid($>),
  28.     'host'      => "localhost",
  29.     'prefix'    => "", #to avoid 'use of uninitialized value...'
  30. );
  31. Getopt::Long::Configure(qw(no_ignore_case)); # disambuguate -p and -P
  32. GetOptions( %opt,
  33.     "help",
  34.     "user|u=s",
  35.     "password|p=s",
  36.     "host|h=s",
  37.     "port|P=s",
  38.     "socket|S=s",
  39.     "tbl-status",
  40.     "col",
  41.     "idx",
  42.     "clear",
  43.     "clear-only",
  44.     "prefix=s",
  45.     "quiet|q",
  46. ) or usage("Invalid option");
  47. if ($opt{'help'}) {usage();}
  48. my ($db_to_write,$db_like_wild,$tbl_like_wild);
  49. if (@ARGV==0)
  50. {
  51.     usage("Not enough arguments");
  52. }
  53. $db_to_write="`$ARGV[0]`"; shift @ARGV;
  54. $db_like_wild=($ARGV[0])?$ARGV[0]:"%"; shift @ARGV;
  55. $tbl_like_wild=($ARGV[0])?$ARGV[0]:"%"; shift @ARGV;
  56. if (@ARGV>0) { usage("Too many arguments"); }
  57. $0 = $1 if $0 =~ m:/([^/]+)$:;
  58. my $info_db="`".$opt{'prefix'}."db`";
  59. my $info_tbl="`".$opt{'prefix'}."tbl".
  60.     (($opt{'tbl-status'})?"_status":"")."`";
  61. my $info_col="`".$opt{'prefix'}."col`";
  62. my $info_idx="`".$opt{'prefix'}."idx`";
  63. # --- connect to the database ---
  64. my $dsn = ";host=$opt{'host'}";
  65. $dsn .= ";port=$opt{'port'}" if $opt{'port'};
  66. $dsn .= ";mysql_socket=$opt{'socket'}" if $opt{'socket'};
  67. my $dbh = DBI->connect("dbi:mysql:$dsn;mysql_read_default_group=perl",
  68.                         $opt{'user'}, $opt{'password'},
  69. {
  70.     RaiseError => 1,
  71.     PrintError => 0,
  72.     AutoCommit => 1,
  73. });
  74. $db_like_wild=$dbh->quote($db_like_wild);
  75. $tbl_like_wild=$dbh->quote($tbl_like_wild);
  76. #Ask
  77. if (!$opt{'quiet'})
  78. {
  79.     print "n!! This program is doing to do:nn";
  80.     print "**DROP** TABLE ...n" if ($opt{'clear'} or $opt{'clear-only'});
  81.     print "**DELETE** FROM ... WHERE `Database` LIKE $db_like_wild AND `Table` LIKE $tbl_like_wild
  82. **INSERT** INTO ...
  83. on the following tables :n";
  84.     foreach  (($info_db, $info_tbl),
  85.       (($opt{'col'})?$info_col:()), 
  86.       (($opt{'idx'})?$info_idx:()))
  87.     {
  88. print("  $db_to_write.$_n");
  89.     }
  90.     print "nContinue (you can skip this confirmation step with --quiet) ? (y|n) [n]";
  91.     if (<STDIN> !~ /^s*ys*$/i) 
  92.     {
  93. print "Nothing done!n";exit;
  94.     }
  95. }
  96. if ($opt{'clear'} or $opt{'clear-only'}) 
  97. {
  98. #do not drop the $db_to_write database !
  99.     foreach  (($info_db, $info_tbl),
  100.       (($opt{'col'})?$info_col:()), 
  101.       (($opt{'idx'})?$info_idx:()))
  102.     {
  103. $dbh->do("DROP TABLE IF EXISTS $db_to_write.$_");
  104.     }
  105.     if ($opt{'clear-only'}) 
  106.     {
  107. print "Wrote to database $db_to_write .n" unless ($opt{'quiet'});
  108. exit; 
  109.     }
  110. }
  111. my %sth;
  112. my %extra_col_desc;
  113. my %row;
  114. my %done_create_table;
  115. #create the $db_to_write database
  116. $dbh->do("CREATE DATABASE IF NOT EXISTS $db_to_write");
  117. $dbh->do("USE $db_to_write");
  118. #get databases
  119. $sth{'db'}=$dbh->prepare("SHOW DATABASES LIKE $db_like_wild");
  120. $sth{'db'}->execute;
  121. #create $info_db which will receive info about databases.
  122. #Ensure that the first column to be called "Database" (as SHOW DATABASES LIKE
  123. #returns a varying
  124. #column name (of the form "Database (%...)") which is not suitable)
  125. $extra_col_desc{'db'}=do_create_table("db",$info_db,undef,"`Database`");
  126. #we'll remember the type of the `Database` column (as returned by
  127. #SHOW DATABASES), which we will need when creating the next tables. 
  128. #clear out-of-date info from this table 
  129. $dbh->do("DELETE FROM $info_db WHERE `Database` LIKE $db_like_wild"); 
  130. while ($row{'db'}=$sth{'db'}->fetchrow_arrayref) #go through all databases
  131. {
  132. #insert the database name
  133.     $dbh->do("INSERT INTO $info_db VALUES("
  134.      .join(',' ,  ( map $dbh->quote($_), @{$row{'db'}} ) ).")" );
  135. #for each database, get tables
  136.     $sth{'tbl'}=$dbh->prepare("SHOW TABLE"
  137.     .( ($opt{'tbl-status'}) ? 
  138.        " STATUS"
  139.        : "S" )
  140.     ." from `$row{'db'}->[0]` LIKE $tbl_like_wild");
  141.     $sth{'tbl'}->execute;
  142.     unless ($done_create_table{$info_tbl})
  143. #tables must be created only once, and out-of-date info must be
  144. #cleared once
  145.     {
  146. $done_create_table{$info_tbl}=1;
  147. $extra_col_desc{'tbl'}=
  148.     do_create_table("tbl",$info_tbl,
  149. #add an extra column (database name) at the left
  150. #and ensure that the table name will be called "Table"
  151. #(this is unncessesary with
  152. #SHOW TABLE STATUS, but necessary with SHOW TABLES (which returns a column
  153. #named "Tables_in_..."))
  154.     "`Database` ".$extra_col_desc{'db'},"`Table`"); 
  155. $dbh->do("DELETE FROM $info_tbl WHERE `Database` LIKE $db_like_wild                          AND `Table` LIKE $tbl_like_wild");
  156.     }
  157.     while ($row{'tbl'}=$sth{'tbl'}->fetchrow_arrayref)
  158.     {
  159. $dbh->do("INSERT INTO $info_tbl VALUES("
  160.  .$dbh->quote($row{'db'}->[0]).","
  161.  .join(',' ,  ( map $dbh->quote($_), @{$row{'tbl'}} ) ).")");
  162. #for each table, get columns...
  163. if ($opt{'col'})
  164. {
  165.     $sth{'col'}=$dbh->prepare("SHOW COLUMNS FROM `$row{'tbl'}->[0]` FROM `$row{'db'}->[0]`"); 
  166.     $sth{'col'}->execute;
  167.     unless ($done_create_table{$info_col})
  168.     {
  169. $done_create_table{$info_col}=1;
  170. do_create_table("col",$info_col,
  171. "`Database` ".$extra_col_desc{'db'}.","
  172. ."`Table` ".$extra_col_desc{'tbl'}.","
  173. ."`Seq_in_table` BIGINT(3)"); 
  174. #We need to add a sequence number (1 for the first column of the table,
  175. #2 for the second etc) so that users are able to retrieve columns in order
  176. #if they want. This is not needed for INDEX 
  177. #(where there is already Seq_in_index)
  178. $dbh->do("DELETE FROM $info_col WHERE `Database` 
  179.                             LIKE $db_like_wild
  180.     AND `Table` LIKE $tbl_like_wild");
  181.     }
  182.     my $col_number=0;
  183.     while ($row{'col'}=$sth{'col'}->fetchrow_arrayref)
  184.     {
  185. $dbh->do("INSERT INTO $info_col VALUES("
  186.  .$dbh->quote($row{'db'}->[0]).","
  187.  .$dbh->quote($row{'tbl'}->[0]).","
  188.  .++$col_number.","
  189.  .join(',' ,  ( map $dbh->quote($_), @{$row{'col'}} ) ).")");
  190.     }
  191. }
  192. #and get index.
  193. if ($opt{'idx'})
  194. {
  195.     $sth{'idx'}=$dbh->prepare("SHOW INDEX FROM `$row{'tbl'}->[0]` FROM `$row{'db'}->[0]`"); 
  196.     $sth{'idx'}->execute;
  197.     unless ($done_create_table{$info_idx})
  198.     {
  199. $done_create_table{$info_idx}=1;
  200. do_create_table("idx",$info_idx,
  201. "`Database` ".$extra_col_desc{'db'});
  202. $dbh->do("DELETE FROM $info_idx WHERE `Database`
  203.  LIKE $db_like_wild
  204.  AND `Table` LIKE $tbl_like_wild");
  205.     }
  206.     while ($row{'idx'}=$sth{'idx'}->fetchrow_arrayref)
  207.     {
  208. $dbh->do("INSERT INTO $info_idx VALUES("
  209.  .$dbh->quote($row{'db'}->[0]).","
  210.  .join(',' ,  ( map $dbh->quote($_), @{$row{'idx'}} ) ).")");
  211.     }
  212. }
  213.     }
  214. }
  215. print "Wrote to database $db_to_write .n" unless ($opt{'quiet'});
  216. exit;
  217. sub do_create_table
  218. {
  219.     my ($sth_key,$target_tbl,$extra_col_desc,$first_col_name)=@_; 
  220.     my $create_table_query=$extra_col_desc;
  221.     my ($i,$first_col_desc,$col_desc);
  222.     for ($i=0;$i<$sth{$sth_key}->{NUM_OF_FIELDS};$i++)
  223.     {
  224. if ($create_table_query) { $create_table_query.=", "; }
  225. $col_desc=$sth{$sth_key}->{mysql_type_name}->[$i];
  226. if ($col_desc =~ /char|int/i)
  227. {
  228.     $col_desc.="($sth{$sth_key}->{PRECISION}->[$i])";
  229. }
  230. elsif ($col_desc =~ /decimal|numeric/i) #(never seen that)
  231. {
  232.     $col_desc.=
  233. "($sth{$sth_key}->{PRECISION}->[$i],$sth{$sth_key}->{SCALE}->[$i])";
  234. }
  235. elsif ($col_desc !~ /date/i) #date and datetime are OK,
  236.                          #no precision or scale for them
  237. {
  238.     warn "unexpected column type '$col_desc' 
  239. (neither 'char','int','decimal|numeric')
  240. when creating $target_tbl, hope table creation will go OKn";
  241. }
  242. if ($i==0) {$first_col_desc=$col_desc};
  243. $create_table_query.=
  244.     ( ($i==0 and $first_col_name) ? 
  245.       "$first_col_name " :"`$sth{$sth_key}->{NAME}->[$i]` " )
  246.     .$col_desc;
  247.     }
  248. if ($create_table_query)
  249. {
  250.     $dbh->do("CREATE TABLE IF NOT EXISTS $target_tbl ($create_table_query)");
  251. }
  252. return $first_col_desc;
  253. }
  254. __END__
  255. =head1 DESCRIPTION
  256. mysql_tableinfo asks a MySQL server information about its
  257. databases, tables, table columns and index, and stores this
  258. in tables called `db`, `tbl` (or `tbl_status`), `col`, `idx` 
  259. (with an optional prefix specified with --prefix).
  260. After that, you can query these information tables, for example
  261. to build your admin scripts with SQL queries, like
  262. SELECT CONCAT("CHECK TABLE ",`database`,".",`table`," EXTENDED;") 
  263. FROM info.tbl WHERE ... ;
  264. as people usually do with some other RDBMS
  265. (note: to increase the speed of your queries on the info tables,
  266. you may add some index on them).
  267. The database_like_wild and table_like_wild instructs the program
  268. to gather information only about databases and tables
  269. whose names match these patterns. If the info
  270. tables already exist, their rows matching the patterns are simply
  271. deleted and replaced by the new ones. That is,
  272. old rows not matching the patterns are not touched.
  273. If the database_like_wild and table_like_wild arguments
  274. are not specified on the command-line they default to "%".
  275. The program :
  276. - does CREATE DATABASE IF NOT EXISTS database_to_write
  277. where database_to_write is the database name specified on the command-line.
  278. - does CREATE TABLE IF NOT EXISTS database_to_write.`db`
  279. - fills database_to_write.`db` with the output of
  280. SHOW DATABASES LIKE database_like_wild
  281. - does CREATE TABLE IF NOT EXISTS database_to_write.`tbl`
  282. (respectively database_to_write.`tbl_status`
  283. if the --tbl-status option is on)
  284. - for every found database,
  285. fills database_to_write.`tbl` (respectively database_to_write.`tbl_status`)
  286. with the output of 
  287. SHOW TABLES FROM found_db LIKE table_like_wild
  288. (respectively SHOW TABLE STATUS FROM found_db LIKE table_like_wild)
  289. - if the --col option is on,
  290.     * does CREATE TABLE IF NOT EXISTS database_to_write.`col`
  291.     * for every found table,
  292.       fills database_to_write.`col` with the output of 
  293.       SHOW COLUMNS FROM found_tbl FROM found_db
  294. - if the --idx option is on,
  295.     * does CREATE TABLE IF NOT EXISTS database_to_write.`idx`
  296.     * for every found table,
  297.       fills database_to_write.`idx` with the output of 
  298.       SHOW INDEX FROM found_tbl FROM found_db
  299. Some options may modify this general scheme (see below).
  300. As mentioned, the contents of the info tables are the output of
  301. SHOW commands. In fact the contents are slightly more complete :
  302. - the `tbl` (or `tbl_status`) info table 
  303.   has an extra column which contains the database name,
  304. - the `col` info table
  305.   has an extra column which contains the table name,
  306.   and an extra column which contains, for each described column,
  307.   the number of this column in the table owning it (this extra column
  308.   is called `Seq_in_table`). `Seq_in_table` makes it possible for you
  309.   to retrieve your columns in sorted order, when you are querying
  310.   the `col` table. 
  311. - the `index` info table
  312.   has an extra column which contains the database name.
  313. Caution: info tables contain certain columns (e.g.
  314. Database, Table, Null...) whose names, as they are MySQL reserved words,
  315. need to be backquoted (`...`) when used in SQL statements.
  316. Caution: as information fetching and info tables filling happen at the
  317. same time, info tables may contain inaccurate information about
  318. themselves.
  319. =head1 OPTIONS
  320. =over 4
  321. =item --clear
  322. Does DROP TABLE on the info tables (only those that the program is
  323. going to fill, for example if you do not use --col it won't drop
  324. the `col` table) and processes normally. Does not drop database_to_write.
  325. =item --clear-only
  326. Same as --clear but exits after the DROPs.
  327. =item --col
  328. Adds columns information (into table `col`).
  329. =item --idx
  330. Adds index information (into table `idx`).
  331. =item --prefix prefix
  332. The info tables are named from the concatenation of prefix and,
  333. respectively, db, tbl (or tbl_status), col, idx. Do not quote ('')
  334. or backquote (``) prefix.
  335. =item -q, --quiet
  336. Does not warn you about what the script is going to do (DROP TABLE etc)
  337. and does not ask for a confirmation before starting.
  338. =item --tbl-status
  339. Instead of using SHOW TABLES, uses SHOW TABLE STATUS
  340. (much more complete information, but slower). 
  341. =item --help
  342. Display helpscreen and exit
  343. =item -u, --user=#         
  344. user for database login if not current user. Give a user
  345. who has sufficient privileges (CREATE, ...).
  346. =item -p, --password=#     
  347. password to use when connecting to server
  348. =item -h, --host=#     
  349. host to connect to
  350. =item -P, --port=#         
  351. port to use when connecting to server
  352. =item -S, --socket=#         
  353. UNIX domain socket to use when connecting to server
  354. =head1 WARRANTY
  355. This software is free and comes without warranty of any kind. You
  356. should never trust backup software without studying the code yourself.
  357. Study the code inside this script and only rely on it if I<you> believe
  358. that it does the right thing for you.
  359. Patches adding bug fixes, documentation and new features are welcome.
  360. =head1 TO DO
  361. Use extended inserts to be faster (for servers with many databases
  362. or tables). But to do that, must care about net-buffer-length.
  363. =head1 AUTHOR
  364. 2002-06-18 Guilhem Bichot (guilhem.bichot@mines-paris.org)
  365. And all the authors of mysqlhotcopy, which served as a model for 
  366. the structure of the program.