← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/db.cc

  • Committer: Brian Aker
  • Date: 2010-12-27 20:20:52 UTC
  • mto: (2023.2.10 bool)
  • mto: This revision was merged to the branch mainline in revision 2037.
  • Revision ID: brian@tangent.org-20101227202052-fzftwyo0vu02zhpa
Refactor boolean function.

Show diffs side-by-side

added added

removed removed

Lines of Context:
drizzled/db.cc
25
25
#include <string>
26
26
#include <fstream>
27
27
 
 
28
#include <drizzled/message/schema.pb.h>
28
29
#include "drizzled/error.h"
29
30
#include <drizzled/gettext.h>
30
31
#include <drizzled/my_hash.h>
45
46
 
46
47
#include <boost/thread/mutex.hpp>
47
48
 
 
49
boost::mutex LOCK_create_db;
 
50
 
48
51
#include "drizzled/internal/my_sys.h"
49
52
 
50
53
#define MAX_DROP_TABLE_Q_LEN      1024
54
57
namespace drizzled
55
58
{
56
59
 
 
60
static long drop_tables_via_filenames(Session *session,
 
61
                                 SchemaIdentifier &schema_identifier,
 
62
                                 TableIdentifier::vector &dropped_tables);
57
63
static void change_db_impl(Session *session);
58
 
static void change_db_impl(Session *session, identifier::Schema &schema_identifier);
 
64
static void change_db_impl(Session *session, SchemaIdentifier &schema_identifier);
59
65
 
60
66
/*
61
67
  Create a database
85
91
 
86
92
  /*
87
93
    Do not create database if another thread is holding read lock.
88
 
    Wait for global read lock before acquiring session->catalog()->schemaLock().
 
94
    Wait for global read lock before acquiring LOCK_create_db.
89
95
    After wait_if_global_read_lock() we have protection against another
90
 
    global read lock. If we would acquire session->catalog()->schemaLock() first,
 
96
    global read lock. If we would acquire LOCK_create_db first,
91
97
    another thread could step in and get the global read lock before we
92
98
    reach wait_if_global_read_lock(). If this thread tries the same as we
93
 
    (admin a db), it would then go and wait on session->catalog()->schemaLock()...
 
99
    (admin a db), it would then go and wait on LOCK_create_db...
94
100
    Furthermore wait_if_global_read_lock() checks if the current thread
95
101
    has the global read lock and refuses the operation with
96
102
    ER_CANT_UPDATE_WITH_READLOCK if applicable.
105
111
 
106
112
  // @todo push this lock down into the engine
107
113
  {
108
 
    boost::mutex::scoped_lock scopedLock(session->catalog().schemaLock());
 
114
    boost::mutex::scoped_lock scopedLock(LOCK_create_db);
109
115
 
110
116
    // Check to see if it exists already.  
111
 
    identifier::Schema schema_identifier(schema_message.name());
 
117
    SchemaIdentifier schema_identifier(schema_message.name());
112
118
    if (plugin::StorageEngine::doesSchemaExist(schema_identifier))
113
119
    {
114
120
      if (not is_if_not_exists)
115
121
      {
116
 
        my_error(ER_DB_CREATE_EXISTS, schema_identifier);
 
122
        my_error(ER_DB_CREATE_EXISTS, MYF(0), schema_message.name().c_str());
117
123
        error= true;
118
124
      }
119
125
      else
131
137
    }
132
138
    else // Created !
133
139
    {
134
 
      transaction_services.createSchema(*session, schema_message);
 
140
      transaction_services.createSchema(session, schema_message);
135
141
      session->my_ok(1);
136
142
    }
137
143
  }
143
149
 
144
150
/* db-name is already validated when we come here */
145
151
 
146
 
bool alter_db(Session *session,
147
 
              const message::Schema &schema_message,
148
 
              const message::schema::shared_ptr &original_schema)
 
152
bool alter_db(Session *session, const message::Schema &schema_message)
149
153
{
150
154
  TransactionServices &transaction_services= TransactionServices::singleton();
151
155
 
152
156
  /*
153
157
    Do not alter database if another thread is holding read lock.
154
 
    Wait for global read lock before acquiring session->catalog()->schemaLock().
 
158
    Wait for global read lock before acquiring LOCK_create_db.
155
159
    After wait_if_global_read_lock() we have protection against another
156
 
    global read lock. If we would acquire session->catalog()->schemaLock() first,
 
160
    global read lock. If we would acquire LOCK_create_db first,
157
161
    another thread could step in and get the global read lock before we
158
162
    reach wait_if_global_read_lock(). If this thread tries the same as we
159
 
    (admin a db), it would then go and wait on session->catalog()->schemaLock()...
 
163
    (admin a db), it would then go and wait on LOCK_create_db...
160
164
    Furthermore wait_if_global_read_lock() checks if the current thread
161
165
    has the global read lock and refuses the operation with
162
166
    ER_CANT_UPDATE_WITH_READLOCK if applicable.
166
170
 
167
171
  bool success;
168
172
  {
169
 
    boost::mutex::scoped_lock scopedLock(session->catalog().schemaLock());
 
173
    boost::mutex::scoped_lock scopedLock(LOCK_create_db);
170
174
 
171
 
    identifier::Schema schema_idenifier(schema_message.name());
 
175
    SchemaIdentifier schema_idenifier(schema_message.name());
172
176
    if (not plugin::StorageEngine::doesSchemaExist(schema_idenifier))
173
177
    {
174
 
      my_error(ER_SCHEMA_DOES_NOT_EXIST, schema_idenifier);
 
178
      my_error(ER_SCHEMA_DOES_NOT_EXIST, MYF(0), schema_message.name().c_str());
175
179
      return false;
176
180
    }
177
181
 
180
184
 
181
185
    if (success)
182
186
    {
183
 
      transaction_services.alterSchema(*session, original_schema, schema_message);
 
187
      transaction_services.rawStatement(session, *session->getQueryString());
184
188
      session->my_ok(1);
185
189
    }
186
190
    else
187
191
    {
188
 
      my_error(ER_ALTER_SCHEMA, schema_idenifier);
 
192
      my_error(ER_ALTER_SCHEMA, MYF(0), schema_message.name().c_str());
189
193
    }
190
194
  }
191
195
  session->startWaitingGlobalReadLock();
211
215
    ERROR Error
212
216
*/
213
217
 
214
 
bool rm_db(Session *session, identifier::Schema &schema_identifier, const bool if_exists)
 
218
bool rm_db(Session *session, SchemaIdentifier &schema_identifier, const bool if_exists)
215
219
{
216
 
  bool error= false;
 
220
  long deleted=0;
 
221
  int error= false;
 
222
  TableIdentifier::vector dropped_tables;
 
223
  message::Schema schema_proto;
217
224
 
218
225
  /*
219
226
    Do not drop database if another thread is holding read lock.
220
 
    Wait for global read lock before acquiring session->catalog()->schemaLock().
 
227
    Wait for global read lock before acquiring LOCK_create_db.
221
228
    After wait_if_global_read_lock() we have protection against another
222
 
    global read lock. If we would acquire session->catalog()->schemaLock() first,
 
229
    global read lock. If we would acquire LOCK_create_db first,
223
230
    another thread could step in and get the global read lock before we
224
231
    reach wait_if_global_read_lock(). If this thread tries the same as we
225
 
    (admin a db), it would then go and wait on session->catalog()->schemaLock()...
 
232
    (admin a db), it would then go and wait on LOCK_create_db...
226
233
    Furthermore wait_if_global_read_lock() checks if the current thread
227
234
    has the global read lock and refuses the operation with
228
235
    ER_CANT_UPDATE_WITH_READLOCK if applicable.
229
236
  */
230
237
  if (session->wait_if_global_read_lock(false, true))
231
238
  {
232
 
    return true;
233
 
  }
234
 
 
235
 
  do
236
 
  {
237
 
    boost::mutex::scoped_lock scopedLock(session->catalog().schemaLock());
 
239
    return -1;
 
240
  }
 
241
 
 
242
  // Lets delete the temporary tables first outside of locks.  
 
243
  set<string> set_of_names;
 
244
  session->doGetTableNames(schema_identifier, set_of_names);
 
245
 
 
246
  for (set<string>::iterator iter= set_of_names.begin(); iter != set_of_names.end(); iter++)
 
247
  {
 
248
    TableIdentifier identifier(schema_identifier, *iter, message::Table::TEMPORARY);
 
249
    Table *table= session->find_temporary_table(identifier);
 
250
    session->close_temporary_table(table);
 
251
  }
 
252
 
 
253
  {
 
254
    boost::mutex::scoped_lock scopedLock(LOCK_create_db);
238
255
 
239
256
    /* See if the schema exists */
240
257
    if (not plugin::StorageEngine::doesSchemaExist(schema_identifier))
241
258
    {
 
259
      std::string path;
 
260
      schema_identifier.getSQLPath(path);
 
261
 
242
262
      if (if_exists)
243
263
      {
244
 
        std::string path;
245
 
        schema_identifier.getSQLPath(path);
246
 
 
247
264
        push_warning_printf(session, DRIZZLE_ERROR::WARN_LEVEL_NOTE,
248
265
                            ER_DB_DROP_EXISTS, ER(ER_DB_DROP_EXISTS),
249
266
                            path.c_str());
250
267
      }
251
268
      else
252
269
      {
253
 
        error= true;
254
 
        my_error(ER_DB_DROP_EXISTS, schema_identifier);
255
 
        break;
256
 
      }
257
 
    }
258
 
    else
259
 
    {
260
 
      error= plugin::StorageEngine::dropSchema(*session, schema_identifier);
261
 
    }
262
 
 
263
 
  } while (0);
264
 
 
265
 
  /*
266
 
    If this database was the client's selected database, we silently
267
 
    change the client's selected database to nothing (to have an empty
268
 
    SELECT DATABASE() in the future). For this we free() session->db and set
269
 
    it to 0.
270
 
  */
271
 
  if (not error and schema_identifier.compare(*session->schema()))
272
 
    change_db_impl(session);
 
270
        error= -1;
 
271
        my_error(ER_DB_DROP_EXISTS, MYF(0), path.c_str());
 
272
        goto exit;
 
273
      }
 
274
    }
 
275
    else
 
276
    {
 
277
      /* After deleting database, remove all cache entries related to schema */
 
278
      table::Cache::singleton().removeSchema(schema_identifier);
 
279
 
 
280
      error= -1;
 
281
      deleted= drop_tables_via_filenames(session, schema_identifier, dropped_tables);
 
282
      if (deleted >= 0)
 
283
      {
 
284
        error= 0;
 
285
        
 
286
        /* We've already verified that the schema does exist, so safe to log it */
 
287
        TransactionServices &transaction_services= TransactionServices::singleton();
 
288
        transaction_services.dropSchema(session, schema_identifier.getSchemaName());
 
289
      }
 
290
    }
 
291
    if (deleted >= 0)
 
292
    {
 
293
      session->clear_error();
 
294
      session->server_status|= SERVER_STATUS_DB_DROPPED;
 
295
      session->my_ok((uint32_t) deleted);
 
296
      session->server_status&= ~SERVER_STATUS_DB_DROPPED;
 
297
    }
 
298
    else
 
299
    {
 
300
      char *query, *query_pos, *query_end, *query_data_start;
 
301
 
 
302
      if (!(query= (char*) session->alloc(MAX_DROP_TABLE_Q_LEN)))
 
303
        goto exit; /* not much else we can do */
 
304
      query_pos= query_data_start= strcpy(query,"drop table ")+11;
 
305
      query_end= query + MAX_DROP_TABLE_Q_LEN;
 
306
 
 
307
      TransactionServices &transaction_services= TransactionServices::singleton();
 
308
      for (TableIdentifier::vector::iterator it= dropped_tables.begin();
 
309
           it != dropped_tables.end();
 
310
           it++)
 
311
      {
 
312
        uint32_t tbl_name_len;
 
313
 
 
314
        /* 3 for the quotes and the comma*/
 
315
        tbl_name_len= (*it).getTableName().length() + 3;
 
316
        if (query_pos + tbl_name_len + 1 >= query_end)
 
317
        {
 
318
          /* These DDL methods and logging protected with LOCK_create_db */
 
319
          transaction_services.rawStatement(session, query);
 
320
          query_pos= query_data_start;
 
321
        }
 
322
 
 
323
        *query_pos++ = '`';
 
324
        query_pos= strcpy(query_pos, (*it).getTableName().c_str()) + (tbl_name_len-3);
 
325
        *query_pos++ = '`';
 
326
        *query_pos++ = ',';
 
327
      }
 
328
 
 
329
      if (query_pos != query_data_start)
 
330
      {
 
331
        /* These DDL methods and logging protected with LOCK_create_db */
 
332
        transaction_services.rawStatement(session, query);
 
333
      }
 
334
    }
 
335
 
 
336
exit:
 
337
    /*
 
338
      If this database was the client's selected database, we silently
 
339
      change the client's selected database to nothing (to have an empty
 
340
      SELECT DATABASE() in the future). For this we free() session->db and set
 
341
      it to 0.
 
342
    */
 
343
    if (schema_identifier.compare(*session->schema()))
 
344
      change_db_impl(session);
 
345
  }
