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

MySQL数据库

开发平台:

Visual C++

  1. /* Copyright (C) 2000 MySQL AB & MySQL Finland AB & TCX DataKonsult 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.   Atomic rename of table;  RENAME TABLE t1 to t2, tmp to t1 [,...]
  15. */
  16. #include "mysql_priv.h"
  17. static TABLE_LIST *rename_tables(THD *thd, TABLE_LIST *table_list,
  18.  bool skip_error);
  19. static TABLE_LIST *reverse_table_list(TABLE_LIST *table_list);
  20. /*
  21.   Every second entry in the table_list is the original name and every
  22.   second entry is the new name.
  23. */
  24. bool mysql_rename_tables(THD *thd, TABLE_LIST *table_list)
  25. {
  26.   bool error= 1;
  27.   TABLE_LIST *ren_table= 0;
  28.   DBUG_ENTER("mysql_rename_tables");
  29.   /*
  30.     Avoid problems with a rename on a table that we have locked or
  31.     if the user is trying to to do this in a transcation context
  32.   */
  33.   if (thd->locked_tables || thd->active_transaction())
  34.   {
  35.     my_error(ER_LOCK_OR_ACTIVE_TRANSACTION,MYF(0));
  36.     DBUG_RETURN(1);
  37.   }
  38.   if (wait_if_global_read_lock(thd,0,1))
  39.     DBUG_RETURN(1);
  40.   VOID(pthread_mutex_lock(&LOCK_open));
  41.   if (lock_table_names(thd, table_list))
  42.     goto err;
  43.   
  44.   error=0;
  45.   if ((ren_table=rename_tables(thd,table_list,0)))
  46.   {
  47.     /* Rename didn't succeed;  rename back the tables in reverse order */
  48.     TABLE_LIST *table;
  49.     /* Reverse the table list */
  50.     table_list= reverse_table_list(table_list);
  51.     /* Find the last renamed table */
  52.     for (table=table_list ;
  53.  table->next != ren_table ;
  54.  table=table->next->next) ;
  55.     table=table->next->next; // Skip error table
  56.     /* Revert to old names */
  57.     rename_tables(thd, table, 1);
  58.     /* Revert the table list (for prepared statements) */
  59.     table_list= reverse_table_list(table_list);
  60.     error= 1;
  61.   }
  62.   /* Lets hope this doesn't fail as the result will be messy */ 
  63.   if (!error)
  64.   {
  65.     mysql_update_log.write(thd,thd->query,thd->query_length);
  66.     if (mysql_bin_log.is_open())
  67.     {
  68.       thd->clear_error();
  69.       Query_log_event qinfo(thd, thd->query, thd->query_length, 0, FALSE);
  70.       mysql_bin_log.write(&qinfo);
  71.     }
  72.     send_ok(thd);
  73.   }
  74.   unlock_table_names(thd,table_list);
  75. err:
  76.   pthread_mutex_unlock(&LOCK_open);
  77.   start_waiting_global_read_lock(thd);
  78.   DBUG_RETURN(error);
  79. }
  80. /*
  81.   reverse table list
  82.   SYNOPSIS
  83.     reverse_table_list()
  84.     table_list pointer to table _list
  85.   RETURN
  86.     pointer to new (reversed) list
  87. */
  88. static TABLE_LIST *reverse_table_list(TABLE_LIST *table_list)
  89. {
  90.   TABLE_LIST *prev= 0;
  91.   while (table_list)
  92.   {
  93.     TABLE_LIST *next= table_list->next;
  94.     table_list->next= prev;
  95.     prev= table_list;
  96.     table_list= next;
  97.   }
  98.   return (prev);
  99. }
  100. /*
  101.   Rename all tables in list; Return pointer to wrong entry if something goes
  102.   wrong.  Note that the table_list may be empty!
  103. */
  104. static TABLE_LIST *
  105. rename_tables(THD *thd, TABLE_LIST *table_list, bool skip_error)
  106. {
  107.   TABLE_LIST *ren_table,*new_table;
  108.   DBUG_ENTER("rename_tables");
  109.   for (ren_table=table_list ; ren_table ; ren_table=new_table->next)
  110.   {
  111.     db_type table_type;
  112.     char name[FN_REFLEN];
  113.     const char *new_alias, *old_alias;
  114.     new_table=ren_table->next;
  115.     if (lower_case_table_names == 2)
  116.     {
  117.       old_alias= ren_table->alias;
  118.       new_alias= new_table->alias;
  119.     }
  120.     else
  121.     {
  122.       old_alias= ren_table->real_name;
  123.       new_alias= new_table->real_name;
  124.     }
  125.     sprintf(name,"%s/%s/%s%s",mysql_data_home,
  126.     new_table->db, new_alias, reg_ext);
  127.     unpack_filename(name, name);
  128.     if (!access(name,F_OK))
  129.     {
  130.       my_error(ER_TABLE_EXISTS_ERROR,MYF(0),new_alias);
  131.       DBUG_RETURN(ren_table); // This can't be skipped
  132.     }
  133.     sprintf(name,"%s/%s/%s%s",mysql_data_home,
  134.     ren_table->db, old_alias,
  135.     reg_ext);
  136.     unpack_filename(name, name);
  137.     if ((table_type=get_table_type(name)) == DB_TYPE_UNKNOWN)
  138.     {
  139.       my_error(ER_FILE_NOT_FOUND, MYF(0), name, my_errno);
  140.       if (!skip_error)
  141. DBUG_RETURN(ren_table);
  142.     }
  143.     else if (mysql_rename_table(table_type,
  144. ren_table->db, old_alias,
  145. new_table->db, new_alias))
  146.     {
  147.       if (!skip_error)
  148. DBUG_RETURN(ren_table);
  149.     }
  150.   }
  151.   DBUG_RETURN(0);
  152. }