← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to mysys/my_handler.c

  • Committer: Monty Taylor
  • Date: 2008-07-09 16:33:52 UTC
  • mto: (77.6.1 glibclient-merge)
  • mto: This revision was merged to the branch mainline in revision 112.
  • Revision ID: monty@inaugust.com-20080709163352-yv7jbu81frc1l4ec
Finished the warnings work!

Show diffs side-by-side

added added

removed removed

Lines of Context:
mysys/my_handler.c
1
 
/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-
2
 
 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
3
 
 *
4
 
 *  Copyright (C) 2008 Sun Microsystems
5
 
 *
6
 
 *  This program is free software; you can redistribute it and/or modify
7
 
 *  it under the terms of the GNU General Public License as published by
8
 
 *  the Free Software Foundation; version 2 of the License.
9
 
 *
10
 
 *  This program is distributed in the hope that it will be useful,
11
 
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12
 
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13
 
 *  GNU General Public License for more details.
14
 
 *
15
 
 *  You should have received a copy of the GNU General Public License
16
 
 *  along with this program; if not, write to the Free Software
17
 
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
18
 
 */
19
 
 
20
 
#include "config.h"
21
 
 
22
 
#include "drizzled/charset_info.h"
23
 
#include <drizzled/base.h>
24
 
#include <plugin/myisam/my_handler.h>
25
 
#include "drizzled/internal/my_sys.h"
26
 
 
27
 
#include <cassert>
28
 
#include <algorithm>
29
 
 
30
 
using namespace drizzled;
31
 
using namespace std;
32
 
 
33
 
template<class T>
34
 
int CMP_NUM(const T& a, const T&b)
35
 
{
36
 
  return (a < b) ? -1 : (a == b) ? 0 : 1;
37
 
}
38
 
 
39
 
 
40
 
int ha_compare_text(const CHARSET_INFO * const charset_info, unsigned char *a, uint32_t a_length,
41
 
                    unsigned char *b, uint32_t b_length, bool part_key,
42
 
                    bool skip_end_space)
 
1
/* Copyright (C) 2002-2006 MySQL AB
 
2
   
 
3
   This library is free software; you can redistribute it and/or
 
4
   modify it under the terms of the GNU Library General Public
 
5
   License as published by the Free Software Foundation; version 2
 
6
   of the License.
 
7
   
 
8
   This library is distributed in the hope that it will be useful,
 
9
   but WITHOUT ANY WARRANTY; without even the implied warranty of
 
10
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 
11
   Library General Public License for more details.
 
12
 
 
13
   You should have received a copy of the GNU Library General Public
 
14
   License along with this library; if not, write to the Free
 
15
   Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
 
16
   MA 02111-1307, USA */
 
17
 
 
18
#include <my_global.h>
 
19
#include <m_ctype.h>
 
20
#include <my_base.h>
 
21
#include <my_handler.h>
 
22
#include <my_sys.h>
 
23
 
 
24
#include "mysys_priv.h"
 
25
#include "my_handler_errors.h"
 
26
 
 
27
int ha_compare_text(CHARSET_INFO *charset_info, uchar *a, uint a_length,
 
28
                    uchar *b, uint b_length, my_bool part_key,
 
29
                    my_bool skip_end_space)
43
30
{
44
31
  if (!part_key)
45
32
    return charset_info->coll->strnncollsp(charset_info, a, a_length,
46
 
                                           b, b_length, (bool)!skip_end_space);
 
33
                                           b, b_length, (my_bool)!skip_end_space);
47
34
  return charset_info->coll->strnncoll(charset_info, a, a_length,
48
35
                                       b, b_length, part_key);
49
36
}
50
37
 
51
38
 
52
 
static int compare_bin(unsigned char *a, uint32_t a_length, unsigned char *b, uint32_t b_length,
53
 
                       bool part_key, bool skip_end_space)
 
39
static int compare_bin(uchar *a, uint a_length, uchar *b, uint b_length,
 
40
                       my_bool part_key, my_bool skip_end_space)
