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

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. #include "myisamdef.h"
  14. #ifdef HAVE_MMAP
  15. #include <sys/mman.h>
  16. #endif
  17. static void mi_extra_keyflag(MI_INFO *info, enum ha_extra_function function);
  18. /*
  19.   Set options and buffers to optimize table handling
  20.   SYNOPSIS
  21.     mi_extra()
  22.     info open table
  23.     function operation
  24.     extra_arg Pointer to extra argument (normally pointer to ulong)
  25.      Used when function is one of:
  26. HA_EXTRA_WRITE_CACHE
  27. HA_EXTRA_CACHE
  28.   RETURN VALUES
  29.     0  ok
  30.     #  error
  31. */
  32. int mi_extra(MI_INFO *info, enum ha_extra_function function, void *extra_arg)
  33. {
  34.   int error=0;
  35.   ulong cache_size;
  36.   MYISAM_SHARE *share=info->s;
  37.   DBUG_ENTER("mi_extra");
  38.   DBUG_PRINT("enter",("function: %d",(int) function));
  39.   switch (function) {
  40.   case HA_EXTRA_RESET:
  41.     /*
  42.       Free buffers and reset the following flags:
  43.       EXTRA_CACHE, EXTRA_WRITE_CACHE, EXTRA_KEYREAD, EXTRA_QUICK
  44.       If the row buffer cache is large (for dynamic tables), reduce it
  45.       to save memory.
  46.     */
  47.     if (info->opt_flag & (READ_CACHE_USED | WRITE_CACHE_USED))
  48.     {
  49.       info->opt_flag&= ~(READ_CACHE_USED | WRITE_CACHE_USED);
  50.       error=end_io_cache(&info->rec_cache);
  51.     }
  52.     if (share->base.blobs)
  53.       mi_alloc_rec_buff(info, -1, &info->rec_buff);
  54. #if defined(HAVE_MMAP) && defined(HAVE_MADVISE)
  55.     if (info->opt_flag & MEMMAP_USED)
  56.       madvise(share->file_map,share->state.state.data_file_length,MADV_RANDOM);
  57. #endif
  58.     info->opt_flag&= ~(KEY_READ_USED | REMEMBER_OLD_POS);
  59.     info->quick_mode=0;
  60.     /* Fall through */
  61.   case HA_EXTRA_RESET_STATE: /* Reset state (don't free buffers) */
  62.     info->lastinx= 0; /* Use first index as def */
  63.     info->last_search_keypage=info->lastpos= HA_OFFSET_ERROR;
  64.     info->page_changed=1;
  65. /* Next/prev gives first/last */
  66.     if (info->opt_flag & READ_CACHE_USED)
  67.     {
  68.       reinit_io_cache(&info->rec_cache,READ_CACHE,0,
  69.       (pbool) (info->lock_type != F_UNLCK),
  70.       (pbool) test(info->update & HA_STATE_ROW_CHANGED)
  71.       );
  72.     }
  73.     info->update= ((info->update & HA_STATE_CHANGED) | HA_STATE_NEXT_FOUND |
  74.    HA_STATE_PREV_FOUND);
  75.     break;
  76.   case HA_EXTRA_CACHE:
  77.     if (info->lock_type == F_UNLCK &&
  78. (share->options & HA_OPTION_PACK_RECORD))
  79.     {
  80.       error=1; /* Not possibly if not locked */
  81.       my_errno=EACCES;
  82.       break;
  83.     }
  84. #if defined(HAVE_MMAP) && defined(HAVE_MADVISE)
  85.     if ((share->options & HA_OPTION_COMPRESS_RECORD))
  86.     {
  87.       pthread_mutex_lock(&share->intern_lock);
  88.       if (_mi_memmap_file(info))
  89.       {
  90. /* We don't nead MADV_SEQUENTIAL if small file */
  91. madvise(share->file_map,share->state.state.data_file_length,
  92. share->state.state.data_file_length <= RECORD_CACHE_SIZE*16 ?
  93. MADV_RANDOM : MADV_SEQUENTIAL);
  94. pthread_mutex_unlock(&share->intern_lock);
  95. break;
  96.       }
  97.       pthread_mutex_unlock(&share->intern_lock);
  98.     }
  99. #endif
  100.     if (info->opt_flag & WRITE_CACHE_USED)
  101.     {
  102.       info->opt_flag&= ~WRITE_CACHE_USED;
  103.       if ((error=end_io_cache(&info->rec_cache)))
  104. break;
  105.     }
  106.     if (!(info->opt_flag &
  107.   (READ_CACHE_USED | WRITE_CACHE_USED | MEMMAP_USED)))
  108.     {
  109.       cache_size= (extra_arg ? *(ulong*) extra_arg :
  110.    my_default_record_cache_size);
  111.       if (!(init_io_cache(&info->rec_cache,info->dfile,
  112.  (uint) min(info->state->data_file_length+1,
  113.     cache_size),
  114.   READ_CACHE,0L,(pbool) (info->lock_type != F_UNLCK),
  115.   MYF(share->write_flag & MY_WAIT_IF_FULL))))
  116.       {
  117. info->opt_flag|=READ_CACHE_USED;
  118. info->update&= ~HA_STATE_ROW_CHANGED;
  119.       }
  120.       if (share->concurrent_insert)
  121. info->rec_cache.end_of_file=info->state->data_file_length;
  122.     }
  123.     break;
  124.   case HA_EXTRA_REINIT_CACHE:
  125.     if (info->opt_flag & READ_CACHE_USED)
  126.     {
  127.       reinit_io_cache(&info->rec_cache,READ_CACHE,info->nextpos,
  128.       (pbool) (info->lock_type != F_UNLCK),
  129.       (pbool) test(info->update & HA_STATE_ROW_CHANGED));
  130.       info->update&= ~HA_STATE_ROW_CHANGED;
  131.       if (share->concurrent_insert)
  132. info->rec_cache.end_of_file=info->state->data_file_length;
  133.     }
  134.     break;
  135.   case HA_EXTRA_WRITE_CACHE:
  136.     if (info->lock_type == F_UNLCK)
  137.     {
  138.       error=1; /* Not possibly if not locked */
  139.       break;
  140.     }
  141.     cache_size= (extra_arg ? *(ulong*) extra_arg :
  142.  my_default_record_cache_size);
  143.     if (!(info->opt_flag &
  144.   (READ_CACHE_USED | WRITE_CACHE_USED | OPT_NO_ROWS)) &&
  145. !share->state.header.uniques)
  146.       if (!(init_io_cache(&info->rec_cache,info->dfile, cache_size,
  147.  WRITE_CACHE,info->state->data_file_length,
  148.   (pbool) (info->lock_type != F_UNLCK),
  149.   MYF(share->write_flag & MY_WAIT_IF_FULL))))
  150.       {
  151. info->opt_flag|=WRITE_CACHE_USED;
  152. info->update&= ~(HA_STATE_ROW_CHANGED |
  153.  HA_STATE_WRITE_AT_END |
  154.  HA_STATE_EXTEND_BLOCK);
  155.       }
  156.     break;
  157.   case HA_EXTRA_PREPARE_FOR_UPDATE:
  158.     if (info->s->data_file_type != DYNAMIC_RECORD)
  159.       break;
  160.     /* Remove read/write cache if dynamic rows */
  161.   case HA_EXTRA_NO_CACHE:
  162.     if (info->opt_flag & (READ_CACHE_USED | WRITE_CACHE_USED))
  163.     {
  164.       info->opt_flag&= ~(READ_CACHE_USED | WRITE_CACHE_USED);
  165.       error=end_io_cache(&info->rec_cache);
  166.       /* Sergei will insert full text index caching here */
  167.     }
  168. #if defined(HAVE_MMAP) && defined(HAVE_MADVISE)
  169.     if (info->opt_flag & MEMMAP_USED)
  170.       madvise(share->file_map,share->state.state.data_file_length,MADV_RANDOM);
  171. #endif
  172.     break;
  173.   case HA_EXTRA_FLUSH_CACHE:
  174.     if (info->opt_flag & WRITE_CACHE_USED)
  175.     {
  176.       if ((error=flush_io_cache(&info->rec_cache)))
  177. mi_mark_crashed(info); /* Fatal error found */
  178.     }
  179.     break;
  180.   case HA_EXTRA_NO_READCHECK:
  181.     info->opt_flag&= ~READ_CHECK_USED; /* No readcheck */
  182.     break;
  183.   case HA_EXTRA_READCHECK:
  184.     info->opt_flag|= READ_CHECK_USED;
  185.     break;
  186.   case HA_EXTRA_KEYREAD: /* Read only keys to record */
  187.   case HA_EXTRA_REMEMBER_POS:
  188.     info->opt_flag |= REMEMBER_OLD_POS;
  189.     bmove((byte*) info->lastkey+share->base.max_key_length*2,
  190.   (byte*) info->lastkey,info->lastkey_length);
  191.     info->save_update= info->update;
  192.     info->save_lastinx= info->lastinx;
  193.     info->save_lastpos= info->lastpos;
  194.     info->save_lastkey_length=info->lastkey_length;
  195.     if (function == HA_EXTRA_REMEMBER_POS)
  196.       break;
  197.     /* fall through */
  198.   case HA_EXTRA_KEYREAD_CHANGE_POS:
  199.     info->opt_flag |= KEY_READ_USED;
  200.     info->read_record=_mi_read_key_record;
  201.     break;
  202.   case HA_EXTRA_NO_KEYREAD:
  203.   case HA_EXTRA_RESTORE_POS:
  204.     if (info->opt_flag & REMEMBER_OLD_POS)
  205.     {
  206.       bmove((byte*) info->lastkey,
  207.     (byte*) info->lastkey+share->base.max_key_length*2,
  208.     info->save_lastkey_length);
  209.       info->update= info->save_update | HA_STATE_WRITTEN;
  210.       info->lastinx= info->save_lastinx;
  211.       info->lastpos= info->save_lastpos;
  212.       info->lastkey_length=info->save_lastkey_length;
  213.     }
  214.     info->read_record= share->read_record;
  215.     info->opt_flag&= ~(KEY_READ_USED | REMEMBER_OLD_POS);
  216.     break;
  217.   case HA_EXTRA_NO_USER_CHANGE: /* Database is somehow locked agains changes */
  218.     info->lock_type= F_EXTRA_LCK; /* Simulate as locked */
  219.     break;
  220.   case HA_EXTRA_WAIT_LOCK:
  221.     info->lock_wait=0;
  222.     break;
  223.   case HA_EXTRA_NO_WAIT_LOCK:
  224.     info->lock_wait=MY_DONT_WAIT;
  225.     break;
  226.   case HA_EXTRA_NO_KEYS:
  227.     if (info->lock_type == F_UNLCK)
  228.     {
  229.       error=1; /* Not possibly if not lock */
  230.       break;
  231.     }
  232.     if (share->state.key_map)
  233.     {
  234.       MI_KEYDEF *key=share->keyinfo;
  235.       uint i;
  236.       for (i=0 ; i < share->base.keys ; i++,key++)
  237.       {
  238.         if (!(key->flag & HA_NOSAME) && info->s->base.auto_key != i+1)
  239.         {
  240.           share->state.key_map&= ~ ((ulonglong) 1 << i);
  241.           info->update|= HA_STATE_CHANGED;
  242.         }
  243.       }
  244.       if (!share->changed)
  245.       {
  246. share->state.changed|= STATE_CHANGED | STATE_NOT_ANALYZED;
  247. share->changed=1; /* Update on close */
  248. if (!share->global_changed)
  249. {
  250.   share->global_changed=1;
  251.   share->state.open_count++;
  252. }
  253.       }
  254.       share->state.state= *info->state;
  255.       error=mi_state_info_write(share->kfile,&share->state,1 | 2);
  256.     }
  257.     break;
  258.   case HA_EXTRA_FORCE_REOPEN:
  259.     pthread_mutex_lock(&THR_LOCK_myisam);
  260.     share->last_version= 0L; /* Impossible version */
  261.     pthread_mutex_unlock(&THR_LOCK_myisam);
  262.     break;
  263.   case HA_EXTRA_PREPARE_FOR_DELETE:
  264.     pthread_mutex_lock(&THR_LOCK_myisam);
  265.     share->last_version= 0L; /* Impossible version */
  266. #ifdef __WIN__
  267.     /* Close the isam and data files as Win32 can't drop an open table */
  268.     pthread_mutex_lock(&share->intern_lock);
  269.     if (flush_key_blocks(share->key_cache, share->kfile,
  270.  (function == HA_EXTRA_FORCE_REOPEN ?
  271.   FLUSH_RELEASE : FLUSH_IGNORE_CHANGED)))
  272.     {
  273.       error=my_errno;
  274.       share->changed=1;
  275.       mi_mark_crashed(info); /* Fatal error found */
  276.     }
  277.     if (info->opt_flag & (READ_CACHE_USED | WRITE_CACHE_USED))
  278.     {
  279.       info->opt_flag&= ~(READ_CACHE_USED | WRITE_CACHE_USED);
  280.       error=end_io_cache(&info->rec_cache);
  281.     }
  282.     if (info->lock_type != F_UNLCK && ! info->was_locked)
  283.     {
  284.       info->was_locked=info->lock_type;
  285.       if (mi_lock_database(info,F_UNLCK))
  286. error=my_errno;
  287.       info->lock_type = F_UNLCK;
  288.     }
  289.     if (share->kfile >= 0)
  290.       _mi_decrement_open_count(info);
  291.     if (share->kfile >= 0 && my_close(share->kfile,MYF(0)))
  292.       error=my_errno;
  293.     {
  294.       LIST *list_element ;
  295.       for (list_element=myisam_open_list ;
  296.    list_element ;
  297.    list_element=list_element->next)
  298.       {
  299. MI_INFO *tmpinfo=(MI_INFO*) list_element->data;
  300. if (tmpinfo->s == info->s)
  301. {
  302.   if (tmpinfo->dfile >= 0 && my_close(tmpinfo->dfile,MYF(0)))
  303.     error = my_errno;
  304.   tmpinfo->dfile= -1;
  305. }
  306.       }
  307.     }
  308.     share->kfile= -1; /* Files aren't open anymore */
  309.     pthread_mutex_unlock(&share->intern_lock);
  310. #endif
  311.     pthread_mutex_unlock(&THR_LOCK_myisam);
  312.     break;
  313.   case HA_EXTRA_FLUSH:
  314.     if (!share->temporary)
  315.       flush_key_blocks(share->key_cache, share->kfile, FLUSH_KEEP);
  316. #ifdef HAVE_PWRITE
  317.     _mi_decrement_open_count(info);
  318. #endif
  319.     if (share->not_flushed)
  320.     {
  321.       share->not_flushed=0;
  322.       if (my_sync(share->kfile, MYF(0)))
  323. error= my_errno;
  324.       if (my_sync(info->dfile, MYF(0)))
  325. error= my_errno;
  326.       if (error)
  327.       {
  328. share->changed=1;
  329. mi_mark_crashed(info); /* Fatal error found */
  330.       }
  331.     }
  332.     if (share->base.blobs)
  333.       mi_alloc_rec_buff(info, -1, &info->rec_buff);
  334.     break;
  335.   case HA_EXTRA_NORMAL: /* Theese isn't in use */
  336.     info->quick_mode=0;
  337.     break;
  338.   case HA_EXTRA_QUICK:
  339.     info->quick_mode=1;
  340.     break;
  341.   case HA_EXTRA_NO_ROWS:
  342.     if (!share->state.header.uniques)
  343.       info->opt_flag|= OPT_NO_ROWS;
  344.     break;
  345.   case HA_EXTRA_PRELOAD_BUFFER_SIZE:
  346.     info->preload_buff_size= *((ulong *) extra_arg); 
  347.     break;
  348.   case HA_EXTRA_CHANGE_KEY_TO_UNIQUE:
  349.   case HA_EXTRA_CHANGE_KEY_TO_DUP:
  350.     mi_extra_keyflag(info, function);
  351.     break;
  352.   case HA_EXTRA_KEY_CACHE:
  353.   case HA_EXTRA_NO_KEY_CACHE:
  354.   default:
  355.     break;
  356.   }
  357.   {
  358.     char tmp[1];
  359.     tmp[0]=function;
  360.     myisam_log_command(MI_LOG_EXTRA,info,(byte*) tmp,1,error);
  361.   }
  362.   DBUG_RETURN(error);
  363. } /* mi_extra */
  364. /*
  365.     Start/Stop Inserting Duplicates Into a Table, WL#1648.
  366.  */
  367. static void mi_extra_keyflag(MI_INFO *info, enum ha_extra_function function)
  368. {
  369.   uint  idx;
  370.   for (idx= 0; idx< info->s->base.keys; idx++)
  371.   {
  372.     switch (function) {
  373.     case HA_EXTRA_CHANGE_KEY_TO_UNIQUE:
  374.       info->s->keyinfo[idx].flag|= HA_NOSAME;
  375.       break;
  376.     case HA_EXTRA_CHANGE_KEY_TO_DUP:
  377.       info->s->keyinfo[idx].flag&= ~(HA_NOSAME);
  378.       break;
  379.     default:
  380.       break;
  381.     }
  382.   }
  383. }