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

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. /* locking functions for mysql */
  14. /*
  15.   Because of the new concurrent inserts, we must first get external locks
  16.   before getting internal locks.  If we do it in the other order, the status
  17.   information is not up to date when called from the lock handler.
  18.   GENERAL DESCRIPTION OF LOCKING
  19.   When not using LOCK TABLES:
  20.   - For each SQL statement mysql_lock_tables() is called for all involved
  21.     tables.
  22.     - mysql_lock_tables() will call
  23.       table_handler->external_lock(thd,locktype) for each table.
  24.       This is followed by a call to thr_multi_lock() for all tables.
  25.   - When statement is done, we call mysql_unlock_tables().
  26.     This will call thr_multi_unlock() followed by
  27.     table_handler->external_lock(thd, F_UNLCK) for each table.
  28.   - Note that mysql_unlock_tables() may be called several times as
  29.     MySQL in some cases can free some tables earlier than others.
  30.   - The above is true both for normal and temporary tables.
  31.   - Temporary non transactional tables are never passed to thr_multi_lock()
  32.     and we never call external_lock(thd, F_UNLOCK) on these.
  33.   When using LOCK TABLES:
  34.   - LOCK TABLE will call mysql_lock_tables() for all tables.
  35.     mysql_lock_tables() will call
  36.     table_handler->external_lock(thd,locktype) for each table.
  37.     This is followed by a call to thr_multi_lock() for all tables.
  38.   - For each statement, we will call table_handler->start_stmt(THD)
  39.     to inform the table handler that we are using the table.
  40.     The tables used can only be tables used in LOCK TABLES or a
  41.     temporary table.
  42.   - When statement is done, we will call ha_commit_stmt(thd);
  43.   - When calling UNLOCK TABLES we call mysql_unlock_tables() for all
  44.     tables used in LOCK TABLES
  45. TODO:
  46.   Change to use my_malloc() ONLY when using LOCK TABLES command or when
  47.   we are forced to use mysql_lock_merge.
  48. */
  49. #include "mysql_priv.h"
  50. #include <hash.h>
  51. #include "ha_myisammrg.h"
  52. #ifndef MASTER
  53. #include "../srclib/myisammrg/myrg_def.h"
  54. #else
  55. #include "../myisammrg/myrg_def.h"
  56. #endif
  57. static MYSQL_LOCK *get_lock_data(THD *thd, TABLE **table,uint count,
  58.  bool unlock, TABLE **write_locked);
  59. static int lock_external(THD *thd, TABLE **table,uint count);
  60. static int unlock_external(THD *thd, TABLE **table,uint count);
  61. static void print_lock_error(int error);
  62. /*
  63.   Lock tables.
  64.   SYNOPSIS
  65.     mysql_lock_tables()
  66.     thd                         The current thread.
  67.     tables                      An array of pointers to the tables to lock.
  68.     count                       The number of tables to lock.
  69.     flags                       Options:
  70.       MYSQL_LOCK_IGNORE_GLOBAL_READ_LOCK      Ignore a global read lock
  71.       MYSQL_LOCK_IGNORE_FLUSH                 Ignore a flush tables.
  72.   RETURN
  73.     A lock structure pointer on success.
  74.     NULL on error.
  75. */
  76. MYSQL_LOCK *mysql_lock_tables(THD *thd, TABLE **tables, uint count, uint flags)
  77. {
  78.   MYSQL_LOCK *sql_lock;
  79.   TABLE *write_lock_used;
  80.   DBUG_ENTER("mysql_lock_tables");
  81.   for (;;)
  82.   {
  83.     if (!(sql_lock = get_lock_data(thd,tables,count, 0,&write_lock_used)))
  84.       break;
  85.     if (global_read_lock && write_lock_used &&
  86.         ! (flags & MYSQL_LOCK_IGNORE_GLOBAL_READ_LOCK))
  87.     {
  88.       /*
  89. Someone has issued LOCK ALL TABLES FOR READ and we want a write lock
  90. Wait until the lock is gone
  91.       */
  92.       if (wait_if_global_read_lock(thd, 1, 1))
  93.       {
  94. my_free((gptr) sql_lock,MYF(0));
  95. sql_lock=0;
  96. break;
  97.       }
  98.       if (thd->version != refresh_version)
  99.       {
  100. my_free((gptr) sql_lock,MYF(0));
  101. goto retry;
  102.       }
  103.     }
  104.     thd->proc_info="System lock";
  105.     if (lock_external(thd, tables, count))
  106.     {
  107.       my_free((gptr) sql_lock,MYF(0));
  108.       sql_lock=0;
  109.       thd->proc_info=0;
  110.       break;
  111.     }
  112.     thd->proc_info="Table lock";
  113.     thd->locked=1;
  114.     if (thr_multi_lock(sql_lock->locks,sql_lock->lock_count))
  115.     {
  116.       thd->some_tables_deleted=1; // Try again
  117.       sql_lock->lock_count=0; // Locks are alread freed
  118.     }
  119.     else if (!thd->some_tables_deleted || (flags & MYSQL_LOCK_IGNORE_FLUSH))
  120.     {
  121.       thd->locked=0;
  122.       break;
  123.     }
  124.     else if (!thd->open_tables)
  125.     {
  126.       // Only using temporary tables, no need to unlock
  127.       thd->some_tables_deleted=0;
  128.       thd->locked=0;
  129.       break;
  130.     }
  131.     thd->proc_info=0;
  132.     /* some table was altered or deleted. reopen tables marked deleted */
  133.     mysql_unlock_tables(thd,sql_lock);
  134.     thd->locked=0;
  135. retry:
  136.     sql_lock=0;
  137.     if (wait_for_tables(thd))
  138.       break; // Couldn't open tables
  139.   }
  140.   thd->proc_info=0;
  141.   if (thd->killed)
  142.   {
  143.     my_error(ER_SERVER_SHUTDOWN,MYF(0));
  144.     if (sql_lock)
  145.     {
  146.       mysql_unlock_tables(thd,sql_lock);
  147.       sql_lock=0;
  148.     }
  149.   }
  150.   thd->lock_time();
  151.   DBUG_RETURN (sql_lock);
  152. }
  153. static int lock_external(THD *thd, TABLE **tables, uint count)
  154. {
  155.   reg1 uint i;
  156.   int lock_type,error;
  157.   DBUG_ENTER("lock_external");
  158.   for (i=1 ; i <= count ; i++, tables++)
  159.   {
  160.     DBUG_ASSERT((*tables)->reginfo.lock_type >= TL_READ);
  161.     lock_type=F_WRLCK; /* Lock exclusive */
  162.     if ((*tables)->db_stat & HA_READ_ONLY ||
  163. ((*tables)->reginfo.lock_type >= TL_READ &&
  164.  (*tables)->reginfo.lock_type <= TL_READ_NO_INSERT))
  165.       lock_type=F_RDLCK;
  166.     if ((error=(*tables)->file->external_lock(thd,lock_type)))
  167.     {
  168.       for (; i-- ; tables--)
  169.       {
  170. (*tables)->file->external_lock(thd, F_UNLCK);
  171. (*tables)->current_lock=F_UNLCK;
  172.       }
  173.       print_lock_error(error);
  174.       DBUG_RETURN(error);
  175.     }
  176.     else
  177.     {
  178.       (*tables)->db_stat &= ~ HA_BLOCK_LOCK;
  179.       (*tables)->current_lock= lock_type;
  180.     }
  181.   }
  182.   DBUG_RETURN(0);
  183. }
  184. void mysql_unlock_tables(THD *thd, MYSQL_LOCK *sql_lock)
  185. {
  186.   DBUG_ENTER("mysql_unlock_tables");
  187.   if (sql_lock->lock_count)
  188.     thr_multi_unlock(sql_lock->locks,sql_lock->lock_count);
  189.   if (sql_lock->table_count)
  190.     VOID(unlock_external(thd,sql_lock->table,sql_lock->table_count));
  191.   my_free((gptr) sql_lock,MYF(0));
  192.   DBUG_VOID_RETURN;
  193. }
  194. /*
  195.   Unlock some of the tables locked by mysql_lock_tables
  196.   This will work even if get_lock_data fails (next unlock will free all)
  197.   */
  198. void mysql_unlock_some_tables(THD *thd, TABLE **table,uint count)
  199. {
  200.   MYSQL_LOCK *sql_lock;
  201.   TABLE *write_lock_used;
  202.   if ((sql_lock = get_lock_data(thd, table, count, 1, &write_lock_used)))
  203.     mysql_unlock_tables(thd, sql_lock);
  204. }
  205. /*
  206. ** unlock all tables locked for read.
  207. */
  208. void mysql_unlock_read_tables(THD *thd, MYSQL_LOCK *sql_lock)
  209. {
  210.   uint i,found;
  211.   DBUG_ENTER("mysql_unlock_read_tables");
  212.   /* Move all write locks first */
  213.   THR_LOCK_DATA **lock=sql_lock->locks;
  214.   for (i=found=0 ; i < sql_lock->lock_count ; i++)
  215.   {
  216.     if (sql_lock->locks[i]->type >= TL_WRITE_ALLOW_READ)
  217.     {
  218.       swap_variables(THR_LOCK_DATA *, *lock, sql_lock->locks[i]);
  219.       lock++;
  220.       found++;
  221.     }
  222.   }
  223.   /* unlock the read locked tables */
  224.   if (i != found)
  225.   {
  226.     thr_multi_unlock(lock,i-found);
  227.     sql_lock->lock_count= found;
  228.   }
  229.   /* Then do the same for the external locks */
  230.   /* Move all write locked tables first */
  231.   TABLE **table=sql_lock->table;
  232.   for (i=found=0 ; i < sql_lock->table_count ; i++)
  233.   {
  234.     if ((uint) sql_lock->table[i]->reginfo.lock_type >= TL_WRITE_ALLOW_READ)
  235.     {
  236.       swap_variables(TABLE *, *table, sql_lock->table[i]);
  237.       table++;
  238.       found++;
  239.     }
  240.   }
  241.   /* Unlock all read locked tables */
  242.   if (i != found)
  243.   {
  244.     VOID(unlock_external(thd,table,i-found));
  245.     sql_lock->table_count=found;
  246.   }
  247.   DBUG_VOID_RETURN;
  248. }
  249. void mysql_lock_remove(THD *thd, MYSQL_LOCK *locked,TABLE *table)
  250. {
  251.   mysql_unlock_some_tables(thd, &table,1);
  252.   if (locked)
  253.   {
  254.     reg1 uint i;
  255.     for (i=0; i < locked->table_count; i++)
  256.     {
  257.       if (locked->table[i] == table)
  258.       {
  259. locked->table_count--;
  260. bmove((char*) (locked->table+i),
  261.       (char*) (locked->table+i+1),
  262.       (locked->table_count-i)* sizeof(TABLE*));
  263. break;
  264.       }
  265.     }
  266.     THR_LOCK_DATA **prev=locked->locks;
  267.     for (i=0 ; i < locked->lock_count ; i++)
  268.     {
  269.       if (locked->locks[i]->type != TL_UNLOCK)
  270. *prev++ = locked->locks[i];
  271.     }
  272.     locked->lock_count=(uint) (prev - locked->locks);
  273.   }
  274. }
  275. /* abort all other threads waiting to get lock in table */
  276. void mysql_lock_abort(THD *thd, TABLE *table)
  277. {
  278.   MYSQL_LOCK *locked;
  279.   TABLE *write_lock_used;
  280.   if ((locked = get_lock_data(thd,&table,1,1,&write_lock_used)))
  281.   {
  282.     for (uint i=0; i < locked->lock_count; i++)
  283.       thr_abort_locks(locked->locks[i]->lock);
  284.     my_free((gptr) locked,MYF(0));
  285.   }
  286. }
  287. /*
  288.   Abort one thread / table combination
  289.   SYNOPSIS
  290.     mysql_lock_abort_for_thread()
  291.     thd Thread handler
  292.     table Table that should be removed from lock queue
  293.   RETURN
  294.     0  Table was not locked by another thread
  295.     1  Table was locked by at least one other thread
  296. */
  297. bool mysql_lock_abort_for_thread(THD *thd, TABLE *table)
  298. {
  299.   MYSQL_LOCK *locked;
  300.   TABLE *write_lock_used;
  301.   bool result= FALSE;
  302.   DBUG_ENTER("mysql_lock_abort_for_thread");
  303.   if ((locked = get_lock_data(thd,&table,1,1,&write_lock_used)))
  304.   {
  305.     for (uint i=0; i < locked->lock_count; i++)
  306.     {
  307.       if (thr_abort_locks_for_thread(locked->locks[i]->lock,
  308.                                      table->in_use->real_id))
  309.         result= TRUE;
  310.     }
  311.     my_free((gptr) locked,MYF(0));
  312.   }
  313.   DBUG_RETURN(result);
  314. }
  315. MYSQL_LOCK *mysql_lock_merge(MYSQL_LOCK *a,MYSQL_LOCK *b)
  316. {
  317.   MYSQL_LOCK *sql_lock;
  318.   DBUG_ENTER("mysql_lock_merge");
  319.   if (!(sql_lock= (MYSQL_LOCK*)
  320. my_malloc(sizeof(*sql_lock)+
  321.   sizeof(THR_LOCK_DATA*)*(a->lock_count+b->lock_count)+
  322.   sizeof(TABLE*)*(a->table_count+b->table_count),MYF(MY_WME))))
  323.     DBUG_RETURN(0); // Fatal error
  324.   sql_lock->lock_count=a->lock_count+b->lock_count;
  325.   sql_lock->table_count=a->table_count+b->table_count;
  326.   sql_lock->locks=(THR_LOCK_DATA**) (sql_lock+1);
  327.   sql_lock->table=(TABLE**) (sql_lock->locks+sql_lock->lock_count);
  328.   memcpy(sql_lock->locks,a->locks,a->lock_count*sizeof(*a->locks));
  329.   memcpy(sql_lock->locks+a->lock_count,b->locks,
  330.  b->lock_count*sizeof(*b->locks));
  331.   memcpy(sql_lock->table,a->table,a->table_count*sizeof(*a->table));
  332.   memcpy(sql_lock->table+a->table_count,b->table,
  333.  b->table_count*sizeof(*b->table));
  334.   my_free((gptr) a,MYF(0));
  335.   my_free((gptr) b,MYF(0));
  336.   DBUG_RETURN(sql_lock);
  337. }
  338. /* unlock a set of external */
  339. static int unlock_external(THD *thd, TABLE **table,uint count)
  340. {
  341.   int error,error_code;
  342.   DBUG_ENTER("unlock_external");
  343.   error_code=0;
  344.   do
  345.   {
  346.     if ((*table)->current_lock != F_UNLCK)
  347.     {
  348.       (*table)->current_lock = F_UNLCK;
  349.       if ((error=(*table)->file->external_lock(thd, F_UNLCK)))
  350. error_code=error;
  351.     }
  352.     table++;
  353.   } while (--count);
  354.   if (error_code)
  355.     print_lock_error(error_code);
  356.   DBUG_RETURN(error_code);
  357. }
  358. /*
  359. ** Get lock structures from table structs and initialize locks
  360. */
  361. static MYSQL_LOCK *get_lock_data(THD *thd, TABLE **table_ptr, uint count,
  362.  bool get_old_locks, TABLE **write_lock_used)
  363. {
  364.   uint i,tables,lock_count;
  365.   MYSQL_LOCK *sql_lock;
  366.   THR_LOCK_DATA **locks;
  367.   TABLE **to;
  368.   *write_lock_used=0;
  369.   for (i=tables=lock_count=0 ; i < count ; i++)
  370.   {
  371.     if (table_ptr[i]->tmp_table != TMP_TABLE)
  372.     {
  373.       tables+=table_ptr[i]->file->lock_count();
  374.       lock_count++;
  375.     }
  376.   }
  377.   if (!(sql_lock= (MYSQL_LOCK*)
  378. my_malloc(sizeof(*sql_lock)+
  379.   sizeof(THR_LOCK_DATA*)*tables+sizeof(table_ptr)*lock_count,
  380.   MYF(0))))
  381.     return 0;
  382.   locks=sql_lock->locks=(THR_LOCK_DATA**) (sql_lock+1);
  383.   to=sql_lock->table=(TABLE**) (locks+tables);
  384.   sql_lock->table_count=lock_count;
  385.   sql_lock->lock_count=tables;
  386.   for (i=0 ; i < count ; i++)
  387.   {
  388.     TABLE *table;
  389.     if ((table=table_ptr[i])->tmp_table == TMP_TABLE)
  390.       continue;
  391.     *to++=table;
  392.     enum thr_lock_type lock_type= table->reginfo.lock_type;
  393.     if (lock_type >= TL_WRITE_ALLOW_WRITE)
  394.     {
  395.       *write_lock_used=table;
  396.       if (table->db_stat & HA_READ_ONLY)
  397.       {
  398. my_error(ER_OPEN_AS_READONLY,MYF(0),table->table_name);
  399. my_free((gptr) sql_lock,MYF(0));
  400. return 0;
  401.       }
  402.     }
  403.     THR_LOCK_DATA **org_locks = locks;
  404.     locks=table->file->store_lock(thd, locks, get_old_locks ? TL_IGNORE :
  405.   lock_type);
  406.     if (locks)
  407.       for ( ; org_locks != locks ; org_locks++)
  408. (*org_locks)->debug_print_param= (void *) table;
  409.   }
  410.   return sql_lock;
  411. }
  412. /*****************************************************************************
  413.   Lock table based on the name.
  414.   This is used when we need total access to a closed, not open table
  415. *****************************************************************************/
  416. /*
  417.   Lock and wait for the named lock.
  418.   SYNOPSIS
  419.     lock_and_wait_for_table_name()
  420.     thd Thread handler
  421.     table_list Lock first table in this list
  422.   NOTES
  423.     Works together with global read lock.
  424.   RETURN
  425.     0 ok
  426.     1 error
  427. */
  428. int lock_and_wait_for_table_name(THD *thd, TABLE_LIST *table_list)
  429. {
  430.   int lock_retcode;
  431.   int error= -1;
  432.   DBUG_ENTER("lock_and_wait_for_table_name");
  433.   if (wait_if_global_read_lock(thd, 0, 1))
  434.     DBUG_RETURN(1);
  435.   VOID(pthread_mutex_lock(&LOCK_open));
  436.   if ((lock_retcode = lock_table_name(thd, table_list)) < 0)
  437.     goto end;
  438.   if (lock_retcode && wait_for_locked_table_names(thd, table_list))
  439.   {
  440.     unlock_table_name(thd, table_list);
  441.     goto end;
  442.   }
  443.   error=0;
  444. end:
  445.   pthread_mutex_unlock(&LOCK_open);
  446.   start_waiting_global_read_lock(thd);
  447.   DBUG_RETURN(error);
  448. }
  449. /*
  450.   Put a not open table with an old refresh version in the table cache.
  451.   SYNPOSIS
  452.     lock_table_name()
  453.     thd Thread handler
  454.     table_list Lock first table in this list
  455.   WARNING
  456.     If you are going to update the table, you should use
  457.     lock_and_wait_for_table_name instead of this function as this works
  458.     together with 'FLUSH TABLES WITH READ LOCK'
  459.   NOTES
  460.     This will force any other threads that uses the table to release it
  461.     as soon as possible.
  462.   REQUIREMENTS
  463.     One must have a lock on LOCK_open !
  464.   RETURN:
  465.     < 0 error
  466.     == 0 table locked
  467.     > 0  table locked, but someone is using it
  468. */
  469. int lock_table_name(THD *thd, TABLE_LIST *table_list)
  470. {
  471.   TABLE *table;
  472.   char  key[MAX_DBKEY_LENGTH];
  473.   char *db= table_list->db;
  474.   uint  key_length;
  475.   DBUG_ENTER("lock_table_name");
  476.   DBUG_PRINT("enter",("db: %s  name: %s", db, table_list->real_name));
  477.   safe_mutex_assert_owner(&LOCK_open);
  478.   key_length=(uint) (strmov(strmov(key,db)+1,table_list->real_name)
  479.      -key)+ 1;
  480.   /* Only insert the table if we haven't insert it already */
  481.   for (table=(TABLE*) hash_search(&open_cache,(byte*) key,key_length) ;
  482.        table ;
  483.        table = (TABLE*) hash_next(&open_cache,(byte*) key,key_length))
  484.     if (table->in_use == thd)
  485.       DBUG_RETURN(0);
  486.   /*
  487.     Create a table entry with the right key and with an old refresh version
  488.     Note that we must use my_malloc() here as this is freed by the table
  489.     cache
  490.   */
  491.   if (!(table= (TABLE*) my_malloc(sizeof(*table)+key_length,
  492.   MYF(MY_WME | MY_ZEROFILL))))
  493.     DBUG_RETURN(-1);
  494.   memcpy((table->table_cache_key= (char*) (table+1)), key, key_length);
  495.   table->key_length=key_length;
  496.   table->in_use=thd;
  497.   table->locked_by_name=1;
  498.   table_list->table=table;
  499.   if (my_hash_insert(&open_cache, (byte*) table))
  500.   {
  501.     my_free((gptr) table,MYF(0));
  502.     DBUG_RETURN(-1);
  503.   }
  504.   
  505.   if (remove_table_from_cache(thd, db, table_list->real_name, RTFC_NO_FLAG))
  506.   {
  507.     DBUG_RETURN(1); // Table is in use
  508.   }
  509.   DBUG_RETURN(0);
  510. }
  511. void unlock_table_name(THD *thd, TABLE_LIST *table_list)
  512. {
  513.   if (table_list->table)
  514.   {
  515.     hash_delete(&open_cache, (byte*) table_list->table);
  516.     (void) pthread_cond_broadcast(&COND_refresh);
  517.   }
  518. }
  519. static bool locked_named_table(THD *thd, TABLE_LIST *table_list)
  520. {
  521.   for (; table_list ; table_list=table_list->next)
  522.   {
  523.     if (table_list->table && table_is_used(table_list->table,0))
  524.       return 1;
  525.   }
  526.   return 0; // All tables are locked
  527. }
  528. bool wait_for_locked_table_names(THD *thd, TABLE_LIST *table_list)
  529. {
  530.   bool result=0;
  531.   DBUG_ENTER("wait_for_locked_table_names");
  532.   safe_mutex_assert_owner(&LOCK_open);
  533.   while (locked_named_table(thd,table_list))
  534.   {
  535.     if (thd->killed)
  536.     {
  537.       result=1;
  538.       break;
  539.     }
  540.     wait_for_refresh(thd);
  541.     pthread_mutex_lock(&LOCK_open);
  542.   }
  543.   DBUG_RETURN(result);
  544. }
  545. /*
  546.   Lock all tables in list with a name lock
  547.   SYNOPSIS
  548.     lock_table_names()
  549.     thd Thread handle
  550.     table_list Names of tables to lock
  551.   NOTES
  552.     If you are just locking one table, you should use
  553.     lock_and_wait_for_table_name().
  554.   REQUIREMENTS
  555.     One must have a lock on LOCK_open when calling this
  556.   RETURN
  557.     0 ok
  558.     1 Fatal error (end of memory ?)
  559. */
  560. bool lock_table_names(THD *thd, TABLE_LIST *table_list)
  561. {
  562.   bool got_all_locks=1;
  563.   TABLE_LIST *lock_table;
  564.   for (lock_table=table_list ; lock_table ; lock_table=lock_table->next)
  565.   {
  566.     int got_lock;
  567.     if ((got_lock=lock_table_name(thd,lock_table)) < 0)
  568.       goto end; // Fatal error
  569.     if (got_lock)
  570.       got_all_locks=0; // Someone is using table
  571.   }
  572.   /* If some table was in use, wait until we got the lock */
  573.   if (!got_all_locks && wait_for_locked_table_names(thd, table_list))
  574.     goto end;
  575.   return 0;
  576. end:
  577.   unlock_table_names(thd, table_list, lock_table);
  578.   return 1;
  579. }
  580. /*
  581.   Unlock all tables in list with a name lock
  582.   SYNOPSIS
  583.     unlock_table_names()
  584.     thd Thread handle
  585.     table_list Names of tables to unlock
  586.     last_table Don't unlock any tables after this one.
  587. (default 0, which will unlock all tables)
  588.   NOTES
  589.     One must have a lock on LOCK_open when calling this
  590.     This function will send a COND_refresh signal to inform other threads
  591.     that the name locks are removed
  592.   RETURN
  593.     0 ok
  594.     1 Fatal error (end of memory ?)
  595. */
  596. void unlock_table_names(THD *thd, TABLE_LIST *table_list,
  597. TABLE_LIST *last_table)
  598. {
  599.   for (TABLE_LIST *table=table_list ; table != last_table ; table=table->next)
  600.     unlock_table_name(thd,table);
  601.   pthread_cond_broadcast(&COND_refresh);
  602. }
  603. static void print_lock_error(int error)
  604. {
  605.   int textno;
  606.   DBUG_ENTER("print_lock_error");
  607.   switch (error) {
  608.   case HA_ERR_LOCK_WAIT_TIMEOUT:
  609.     textno=ER_LOCK_WAIT_TIMEOUT;
  610.     break;
  611.   case HA_ERR_READ_ONLY_TRANSACTION:
  612.     textno=ER_READ_ONLY_TRANSACTION;
  613.     break;
  614.   default:
  615.     textno=ER_CANT_LOCK;
  616.     break;
  617.   }
  618.   my_error(textno,MYF(ME_BELL+ME_OLDWIN+ME_WAITTANG),error);
  619.   DBUG_VOID_RETURN;
  620. }
  621. /****************************************************************************
  622.   Handling of global read locks
  623.   Taking the global read lock is TWO steps (2nd step is optional; without
  624.   it, COMMIT of existing transactions will be allowed):
  625.   lock_global_read_lock() THEN make_global_read_lock_block_commit().
  626.   The global locks are handled through the global variables:
  627.   global_read_lock
  628.     count of threads which have the global read lock (i.e. have completed at
  629.     least the first step above)
  630.   global_read_lock_blocks_commit
  631.     count of threads which have the global read lock and block
  632.     commits (i.e. are in or have completed the second step above)
  633.   waiting_for_read_lock
  634.     count of threads which want to take a global read lock but cannot
  635.   protect_against_global_read_lock
  636.     count of threads which have set protection against global read lock.
  637.   How blocking of threads by global read lock is achieved: that's
  638.   advisory. Any piece of code which should be blocked by global read lock must
  639.   be designed like this:
  640.   - call to wait_if_global_read_lock(). When this returns 0, no global read
  641.   lock is owned; if argument abort_on_refresh was 0, none can be obtained.
  642.   - job
  643.   - if abort_on_refresh was 0, call to start_waiting_global_read_lock() to
  644.   allow other threads to get the global read lock. I.e. removal of the
  645.   protection.
  646.   (Note: it's a bit like an implementation of rwlock).
  647.   [ I am sorry to mention some SQL syntaxes below I know I shouldn't but found
  648.   no better descriptive way ]
  649.   Why does FLUSH TABLES WITH READ LOCK need to block COMMIT: because it's used
  650.   to read a non-moving SHOW MASTER STATUS, and a COMMIT writes to the binary
  651.   log.
  652.   Why getting the global read lock is two steps and not one. Because FLUSH
  653.   TABLES WITH READ LOCK needs to insert one other step between the two:
  654.   flushing tables. So the order is
  655.   1) lock_global_read_lock() (prevents any new table write locks, i.e. stalls
  656.   all new updates)
  657.   2) close_cached_tables() (the FLUSH TABLES), which will wait for tables
  658.   currently opened and being updated to close (so it's possible that there is
  659.   a moment where all new updates of server are stalled *and* FLUSH TABLES WITH
  660.   READ LOCK is, too).
  661.   3) make_global_read_lock_block_commit().
  662.   If we have merged 1) and 3) into 1), we would have had this deadlock:
  663.   imagine thread 1 and 2, in non-autocommit mode, thread 3, and an InnoDB
  664.   table t.
  665.   thd1: SELECT * FROM t FOR UPDATE;
  666.   thd2: UPDATE t SET a=1; # blocked by row-level locks of thd1
  667.   thd3: FLUSH TABLES WITH READ LOCK; # blocked in close_cached_tables() by the
  668.   table instance of thd2
  669.   thd1: COMMIT; # blocked by thd3.
  670.   thd1 blocks thd2 which blocks thd3 which blocks thd1: deadlock.
  671.   
  672.   Note that we need to support that one thread does
  673.   FLUSH TABLES WITH READ LOCK; and then COMMIT;
  674.   (that's what innobackup does, for some good reason).
  675.   So in this exceptional case the COMMIT should not be blocked by the FLUSH
  676.   TABLES WITH READ LOCK.
  677.   TODO in MySQL 5.x: make_global_read_lock_block_commit() should be
  678.   killable. Normally CPU does not spend a long time in this function (COMMITs
  679.   are quite fast), but it would still be nice.
  680. ****************************************************************************/
  681. volatile uint global_read_lock=0;
  682. volatile uint global_read_lock_blocks_commit=0;
  683. static volatile uint protect_against_global_read_lock=0;
  684. static volatile uint waiting_for_read_lock=0;
  685. #define GOT_GLOBAL_READ_LOCK               1
  686. #define MADE_GLOBAL_READ_LOCK_BLOCK_COMMIT 2
  687. bool lock_global_read_lock(THD *thd)
  688. {
  689.   DBUG_ENTER("lock_global_read_lock");
  690.   if (!thd->global_read_lock)
  691.   {
  692.     (void) pthread_mutex_lock(&LOCK_open);
  693.     const char *old_message=thd->enter_cond(&COND_refresh, &LOCK_open,
  694.     "Waiting to get readlock");
  695.     DBUG_PRINT("info",
  696.        ("waiting_for: %d  protect_against: %d",
  697. waiting_for_read_lock, protect_against_global_read_lock));
  698.     waiting_for_read_lock++;
  699.     while (protect_against_global_read_lock && !thd->killed)
  700.       pthread_cond_wait(&COND_refresh, &LOCK_open);
  701.     waiting_for_read_lock--;
  702.     if (thd->killed)
  703.     {
  704.       thd->exit_cond(old_message);
  705.       DBUG_RETURN(1);
  706.     }
  707.     thd->global_read_lock= GOT_GLOBAL_READ_LOCK;
  708.     global_read_lock++;
  709.     thd->exit_cond(old_message);
  710.   }
  711.   /*
  712.     We DON'T set global_read_lock_blocks_commit now, it will be set after
  713.     tables are flushed (as the present function serves for FLUSH TABLES WITH
  714.     READ LOCK only). Doing things in this order is necessary to avoid
  715.     deadlocks (we must allow COMMIT until all tables are closed; we should not
  716.     forbid it before, or we can have a 3-thread deadlock if 2 do SELECT FOR
  717.     UPDATE and one does FLUSH TABLES WITH READ LOCK).
  718.   */
  719.   DBUG_RETURN(0);
  720. }
  721. void unlock_global_read_lock(THD *thd)
  722. {
  723.   uint tmp;
  724.   pthread_mutex_lock(&LOCK_open);
  725.   tmp= --global_read_lock;
  726.   if (thd->global_read_lock == MADE_GLOBAL_READ_LOCK_BLOCK_COMMIT)
  727.     --global_read_lock_blocks_commit;
  728.   pthread_mutex_unlock(&LOCK_open);
  729.   /* Send the signal outside the mutex to avoid a context switch */
  730.   if (!tmp)
  731.     pthread_cond_broadcast(&COND_refresh);
  732.   thd->global_read_lock= 0;
  733. }
  734. #define must_wait (global_read_lock &&                             
  735.                    (is_not_commit ||                               
  736.                     global_read_lock_blocks_commit))
  737. bool wait_if_global_read_lock(THD *thd, bool abort_on_refresh,
  738.                               bool is_not_commit)
  739. {
  740.   const char *old_message;
  741.   bool result= 0, need_exit_cond;
  742.   DBUG_ENTER("wait_if_global_read_lock");
  743.   LINT_INIT(old_message);
  744.   (void) pthread_mutex_lock(&LOCK_open);
  745.   if ((need_exit_cond= must_wait))
  746.   {
  747.     if (thd->global_read_lock) // This thread had the read locks
  748.     {
  749.       if (is_not_commit)
  750.         my_error(ER_CANT_UPDATE_WITH_READLOCK,MYF(0));
  751.       (void) pthread_mutex_unlock(&LOCK_open);
  752.       /*
  753.         We allow FLUSHer to COMMIT; we assume FLUSHer knows what it does.
  754.         This allowance is needed to not break existing versions of innobackup
  755.         which do a BEGIN; INSERT; FLUSH TABLES WITH READ LOCK; COMMIT.
  756.       */
  757.       DBUG_RETURN(is_not_commit);
  758.     }
  759.     old_message=thd->enter_cond(&COND_refresh, &LOCK_open,
  760. "Waiting for release of readlock");
  761.     while (must_wait && ! thd->killed &&
  762.    (!abort_on_refresh || thd->version == refresh_version))
  763.       (void) pthread_cond_wait(&COND_refresh,&LOCK_open);
  764.     if (thd->killed)
  765.       result=1;
  766.   }
  767.   if (!abort_on_refresh && !result)
  768.     protect_against_global_read_lock++;
  769.   /*
  770.     The following is only true in case of a global read locks (which is rare)
  771.     and if old_message is set
  772.   */
  773.   if (unlikely(need_exit_cond)) 
  774.     thd->exit_cond(old_message);
  775.   else
  776.     pthread_mutex_unlock(&LOCK_open);
  777.   DBUG_RETURN(result);
  778. }
  779. void start_waiting_global_read_lock(THD *thd)
  780. {
  781.   bool tmp;
  782.   DBUG_ENTER("start_waiting_global_read_lock");
  783.   if (unlikely(thd->global_read_lock))
  784.     DBUG_VOID_RETURN;
  785.   (void) pthread_mutex_lock(&LOCK_open);
  786.   tmp= (!--protect_against_global_read_lock &&
  787.         (waiting_for_read_lock || global_read_lock_blocks_commit));
  788.   (void) pthread_mutex_unlock(&LOCK_open);
  789.   if (tmp)
  790.     pthread_cond_broadcast(&COND_refresh);
  791.   DBUG_VOID_RETURN;
  792. }
  793. void make_global_read_lock_block_commit(THD *thd)
  794. {
  795.   /*
  796.     If we didn't succeed lock_global_read_lock(), or if we already suceeded
  797.     make_global_read_lock_block_commit(), do nothing.
  798.   */
  799.   if (thd->global_read_lock != GOT_GLOBAL_READ_LOCK)
  800.     return;
  801.   pthread_mutex_lock(&LOCK_open);
  802.   /* increment this BEFORE waiting on cond (otherwise race cond) */
  803.   global_read_lock_blocks_commit++;
  804.   while (protect_against_global_read_lock)
  805.     pthread_cond_wait(&COND_refresh, &LOCK_open);
  806.   pthread_mutex_unlock(&LOCK_open);
  807.   thd->global_read_lock= MADE_GLOBAL_READ_LOCK_BLOCK_COMMIT;
  808. }