← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/db.cc

  • Committer: Stewart Smith
  • Date: 2010-11-03 03:28:23 UTC
  • mto: (1902.1.1 build) (1910.1.2 build)
  • mto: This revision was merged to the branch mainline in revision 1903.
  • Revision ID: stewart@flamingspork.com-20101103032823-44k21f0njmk97omr
fix docs warning: Title underline (and overline) is too short in brief_history_of_drizzle.rst

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
 
57
 
static void change_db_impl(Session *session);
58
 
static void change_db_impl(Session *session, identifier::Schema &schema_identifier);
 
60
static long drop_tables_via_filenames(Session *session,
 
61
                                 SchemaIdentifier &schema_identifier,
 
62
                                 TableIdentifiers &dropped_tables);
 
63
static void mysql_change_db_impl(Session *session);
 
64
static void mysql_change_db_impl(Session *session, SchemaIdentifier &schema_identifier);
59
65
 
60
66
/*
61
67
  Create a database
62
68
 
63
69
  SYNOPSIS
64
 
  create_db()
 
70
  mysql_create_db()
65
71
  session               Thread handler
66
72
  db            Name of database to create
67
73
                Function assumes that this is already validated.
78
84
 
79
85
*/
80
86
 
81
 
bool create_db(Session *session, const message::Schema &schema_message, const bool is_if_not_exists)
 
87
bool mysql_create_db(Session *session, const message::Schema &schema_message, const bool is_if_not_exists)
82
88
{
83
89
  TransactionServices &transaction_services= TransactionServices::singleton();
84
90
  bool error= false;
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.
97
103
  */
98
 
  if (session->wait_if_global_read_lock(false, true))
 
104
  if (wait_if_global_read_lock(session, 0, 1))
99
105
  {
100
106
    return false;
101
107
  }
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
  }
138
 
  session->startWaitingGlobalReadLock();
 
144
  start_waiting_global_read_lock(session);
139
145
 
140
146
  return error;
141
147
}
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 mysql_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.
163
167
  */
164
 
  if ((session->wait_if_global_read_lock(false, true)))
 
168
  if ((wait_if_global_read_lock(session, 0, 1)))
165
169
    return false;
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
 
  session->startWaitingGlobalReadLock();
 
195
  start_waiting_global_read_lock(session);
192
196
 
193
197
  return success;
194
198
}
198
202
  Drop all tables in a database and the database itself
199
203
 
200
204
  SYNOPSIS
201
 
    rm_db()
 
205
    mysql_rm_db()
202
206
    session                     Thread handle
203
207
    db                  Database name in the case given by user
204
208
                        It's already validated and set to lower case
211
215
    ERROR Error
212
216
*/
213
217
 
214
 
bool rm_db(Session *session, identifier::Schema &schema_identifier, const bool if_exists)
 
218
bool mysql_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
  TableIdentifiers 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
 
  if (session->wait_if_global_read_lock(false, true))
231
 
  {
232
 
    return true;
233
 
  }
234
 
 
235
 
  do
