← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/identifier/table.cc

  • Committer: Brian Aker
  • Date: 2010-10-24 01:34:39 UTC
  • mto: This revision was merged to the branch mainline in revision 1875.
  • Revision ID: brian@tangent.org-20101024013439-w9r8qtmt5ihnqtkd
Encapsulate table_name from table_list.

Show diffs side-by-side

added added

removed removed

Lines of Context:
drizzled/identifier/table.cc
29
29
#include "drizzled/table.h"
30
30
 
31
31
#include "drizzled/util/string.h"
32
 
#include "drizzled/util/tablename_to_filename.h"
33
32
 
34
33
#include <algorithm>
35
34
#include <sstream>
42
41
namespace drizzled
43
42
{
44
43
 
45
 
class SchemaIdentifier;
46
 
 
47
44
extern std::string drizzle_tmpdir;
48
45
extern pid_t current_pid;
49
46
 
50
47
static const char hexchars[]= "0123456789abcdef";
51
48
 
 
49
static bool tablename_to_filename(const string &from, string &to);
 
50
 
52
51
/*
53
52
  Translate a cursor name to a table name (WL #1324).
54
53
 
204
203
    path length on success, 0 on failure
205
204
*/
206
205
 
207
 
size_t TableIdentifier::build_table_filename(std::string &in_path, const std::string &in_db, const std::string &in_table_name, bool is_tmp)
 
206
size_t TableIdentifier::build_table_filename(std::string &path, const std::string &db, const std::string &table_name, bool is_tmp)
208
207
{
209
208
  bool conversion_error= false;
210
209
 
211
 
  conversion_error= util::tablename_to_filename(in_db, in_path);
 
210
  conversion_error= tablename_to_filename(db, path);
212
211
  if (conversion_error)
213
212
  {
214
213
    errmsg_printf(ERRMSG_LVL_ERROR,
217
216
    return 0;
218
217
  }
219
218
 
220
 
  in_path.append(FN_ROOTDIR);
 
219
  path.append(FN_ROOTDIR);
221
220
 
222
221
  if (is_tmp) // It a conversion tmp
223
222
  {
224
 
    in_path.append(in_table_name);
 
223
    path.append(table_name);
225
224
  }
226
225
  else
227
226
  {
228
 
    conversion_error= util::tablename_to_filename(in_table_name, in_path);
 
227
    conversion_error= tablename_to_filename(table_name, path);
229
228
    if (conversion_error)
230
229
    {
231
230
      errmsg_printf(ERRMSG_LVL_ERROR,
235
234
    }
236
235
  }
237
236
   
238
 
  return in_path.length();
 
237
  return path.length();
 
238
}
 
239
 
 
240
 
 
241
/*
 
242
  Translate a table name to a cursor name (WL #1324).
 
243
 
 
244
  SYNOPSIS
 
245
    tablename_to_filename()
 
246
      from                      The table name
 
247
      to                OUT     The cursor name
 
248
      to_length                 The size of the cursor name buffer.
 
249
 
 
250
  RETURN
 
251
    true if errors happen. false on success.
 
252
*/
 
253
static bool tablename_to_filename(const string &from, string &to)
 
254
{
 
255
  
 
256
  string::const_iterator iter= from.begin();
 
257
  for (; iter != from.end(); ++iter)
 
258
  {
 
259
    if (isascii(*iter))
 
260
    {
 
261
      if ((isdigit(*iter)) ||
 
262
          (islower(*iter)) ||
 
263
          (*iter == '_') ||
 
264
          (*iter == ' ') ||
 
265
          (*iter == '-'))
 
266
      {
 
267
        to.push_back(*iter);
 
268
        continue;
 
269
      }
 
270
 
 
271
      if (isupper(*iter))
 
272
      {
 
273
        to.push_back(tolower(*iter));
 
274
        continue;
 
275
      }
 
276
    }
 
277
   
 
278
    /* We need to escape this char in a way that can be reversed */
 
279
    to.push_back('@');
 
280
    to.push_back(hexchars[(*iter >> 4) & 15]);
 
281
    to.push_back(hexchars[(*iter) & 15]);
 
282
  }
 
283
 
 
284
  if (internal::check_if_legal_tablename(to.c_str()))
 
285
  {
 
286
    to.append("@@@");
 
287
  }
 
288
  return false;
239
289
}
240
290
 
241
291
TableIdentifier::TableIdentifier(const drizzled::Table &table) :
272
322
  util::insensitive_hash hasher;
273
323
  hash_value= hasher(path);
274
324
 
275
 
  std::string tb_name(getTableName());
276
 
  std::transform(tb_name.begin(), tb_name.end(), tb_name.begin(), ::tolower);
 
325
  key.resize(getKeySize());
277
326
 
278
 
  key.set(getKeySize(), getSchemaName(), tb_name);
 
327
  std::copy(getSchemaName().begin(), getSchemaName().end(), key.begin());
 
328
  std::copy(getTableName().begin(), getTableName().end(), key.begin() + getSchemaName().length() + 1);
279
329
}
280
330
 
281
331
 
284
334
  return path;
285
335
}
286
336
 
287
 
void TableIdentifier::getSQLPath(std::string &sql_path) const  // @todo this is just used for errors, we should find a way to optimize it
288
 
{
289
 
  switch (type) {
290
 
  case message::Table::FUNCTION:
291
 
  case message::Table::STANDARD:
292
 
    sql_path.append(getSchemaName());
293
 
    sql_path.append(".");
294
 
    sql_path.append(table_name);
295
 
    break;
296
 
  case message::Table::INTERNAL:
297
 
    sql_path.append("temporary.");
298
 
    sql_path.append(table_name);
299
 
    break;
300
 
  case message::Table::TEMPORARY:
301
 
    sql_path.append(getSchemaName());
302
 
    sql_path.append(".#");
303
 
    sql_path.append(table_name);
304
 
    break;
305
 
  }
306
 
}
307
 
 
308
 
bool TableIdentifier::isValid() const
309
 
{
310
 
  if (not SchemaIdentifier::isValid())
311
 
    return false;
312
 
 
313
 
  bool error= false;
314
 
  do
315
 
  {
316
 
    if (table_name.empty())
317
 
    {
318
 
      error= true;
319
 
      break;
320
 
    }
321
 
 
322
 
    if (table_name.size() > NAME_LEN)
323
 
    {
324
 
      error= true;
325
 
      break;
326
 
    }
327
 
 
328
 
    if (table_name.at(table_name.length() -1) == ' ')
329
 
    {
330
 
      error= true;
331
 
      break;
332
 
    }
333
 
 
334
 
    if (table_name.at(0) == '.')
335
 
    {
336
 
      error= true;
337
 
      break;
338
 
    }
339
 
 
340
 
    {
341
 
      const CHARSET_INFO * const cs= &my_charset_utf8mb4_general_ci;
342
 
 
343
 
      int well_formed_error;
344
 
      uint32_t res= cs->cset->well_formed_len(cs, table_name.c_str(), table_name.c_str() + table_name.length(),
345
 
                                              NAME_CHAR_LEN, &well_formed_error);
346
 
      if (well_formed_error or table_name.length() != res)
347
 
      {
348
 
        error= true;
349
 
        break;
350
 
      }
351
 
    }
352
 
  } while (0);
353
 
 
354
 
  if (error)
355
 
  {
356
 
    std::string name;
357
 
 
358
 
    getSQLPath(name);
359
 
    my_error(ER_WRONG_TABLE_NAME, MYF(0), name.c_str());
360
 
 
361
 
    return false;
362
 
  }
363
 
 
364
 
  return true;
 
337
const std::string &TableIdentifier::getSQLPath()  // @todo this is just used for errors, we should find a way to optimize it
 
338
{
 
339
  if (sql_path.empty())
 
340
  {
 
341
    switch (type) {
 
342
    case message::Table::FUNCTION:
 
343
    case message::Table::STANDARD:
 
344
      sql_path.append(getSchemaName());
 
345
      sql_path.append(".");
 
346
      sql_path.append(table_name);
 
347
      break;
 
348
    case message::Table::INTERNAL:
 
349
      sql_path.append("temporary.");
 
350
      sql_path.append(table_name);
 
351
      break;
 
352
    case message::Table::TEMPORARY:
 
353
      sql_path.append(getSchemaName());
 
354
      sql_path.append(".#");
 
355
      sql_path.append(table_name);
 
356
      break;
 
357
    }
 
358
  }
 
359
 
 
360
  return sql_path;
365
361
}
366
362
 
367
363
 
371
367
  message.set_schema(getSchemaName());
372
368
}
373
369
 
374
 
void TableIdentifier::Key::set(size_t resize_arg, const std::string &a, const std::string &b)
375
 
{
376
 
  key_buffer.resize(resize_arg);
377
 
 
378
 
  std::copy(a.begin(), a.end(), key_buffer.begin());
379
 
  std::copy(b.begin(), b.end(), key_buffer.begin() + a.length() + 1);
380
 
 
381
 
  util::sensitive_hash hasher;
382
 
  hash_value= hasher(key_buffer);
383
 
}
384
 
 
385
370
std::size_t hash_value(TableIdentifier const& b)
386
371
{
387
372
  return b.getHashValue();
388
373
}
389
374
 
390
 
std::size_t hash_value(TableIdentifier::Key const& b)
391
 
{
392
 
  return b.getHashValue();
393
 
}
394
 
 
395
375
} /* namespace drizzled */