273
346
 
274
347
  session->startWaitingGlobalReadLock();
275
348
 
276
349
  return error;
277
350
}
278
351
 
 
352
 
 
353
static int rm_table_part2(Session *session, TableList *tables)
 
354
{
 
355
  TransactionServices &transaction_services= TransactionServices::singleton();
 
356
 
 
357
  TableList *table;
 
358
  String wrong_tables;
 
359
  int error= 0;
 
360
  bool foreign_key_error= false;
 
361
 
 
362
  {
 
363
    table::Cache::singleton().mutex().lock(); /* Part 2 of rm a table */
 
364
 
 
365
    if (session->lock_table_names_exclusively(tables))
 
366
    {
 
367
      table::Cache::singleton().mutex().unlock();
 
368
      return 1;
 
369
    }
 
370
 
 
371
    /* Don't give warnings for not found errors, as we already generate notes */
 
372
    session->no_warnings_for_error= 1;
 
373
 
 
374
    for (table= tables; table; table= table->next_local)
 
375
    {
 
376
      const char *db=table->getSchemaName();
 
377
      TableIdentifier identifier(table->getSchemaName(), table->getTableName());
 
378
 
 
379
      plugin::StorageEngine *table_type;
 
380
 
 
381
      error= session->drop_temporary_table(identifier);
 
382
 
 
383
      switch (error) {
 
384
      case  0:
 
385
        // removed temporary table
 
386
        continue;
 
387
      case -1:
 
388
        error= 1;
 
389
        tables->unlock_table_names();
 
390
        table::Cache::singleton().mutex().unlock();
 
391
        session->no_warnings_for_error= 0;
 
392
 
 
393
        return(error);
 
394
      default:
 
395
        // temporary table not found
 
396
        error= 0;
 
397
      }
 
398
 
 
399
      table_type= table->getDbType();
 
400
 
 
401
      {
 
402
        Table *locked_table;
 
403
        abort_locked_tables(session, identifier);
 
404
        table::Cache::singleton().removeTable(session, identifier,
 
405
                                              RTFC_WAIT_OTHER_THREAD_FLAG |
 
406
                                              RTFC_CHECK_KILLED_FLAG);
 
407
        /*
 
408
          If the table was used in lock tables, remember it so that
 
409
          unlock_table_names can free it
 
410
        */
 
411
        if ((locked_table= drop_locked_tables(session, identifier)))
 
412
          table->table= locked_table;
 
413
 
 
414
        if (session->getKilled())
 
415
        {
 
416
          error= -1;
 
417
          tables->unlock_table_names();
 
418
          table::Cache::singleton().mutex().unlock();
 
419
          session->no_warnings_for_error= 0;
 
420
 
 
421
          return(error);
 
422
        }
 
423
      }
 
424
      identifier.getPath();
 
425
 
 
426
      if (table_type == NULL && not plugin::StorageEngine::doesTableExist(*session, identifier))
 
427
      {
 
428
        // Table was not found on disk and table can't be created from engine
 
429
        push_warning_printf(session, DRIZZLE_ERROR::WARN_LEVEL_NOTE,
 
430
                            ER_BAD_TABLE_ERROR, ER(ER_BAD_TABLE_ERROR),
 
431
                            table->getTableName());
 
432
      }
 
433
      else
 
434
      {
 
435
        error= plugin::StorageEngine::dropTable(*session, identifier);
 
436
 
 
437
        /* Generate transaction event ONLY when we successfully drop */ 
 
438
        if (error == 0)
 
439
        {
 
440
          transaction_services.dropTable(session, string(db), string(table->getTableName()));
 
441
        }
 
442
 
 
443
        if ((error == ENOENT || error == HA_ERR_NO_SUCH_TABLE))
 
444
        {
 
445
          error= 0;
 
446
          session->clear_error();
 
447
        }
 
448
 
 
449
        if (error == HA_ERR_ROW_IS_REFERENCED)
 
450
        {
 
451
          /* the table is referenced by a foreign key constraint */
 
452
          foreign_key_error= true;
 
453
        }
 
454
      }
 
455
 
 
456
      if (error)
 
457
      {
 
458
        if (wrong_tables.length())
 
459
          wrong_tables.append(',');
 
460
        wrong_tables.append(String(table->getTableName(),system_charset_info));
 
461
      }
 
462
    }
 
463
    /*
 
464
      It's safe to unlock table::Cache::singleton().mutex(): we have an exclusive lock
 
465
      on the table name.
 
466
    */
 
467
    table::Cache::singleton().mutex().unlock();
 
468
  }
 
469
 
 
470
  error= 0;
 
471
  if (wrong_tables.length())
 
472
  {
 
473
    if (not foreign_key_error)
 
474
      my_printf_error(ER_BAD_TABLE_ERROR, ER(ER_BAD_TABLE_ERROR), MYF(0),
 
475
                      wrong_tables.c_ptr());
 
476
    else
 
477
    {
 
478
      my_message(ER_ROW_IS_REFERENCED, ER(ER_ROW_IS_REFERENCED), MYF(0));
 
479
    }
 
480
    error= 1;
 
481
  }
 
482
 
 
483
  {
 
484
    boost::mutex::scoped_lock scopedLock(table::Cache::singleton().mutex()); /* final bit in rm table lock */
 
485
    tables->unlock_table_names();
 
486
  }
 
487
  session->no_warnings_for_error= 0;
 
488
 
 
489
  return(error);
 
490
}
 