236
 
  {
237
 
    boost::mutex::scoped_lock scopedLock(session->catalog().schemaLock());
 
237
  if (wait_if_global_read_lock(session, 0, 1))
 
238
  {
 
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
    {
242
259
      if (if_exists)
243
260
      {
244
 
        std::string path;
245
 
        schema_identifier.getSQLPath(path);
246
 
 
247
261
        push_warning_printf(session, DRIZZLE_ERROR::WARN_LEVEL_NOTE,
248
262
                            ER_DB_DROP_EXISTS, ER(ER_DB_DROP_EXISTS),
249
 
                            path.c_str());
 
263
                            schema_identifier.getSQLPath().c_str());
250
264
      }
251
265
      else
252
266
      {
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
 
 
 
267
        error= -1;
 
268
        my_error(ER_DB_DROP_EXISTS, MYF(0), schema_identifier.getSQLPath().c_str());
 
269
        goto exit;
 
270
      }
 
271
    }
 
272
    else
 
273
    {
 
274
      LOCK_open.lock(); /* After deleting database, remove all cache entries related to schema */
 
275
      table::Cache::singleton().removeSchema(schema_identifier);
 
276
      LOCK_open.unlock();
 
277
 
 
278
 
 
279
      error= -1;
 
280
      deleted= drop_tables_via_filenames(session, schema_identifier, dropped_tables);
 
281
      if (deleted >= 0)
 
282
      {
 
283
        error= 0;
 
284
      }
 
285
    }
 
286
    if (deleted >= 0)
 
287
    {
 
288
      assert(! session->query.empty());
 
289
 
 
290
      TransactionServices &transaction_services= TransactionServices::singleton();
 
291
      transaction_services.dropSchema(session, schema_identifier.getSchemaName());
 
292
      session->clear_error();
 
293
      session->server_status|= SERVER_STATUS_DB_DROPPED;
 
294
      session->my_ok((uint32_t) deleted);
 
295
      session->server_status&= ~SERVER_STATUS_DB_DROPPED;
 
296
    }
 
297
    else
 
298
    {
 
299
      char *query, *query_pos, *query_end, *query_data_start;
 
300
 
 
301
      if (!(query= (char*) session->alloc(MAX_DROP_TABLE_Q_LEN)))
 
302
        goto exit; /* not much else we can do */
 
303
      query_pos= query_data_start= strcpy(query,"drop table ")+11;
 
304
      query_end= query + MAX_DROP_TABLE_Q_LEN;
 
305
 
 
306
      TransactionServices &transaction_services= TransactionServices::singleton();
 
307
      for (TableIdentifiers::iterator it= dropped_tables.begin();
 
308
           it != dropped_tables.end();
 
309
           it++)
 
310
      {
 
311
        uint32_t tbl_name_len;
 
312
 
 
313
        /* 3 for the quotes and the comma*/
 
314
        tbl_name_len= (*it).getTableName().length() + 3;
 
315
        if (query_pos + tbl_name_len + 1 >= query_end)
 
316
        {
 
317
          /* These DDL methods and logging protected with LOCK_create_db */
 
318
          transaction_services.rawStatement(session, query);
 
319
          query_pos= query_data_start;
 
320
        }
 
321
 
 
322
        *query_pos++ = '`';
 
323
        query_pos= strcpy(query_pos, (*it).getTableName().c_str()) + (tbl_name_len-3);
 
324
        *query_pos++ = '`';
 
325
        *query_pos++ = ',';
 
326
      }
 
327
 
 
328
      if (query_pos != query_data_start)
 
329
      {
 
330
        /* These DDL methods and logging protected with LOCK_create_db */
 
331
        transaction_services.rawStatement(session, query);
 
332
      }
 
333
    }
 
334
 
 
335
exit:
 
336
    /*
 
337
      If this database was the client's selected database, we silently
 
338
      change the client's selected database to nothing (to have an empty
 
339
      SELECT DATABASE() in the future). For this we free() session->db and set
 
340
      it to 0.
 
341
    */
 
342
    if (schema_identifier.compare(session->db))
 
343
      mysql_change_db_impl(session);
 
344
  }
 
345
 
 
346
  start_waiting_global_read_lock(session);
 
347
 
 
348
  return error;
 
349
}
 
350
 
 
351
 
 
352
static int rm_table_part2(Session *session, TableList *tables)
 