54
41
{
55
 
  uint32_t length= min(a_length,b_length);
56
 
  unsigned char *end= a+ length;
 
42
  uint length= min(a_length,b_length);
 
43
  uchar *end= a+ length;
57
44
  int flag;
58
45
 
59
46
  while (a < end)
104
91
    next_flag   How keys should be compared
105
92
                If bit SEARCH_FIND is not set the keys includes the row
106
93
                position and this should also be compared
107
 
    diff_pos    OUT Number of first keypart where values differ, counting
 
94
    diff_pos    OUT Number of first keypart where values differ, counting 
108
95
                from one.
109
96
    diff_pos[1] OUT  (b + diff_pos[1]) points to first value in tuple b
110
97
                      that is different from corresponding value in tuple a.
111
 
 
112
 
  EXAMPLES
 
98
  
 
99
  EXAMPLES 
113
100
   Example1: if the function is called for tuples
114
101
     ('aaa','bbb') and ('eee','fff'), then
115
102
     diff_pos[0] = 1 (as 'aaa' != 'eee')
132
119
 
133
120
#define FCMP(A,B) ((int) (A) - (int) (B))
134
121
 
135
 
int ha_key_cmp(register HA_KEYSEG *keyseg, register unsigned char *a,
136
 
               register unsigned char *b, uint32_t key_length, uint32_t nextflag,
137
 
               uint32_t *diff_pos)
 
122
int ha_key_cmp(register HA_KEYSEG *keyseg, register uchar *a,
 
123
               register uchar *b, uint key_length, uint nextflag,
 
124
               uint *diff_pos)
138
125
{
139
126
  int flag;
140
 
  int32_t l_1,l_2;
141
 
  uint32_t u_1,u_2;
 
127
  int16 s_1,s_2;
 
128
  int32 l_1,l_2;
 
129
  uint32 u_1,u_2;
 
130
  float f_1,f_2;
142
131
  double d_1,d_2;
143
 
  uint32_t next_key_length;
144
 
  unsigned char *orig_b= b;
 
132
  uint next_key_length;
 
133
  uchar *orig_b= b;
145
134
 
146
135
  *diff_pos=0;
147
136
  for ( ; (int) key_length >0 ; key_length=next_key_length, keyseg++)
148
137
  {
149
 
    unsigned char *end;
150
 
    uint32_t piks=! (keyseg->flag & HA_NO_SORT);
 
138
    uchar *end;
 
139
    uint piks=! (keyseg->flag & HA_NO_SORT);
151
140
    (*diff_pos)++;
152
141
    diff_pos[1]= (uint)(b - orig_b);
153
142
 
178
167
        continue;                               /* To next key part */
179
168
      }
180
169
    }
181
 
    end= a+ min((uint32_t)keyseg->length,key_length);
 
170
    end= a+ min(keyseg->length,key_length);
182
171
    next_key_length=key_length-keyseg->length;
183
172
 
184
173
    switch ((enum ha_base_keytype) keyseg->type) {
192
181
 
193
182
        if (piks &&
194
183
            (flag=ha_compare_text(keyseg->charset,a,a_length,b,b_length,
195
 
                                  (bool) ((nextflag & SEARCH_PREFIX) &&
 
184
                                  (my_bool) ((nextflag & SEARCH_PREFIX) &&
196
185
                                             next_key_length <= 0),
197
 
                                  (bool)!(nextflag & SEARCH_PREFIX))))
 
186
                                  (my_bool)!(nextflag & SEARCH_PREFIX))))
198
187
          return ((keyseg->flag & HA_REVERSE_SORT) ? -flag : flag);
199
188
        a+=a_length;
200
189
        b+=b_length;
202
191
      }
203
192
      else
204
193
      {
205
 
        uint32_t length=(uint) (end-a), a_length=length, b_length=length;
 
194
        uint length=(uint) (end-a), a_length=length, b_length=length;
206
195
        if (piks &&
207
196
            (flag= ha_compare_text(keyseg->charset, a, a_length, b, b_length,
208
 
                                   (bool) ((nextflag & SEARCH_PREFIX) &&
 
197
                                   (my_bool) ((nextflag & SEARCH_PREFIX) &&
209
198
                                              next_key_length <= 0),
210
 
                                   (bool)!(nextflag & SEARCH_PREFIX))))
 
199
                                   (my_bool)!(nextflag & SEARCH_PREFIX))))
211
200
          return ((keyseg->flag & HA_REVERSE_SORT) ? -flag : flag);
212
201
        a=end;
213
202
        b+=length;
214
203
      }
215
204
      break;
216
205
    case HA_KEYTYPE_BINARY:
 
206
    case HA_KEYTYPE_BIT:
