mi_rkey.c
上传用户: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. /* Read record based on a key */
  14. #include "myisamdef.h"
  15. #include "rt_index.h"
  16. /* Read a record using key */
  17. /* Ordinary search_flag is 0 ; Give error if no record with key */
  18. int mi_rkey(MI_INFO *info, byte *buf, int inx, const byte *key, uint key_len,
  19.     enum ha_rkey_function search_flag)
  20. {
  21.   uchar *key_buff;
  22.   MYISAM_SHARE *share=info->s;
  23.   MI_KEYDEF *keyinfo;
  24.   HA_KEYSEG *last_used_keyseg;
  25.   uint pack_key_length, use_key_length, nextflag;
  26.   DBUG_ENTER("mi_rkey");
  27.   DBUG_PRINT("enter", ("base: %lx  buf: %lx  inx: %d  search_flag: %d",
  28.                        (long) info, (long) buf, inx, search_flag));
  29.   if ((inx = _mi_check_index(info,inx)) < 0)
  30.     DBUG_RETURN(my_errno);
  31.   info->update&= (HA_STATE_CHANGED | HA_STATE_ROW_CHANGED);
  32.   info->last_key_func= search_flag;
  33.   keyinfo= share->keyinfo + inx;
  34.   if (info->once_flags & USE_PACKED_KEYS)
  35.   {
  36.     info->once_flags&= ~USE_PACKED_KEYS; /* Reset flag */
  37.     /*
  38.       key is already packed!;  This happens when we are using a MERGE TABLE
  39.     */
  40.     key_buff=info->lastkey+info->s->base.max_key_length;
  41.     pack_key_length= key_len;
  42.     bmove(key_buff,key,key_len);
  43.     last_used_keyseg= 0;
  44.   }
  45.   else
  46.   {
  47.     if (key_len == 0)
  48.       key_len=USE_WHOLE_KEY;
  49.     /* Save the packed key for later use in the second buffer of lastkey. */
  50.     key_buff=info->lastkey+info->s->base.max_key_length;
  51.     pack_key_length=_mi_pack_key(info,(uint) inx, key_buff, (uchar*) key,
  52.  key_len, &last_used_keyseg);
  53.     /* Save packed_key_length for use by the MERGE engine. */
  54.     info->pack_key_length= pack_key_length;
  55.     DBUG_EXECUTE("key",_mi_print_key(DBUG_FILE, keyinfo->seg,
  56.      key_buff, pack_key_length););
  57.   }
  58.   if (fast_mi_readinfo(info))
  59.     goto err;
  60.   if (share->concurrent_insert)
  61.     rw_rdlock(&share->key_root_lock[inx]);
  62.   nextflag=myisam_read_vec[search_flag];
  63.   use_key_length=pack_key_length;
  64.   if (!(nextflag & (SEARCH_FIND | SEARCH_NO_FIND | SEARCH_LAST)))
  65.     use_key_length=USE_WHOLE_KEY;
  66.   switch (info->s->keyinfo[inx].key_alg) {
  67. #ifdef HAVE_RTREE_KEYS
  68.   case HA_KEY_ALG_RTREE:
  69.     if (rtree_find_first(info,inx,key_buff,use_key_length,nextflag) < 0)
  70.     {
  71.       my_errno=HA_ERR_CRASHED;
  72.       goto err;
  73.     }
  74.     break;
  75. #endif
  76.   case HA_KEY_ALG_BTREE:
  77.   default:
  78.     if (!_mi_search(info, keyinfo, key_buff, use_key_length,
  79.   myisam_read_vec[search_flag], info->s->state.key_root[inx]))
  80.     {
  81.       while (info->lastpos >= info->state->data_file_length)
  82.       {
  83.         /*
  84.   Skip rows that are inserted by other threads since we got a lock
  85.   Note that this can only happen if we are not searching after an
  86.   exact key, because the keys are sorted according to position
  87.         */
  88.         if  (_mi_search_next(info, keyinfo, info->lastkey,
  89.    info->lastkey_length,
  90.    myisam_readnext_vec[search_flag],
  91.    info->s->state.key_root[inx]))
  92.   break;
  93.       }
  94.     }
  95.   }
  96.   if (share->concurrent_insert)
  97.     rw_unlock(&share->key_root_lock[inx]);
  98.   /* Calculate length of the found key;  Used by mi_rnext_same */
  99.   if ((keyinfo->flag & HA_VAR_LENGTH_KEY) && last_used_keyseg &&
  100.       info->lastpos != HA_OFFSET_ERROR)
  101.     info->last_rkey_length= _mi_keylength_part(keyinfo, info->lastkey,
  102.        last_used_keyseg);
  103.   else
  104.     info->last_rkey_length= pack_key_length;
  105.   /* Check if we don't want to have record back, only error message */
  106.   if (!buf)
  107.     DBUG_RETURN(info->lastpos == HA_OFFSET_ERROR ? my_errno : 0);
  108.   if (!(*info->read_record)(info,info->lastpos,buf))
  109.   {
  110.     info->update|= HA_STATE_AKTIV; /* Record is read */
  111.     DBUG_RETURN(0);
  112.   }
  113.   info->lastpos = HA_OFFSET_ERROR; /* Didn't find key */
  114.   /* Store last used key as a base for read next */
  115.   memcpy(info->lastkey,key_buff,pack_key_length);
  116.   info->last_rkey_length= pack_key_length;
  117.   bzero((char*) info->lastkey+pack_key_length,info->s->base.rec_reflength);
  118.   info->lastkey_length=pack_key_length+info->s->base.rec_reflength;
  119.   if (search_flag == HA_READ_AFTER_KEY)
  120.     info->update|=HA_STATE_NEXT_FOUND; /* Previous gives last row */
  121. err:
  122.   DBUG_RETURN(my_errno);
  123. } /* _mi_rkey */