491
 
 
492
/*
 
493
  Removes files with known extensions plus.
 
494
  session MUST be set when calling this function!
 
495
*/
 
496
 
 
497
static long drop_tables_via_filenames(Session *session,
 
498
                                      SchemaIdentifier &schema_identifier,
 
499
                                      TableIdentifier::vector &dropped_tables)
 
500
{
 
501
  long deleted= 0;
 
502
  TableList *tot_list= NULL, **tot_list_next;
 
503
 
 
504
  tot_list_next= &tot_list;
 
505
 
 
506
  plugin::StorageEngine::getIdentifiers(*session, schema_identifier, dropped_tables);
 
507
 
 
508
  for (TableIdentifier::vector::iterator it= dropped_tables.begin();
 
509
       it != dropped_tables.end();
 
510
       it++)
 
511
  {
 
512
    size_t db_len= schema_identifier.getSchemaName().size();
 
513
 
 
514
    /* Drop the table nicely */
 
515
    TableList *table_list=(TableList*)
 
516
      session->calloc(sizeof(*table_list) +
 
517
                      db_len + 1 +
 
518
                      (*it).getTableName().length() + 1);
 
519
 
 
520
    if (not table_list)
 
521
      return -1;
 
522
 
 
523
    table_list->setSchemaName((char*) (table_list+1));
 
524
    table_list->setTableName(strcpy((char*) (table_list+1), schema_identifier.getSchemaName().c_str()) + db_len + 1);
 
525
    TableIdentifier::filename_to_tablename((*it).getTableName().c_str(), const_cast<char *>(table_list->getTableName()), (*it).getTableName().size() + 1);
 
526
    table_list->alias= table_list->getTableName();  // If lower_case_table_names=2
 
527
    table_list->setInternalTmpTable((strncmp((*it).getTableName().c_str(),
 
528
                                             TMP_FILE_PREFIX,
 
529
                                             strlen(TMP_FILE_PREFIX)) == 0));
 
530
    /* Link into list */
 
531
    (*tot_list_next)= table_list;
 
532
    tot_list_next= &table_list->next_local;
 
533
    deleted++;
 
534
  }
 
535
  if (session->getKilled())
 
536
    return -1;
 
537
 
 
538
  if (tot_list)
 
539
  {
 
540
    if (rm_table_part2(session, tot_list))
 
541
      return -1;
 
542
  }
 
543
 
 
544
 
 
545
  if (not plugin::StorageEngine::dropSchema(schema_identifier))
 
546
  {
 
547
    std::string path;
 
548
    schema_identifier.getSQLPath(path);
 
549
    my_error(ER_DROP_SCHEMA, MYF(0), path.c_str());
 
550
 
 
551
    return -1;
 
552
  }
 
553
 
 
554
  return deleted;
 
555
}
 
