← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/sql_string.h

  • Committer: Mark Atwood
  • Date: 2011-10-27 05:08:12 UTC
  • mfrom: (2445.1.11 rf)
  • Revision ID: me@mark.atwood.name-20111027050812-1icvs72lb0u4xdc4
merge lp:~olafvdspek/drizzle/refactor8

Show diffs side-by-side

added added

removed removed

Lines of Context:
drizzled/sql_string.h
17
17
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
18
18
 */
19
19
 
20
 
#ifndef DRIZZLED_SQL_STRING_H
21
 
#define DRIZZLED_SQL_STRING_H
 
20
#pragma once
22
21
 
23
22
/* This file is originally from the mysql distribution. Coded by monty */
24
23
 
25
24
#include <drizzled/common.h>
 
25
#include <drizzled/util/data_ref.h>
 
26
 
26
27
#include <cassert>
27
28
#include <cstdlib>
28
29
#include <cstring>
29
30
#include <string>
30
31
 
 
32
#include <drizzled/visibility.h>
 
33
 
31
34
#ifndef NOT_FIXED_DEC
32
35
#define NOT_FIXED_DEC                   (uint8_t)31
33
36
#endif
34
37
 
35
 
namespace drizzled
36
 
{
37
 
 
38
 
class String;
39
 
 
40
 
extern String my_empty_string;
 
38
namespace drizzled {
 
39
 
 
40
extern DRIZZLED_API String my_empty_string;
41
41
extern const String my_null_string;
42
 
namespace memory { class Root; }
43
 
typedef struct charset_info_st CHARSET_INFO;
44
 
 
45
 
std::string String_to_std_string(String const& s);
46
 
String* set_String_from_std_string(String* s, std::string const& cs);
47
 
 
48
 
int sortcmp(const String *a,const String *b, const CHARSET_INFO * const cs);
 
42
 
 
43
int sortcmp(const String *a,const String *b, const charset_info_st * const cs);
49
44
int stringcmp(const String *a,const String *b);
50
45
String *copy_if_not_alloced(String *a,String *b,size_t arg_length);
51
 
size_t well_formed_copy_nchars(const CHARSET_INFO * const to_cs,
 
46
size_t well_formed_copy_nchars(const charset_info_st * const to_cs,
52
47
                                 char *to, size_t to_length,
53
 
                                 const CHARSET_INFO * const from_cs,
 
48
                                 const charset_info_st * const from_cs,
54
49
                                 const char *from, size_t from_length,
55
50
                                 size_t nchars,
56
51
                                 const char **well_formed_error_pos,
58
53
                                 const char **from_end_pos);
59
54
 
60
55
 
61
 
class String
 
56
class DRIZZLED_API String
62
57
{
63
58
  char *Ptr;
64
59
  size_t str_length,Alloced_length;
65
60
  bool alloced;
66
 
  const CHARSET_INFO *str_charset;
 
61
  const charset_info_st *str_charset;
67
62
 
68
63
public:
69
64
  String();
70
65
  String(size_t length_arg);
71
 
  String(const char *str, const CHARSET_INFO * const cs);
72
 
  String(const char *str, size_t len, const CHARSET_INFO * const cs);
73
 
  String(char *str, size_t len, const CHARSET_INFO * const cs);
74
 
  String(const String &str);
 
66
  String(const char*, const charset_info_st*);
 
67
  String(const char*, size_t, const charset_info_st*);
 
68
  String(char *str, size_t len, const charset_info_st*);
 
69
  String(str_ref, const charset_info_st*);
 
70
  String(const String&);
75
71
 
76
72
  static void *operator new(size_t size, memory::Root *mem_root);
77
73
  static void operator delete(void *, size_t)
80
76
  { }
81
77
  ~String();
82
78
 
83
 
  inline void set_charset(const CHARSET_INFO * const charset_arg)
 
79
  inline void set_charset(const charset_info_st * const charset_arg)
84
80
  { str_charset= charset_arg; }
85
 
  inline const CHARSET_INFO *charset() const { return str_charset; }
 
81
  inline const charset_info_st *charset() const { return str_charset; }
86
82
  inline size_t length() const { return str_length;}
87
83
  inline size_t alloced_length() const { return Alloced_length;}
88
84
  inline char& operator [] (size_t i) const { return Ptr[i]; }
89
 
  inline void length(size_t len) { str_length=len ; }
90
 
  inline bool is_empty() { return (str_length == 0); }
91
 
  inline void mark_as_const() { Alloced_length= 0;}
 
85
  inline void length(size_t len) { str_length=len; }
 
86
  inline bool empty() const { return str_length == 0; }
 
87
  inline void mark_as_const() { Alloced_length= 0; }
92
88
  inline char *ptr() { return Ptr; }
93
89
  inline const char *ptr() const { return Ptr; }
94
90
  inline char *c_ptr()
95
91
  {
96
92
    if (str_length == Alloced_length)
97
 
      (void) realloc(str_length);
 
93
      realloc(str_length);
98
94
    else
99
95
      Ptr[str_length]= 0;
100
96
 
101
97
    return Ptr;
102
98
  }
103
 
  inline char *c_ptr_quick()
104
 
  {
105
 
    if (Ptr && str_length < Alloced_length)
106
 
      Ptr[str_length]=0;
107
 
    return Ptr;
108
 
  }
109
 
  inline char *c_ptr_safe()
110
 
  {
111
 
    if (Ptr && str_length < Alloced_length)
112
 
      Ptr[str_length]=0;
113
 
    else
114
 
      (void) realloc(str_length);
115
 
    return Ptr;
116
 
  }
117
 
  inline char *c_str()
118
 
  {
119
 
    if (Ptr && str_length < Alloced_length)
120
 
      Ptr[str_length]=0;
121
 
    else
122
 
      (void) realloc(str_length);
 
99
  inline const char* begin() const
 
100
  {
 
101
    return Ptr;
 
102
  }
 
103
  inline const char* end() const
 
104
  {
 
105
    return begin() + size();
 
106
  }
 
107
  inline const char* data() const
 
108
  {
 
109
    return Ptr;
 
110
  }
 
111
  inline size_t size() const
 
112
  {
 
113
    return length();
 
114
  }
 
115
  inline const char* c_str()
 
116
  {
 
117
    if (Ptr && str_length < Alloced_length)
 
118
      Ptr[str_length]=0;
 
119
    else
 
120
      realloc(str_length);
123
121
    return Ptr;
124
122
  }
125
123
  void append_identifier(const char *name, size_t length);
 
124
  void append_identifier(str_ref);
126
125
 
127
126
  void set(String &str,size_t offset,size_t arg_length)
128
127
  {
135
134
      Alloced_length=0;
136
135
    str_charset=str.str_charset;
137
136
  }
138
 
  inline void set(char *str,size_t arg_length, const CHARSET_INFO * const cs)
 
137
  inline void set(char *str,size_t arg_length, const charset_info_st * const cs)
139
138
  {
140
139
    free();
141
140
    Ptr= str; str_length=Alloced_length=arg_length ; alloced=0;
142
141
    str_charset=cs;
143
142
  }
144
 
  inline void set(const char *str,size_t arg_length, const CHARSET_INFO * const cs)
 
143
 
 
144
  inline void set(const char *str,size_t arg_length, const charset_info_st * const cs)
145
145
  {
146
146
    free();
147
147
    Ptr= const_cast<char*>(str);
148
148
    str_length=arg_length; Alloced_length=0 ; alloced=0;
149
149
    str_charset=cs;
150
150
  }
151
 
  bool set_ascii(const char *str, size_t arg_length);
152
 
  inline void set_quick(char *str,size_t arg_length, const CHARSET_INFO * const cs)
 
151
  void set_ascii(const char *str, size_t arg_length);
 
152
 
 
153
  inline void set_quick(char *str,size_t arg_length, const charset_info_st * const cs)
153
154
  {
154
155
    if (!alloced)
155
156
    {
157
158
    }
158
159
    str_charset= cs;
159
160
  }
160
 
  bool set_int(int64_t num, bool unsigned_flag, const CHARSET_INFO * const cs);
161
 
  bool set(int64_t num, const CHARSET_INFO * const cs)
162
 
  { return set_int(num, false, cs); }
163
 
  bool set(uint64_t num, const CHARSET_INFO * const cs)
164
 
  { return set_int(static_cast<int64_t>(num), true, cs); }
165
 
  bool set_real(double num,size_t decimals, const CHARSET_INFO * const cs);
 
161
 
 
162
  void set_int(int64_t num, bool unsigned_flag, const charset_info_st * const cs);
 
163
  void set(int64_t num, const charset_info_st * const cs)
 
164
  { set_int(num, false, cs); }
 
165
  void set(uint64_t num, const charset_info_st * const cs)
 
166
  { set_int(static_cast<int64_t>(num), true, cs); }
 
167
  void set_real(double num,size_t decimals, const charset_info_st* cs);
166
168
 
167
169
  /*
168
170
    PMG 2004.11.12
202
204
      str_length=0;                             /* Safety */
203
205
    }
204
206
  }
205
 
  inline bool alloc(size_t arg_length)
 
207
  inline void alloc(size_t arg_length)
206
208
  {
207
 
    if (arg_length < Alloced_length)
208
 
      return 0;
209
 
    return real_alloc(arg_length);
 
209
    if (arg_length >= Alloced_length)
 
210
      real_alloc(arg_length);
210
211
  }
211
 
  bool real_alloc(size_t arg_length);                   // Empties old string
212
 
  bool realloc(size_t arg_length);
 
212
  void real_alloc(size_t arg_length);                   // Empties old string
 
213
  void realloc(size_t arg_length);
213
214
  inline void shrink(size_t arg_length)         // Shrink buffer
214
215
  {
215
216
    if (arg_length < Alloced_length)
217
218
      char *new_ptr;
218
219
      if (!(new_ptr= reinterpret_cast<char*>(::realloc(Ptr,arg_length))))
219
220
      {
220
 
        Alloced_length = 0;
221
 
        real_alloc(arg_length);
 
221
        Alloced_length = 0;
 
222
        real_alloc(arg_length);
222
223
      }
223
224
      else
224
225
      {
225
 
        Ptr=new_ptr;
226
 
        Alloced_length=arg_length;
 
226
        Ptr=new_ptr;
 
227
        Alloced_length=arg_length;
227
228
      }
228
229
    }
229
230
  }
230
 
  bool is_alloced() { return alloced; }
 
231
  bool is_alloced() { return alloced; } const
231
232
  inline String& operator = (const String &s)
232
233
  {
233
234
    if (&s != this)
244
245
    return *this;
245
246
  }
246
247
 
247
 
  bool copy();                                  // Alloc string if not alloced
248
 
  bool copy(const String &s);                   // Allocate new string
249
 
  bool copy(const char *s,size_t arg_length, const CHARSET_INFO * const cs);    // Allocate new string
250
 
  static bool needs_conversion(size_t arg_length,
251
 
                               const CHARSET_INFO * const cs_from, const CHARSET_INFO * const cs_to,
252
 
                               size_t *offset);
253
 
  bool set_or_copy_aligned(const char *s, size_t arg_length, const CHARSET_INFO * const cs);
254
 
  bool copy(const char*s,size_t arg_length, const CHARSET_INFO * const csfrom,
255
 
            const CHARSET_INFO * const csto, size_t *errors);
256
 
  bool append(const String &s);
257
 
  bool append(const char *s);
258
 
  bool append(const char *s,size_t arg_length);
259
 
  bool append(const char *s,size_t arg_length, const CHARSET_INFO * const cs);
260
 
  bool append_with_prefill(const char *s, size_t arg_length,
261
 
                           size_t full_length, char fill_char);
 
248
  void copy();                                  // Alloc string if not alloced
 
249
  void copy(const String&);                     // Allocate new string
 
250
  void copy(const std::string&, const charset_info_st*);        // Allocate new string
 
251
  void copy(const char*, size_t, const charset_info_st*); // Allocate new string
 
252
  static bool needs_conversion(size_t arg_length, const charset_info_st* cs_from, const charset_info_st* cs_to);
 
253
  void set_or_copy_aligned(const char *s, size_t arg_length, const charset_info_st*);
 
254
  void copy(const char*s,size_t arg_length, const charset_info_st& csto);
 
255
  void append(const char*);
 
256
  void append(const char*, size_t);
 
257
  void append(str_ref);
 
258
  void append_with_prefill(const char *s, size_t arg_length, size_t full_length, char fill_char);
262
259
  int strstr(const String &search,size_t offset=0); // Returns offset to substring or -1
263
260
  int strrstr(const String &search,size_t offset=0); // Returns offset to substring or -1
264
 
  bool replace(size_t offset,size_t arg_length,const char *to,size_t length);
265
 
  bool replace(size_t offset,size_t arg_length,const String &to);
266
 
  inline bool append(char chr)
 
261
  void replace(size_t offset,size_t arg_length,const char *to,size_t length);
 
262
  void replace(size_t offset,size_t arg_length,const String &to);
 
263
 
 
264
  inline void append(char chr)
267
265
  {
268
266
    if (str_length < Alloced_length)
269
267
    {
271
269
    }
272
270
    else
273
271
    {
274
 
      if (realloc(str_length+1))
275
 
        return 1;
 
272
      realloc(str_length+1);
276
273
      Ptr[str_length++]=chr;
277
274
    }
278
 
    return 0;
279
275
  }
280
 
  friend int sortcmp(const String *a,const String *b, const CHARSET_INFO * const cs);
 
276
  friend int sortcmp(const String *a,const String *b, const charset_info_st * const cs);
281
277
  friend int stringcmp(const String *a,const String *b);
282
278
  friend String *copy_if_not_alloced(String *a,String *b,size_t arg_length);
283
 
  size_t numchars();
284
 
  int charpos(int i,size_t offset=0);
 
279
  size_t numchars() const;
 
280
  int charpos(int i, size_t offset= 0) const;
285
281
 
286
 
  int reserve(size_t space_needed)
 
282
  void reserve(size_t space_needed)
287
283
  {
288
 
    return realloc(str_length + space_needed);
 
284
    realloc(str_length + space_needed);
289
285
  }
290
 
  int reserve(size_t space_needed, size_t grow_by);
291
 
 
292
 
  /*
293
 
    The following append operations do NOT check alloced memory
294
 
    q_*** methods writes values of parameters itself
295
 
    qs_*** methods writes string representation of value
296
 
  */
297
 
  void q_append(const char c);
298
 
  void q_append(const size_t n);
299
 
  void q_append(double d);
300
 
  void q_append(double *d);
301
 
  void q_append(const char *data, size_t data_len);
302
 
  void write_at_position(int position, size_t value);
303
 
 
304
 
  /* Inline (general) functions used by the protocol functions */
305
 
 
306
 
  inline char *prep_append(size_t arg_length, size_t step_alloc)
 
286
  void reserve(size_t space_needed, size_t grow_by);
 
287
 
 
288
  inline void append(const char *s, size_t arg_length, size_t step_alloc)
307
289
  {
308
290
    size_t new_length= arg_length + str_length;
309
291
    if (new_length > Alloced_length)
310
 
    {
311
 
      if (realloc(new_length + step_alloc))
312
 
        return 0;
313
 
    }
314
 
    size_t old_length= str_length;
315
 
    str_length+= arg_length;
316
 
    return Ptr+ old_length;                     /* Area to use */
317
 
  }
318
 
 
319
 
  inline bool append(const char *s, size_t arg_length, size_t step_alloc)
320
 
  {
321
 
    size_t new_length= arg_length + str_length;
322
 
    if (new_length > Alloced_length && realloc(new_length + step_alloc))
323
 
      return true;
 
292
                        realloc(new_length + step_alloc);
324
293
    memcpy(Ptr+str_length, s, arg_length);
325
294
    str_length+= arg_length;
326
 
    return false;
327
295
  }
328
 
  void print(String *print);
 
296
 
 
297
  void print(String&) const;
329
298
 
330
299
  /* Swap two string objects. Efficient way to exchange data without memcpy. */
331
300
  void swap(String &s);
336
305
  }
337
306
};
338
307
 
339
 
bool check_if_only_end_space(const CHARSET_INFO * const cs, char *str,
340
 
                             char *end);
 
308
bool check_if_only_end_space(const charset_info_st* const, char *str, char *end);
 
309
 
 
310
std::ostream& operator<<(std::ostream&, const String&);
341
311
 
342
312
} /* namespace drizzled */
343
313
 
344
314
bool operator==(const drizzled::String &s1, const drizzled::String &s2);
345
315
bool operator!=(const drizzled::String &s1, const drizzled::String &s2);
346
 
 
347
 
 
348
 
#endif /* DRIZZLED_SQL_STRING_H */