← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/function/find_in_set.cc

  • Committer: Brian Aker
  • Date: 2010-02-10 18:04:24 UTC
  • mfrom: (1286.1.5 build)
  • Revision ID: brian@gaz-20100210180424-03ypoyifmlc2lgcp
Merge of Brian/Padraig

Show diffs side-by-side

added added

removed removed

Lines of Context:
drizzled/function/find_in_set.cc
 
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/function/find_in_set.h>
 
23
 
 
24
/* Search after a string in a string of strings separated by ',' */
 
25
/* Returns number of found type >= 1 or 0 if not found */
 
26
/* This optimizes searching in enums to bit testing! */
 
27
 
 
28
namespace drizzled
 
29
{
 
30
 
 
31
void Item_func_find_in_set::fix_length_and_dec()
 
32
{
 
33
  decimals=0;
 
34
  max_length=3;                                 // 1-999
 
35
  agg_arg_charsets(cmp_collation, args, 2, MY_COLL_CMP_CONV, 1);
 
36
}
 
37
 
 
38
static const char separator=',';
 
39
 
 
40
int64_t Item_func_find_in_set::val_int()
 
41
{
 
42
  assert(fixed == 1);
 
43
  if (enum_value)
 
44
  {
 
45
    uint64_t tmp=(uint64_t) args[1]->val_int();
 
46
    if (!(null_value=args[1]->null_value || args[0]->null_value))
 
47
    {
 
48
      if (tmp & enum_bit)
 
49
        return enum_value;
 
50
    }
 
51
    return 0L;
 
52
  }
 
53
 
 
54
  String *find=args[0]->val_str(&value);
 
55
  String *buffer=args[1]->val_str(&value2);
 
56
  if (!find || !buffer)
 
57
  {
 
58
    null_value=1;
 
59
    return 0;
 
60
  }
 
61
  null_value=0;
 
62
 
 
63
  int diff;
 
64
  if ((diff=buffer->length() - find->length()) >= 0)
 
65
  {
 
66
    my_wc_t wc;
 
67
    const CHARSET_INFO * const cs= cmp_collation.collation;
 
68
    const char *str_begin= buffer->ptr();
 
69
    const char *str_end= buffer->ptr();
 
70
    const char *real_end= str_end+buffer->length();
 
71
    const unsigned char *find_str= (const unsigned char *) find->ptr();
 
72
    uint32_t find_str_len= find->length();
 
73
    int position= 0;
 
74
    while (1)
 
75
    {
 
76
      int symbol_len;
 
77
      if ((symbol_len= cs->cset->mb_wc(cs, &wc, (unsigned char*) str_end,
 
78
                                       (unsigned char*) real_end)) > 0)
 
79
      {
 
80
        const char *substr_end= str_end + symbol_len;
 
81
        bool is_last_item= (substr_end == real_end);
 
82
        bool is_separator= (wc == (my_wc_t) separator);
 
83
        if (is_separator || is_last_item)
 
84
        {
 
85
          position++;
 
86
          if (is_last_item && !is_separator)
 
87
            str_end= substr_end;
 
88
          if (!my_strnncoll(cs, (const unsigned char *) str_begin,
 
89
                            str_end - str_begin,
 
90
                            find_str, find_str_len))
 
91
            return (int64_t) position;
 
92
          else
 
93
            str_begin= substr_end;
 
94
        }
 
95
        str_end= substr_end;
 
96
      }
 
97
      else if (str_end - str_begin == 0 &&
 
98
               find_str_len == 0 &&
 
99
               wc == (my_wc_t) separator)
 
100
        return (int64_t) ++position;
 
101
      else
 
102
        return 0L;
 
103
    }
 
104
  }
 
105
  return 0;
 
106
}
 
107
 
 
108
} /* namespace drizzled */