217
207
      if (keyseg->flag & HA_SPACE_PACK)
218
208
      {
219
209
        int a_length,b_length,pack_length;
223
213
 
224
214
        if (piks &&
225
215
            (flag=compare_bin(a,a_length,b,b_length,
226
 
                              (bool) ((nextflag & SEARCH_PREFIX) &&
 
216
                              (my_bool) ((nextflag & SEARCH_PREFIX) &&
227
217
                                         next_key_length <= 0),1)))
228
218
          return ((keyseg->flag & HA_REVERSE_SORT) ? -flag : flag);
229
219
        a+=a_length;
232
222
      }
233
223
      else
234
224
      {
235
 
        uint32_t length=keyseg->length;
 
225
        uint length=keyseg->length;
236
226
        if (piks &&
237
227
            (flag=compare_bin(a,length,b,length,
238
 
                              (bool) ((nextflag & SEARCH_PREFIX) &&
 
228
                              (my_bool) ((nextflag & SEARCH_PREFIX) &&
239
229
                                         next_key_length <= 0),0)))
240
230
          return ((keyseg->flag & HA_REVERSE_SORT) ? -flag : flag);
241
231
        a+=length;
252
242
 
253
243
        if (piks &&
254
244
            (flag= ha_compare_text(keyseg->charset,a,a_length,b,b_length,
255
 
                                   (bool) ((nextflag & SEARCH_PREFIX) &&
 
245
                                   (my_bool) ((nextflag & SEARCH_PREFIX) &&
256
246
                                              next_key_length <= 0),
257
 
                                   (bool) ((nextflag & (SEARCH_FIND |
 
247
                                   (my_bool) ((nextflag & (SEARCH_FIND |
258
248
                                                           SEARCH_UPDATE)) ==
259
249
                                              SEARCH_FIND &&
260
250
                                              ! (keyseg->flag &
264
254
        b+= b_length;
265
255
        break;
266
256
      }
 
257
      break;
267
258
    case HA_KEYTYPE_VARBINARY1:
268
259
    case HA_KEYTYPE_VARBINARY2:
269
260
      {
274
265
 
275
266
        if (piks &&
276
267
            (flag=compare_bin(a,a_length,b,b_length,
277
 
                              (bool) ((nextflag & SEARCH_PREFIX) &&
 
268
                              (my_bool) ((nextflag & SEARCH_PREFIX) &&
278
269
                                         next_key_length <= 0), 0)))
279
270
          return ((keyseg->flag & HA_REVERSE_SORT) ? -flag : flag);
280
271
        a+=a_length;
281
272
        b+=b_length;
282
273
        break;
283
274
      }
 
275
      break;
 
276
    case HA_KEYTYPE_INT8:
 
277
    {
 
278
      int i_1= (int) *((signed char*) a);
 
279
      int i_2= (int) *((signed char*) b);
 
280
      if (piks && (flag = CMP_NUM(i_1,i_2)))
 
281
        return ((keyseg->flag & HA_REVERSE_SORT) ? -flag : flag);
 
282
      a= end;
 
283
      b++;
 
284
      break;
 
285
    }
 
286
    case HA_KEYTYPE_SHORT_INT:
 
287
      s_1= mi_sint2korr(a);
 
288
      s_2= mi_sint2korr(b);
 
289
      if (piks && (flag = CMP_NUM(s_1,s_2)))
 
290
        return ((keyseg->flag & HA_REVERSE_SORT) ? -flag : flag);
 
291
      a=  end;
 
292
      b+= 2; /* sizeof(short int); */
 
293
      break;
 
294
    case HA_KEYTYPE_USHORT_INT:
 
295
      {
 
296
        uint16 us_1,us_2;
 
297
        us_1= mi_sint2korr(a);
 
298
        us_2= mi_sint2korr(b);
 
299
        if (piks && (flag = CMP_NUM(us_1,us_2)))
 
300
          return ((keyseg->flag & HA_REVERSE_SORT) ? -flag : flag);
 
301
        a=  end;
 
302
        b+=2; /* sizeof(short int); */
 
303
        break;
 
304
      }
284
305
    case HA_KEYTYPE_LONG_INT:
285
306
      l_1= mi_sint4korr(a);
286
307
      l_2= mi_sint4korr(b);
297
318
      a=  end;
298
319
      b+= 4; /* sizeof(long int); */
299
320
      break;
 
321
    case HA_KEYTYPE_INT24:
 
322
      l_1=mi_sint3korr(a);
 
323
      l_2=mi_sint3korr(b);
 
324
      if (piks && (flag = CMP_NUM(l_1,l_2)))
 
325
        return ((keyseg->flag & HA_REVERSE_SORT) ? -flag : flag);
 
326
      a=  end;
 
327
      b+= 3;
 
328
      break;
300
329
    case HA_KEYTYPE_UINT24:
301
330
      l_1=mi_uint3korr(a);
302
331
      l_2=mi_uint3korr(b);
305
334
      a=  end;
306
335
      b+= 3;
307
336
      break;
 
337
    case HA_KEYTYPE_FLOAT:
 
338
      mi_float4get(f_1,a);
 
339
      mi_float4get(f_2,b);
 
340
      /*
 
341
        The following may give a compiler warning about floating point
 
342
        comparison not being safe, but this is ok in this context as
 
343
        we are bascily doing sorting
 
344
      */
 
345
      if (piks && (flag = CMP_NUM(f_1,f_2)))
 
346
        return ((keyseg->flag & HA_REVERSE_SORT) ? -flag : flag);
 
347
      a=  end;
 
348
      b+= 4; /* sizeof(float); */
 
349
      break;
308
350
    case HA_KEYTYPE_DOUBLE:
309
351
      mi_float8get(d_1,a);
310
352
      mi_float8get(d_2,b);
318
360
      a=  end;
319
361
      b+= 8;  /* sizeof(double); */
320
362
      break;
 
363
    case HA_KEYTYPE_NUM:                                /* Numeric key */
 
364
    {
 
365
      int swap_flag= 0;
 
366
      int alength,blength;
 
367
 
 
368
      if (keyseg->flag & HA_REVERSE_SORT)
 
369
      {
 
370
        swap_variables(uchar*, a, b);
 
371
        swap_flag=1;                            /* Remember swap of a & b */
 
372
        end= a+ (int) (end-b);
 
373
      }
 
374
      if (keyseg->flag & HA_SPACE_PACK)
 
375
      {
 
376
        alength= *a++; blength= *b++;
 
377
        end=a+alength;
 
378
        next_key_length=key_length-blength-1;
 
379
      }
 
380
      else
 
381
      {
 
382
        alength= (int) (end-a);
 
383
        blength=keyseg->length;
 
384
        /* remove pre space from keys */
 
385
        for ( ; alength && *a == ' ' ; a++, alength--) ;
 
386
        for ( ; blength && *b == ' ' ; b++, blength--) ;
 
387
      }
 
388
      if (piks)
 
389
      {
 
390
        if (*a == '-')
 
391
        {
 
392
          if (*b != '-')
 
393
            return -1;
 
394
          a++; b++;
 
395
          swap_variables(uchar*, a, b);
 
396
          swap_variables(int, alength, blength);
 
397
          swap_flag=1-swap_flag;
 
398
          alength--; blength--;
 
399
          end=a+alength;
 
400
        }
 
401
        else if (*b == '-')
 
402
          return 1;
 
403
        while (alength && (*a == '+' || *a == '0'))
 
404
        {
 
405
          a++; alength--;
 
406
        }
 
407
        while (blength && (*b == '+' || *b == '0'))
 
408
        {
 
409
          b++; blength--;
 
410
        }
 
411
        if (alength != blength)
 
412
          return (alength < blength) ? -1 : 1;
 
413
        while (a < end)
 
414
          if (*a++ !=  *b++)
 
415
            return ((int) a[-1] - (int) b[-1]);
 
416
      }
 
417
      else
 
418
      {
 
419
        b+=(end-a);
 
420
        a=end;
 
421
      }
 
422
 
 
423
      if (swap_flag)                            /* Restore pointers */
 
424
        swap_variables(uchar*, a, b);
 
425
      break;
 
426
    }
321
427
    case HA_KEYTYPE_LONGLONG:
322
428
    {
323
 
      int64_t ll_a,ll_b;
 
429
      longlong ll_a,ll_b;
324
430
      ll_a= mi_sint8korr(a);
325
431
      ll_b= mi_sint8korr(b);
326
432
      if (piks && (flag = CMP_NUM(ll_a,ll_b)))
331
437
    }
332
438
    case HA_KEYTYPE_ULONGLONG:
333
439
    {
334
 
      uint64_t ll_a,ll_b;
 
440
      ulonglong ll_a,ll_b;
335
441
      ll_a= mi_uint8korr(a);
336
442
      ll_b= mi_uint8korr(b);
337
443
      if (piks && (flag = CMP_NUM(ll_a,ll_b)))
348
454
end:
349
455
  if (!(nextflag & SEARCH_FIND))
350
456
  {
351
 
    uint32_t i;
 
457
    uint i;
352
458
    if (nextflag & (SEARCH_NO_FIND | SEARCH_LAST)) /* Find record after key */
353
459
      return (nextflag & (SEARCH_BIGGER | SEARCH_LAST)) ? -1 : 1;
354
460
    flag=0;
391
497
    NULLs.
392
498
*/
393
499
 
394
 
HA_KEYSEG *ha_find_null(HA_KEYSEG *keyseg, unsigned char *a)
 
500
HA_KEYSEG *ha_find_null(HA_KEYSEG *keyseg, uchar *a)
395
501
{
396
502
  for (; (enum ha_base_keytype) keyseg->type != HA_KEYTYPE_END; keyseg++)
397
503
  {
398
 
    unsigned char *end;
 
504
    uchar *end;
399
505
    if (keyseg->null_bit)
400
506
    {
401
507
      if (!*a++)
406
512
    switch ((enum ha_base_keytype) keyseg->type) {
407
513
    case HA_KEYTYPE_TEXT:
408
514
    case HA_KEYTYPE_BINARY:
 
515
    case HA_KEYTYPE_BIT:
409
516
      if (keyseg->flag & HA_SPACE_PACK)
410
517
      {
411
518
        int a_length;
426
533
        a+= a_length;
427
534
        break;
428
535
      }
 
536
    case HA_KEYTYPE_NUM:
 
537
      if (keyseg->flag & HA_SPACE_PACK)
 
538
      {
 
539
        int alength= *a++;
 
540
        end= a+alength;
 
541
      }
 
542
      a= end;
 
543
      break;
 
544
    case HA_KEYTYPE_INT8:
 
545
    case HA_KEYTYPE_SHORT_INT:
 
546
    case HA_KEYTYPE_USHORT_INT:
429
547
    case HA_KEYTYPE_LONG_INT:
430
548
    case HA_KEYTYPE_ULONG_INT:
 
549
    case HA_KEYTYPE_INT24:
431
550
    case HA_KEYTYPE_UINT24:
432
551
    case HA_KEYTYPE_LONGLONG:
433
552
    case HA_KEYTYPE_ULONGLONG:
 
553
    case HA_KEYTYPE_FLOAT:
434
554
    case HA_KEYTYPE_DOUBLE:
435
555
      a= end;
436
556
      break;
437
 
    case HA_KEYTYPE_END:
 
557
    case HA_KEYTYPE_END:                        /* purecov: inspected */
438
558
      /* keep compiler happy */
439
 
      assert(0);
 
559
      DBUG_ASSERT(0);
440
560
      break;
441
561
    }
442
562
  }
445
565
 
446
566
 
447
567
 
 
568
/*
 
569
  Register handler error messages for usage with my_error()
 
570
 
 
571
  NOTES
 
572
    This is safe to call multiple times as my_error_register()
 
573
    will ignore calls to register already registered error numbers.
 
574
*/
 
575
 
 
576
 
 
577
void my_handler_error_register(void)
 
578
{
 
579
  /*
 
580
    If you got compilation error here about compile_time_assert array, check
 
581
    that every HA_ERR_xxx constant has a corresponding error message in
 
582
    handler_error_messages[] list (check mysys/ma_handler_errors.h and
 
583
    include/my_base.h).
 
584
  */
 
585
  compile_time_assert(HA_ERR_FIRST + array_elements(handler_error_messages) ==
 
586
                      HA_ERR_LAST + 1);
 
587
  my_error_register(handler_error_messages, HA_ERR_FIRST,
 
588
                    HA_ERR_FIRST+ array_elements(handler_error_messages)-1);
 
589
}
 
590
 
 
591
 
 
592
void my_handler_error_unregister(void)
 
593
{
 
594
  my_error_unregister(HA_ERR_FIRST,
 
595
                      HA_ERR_FIRST+ array_elements(handler_error_messages)-1);
 
596
}