556
 
279
557
/**
280
558
  @brief Change the current database and its attributes unconditionally.
281
559
 
338
616
    @retval true  Error
339
617
*/
340
618
 
341
 
bool change_db(Session *session, identifier::Schema &schema_identifier)
 
619
bool change_db(Session *session, SchemaIdentifier &schema_identifier)
342
620
{
343
621
 
344
622
  if (not plugin::Authorization::isAuthorized(session->user(), schema_identifier))
349
627
 
350
628
  if (not check_db_name(session, schema_identifier))
351
629
  {
352
 
    my_error(ER_WRONG_DB_NAME, schema_identifier);
 
630
    std::string path;
 
631
    schema_identifier.getSQLPath(path);
 
632
    my_error(ER_WRONG_DB_NAME, MYF(0), path.c_str());
353
633
 
354
634
    return true;
355
635
  }
356
636
 
357
637
  if (not plugin::StorageEngine::doesSchemaExist(schema_identifier))
358
638
  {
359
 
    my_error(ER_BAD_DB_ERROR, schema_identifier);
 
639
    /* Report an error and free new_db_file_name. */
 
640
    std::string path;
 
641
    schema_identifier.getSQLPath(path);
 
642
 
 
643
    my_error(ER_BAD_DB_ERROR, MYF(0), path.c_str());
360
644
 
361
645
    /* The operation failed. */
362
646
 
379
663
  @param new_db_charset Character set of the new database.
380
664
*/
381
665
 
382
 
static void change_db_impl(Session *session, identifier::Schema &schema_identifier)
 
666
static void change_db_impl(Session *session, SchemaIdentifier &schema_identifier)
383
667
{
384
668
  /* 1. Change current database in Session. */
385
669