353
{
 
354
  TransactionServices &transaction_services= TransactionServices::singleton();
 
355
 
 
356
  TableList *table;
 
357
  String wrong_tables;
 
358
  int error= 0;
 
359
  bool foreign_key_error= false;
 
360
 
 
361
  LOCK_open.lock(); /* Part 2 of rm a table */
 
362
 
 
363
  if (lock_table_names_exclusively(session, tables))
 
364
  {
 
365
    LOCK_open.unlock();
 
366
    return 1;
 
367
  }
 
368
 
 
369
  /* Don't give warnings for not found errors, as we already generate notes */
 
370
  session->no_warnings_for_error= 1;
 
371
 
 
372
  for (table= tables; table; table= table->next_local)
 
373
  {
 
374
    const char *db=table->getSchemaName();
 
375
    TableIdentifier identifier(table->getSchemaName(), table->getTableName());
 
376
 
 
377
    plugin::StorageEngine *table_type;
 
378
 
 
379
    error= session->drop_temporary_table(identifier);
 
380
 
 
381
    switch (error) {
 
382
    case  0:
 
383
      // removed temporary table
 
384
      continue;
 
385
    case -1:
 
386
      error= 1;
 
387
      unlock_table_names(tables, NULL);
 
388
      LOCK_open.unlock();
 
389
      session->no_warnings_for_error= 0;
 
390
 
 
391
      return(error);
 
392
    default:
 
393
      // temporary table not found
 
394
      error= 0;
 
395
    }
 
396
 
 
397
    table_type= table->getDbType();
 
398
 
 
399
    {
 
400
      Table *locked_table;
 
401
      abort_locked_tables(session, identifier);
 
402
      table::Cache::singleton().removeTable(session, identifier,
 
403
                                            RTFC_WAIT_OTHER_THREAD_FLAG |
 
404
                                            RTFC_CHECK_KILLED_FLAG);
 
405
      /*
 
406
        If the table was used in lock tables, remember it so that
 
407
        unlock_table_names can free it
 
408
      */
 
409
      if ((locked_table= drop_locked_tables(session, identifier)))
 
410
        table->table= locked_table;
 
411
 
 
412
      if (session->killed)
 
413
      {
 
414
        error= -1;
 
415
        unlock_table_names(tables, NULL);
 
416
        LOCK_open.unlock();
 
417
        session->no_warnings_for_error= 0;
 
418
 
 
419
        return(error);
 
420
      }
 
421
    }
 
422
    identifier.getPath();
 
423
 
 
424
    if (table_type == NULL && not plugin::StorageEngine::doesTableExist(*session, identifier))
 
425
    {
 
426
      // Table was not found on disk and table can't be created from engine
 
427
      push_warning_printf(session, DRIZZLE_ERROR::WARN_LEVEL_NOTE,
 
428
                          ER_BAD_TABLE_ERROR, ER(ER_BAD_TABLE_ERROR),
 
429
                          table->getTableName());
 
430
    }
 
431
    else
 
432
    {
 
433
      error= plugin::StorageEngine::dropTable(*session, identifier);
 
434
 
 
435
      if ((error == ENOENT || error == HA_ERR_NO_SUCH_TABLE))
 
436
      {
 
437
        error= 0;
 
438
        session->clear_error();
 
439
      }
 
440
 
 
441
      if (error == HA_ERR_ROW_IS_REFERENCED)
 
442
      {
 
443
        /* the table is referenced by a foreign key constraint */
 
444
        foreign_key_error= true;
 
445
      }
 
446
    }
 
447
 
 
448
    if (error == 0 || (foreign_key_error == false))
 
449
    {
 
450
      transaction_services.dropTable(session, string(db), string(table->getTableName()), true);
 
451
    }
 
452
 
 
453
    if (error)
 
454
    {
 
455
      if (wrong_tables.length())
 
456
        wrong_tables.append(',');
 
457
      wrong_tables.append(String(table->getTableName(),system_charset_info));
 
458
    }
 
459
  }
265
460
  /*
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.
 
461
    It's safe to unlock LOCK_open: we have an exclusive lock
 
462
    on the table name.
270
463
  */
271
 
  if (not error and schema_identifier.compare(*session->schema()))
272
 
    change_db_impl(session);
273
 
 
274
 
  session->startWaitingGlobalReadLock();
275
 
 
276
 
  return error;
 
464
  LOCK_open.unlock();
 
465
  error= 0;
 
466
  if (wrong_tables.length())
 
467
  {
 
468
    if (not foreign_key_error)
 
469
      my_printf_error(ER_BAD_TABLE_ERROR, ER(ER_BAD_TABLE_ERROR), MYF(0),
 
470
                      wrong_tables.c_ptr());
 
471
    else
 
472
    {
 
473
      my_message(ER_ROW_IS_REFERENCED, ER(ER_ROW_IS_REFERENCED), MYF(0));
 
474
    }
 
475
    error= 1;
 
476
  }
 
477
 
 
478
  LOCK_open.lock(); /* final bit in rm table lock */
 
479
  unlock_table_names(tables, NULL);
 
480
  LOCK_open.unlock();
 
481
  session->no_warnings_for_error= 0;
 
482
 
 
483
  return(error);
 
484
}
 
485
 
 
486
/*
 
487
  Removes files with known extensions plus.
 
488
  session MUST be set when calling this function!
 
489
*/
 
490
 
 
491
static long drop_tables_via_filenames(Session *session,
 
492
                                      SchemaIdentifier &schema_identifier,
 
493
                                      TableIdentifiers &dropped_tables)
 
