← 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: Monty Taylor
  • Date: 2008-08-02 00:06:32 UTC
  • mto: (236.1.42 codestyle)
  • mto: This revision was merged to the branch mainline in revision 261.
  • Revision ID: monty@inaugust.com-20080802000632-jsse0zdd9r6ic5ku
Actually turn gettext on...

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 */