← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to plugin/heap/hp_rnext.cc

  • Committer: Brian Aker
  • Date: 2009-12-18 18:31:01 UTC
  • mfrom: (1241.2.7 build)
  • Revision ID: brian@gaz-20091218183101-igqg1dtowpa0o70s
Fixed from Monty.

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugin/heap/hp_rnext.cc
11
11
 
12
12
   You should have received a copy of the GNU General Public License
13
13
   along with this program; if not, write to the Free Software
14
 
   Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA */
 
14
   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
15
15
 
16
16
#include "heap_priv.h"
17
17
 
18
18
#include <string.h>
19
19
 
20
 
using namespace drizzled;
21
 
 
22
20
/* Read next record with the same key */
23
21
 
24
22
int heap_rnext(HP_INFO *info, unsigned char *record)
25
23
{
26
24
  unsigned char *pos;
27
 
  HP_SHARE *share=info->getShare();
 
25
  HP_SHARE *share=info->s;
28
26
  HP_KEYDEF *keyinfo;
29
27
 
30
28
  if (info->lastinx < 0)
31
 
    return(errno=HA_ERR_WRONG_INDEX);
 
29
    return(my_errno=HA_ERR_WRONG_INDEX);
32
30
 
33
31
  keyinfo = share->keydef + info->lastinx;
 
32
  if (keyinfo->algorithm == HA_KEY_ALG_BTREE)
 
33
  {
 
34
    heap_rb_param custom_arg;
 
35
 
 
36
    if (info->last_pos)
 
37
    {
 
38
      /*
 
39
        We enter this branch for non-DELETE queries after heap_rkey()
 
40
        or heap_rfirst(). As last key position (info->last_pos) is available,
 
41
        we only need to climb the tree using tree_search_next().
 
42
      */
 
43
      pos = (unsigned char *)tree_search_next(&keyinfo->rb_tree,
 
44
                                              &info->last_pos,
 
45
                                              offsetof(TREE_ELEMENT, left),
 
46
                                              offsetof(TREE_ELEMENT, right));
 
47
    }
 
48
    else if (!info->lastkey_len)
 
49
    {
 
50
      /*
 
51
        We enter this branch only for DELETE queries after heap_rfirst(). E.g.
 
52
        DELETE FROM t1 WHERE a<10. As last key position is not available
 
53
        (last key is removed by heap_delete()), we must restart search as it
 
54
        is done in heap_rfirst().
 
55
 
 
56
        It should be safe to handle this situation without this branch. That is
 
57
        branch below should find smallest element in a tree as lastkey_len is
 
58
        zero. tree_search_edge() is a kind of optimisation here as it should be
 
59
        faster than tree_search_key().
 
60
      */
 
61
      pos= (unsigned char *)tree_search_edge(&keyinfo->rb_tree, info->parents,
 
62
                                             &info->last_pos,
 
63
                                             offsetof(TREE_ELEMENT, left));
 
64
    }
 
65
    else
 
66
    {
 
67
      /*
 
68
        We enter this branch only for DELETE queries after heap_rkey(). E.g.
 
69
        DELETE FROM t1 WHERE a=10. As last key position is not available
 
70
        (last key is removed by heap_delete()), we must restart search as it
 
71
        is done in heap_rkey().
 
72
      */
 
73
      custom_arg.keyseg = keyinfo->seg;
 
74
      custom_arg.key_length = info->lastkey_len;
 
75
      custom_arg.search_flag = SEARCH_SAME | SEARCH_FIND;
 
76
      pos = (unsigned char *)tree_search_key(&keyinfo->rb_tree,
 
77
                                             info->lastkey, info->parents,
 
78
                                             &info->last_pos,
 
79
                                             info->last_find_flag,
 
80
                                             &custom_arg);
 
81
    }
 
82
    if (pos)
 
83
    {
 
84
      memcpy(&pos, pos + (*keyinfo->get_key_length)(keyinfo, pos),
 
85
             sizeof(unsigned char*));
 
86
      info->current_ptr = pos;
 
87
    }
 
88
    else
 
89
    {
 
90
      my_errno = HA_ERR_KEY_NOT_FOUND;
 
91
    }
 
92
  }
 
93
  else
34
94
  {
35
95
    if (info->current_hash_ptr)
36
 
      pos= hp_search_next(info, keyinfo, &info->lastkey[0],
 
96
      pos= hp_search_next(info, keyinfo, info->lastkey,
37
97
                           info->current_hash_ptr);
38
98
    else
39
99
    {
40
100
      if (!info->current_ptr && (info->update & HA_STATE_NEXT_FOUND))
41
101
      {
42
102
        pos=0;                                  /* Read next after last */
43
 
        errno=HA_ERR_KEY_NOT_FOUND;
 
103
        my_errno=HA_ERR_KEY_NOT_FOUND;
44
104
      }
45
105
      else if (!info->current_ptr)              /* Deleted or first call */
46
 
        pos= hp_search(info, keyinfo, &info->lastkey[0], 0);
 
106
        pos= hp_search(info, keyinfo, info->lastkey, 0);
47
107
      else
48
 
        pos= hp_search(info, keyinfo, &info->lastkey[0], 1);
 
108
        pos= hp_search(info, keyinfo, info->lastkey, 1);
49
109
    }
50
110
  }
51
111
  if (!pos)
52
112
  {
53
113
    info->update=HA_STATE_NEXT_FOUND;           /* For heap_rprev */
54
 
    if (errno == HA_ERR_KEY_NOT_FOUND)
55
 
      errno=HA_ERR_END_OF_FILE;
56
 
    return(errno);
 
114
    if (my_errno == HA_ERR_KEY_NOT_FOUND)
 
115
      my_errno=HA_ERR_END_OF_FILE;
 
116
    return(my_errno);
57
117
  }
58
118
  hp_extract_record(share, record, pos);
59
119
  info->update=HA_STATE_AKTIV | HA_STATE_NEXT_FOUND;