494
{
 
495
  long deleted= 0;
 
496
  TableList *tot_list= NULL, **tot_list_next;
 
497
 
 
498
  tot_list_next= &tot_list;
 
499
 
 
500
  plugin::StorageEngine::getIdentifiers(*session, schema_identifier, dropped_tables);
 
501
 
 
502
  for (TableIdentifiers::iterator it= dropped_tables.begin();
 
503
       it != dropped_tables.end();
 
504
       it++)
 
505
  {
 
506
    size_t db_len= schema_identifier.getSchemaName().size();
 
507
 
 
508
    /* Drop the table nicely */
 
509
    TableList *table_list=(TableList*)
 
510
      session->calloc(sizeof(*table_list) +
 
511
                      db_len + 1 +
 
512
                      (*it).getTableName().length() + 1);
 
513
 
 
514
    if (not table_list)
 
515
      return -1;
 
516
 
 
517
    table_list->setSchemaName((char*) (table_list+1));
 
518
    table_list->setTableName(strcpy((char*) (table_list+1), schema_identifier.getSchemaName().c_str()) + db_len + 1);
 
519
    TableIdentifier::filename_to_tablename((*it).getTableName().c_str(), const_cast<char *>(table_list->getTableName()), (*it).getTableName().size() + 1);
 
520
    table_list->alias= table_list->getTableName();  // If lower_case_table_names=2
 
521
    table_list->setInternalTmpTable((strncmp((*it).getTableName().c_str(),
 
522
                                             TMP_FILE_PREFIX,
 
523
                                             strlen(TMP_FILE_PREFIX)) == 0));
 
524
    /* Link into list */
 
525
    (*tot_list_next)= table_list;
 
526
    tot_list_next= &table_list->next_local;
 
527
    deleted++;
 
528
  }
 
529
  if (session->killed)
 
530
    return -1;
 
531
 
 
532
  if (tot_list)
 
533
  {
 
534
    if (rm_table_part2(session, tot_list))
 
535
      return -1;
 
536
  }
 
537
 
 
538
 
 
539
  if (not plugin::StorageEngine::dropSchema(schema_identifier))
 
540
  {
 
541
    my_error(ER_DROP_SCHEMA, MYF(0), schema_identifier.getSQLPath().c_str());
 
542
    return -1;
 
543
  }
 
544
 
 
545
  return deleted;
277
546
}
278
547
 
279
548
/**
338
607
    @retval true  Error
339
608
*/
340
609
 
341
 
bool change_db(Session *session, identifier::Schema &schema_identifier)
 
610
bool mysql_change_db(Session *session, SchemaIdentifier &schema_identifier)
342
611
{
343
612
 
344
 
  if (not plugin::Authorization::isAuthorized(session->user(), schema_identifier))
 
613
  if (not plugin::Authorization::isAuthorized(session->getSecurityContext(), schema_identifier))
345
614
  {
346
615
    /* Error message is set in isAuthorized */
347
616
    return true;
349
618
 
350
619
  if (not check_db_name(session, schema_identifier))
351
620
  {
352
 
    my_error(ER_WRONG_DB_NAME, schema_identifier);
 
621
    my_error(ER_WRONG_DB_NAME, MYF(0), schema_identifier.getSQLPath().c_str());
353
622
 
354
623
    return true;
355
624
  }
356
625
 
357
626
  if (not plugin::StorageEngine::doesSchemaExist(schema_identifier))
358
627
  {
359
 
    my_error(ER_BAD_DB_ERROR, schema_identifier);
 
628
    /* Report an error and free new_db_file_name. */
 
629
 
 
630
    my_error(ER_BAD_DB_ERROR, MYF(0), schema_identifier.getSQLPath().c_str());
360
631
 
361
632
    /* The operation failed. */
362
633
 
363
634
    return true;
364
635
  }
365
636
 
366
 
  change_db_impl(session, schema_identifier);
 
637
  mysql_change_db_impl(session, schema_identifier);
367
638
 
368
639
  return false;
369
640
}
379
650
  @param new_db_charset Character set of the new database.
380
651
*/
381
652
 
382
 
static void change_db_impl(Session *session, identifier::Schema &schema_identifier)
 
653
static void mysql_change_db_impl(Session *session, SchemaIdentifier &schema_identifier)
383
654
{
384
655
  /* 1. Change current database in Session. */
385
656
 
406
677
  }
407
678
}
408
679
 
409
 
static void change_db_impl(Session *session)
 
680
static void mysql_change_db_impl(Session *session)
410
681
{
411
682
  session->set_db(string());
412
683
}