← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/handlerton.cc

  • Committer: Stewart Smith
  • Date: 2008-12-18 23:54:47 UTC
  • mto: This revision was merged to the branch mainline in revision 719.
  • Revision ID: stewart@flamingspork.com-20081218235447-qf6dofgd6guwefo2
fix RENAME TABLE

(problem was missing hton to mysql_rename_table)

If engine implements table_exists_in_engine we get the hton from there.

Else, we reintroduce the mysql_frm_type hackery (temporarily, it *will* go away again with FRM removal).

also fix rename test: no views, plus current FLUSH LOCKS bug.

Show diffs side-by-side

added added

removed removed

Lines of Context:
drizzled/handlerton.cc
1
1
/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-
2
2
 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
3
3
 *
4
 
 *  Copyright (C) 2008 Sun Microsystems, Inc.
 
4
 *  Copyright (C) 2008 Sun Microsystems
5
5
 *
6
6
 *  This program is free software; you can redistribute it and/or modify
7
7
 *  it under the terms of the GNU General Public License as published by
17
17
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
18
18
 */
19
19
 
20
 
#include "config.h"
21
 
 
22
 
#include <fcntl.h>
23
 
#include <unistd.h>
24
 
 
25
 
#include <string>
26
 
#include <vector>
27
 
#include <set>
28
 
#include <fstream>
29
 
#include <algorithm>
30
 
#include <functional>
31
 
 
32
 
#include <google/protobuf/io/zero_copy_stream.h>
33
 
#include <google/protobuf/io/zero_copy_stream_impl.h>
34
 
 
35
 
#include "drizzled/cached_directory.h"
36
 
 
 
20
#include <drizzled/server_includes.h>
37
21
#include <drizzled/definitions.h>
38
22
#include <drizzled/base.h>
39
 
#include <drizzled/cursor.h>
40
 
#include <drizzled/plugin/storage_engine.h>
 
23
#include <drizzled/handler.h>
 
24
#include <drizzled/handlerton.h>
41
25
#include <drizzled/session.h>
42
26
#include <drizzled/error.h>
43
27
#include <drizzled/gettext.h>
44
 
#include <drizzled/unireg.h>
45
 
#include <drizzled/data_home.h>
46
 
#include "drizzled/errmsg_print.h"
47
 
#include "drizzled/xid.h"
48
 
#include "drizzled/sql_table.h"
49
 
#include "drizzled/global_charset_info.h"
50
 
#include "drizzled/charset.h"
51
 
#include "drizzled/internal/my_sys.h"
52
 
#include "drizzled/db.h"
53
 
 
54
 
#include <drizzled/table_proto.h>
55
 
#include <drizzled/plugin/event_observer.h>
56
 
 
57
 
#include <drizzled/table/shell.h>
58
 
 
59
 
#include "drizzled/message/cache.h"
60
 
 
61
 
#include <boost/algorithm/string/compare.hpp>
62
 
 
63
 
static bool shutdown_has_begun= false; // Once we put in the container for the vector/etc for engines this will go away.
64
 
 
65
 
namespace drizzled
66
 
{
67
 
 
68
 
namespace plugin
69
 
{
70
 
 
71
 
static EngineVector vector_of_engines;
72
 
static EngineVector vector_of_schema_engines;
73
 
 
74
 
const std::string DEFAULT_STRING("default");
75
 
const std::string UNKNOWN_STRING("UNKNOWN");
76
 
const std::string DEFAULT_DEFINITION_FILE_EXT(".dfe");
77
 
 
78
 
static std::set<std::string> set_of_table_definition_ext;
79
 
 
80
 
EngineVector &StorageEngine::getSchemaEngines()
81
 
{
82
 
  return vector_of_schema_engines;
83
 
}
84
 
 
85
 
StorageEngine::StorageEngine(const std::string name_arg,
86
 
                             const std::bitset<HTON_BIT_SIZE> &flags_arg) :
87
 
  Plugin(name_arg, "StorageEngine"),
88
 
  MonitoredInTransaction(), /* This gives the storage engine a "slot" or ID */
89
 
  flags(flags_arg)
90
 
{
91
 
}
92
 
 
93
 
StorageEngine::~StorageEngine()
94
 
{
95
 
}
96
 
 
97
 
void StorageEngine::setTransactionReadWrite(Session& session)
98
 
{
99
 
  TransactionContext &statement_ctx= session.transaction.stmt;
100
 
  statement_ctx.markModifiedNonTransData();
101
 
}
102
 
 
103
 
 
104
 
int StorageEngine::renameTable(Session &session, const identifier::Table &from, const identifier::Table &to)
105
 
{
106
 
  int error;
107
 
  setTransactionReadWrite(session);
108
 
 
109
 
  if (unlikely(plugin::EventObserver::beforeRenameTable(session, from, to)))
110
 
  {
111
 
    error= ER_EVENT_OBSERVER_PLUGIN;
112
 
  }
113
 
  else
114
 
  {
115
 
    error =  doRenameTable(session, from, to);
116
 
    if (unlikely(plugin::EventObserver::afterRenameTable(session, from, to, error)))
117
 
    {
118
 
      error= ER_EVENT_OBSERVER_PLUGIN;
119
 
    }
120
 
  }
121
 
  
122
 
  return error;
123
 
}
124
 
 
125
 
/**
126
 
  Delete all files with extension from bas_ext().
127
 
 
128
 
  @param name           Base name of table
129
 
 
130
 
  @note
131
 
    We assume that the Cursor may return more extensions than
132
 
    was actually used for the file.
133
 
 
134
 
  @retval
135
 
    0   If we successfully deleted at least one file from base_ext and
136
 
    didn't get any other errors than ENOENT
137
 
  @retval
138
 
    !0  Error
139
 
*/
140
 
int StorageEngine::doDropTable(Session&, const identifier::Table &identifier)
141
 
                               
142
 
{
143
 
  int error= 0;
144
 
  int enoent_or_zero= ENOENT;                   // Error if no file was deleted
145
 
  char buff[FN_REFLEN];
146
 
 
147
 
  for (const char **ext= bas_ext(); *ext ; ext++)
148
 
  {
149
 
    internal::fn_format(buff, identifier.getPath().c_str(), "", *ext,
150
 
                        MY_UNPACK_FILENAME|MY_APPEND_EXT);
151
 
    if (internal::my_delete_with_symlink(buff, MYF(0)))
152
 
    {
153
 
      if ((error= errno) != ENOENT)
154
 
        break;
155
 
    }
156
 
    else
157
 
    {
158
 
      enoent_or_zero= 0;                        // No error for ENOENT
159
 
    }
160
 
 
161
 
    error= enoent_or_zero;
162
 
  }
163
 
  return error;
164
 
}
165
 
 
166
 
bool StorageEngine::addPlugin(StorageEngine *engine)
167
 
{
168
 
 
169
 
  vector_of_engines.push_back(engine);
170
 
 
171
 
  if (engine->getTableDefinitionFileExtension().length())
172
 
  {
173
 
    assert(engine->getTableDefinitionFileExtension().length() == DEFAULT_DEFINITION_FILE_EXT.length());
174
 
    set_of_table_definition_ext.insert(engine->getTableDefinitionFileExtension());
175
 
  }
176
 
 
177
 
  if (engine->check_flag(HTON_BIT_SCHEMA_DICTIONARY))
178
 
    vector_of_schema_engines.push_back(engine);
179
 
 
180
 
  return false;
181
 
}
182
 
 
183
 
void StorageEngine::removePlugin(StorageEngine *)
184
 
{
185
 
  if (shutdown_has_begun == false)
186
 
  {
187
 
    vector_of_engines.clear();
188
 
    vector_of_schema_engines.clear();
189
 
 
190
 
    shutdown_has_begun= true;
191
 
  }
192
 
}
193
 
 
194
 
class FindEngineByName
195
 
  : public std::unary_function<StorageEngine *, bool>
196
 
{
197
 
  const std::string &predicate;
198
 
 
199
 
public:
200
 
  explicit FindEngineByName(const std::string &target_arg) :
201
 
    predicate(target_arg)
202
 
  {
203
 
  }
204
 
 
205
 
  result_type operator() (argument_type engine)
206
 
  {
207
 
    return boost::iequals(engine->getName(), predicate);
208
 
  }
209
 
};
210
 
 
211
 
StorageEngine *StorageEngine::findByName(const std::string &predicate)
212
 
{
213
 
  EngineVector::iterator iter= std::find_if(vector_of_engines.begin(),
214
 
                                            vector_of_engines.end(),
215
 
                                            FindEngineByName(predicate));
216
 
  if (iter != vector_of_engines.end())
217
 
  {
218
 
    StorageEngine *engine= *iter;
219
 
    if (engine->is_user_selectable())
220
 
      return engine;
221
 
  }
222
 
 
223
 
  return NULL;
224
 
}
225
 
 
226
 
StorageEngine *StorageEngine::findByName(Session& session, const std::string &predicate)
227
 
{
228
 
  if (boost::iequals(predicate, DEFAULT_STRING))
229
 
    return session.getDefaultStorageEngine();
230
 
 
231
 
  EngineVector::iterator iter= std::find_if(vector_of_engines.begin(),
232
 
                                            vector_of_engines.end(),
233
 
                                            FindEngineByName(predicate));
234
 
  if (iter != vector_of_engines.end())
235
 
  {
236
 
    StorageEngine *engine= *iter;
237
 
    if (engine->is_user_selectable())
238
 
      return engine;
239
 
  }
240
 
 
241
 
  return NULL;
242
 
}
243
 
 
244
 
class StorageEngineCloseConnection : public std::unary_function<StorageEngine *, void>
245
 
{
246
 
  Session *session;
247
 
public:
248
 
  StorageEngineCloseConnection(Session *session_arg) : session(session_arg) {}
249
 
  /*
250
 
    there's no need to rollback here as all transactions must
251
 
    be rolled back already
252
 
  */
253
 
  inline result_type operator() (argument_type engine)
254
 
  {
255
 
    if (*session->getEngineData(engine))
256
 
      engine->close_connection(session);
257
 
  }
258
 
};
259
 
 
260
 
/**
261
 
  @note
262
 
    don't bother to rollback here, it's done already
263
 
*/
264
 
void StorageEngine::closeConnection(Session* session)
265
 
{
266
 
  std::for_each(vector_of_engines.begin(), vector_of_engines.end(),
267
 
                StorageEngineCloseConnection(session));
268
 
}
269
 
 
270
 
bool StorageEngine::flushLogs(StorageEngine *engine)
271
 
{
272
 
  if (engine == NULL)
273
 
  {
274
 
    if (std::find_if(vector_of_engines.begin(), vector_of_engines.end(),
275
 
                     std::mem_fun(&StorageEngine::flush_logs))
276
 
        != vector_of_engines.begin())
277
 
      return true;
278
 
  }
279
 
  else
280
 
  {
281
 
    if (engine->flush_logs())
282
 
      return true;
283
 
  }
284
 
  return false;
285
 
}
286
 
 
287
 
class StorageEngineGetTableDefinition: public std::unary_function<StorageEngine *,bool>
288
 
{
289
 
  Session& session;
290
 
  const identifier::Table &identifier;
291
 
  message::Table &table_message;
292
 
  drizzled::error_t &err;
293
 
 
294
 
public:
295
 
  StorageEngineGetTableDefinition(Session& session_arg,
296
 
                                  const identifier::Table &identifier_arg,
297
 
                                  message::Table &table_message_arg,
298
 
                                  drizzled::error_t &err_arg) :
299
 
    session(session_arg), 
300
 
    identifier(identifier_arg),
301
 
    table_message(table_message_arg), 
302
 
    err(err_arg) {}
303
 
 
304
 
  result_type operator() (argument_type engine)
305
 
  {
306
 
    int ret= engine->doGetTableDefinition(session, identifier, table_message);
307
 
 
308
 
    if (ret != ENOENT)
309
 
      err= static_cast<drizzled::error_t>(ret);
310
 
 
311
 
    return err == static_cast<drizzled::error_t>(EEXIST) or err != static_cast<drizzled::error_t>(ENOENT);
312
 
  }
313
 
};
314
 
 
315
 
class StorageEngineDoesTableExist: public std::unary_function<StorageEngine *, bool>
316
 
{
317
 
  Session& session;
318
 
  const identifier::Table &identifier;
319
 
 
320
 
public:
321
 
  StorageEngineDoesTableExist(Session& session_arg, const identifier::Table &identifier_arg) :
322
 
    session(session_arg), 
323
 
    identifier(identifier_arg) 
324
 
  { }
325
 
 
326
 
  result_type operator() (argument_type engine)
327
 
  {
328
 
    return engine->doDoesTableExist(session, identifier);
329
 
  }
330
 
};
331
 
 
332
 
/**
333
 
  Utility method which hides some of the details of getTableDefinition()
334
 
*/
335
 
bool plugin::StorageEngine::doesTableExist(Session &session,
336
 
                                           const identifier::Table &identifier,
337
 
                                           bool include_temporary_tables)
338
 
{
339
 
  if (include_temporary_tables)
340
 
  {
341
 
    if (session.doDoesTableExist(identifier))
342
 
      return true;
343
 
  }
344
 
 
345
 
  EngineVector::iterator iter=
346
 
    std::find_if(vector_of_engines.begin(), vector_of_engines.end(),
347
 
                 StorageEngineDoesTableExist(session, identifier));
348
 
 
349
 
  if (iter == vector_of_engines.end())
350
 
  {
351
 
    return false;
352
 
  }
353
 
 
354
 
  return true;
355
 
}
356
 
 
357
 
bool plugin::StorageEngine::doDoesTableExist(Session&, const drizzled::identifier::Table&)
358
 
{
359
 
  std::cerr << " Engine was called for doDoesTableExist() and does not implement it: " << this->getName() << "\n";
360
 
  assert(0);
361
 
  return false;
362
 
}
363
 
 
364
 
/**
365
 
  Call this function in order to give the Cursor the possiblity
366
 
  to ask engine if there are any new tables that should be written to disk
367
 
  or any dropped tables that need to be removed from disk
368
 
*/
369
 
int StorageEngine::getTableDefinition(Session& session,
370
 
                                      const identifier::Table &identifier,
371
 
                                      message::table::shared_ptr &table_message,
372
 
                                      bool include_temporary_tables)
373
 
{
374
 
  drizzled::error_t err= static_cast<drizzled::error_t>(ENOENT);
375
 
 
376
 
  if (include_temporary_tables)
377
 
  {
378
 
    Table *table= session.find_temporary_table(identifier);
379
 
    if (table)
380
 
    {
381
 
      table_message.reset(new message::Table(*table->getShare()->getTableMessage()));
382
 
      return EEXIST;
383
 
    }
384
 
  }
385
 
 
386
 
  drizzled::message::table::shared_ptr table_ptr;
387
 
  if ((table_ptr= drizzled::message::Cache::singleton().find(identifier)))
388
 
  {
389
 
    table_message= table_ptr;
390
 
  }
391
 
 
392
 
  message::Table message;
393
 
  EngineVector::iterator iter=
394
 
    std::find_if(vector_of_engines.begin(), vector_of_engines.end(),
395
 
                 StorageEngineGetTableDefinition(session, identifier, message, err));
396
 
 
397
 
  if (iter == vector_of_engines.end())
398
 
  {
399
 
    return ENOENT;
400
 
  }
401
 
  table_message.reset(new message::Table(message));
402
 
 
403
 
 drizzled::message::Cache::singleton().insert(identifier, table_message);
404
 
 
405
 
  return err;
406
 
}
407
 
 
408
 
message::table::shared_ptr StorageEngine::getTableMessage(Session& session,
409
 
                                                          identifier::Table::const_reference identifier,
410
 
                                                          drizzled::error_t &error,
411
 
                                                          bool include_temporary_tables)
412
 
{
413
 
  error= static_cast<drizzled::error_t>(ENOENT);
414
 
 
415
 
  if (include_temporary_tables)
416
 
  {
417
 
    Table *table= session.find_temporary_table(identifier);
418
 
    if (table)
419
 
    {
420
 
      error= EE_OK;
421
 
      return message::table::shared_ptr(new message::Table(*table->getShare()->getTableMessage()));
422
 
    }
423
 
  }
424
 
 
425
 
  drizzled::message::table::shared_ptr table_ptr;
426
 
  if ((table_ptr= drizzled::message::Cache::singleton().find(identifier)))
427
 
  {
428
 
    (void)table_ptr;
429
 
  }
430
 
 
431
 
  message::Table message;
432
 
  EngineVector::iterator iter=
433
 
    std::find_if(vector_of_engines.begin(), vector_of_engines.end(),
434
 
                 StorageEngineGetTableDefinition(session, identifier, message, error));
435
 
 
436
 
  if (iter == vector_of_engines.end())
437
 
  {
438
 
    error= static_cast<drizzled::error_t>(ENOENT);
439
 
    return message::table::shared_ptr();
440
 
  }
441
 
  message::table::shared_ptr table_message(new message::Table(message));
442
 
 
443
 
  drizzled::message::Cache::singleton().insert(identifier, table_message);
444
 
 
445
 
  return table_message;
446
 
}
447
 
 
448
 
/**
449
 
  An interceptor to hijack the text of the error message without
450
 
  setting an error in the thread. We need the text to present it
451
 
  in the form of a warning to the user.
452
 
*/
453
 
 
454
 
class Ha_delete_table_error_handler: public Internal_error_handler
455
 
{
456
 
public:
457
 
  Ha_delete_table_error_handler() : Internal_error_handler() {}
458
 
  virtual bool handle_error(drizzled::error_t sql_errno,
459
 
                            const char *message,
460
 
                            DRIZZLE_ERROR::enum_warning_level level,
461
 
                            Session *session);
462
 
  char buff[DRIZZLE_ERRMSG_SIZE];
463
 
};
464
 
 
465
 
 
466
 
bool
467
 
Ha_delete_table_error_handler::
468
 
handle_error(drizzled::error_t ,
469
 
             const char *message,
470
 
             DRIZZLE_ERROR::enum_warning_level ,
471
 
             Session *)
472
 
{
473
 
  /* Grab the error message */
474
 
  strncpy(buff, message, sizeof(buff)-1);
475
 
  return true;
476
 
}
477
 
 
478
 
class DropTableByIdentifier: public std::unary_function<EngineVector::value_type, bool>
479
 
{
480
 
  Session::reference session;
481
 
  identifier::Table::const_reference identifier;
482
 
  drizzled::error_t &error;
483
 
 
484
 
public:
485
 
 
486
 
  DropTableByIdentifier(Session::reference session_arg,
487
 
                        identifier::Table::const_reference identifier_arg,
488
 
                        drizzled::error_t &error_arg) :
489
 
    session(session_arg),
490
 
    identifier(identifier_arg),
491
 
    error(error_arg)
492
 
  { }
493
 
 
494
 
  result_type operator() (argument_type engine)
495
 
  {
496
 
    if (not engine->doDoesTableExist(session, identifier))
497
 
      return false;
498
 
 
499
 
    int local_error= engine->doDropTable(session, identifier);
500
 
 
501
 
 
502
 
    if (not local_error)
503
 
      return true;
504
 
 
505
 
    switch (local_error)
506
 
    {
507
 
    case HA_ERR_NO_SUCH_TABLE:
508
 
    case ENOENT:
509
 
      error= static_cast<drizzled::error_t>(HA_ERR_NO_SUCH_TABLE);
510
 
      return false;
511
 
 
512
 
    default:
513
 
      error= static_cast<drizzled::error_t>(local_error);
514
 
      return true;
515
 
    }
516
 
  } 
517
 
};
518
 
 
519
 
 
520
 
bool StorageEngine::dropTable(Session::reference session,
521
 
                              identifier::Table::const_reference identifier,
522
 
                              drizzled::error_t &error)
523
 
{
524
 
  error= EE_OK;
525
 
 
526
 
  EngineVector::const_iterator iter= std::find_if(vector_of_engines.begin(), vector_of_engines.end(),
527
 
                                                  DropTableByIdentifier(session, identifier, error));
528
 
 
529
 
  if (error)
530
 
  {
531
 
    return false;
532
 
  }
533
 
  else if (iter == vector_of_engines.end())
534
 
  {
535
 
    error= ER_BAD_TABLE_ERROR;
536
 
    return false;
537
 
  }
538
 
 
539
 
  drizzled::message::Cache::singleton().erase(identifier);
540
 
 
541
 
  return true;
542
 
}
543
 
 
544
 
bool StorageEngine::dropTable(Session& session,
545
 
                              const identifier::Table &identifier)
546
 
{
547
 
  drizzled::error_t error;
548
 
 
549
 
  if (not dropTable(session, identifier, error))
550
 
  {
551
 
    return false;
552
 
  }
553
 
 
554
 
  return true;
555
 
}
556
 
 
557
 
bool StorageEngine::dropTable(Session::reference session,
558
 
                              StorageEngine &engine,
559
 
                              identifier::Table::const_reference identifier,
560
 
                              drizzled::error_t &error)
561
 
{
562
 
  error= EE_OK;
563
 
  engine.setTransactionReadWrite(session);
564
 
 
565
 
  assert(identifier.isTmp());
566
 
  
567
 
  if (unlikely(plugin::EventObserver::beforeDropTable(session, identifier)))
568
 
  {
569
 
    error= ER_EVENT_OBSERVER_PLUGIN;
570
 
  }
571
 
  else
572
 
  {
573
 
    error= static_cast<drizzled::error_t>(engine.doDropTable(session, identifier));
574
 
 
575
 
    if (unlikely(plugin::EventObserver::afterDropTable(session, identifier, error)))
576
 
    {
577
 
      error= ER_EVENT_OBSERVER_PLUGIN;
578
 
    }
579
 
  }
580
 
 
581
 
  drizzled::message::Cache::singleton().erase(identifier);
582
 
 
583
 
  if (error)
584
 
  {
585
 
    return false;
586
 
  }
587
 
 
588
 
  return true;
589
 
}
590
 
 
591
 
 
592
 
/**
593
 
  Initiates table-file and calls appropriate database-creator.
594
 
 
595
 
  @retval
596
 
   0  ok
597
 
  @retval
598
 
   1  error
599
 
*/
600
 
bool StorageEngine::createTable(Session &session,
601
 
                                const identifier::Table &identifier,
602
 
                                message::Table& table_message)
603
 
{
604
 
  drizzled::error_t error= EE_OK;
605
 
 
606
 
  TableShare share(identifier);
607
 
  table::Shell table(share);
608
 
  message::Table tmp_proto;
609
 
 
610
 
  if (share.parse_table_proto(session, table_message) || share.open_table_from_share(&session, identifier, "", 0, 0, table))
611
 
  { 
612
 
    // @note Error occured, we should probably do a little more here.
613
 
    // ER_CORRUPT_TABLE_DEFINITION,ER_CORRUPT_TABLE_DEFINITION_ENUM 
614
 
    
615
 
    my_error(ER_CORRUPT_TABLE_DEFINITION_UNKNOWN, identifier);
616
 
 
617
 
    return false;
618
 
  }
619
 
  else
620
 
  {
621
 
    /* Check for legal operations against the Engine using the proto (if used) */
622
 
    if (table_message.type() == message::Table::TEMPORARY &&
623
 
        share.storage_engine->check_flag(HTON_BIT_TEMPORARY_NOT_SUPPORTED) == true)
624
 
    {
625
 
      error= HA_ERR_UNSUPPORTED;
626
 
    }
627
 
    else if (table_message.type() != message::Table::TEMPORARY &&
628
 
             share.storage_engine->check_flag(HTON_BIT_TEMPORARY_ONLY) == true)
629
 
    {
630
 
      error= HA_ERR_UNSUPPORTED;
631
 
    }
632
 
    else
633
 
    {
634
 
      share.storage_engine->setTransactionReadWrite(session);
635
 
 
636
 
      error= static_cast<drizzled::error_t>(share.storage_engine->doCreateTable(session,
637
 
                                                                                table,
638
 
                                                                                identifier,
639
 
                                                                                table_message));
640
 
    }
641
 
 
642
 
    if (error == ER_TABLE_PERMISSION_DENIED)
643
 
    {
644
 
      my_error(ER_TABLE_PERMISSION_DENIED, identifier);
645
 
    }
646
 
    else if (error)
647
 
    {
648
 
      std::string path;
649
 
      identifier.getSQLPath(path);
650
 
      my_error(ER_CANT_CREATE_TABLE, MYF(ME_BELL+ME_WAITTANG), path.c_str(), error);
651
 
    }
652
 
 
653
 
    table.delete_table();
654
 
  }
655
 
 
656
 
  return(error == EE_OK);
657
 
}
658
 
 
659
 
Cursor *StorageEngine::getCursor(Table &arg)
660
 
{
661
 
  return create(arg);
662
 
}
663
 
 
664
 
class AddTableIdentifier : 
665
 
  public std::unary_function<StorageEngine *, void>
666
 
{
667
 
  CachedDirectory &directory;
668
 
  const identifier::Schema &identifier;
669
 
  identifier::Table::vector &set_of_identifiers;
670
 
 
671
 
public:
672
 
 
673
 
  AddTableIdentifier(CachedDirectory &directory_arg, const identifier::Schema &identifier_arg, identifier::Table::vector &of_names) :
674
 
    directory(directory_arg),
675
 
    identifier(identifier_arg),
676
 
    set_of_identifiers(of_names)
677
 
  {
678
 
  }
679
 
 
680
 
  result_type operator() (argument_type engine)
681
 
  {
682
 
    engine->doGetTableIdentifiers(directory, identifier, set_of_identifiers);
683
 
  }
684
 
};
685
 
 
686
 
 
687
 
void StorageEngine::getIdentifiers(Session &session, const identifier::Schema &schema_identifier, identifier::Table::vector &set_of_identifiers)
688
 
{
689
 
  CachedDirectory directory(schema_identifier.getPath(), set_of_table_definition_ext);
690
 
 
691
 
  if (schema_identifier == INFORMATION_SCHEMA_IDENTIFIER)
692
 
  { }
693
 
  else if (schema_identifier == DATA_DICTIONARY_IDENTIFIER)
694
 
  { }
695
 
  else
696
 
  {
697
 
    if (directory.fail())
698
 
    {
699
 
      errno= directory.getError();
700
 
      if (errno == ENOENT)
701
 
      {
702
 
        std::string path;
703
 
        schema_identifier.getSQLPath(path);
704
 
        my_error(ER_BAD_DB_ERROR, MYF(ME_BELL+ME_WAITTANG), path.c_str());
705
 
      }
706
 
      else
707
 
      {
708
 
        my_error(ER_CANT_READ_DIR, MYF(ME_BELL+ME_WAITTANG), directory.getPath(), errno);
709
 
      }
710
 
 
711
 
      return;
712
 
    }
713
 
  }
714
 
 
715
 
  std::for_each(vector_of_engines.begin(), vector_of_engines.end(),
716
 
                AddTableIdentifier(directory, schema_identifier, set_of_identifiers));
717
 
 
718
 
  session.doGetTableIdentifiers(directory, schema_identifier, set_of_identifiers);
719
 
}
720
 
 
721
 
class DropTable: public std::unary_function<identifier::Table&, bool>
722
 
{
723
 
  Session &session;
724
 
  StorageEngine *engine;
725
 
 
726
 
public:
727
 
 
728
 
  DropTable(Session &session_arg, StorageEngine *engine_arg) :
729
 
    session(session_arg),
730
 
    engine(engine_arg)
731
 
  { }
732
 
 
733
 
  result_type operator() (argument_type identifier)
734
 
  {
735
 
    return engine->doDropTable(session, identifier) == 0;
736
 
  } 
737
 
};
738
 
 
739
 
/* This will later be converted to identifier::Tables */
740
 
class DropTables: public std::unary_function<StorageEngine *, void>
741
 
{
742
 
  Session &session;
743
 
  identifier::Table::vector &table_identifiers;
744
 
 
745
 
public:
746
 
 
747
 
  DropTables(Session &session_arg, identifier::Table::vector &table_identifiers_arg) :
748
 
    session(session_arg),
749
 
    table_identifiers(table_identifiers_arg)
750
 
  { }
751
 
 
752
 
  result_type operator() (argument_type engine)
753
 
  {
754
 
    // True returning from DropTable means the table has been successfully
755
 
    // deleted, so it should be removed from the list of tables to drop
756
 
    table_identifiers.erase(std::remove_if(table_identifiers.begin(),
757
 
                                           table_identifiers.end(),
758
 
                                           DropTable(session, engine)),
759
 
                            table_identifiers.end());
760
 
  }
761
 
};
 
28
 
 
29
#include CSTDINT_H
762
30
 
763
31
/*
764
 
  This only works for engines which use file based DFE.
765
 
 
766
 
  Note-> Unlike MySQL, we do not, on purpose, delete files that do not match any engines. 
767
 
*/
768
 
void StorageEngine::removeLostTemporaryTables(Session &session, const char *directory)
769
 
{
770
 
  CachedDirectory dir(directory, set_of_table_definition_ext);
771
 
  identifier::Table::vector table_identifiers;
772
 
 
773
 
  if (dir.fail())
774
 
  {
775
 
    errno= dir.getError();
776
 
    my_error(ER_CANT_READ_DIR, MYF(0), directory, errno);
777
 
 
778
 
    return;
779
 
  }
780
 
 
781
 
  CachedDirectory::Entries files= dir.getEntries();
782
 
 
783
 
  for (CachedDirectory::Entries::iterator fileIter= files.begin();
784
 
       fileIter != files.end(); fileIter++)
785
 
  {
786
 
    size_t length;
787
 
    std::string path;
788
 
    CachedDirectory::Entry *entry= *fileIter;
789
 
 
790
 
    /* We remove the file extension. */
791
 
    length= entry->filename.length();
792
 
    entry->filename.resize(length - DEFAULT_DEFINITION_FILE_EXT.length());
793
 
 
794
 
    path+= directory;
795
 
    path+= FN_LIBCHAR;
796
 
    path+= entry->filename;
797
 
    message::Table definition;
798
 
    if (StorageEngine::readTableFile(path, definition))
799
 
    {
800
 
      identifier::Table identifier(definition.schema(), definition.name(), path);
801
 
      table_identifiers.push_back(identifier);
802
 
    }
803
 
  }
804
 
 
805
 
  std::for_each(vector_of_engines.begin(), vector_of_engines.end(),
806
 
                DropTables(session, table_identifiers));
807
 
 
808
 
  /*
809
 
    Now we just clean up anything that might left over.
810
 
 
811
 
    We rescan because some of what might have been there should
812
 
    now be all nice and cleaned up.
813
 
  */
814
 
  std::set<std::string> all_exts= set_of_table_definition_ext;
815
 
 
816
 
  for (EngineVector::iterator iter= vector_of_engines.begin();
817
 
       iter != vector_of_engines.end() ; iter++)
818
 
  {
819
 
    for (const char **ext= (*iter)->bas_ext(); *ext ; ext++)
820
 
      all_exts.insert(*ext);
821
 
  }
822
 
 
823
 
  CachedDirectory rescan(directory, all_exts);
824
 
 
825
 
  files= rescan.getEntries();
826
 
  for (CachedDirectory::Entries::iterator fileIter= files.begin();
827
 
       fileIter != files.end(); fileIter++)
828
 
  {
829
 
    std::string path;
830
 
    CachedDirectory::Entry *entry= *fileIter;
831
 
 
832
 
    path+= directory;
833
 
    path+= FN_LIBCHAR;
834
 
    path+= entry->filename;
835
 
 
836
 
    unlink(path.c_str());
837
 
  }
838
 
}
839
 
 
840
 
 
841
 
/**
842
 
  Print error that we got from Cursor function.
843
 
 
844
 
  @note
845
 
    In case of delete table it's only safe to use the following parts of
846
 
    the 'table' structure:
847
 
    - table->getShare()->path
848
 
    - table->alias
849
 
*/
850
 
void StorageEngine::print_error(int error, myf errflag, const Table &table) const
851
 
{
852
 
  drizzled::error_t textno= ER_GET_ERRNO;
853
 
  switch (error) {
854
 
  case EACCES:
855
 
    textno=ER_OPEN_AS_READONLY;
856
 
    break;
857
 
  case EAGAIN:
858
 
    textno=ER_FILE_USED;
859
 
    break;
860
 
  case ENOENT:
861
 
    textno=ER_FILE_NOT_FOUND;
862
 
    break;
863
 
  case HA_ERR_KEY_NOT_FOUND:
864
 
  case HA_ERR_NO_ACTIVE_RECORD:
865
 
  case HA_ERR_END_OF_FILE:
866
 
    textno=ER_KEY_NOT_FOUND;
867
 
    break;
868
 
  case HA_ERR_WRONG_MRG_TABLE_DEF:
869
 
    textno=ER_WRONG_MRG_TABLE;
870
 
    break;
871
 
  case HA_ERR_FOUND_DUPP_KEY:
872
 
  {
873
 
    uint32_t key_nr= table.get_dup_key(error);
874
 
    if ((int) key_nr >= 0)
875
 
    {
876
 
      const char *err_msg= ER(ER_DUP_ENTRY_WITH_KEY_NAME);
877
 
 
878
 
      print_keydup_error(key_nr, err_msg, table);
879
 
 
880
 
      return;
881
 
    }
882
 
    textno=ER_DUP_KEY;
883
 
    break;
884
 
  }
885
 
  case HA_ERR_FOREIGN_DUPLICATE_KEY:
886
 
  {
887
 
    uint32_t key_nr= table.get_dup_key(error);
888
 
    if ((int) key_nr >= 0)
889
 
    {
890
 
      uint32_t max_length;
891
 
 
892
 
      /* Write the key in the error message */
893
 
      char key[MAX_KEY_LENGTH];
894
 
      String str(key,sizeof(key),system_charset_info);
895
 
 
896
 
      /* Table is opened and defined at this point */
897
 
      key_unpack(&str, &table,(uint32_t) key_nr);
898
 
      max_length= (DRIZZLE_ERRMSG_SIZE-
899
 
                   (uint32_t) strlen(ER(ER_FOREIGN_DUPLICATE_KEY)));
900
 
      if (str.length() >= max_length)
901
 
      {
902
 
        str.length(max_length-4);
903
 
        str.append(STRING_WITH_LEN("..."));
904
 
      }
905
 
      my_error(ER_FOREIGN_DUPLICATE_KEY, MYF(0), table.getShare()->getTableName(),
906
 
        str.c_ptr(), key_nr+1);
907
 
      return;
908
 
    }
909
 
    textno= ER_DUP_KEY;
910
 
    break;
911
 
  }
912
 
  case HA_ERR_FOUND_DUPP_UNIQUE:
913
 
    textno=ER_DUP_UNIQUE;
914
 
    break;
915
 
  case HA_ERR_RECORD_CHANGED:
916
 
    textno=ER_CHECKREAD;
917
 
    break;
918
 
  case HA_ERR_CRASHED:
919
 
    textno=ER_NOT_KEYFILE;
920
 
    break;
921
 
  case HA_ERR_WRONG_IN_RECORD:
922
 
    textno= ER_CRASHED_ON_USAGE;
923
 
    break;
924
 
  case HA_ERR_CRASHED_ON_USAGE:
925
 
    textno=ER_CRASHED_ON_USAGE;
926
 
    break;
927
 
  case HA_ERR_NOT_A_TABLE:
928
 
    textno= static_cast<drizzled::error_t>(error);
929
 
    break;
930
 
  case HA_ERR_CRASHED_ON_REPAIR:
931
 
    textno=ER_CRASHED_ON_REPAIR;
932
 
    break;
933
 
  case HA_ERR_OUT_OF_MEM:
934
 
    textno=ER_OUT_OF_RESOURCES;
935
 
    break;
936
 
  case HA_ERR_WRONG_COMMAND:
937
 
    textno=ER_ILLEGAL_HA;
938
 
    break;
939
 
  case HA_ERR_OLD_FILE:
940
 
    textno=ER_OLD_KEYFILE;
941
 
    break;
942
 
  case HA_ERR_UNSUPPORTED:
943
 
    textno=ER_UNSUPPORTED_EXTENSION;
944
 
    break;
945
 
  case HA_ERR_RECORD_FILE_FULL:
946
 
  case HA_ERR_INDEX_FILE_FULL:
947
 
    textno=ER_RECORD_FILE_FULL;
948
 
    break;
949
 
  case HA_ERR_LOCK_WAIT_TIMEOUT:
950
 
    textno=ER_LOCK_WAIT_TIMEOUT;
951
 
    break;
952
 
  case HA_ERR_LOCK_TABLE_FULL:
953
 
    textno=ER_LOCK_TABLE_FULL;
954
 
    break;
955
 
  case HA_ERR_LOCK_DEADLOCK:
956
 
    textno=ER_LOCK_DEADLOCK;
957
 
    break;
958
 
  case HA_ERR_READ_ONLY_TRANSACTION:
959
 
    textno=ER_READ_ONLY_TRANSACTION;
960
 
    break;
961
 
  case HA_ERR_CANNOT_ADD_FOREIGN:
962
 
    textno=ER_CANNOT_ADD_FOREIGN;
963
 
    break;
964
 
  case HA_ERR_ROW_IS_REFERENCED:
965
 
  {
966
 
    String str;
967
 
    get_error_message(error, &str);
968
 
    my_error(ER_ROW_IS_REFERENCED_2, MYF(0), str.c_ptr_safe());
969
 
    return;
970
 
  }
971
 
  case HA_ERR_NO_REFERENCED_ROW:
972
 
  {
973
 
    String str;
974
 
    get_error_message(error, &str);
975
 
    my_error(ER_NO_REFERENCED_ROW_2, MYF(0), str.c_ptr_safe());
976
 
    return;
977
 
  }
978
 
  case HA_ERR_TABLE_DEF_CHANGED:
979
 
    textno=ER_TABLE_DEF_CHANGED;
980
 
    break;
981
 
  case HA_ERR_NO_SUCH_TABLE:
982
 
    {
983
 
      identifier::Table identifier(table.getShare()->getSchemaName(), table.getShare()->getTableName());
984
 
      my_error(ER_TABLE_UNKNOWN, identifier);
985
 
      return;
986
 
    }
987
 
  case HA_ERR_RBR_LOGGING_FAILED:
988
 
    textno= ER_BINLOG_ROW_LOGGING_FAILED;
989
 
    break;
990
 
  case HA_ERR_DROP_INDEX_FK:
991
 
  {
992
 
    const char *ptr= "???";
993
 
    uint32_t key_nr= table.get_dup_key(error);
994
 
    if ((int) key_nr >= 0)
995
 
      ptr= table.key_info[key_nr].name;
996
 
    my_error(ER_DROP_INDEX_FK, MYF(0), ptr);
997
 
    return;
998
 
  }
999
 
  case HA_ERR_TABLE_NEEDS_UPGRADE:
1000
 
    textno=ER_TABLE_NEEDS_UPGRADE;
1001
 
    break;
1002
 
  case HA_ERR_TABLE_READONLY:
1003
 
    textno= ER_OPEN_AS_READONLY;
1004
 
    break;
1005
 
  case HA_ERR_AUTOINC_READ_FAILED:
1006
 
    textno= ER_AUTOINC_READ_FAILED;
1007
 
    break;
1008
 
  case HA_ERR_AUTOINC_ERANGE:
1009
 
    textno= ER_WARN_DATA_OUT_OF_RANGE;
1010
 
    break;
1011
 
  case HA_ERR_LOCK_OR_ACTIVE_TRANSACTION:
1012
 
    my_message(ER_LOCK_OR_ACTIVE_TRANSACTION,
1013
 
               ER(ER_LOCK_OR_ACTIVE_TRANSACTION), MYF(0));
1014
 
    return;
1015
 
  default:
1016
 
    {
1017
 
      /* 
1018
 
        The error was "unknown" to this function.
1019
 
        Ask Cursor if it has got a message for this error 
1020
 
      */
1021
 
      bool temporary= false;
1022
 
      String str;
1023
 
      temporary= get_error_message(error, &str);
1024
 
      if (!str.is_empty())
1025
 
      {
1026
 
        const char* engine_name= getName().c_str();
1027
 
        if (temporary)
1028
 
          my_error(ER_GET_TEMPORARY_ERRMSG, MYF(0), error, str.ptr(),
1029
 
                   engine_name);
1030
 
        else
1031
 
          my_error(ER_GET_ERRMSG, MYF(0), error, str.ptr(), engine_name);
1032
 
      }
1033
 
      else
1034
 
      {
1035
 
              my_error(ER_GET_ERRNO,errflag,error);
1036
 
      }
1037
 
      return;
1038
 
    }
1039
 
  }
1040
 
 
1041
 
  my_error(textno, errflag, table.getShare()->getTableName(), error);
1042
 
}
1043
 
 
1044
 
 
1045
 
/**
1046
 
  Return an error message specific to this Cursor.
1047
 
 
1048
 
  @param error  error code previously returned by Cursor
1049
 
  @param buf    pointer to String where to add error message
1050
 
 
1051
 
  @return
1052
 
    Returns true if this is a temporary error
1053
 
*/
1054
 
bool StorageEngine::get_error_message(int , String* ) const
1055
 
{
1056
 
  return false;
1057
 
}
1058
 
 
1059
 
 
1060
 
void StorageEngine::print_keydup_error(uint32_t key_nr, const char *msg, const Table &table) const
1061
 
{
1062
 
  /* Write the duplicated key in the error message */
1063
 
  char key[MAX_KEY_LENGTH];
1064
 
  String str(key,sizeof(key),system_charset_info);
1065
 
 
1066
 
  if (key_nr == MAX_KEY)
1067
 
  {
1068
 
    /* Key is unknown */
1069
 
    str.copy("", 0, system_charset_info);
1070
 
    my_printf_error(ER_DUP_ENTRY, msg, MYF(0), str.c_ptr(), "*UNKNOWN*");
1071
 
  }
1072
 
  else
1073
 
  {
1074
 
    /* Table is opened and defined at this point */
1075
 
    key_unpack(&str, &table, (uint32_t) key_nr);
1076
 
    uint32_t max_length=DRIZZLE_ERRMSG_SIZE-(uint32_t) strlen(msg);
1077
 
    if (str.length() >= max_length)
1078
 
    {
1079
 
      str.length(max_length-4);
1080
 
      str.append(STRING_WITH_LEN("..."));
1081
 
    }
1082
 
    my_printf_error(ER_DUP_ENTRY, msg,
1083
 
                    MYF(0), str.c_ptr(), table.key_info[key_nr].name);
1084
 
  }
1085
 
}
1086
 
 
1087
 
 
1088
 
int StorageEngine::deleteDefinitionFromPath(const identifier::Table &identifier)
1089
 
{
1090
 
  std::string path(identifier.getPath());
1091
 
 
1092
 
  path.append(DEFAULT_DEFINITION_FILE_EXT);
1093
 
 
1094
 
  return internal::my_delete(path.c_str(), MYF(0));
1095
 
}
1096
 
 
1097
 
int StorageEngine::renameDefinitionFromPath(const identifier::Table &dest, const identifier::Table &src)
1098
 
{
1099
 
  message::Table table_message;
1100
 
  std::string src_path(src.getPath());
1101
 
  std::string dest_path(dest.getPath());
1102
 
 
1103
 
  src_path.append(DEFAULT_DEFINITION_FILE_EXT);
1104
 
  dest_path.append(DEFAULT_DEFINITION_FILE_EXT);
1105
 
 
1106
 
  bool was_read= StorageEngine::readTableFile(src_path.c_str(), table_message);
1107
 
 
1108
 
  if (not was_read)
1109
 
  {
1110
 
    return ENOENT;
1111
 
  }
1112
 
 
1113
 
  dest.copyToTableMessage(table_message);
1114
 
 
1115
 
  int error= StorageEngine::writeDefinitionFromPath(dest, table_message);
1116
 
 
1117
 
  if (not error)
1118
 
  {
1119
 
    if (unlink(src_path.c_str()))
1120
 
      perror(src_path.c_str());
1121
 
  }
1122
 
 
1123
 
  return error;
1124
 
}
1125
 
 
1126
 
int StorageEngine::writeDefinitionFromPath(const identifier::Table &identifier, message::Table &table_message)
1127
 
{
1128
 
  char definition_file_tmp[FN_REFLEN];
1129
 
  std::string file_name(identifier.getPath());
1130
 
 
1131
 
  file_name.append(DEFAULT_DEFINITION_FILE_EXT);
1132
 
 
1133
 
  snprintf(definition_file_tmp, sizeof(definition_file_tmp), "%sXXXXXX", file_name.c_str());
1134
 
 
1135
 
  int fd= mkstemp(definition_file_tmp);
1136
 
 
1137
 
  if (fd == -1)
1138
 
  {
1139
 
    perror(definition_file_tmp);
1140
 
    return errno;
1141
 
  }
1142
 
 
1143
 
  google::protobuf::io::ZeroCopyOutputStream* output=
1144
 
    new google::protobuf::io::FileOutputStream(fd);
1145
 
 
1146
 
  bool success;
1147
 
 
1148
 
  try
1149
 
  {
1150
 
    success= table_message.SerializeToZeroCopyStream(output);
1151
 
  }
1152
 
  catch (...)
1153
 
  {
1154
 
    success= false;
1155
 
  }
1156
 
 
1157
 
  if (not success)
1158
 
  {
1159
 
    std::string error_message;
1160
 
    identifier.getSQLPath(error_message);
1161
 
 
1162
 
    my_error(ER_CORRUPT_TABLE_DEFINITION, MYF(0),
1163
 
             error_message.c_str(),
1164
 
             table_message.InitializationErrorString().c_str());
1165
 
    delete output;
1166
 
 
1167
 
    if (close(fd) == -1)
1168
 
      perror(definition_file_tmp);
1169
 
 
1170
 
    if (unlink(definition_file_tmp) == -1)
1171
 
      perror(definition_file_tmp);
1172
 
 
1173
 
    return ER_CORRUPT_TABLE_DEFINITION;
1174
 
  }
1175
 
 
1176
 
  delete output;
1177
 
 
1178
 
  if (close(fd) == -1)
1179
 
  {
1180
 
    int error= errno;
1181
 
    perror(definition_file_tmp);
1182
 
 
1183
 
    if (unlink(definition_file_tmp))
1184
 
      perror(definition_file_tmp);
1185
 
 
1186
 
    return error;
1187
 
  }
1188
 
 
1189
 
  if (rename(definition_file_tmp, file_name.c_str()) == -1)
1190
 
  {
1191
 
    int error= errno;
1192
 
    perror(definition_file_tmp);
1193
 
 
1194
 
    if (unlink(definition_file_tmp))
1195
 
      perror(definition_file_tmp);
1196
 
 
1197
 
    return error;
1198
 
  }
1199
 
 
1200
 
  return 0;
1201
 
}
1202
 
 
1203
 
class CanCreateTable: public std::unary_function<StorageEngine *, bool>
1204
 
{
1205
 
  const identifier::Table &identifier;
1206
 
 
1207
 
public:
1208
 
  CanCreateTable(const identifier::Table &identifier_arg) :
1209
 
    identifier(identifier_arg)
1210
 
  { }
1211
 
 
1212
 
  result_type operator() (argument_type engine)
1213
 
  {
1214
 
    return not engine->doCanCreateTable(identifier);
1215
 
  }
 
32
  While we have legacy_db_type, we have this array to
 
33
  check for dups and to find handlerton from legacy_db_type.
 
34
  Remove when legacy_db_type is finally gone
 
35
*/
 
36
st_plugin_int *hton2plugin[MAX_HA];
 
37
 
 
38
static handlerton *installed_htons[128];
 
39
 
 
40
static const LEX_STRING sys_table_aliases[]=
 
41
{
 
42
  { C_STRING_WITH_LEN("INNOBASE") },  { C_STRING_WITH_LEN("INNODB") },
 
43
  { C_STRING_WITH_LEN("HEAP") },      { C_STRING_WITH_LEN("MEMORY") },
 
44
  {NULL, 0}
1216
45
};
1217
46
 
1218
47
 
1219
 
/**
1220
 
  @note on success table can be created.
1221
 
*/
1222
 
bool StorageEngine::canCreateTable(const identifier::Table &identifier)
1223
 
{
1224
 
  EngineVector::iterator iter=
1225
 
    std::find_if(vector_of_engines.begin(), vector_of_engines.end(),
1226
 
                 CanCreateTable(identifier));
1227
 
 
1228
 
  if (iter == vector_of_engines.end())
1229
 
  {
1230
 
    return true;
1231
 
  }
1232
 
 
1233
 
  return false;
1234
 
}
1235
 
 
1236
 
bool StorageEngine::readTableFile(const std::string &path, message::Table &table_message)
1237
 
{
1238
 
  std::fstream input(path.c_str(), std::ios::in | std::ios::binary);
1239
 
 
1240
 
  if (input.good())
1241
 
  {
1242
 
    try {
1243
 
      if (table_message.ParseFromIstream(&input))
 
48
handlerton *ha_resolve_by_legacy_type(Session *session,
 
49
                                      enum legacy_db_type db_type)
 
50
{
 
51
  plugin_ref plugin;
 
52
  switch (db_type) {
 
53
  case DB_TYPE_DEFAULT:
 
54
    return ha_default_handlerton(session);
 
55
  default:
 
56
    if (db_type > DB_TYPE_UNKNOWN && db_type < DB_TYPE_DEFAULT &&
 
57
        (plugin= ha_lock_engine(session, installed_htons[db_type])))
 
58
      return plugin_data(plugin, handlerton*);
 
59
    /* fall through */
 
60
  case DB_TYPE_UNKNOWN:
 
61
    return NULL;
 
62
  }
 
63
}
 
64
 
 
65
 
 
66
static plugin_ref ha_default_plugin(Session *session)
 
67
{
 
68
  if (session->variables.table_plugin)
 
69
    return session->variables.table_plugin;
 
70
  return my_plugin_lock(session, &global_system_variables.table_plugin);
 
71
}
 
72
 
 
73
 
 
74
/**
 
75
  Return the default storage engine handlerton for thread
 
76
 
 
77
  @param ha_default_handlerton(session)
 
78
  @param session         current thread
 
79
 
 
80
  @return
 
81
    pointer to handlerton
 
82
*/
 
83
handlerton *ha_default_handlerton(Session *session)
 
84
{
 
85
  plugin_ref plugin= ha_default_plugin(session);
 
86
  assert(plugin);
 
87
  handlerton *hton= plugin_data(plugin, handlerton*);
 
88
  assert(hton);
 
89
  return hton;
 
90
}
 
91
 
 
92
 
 
93
/**
 
94
  Return the storage engine handlerton for the supplied name
 
95
 
 
96
  @param session         current thread
 
97
  @param name        name of storage engine
 
98
 
 
99
  @return
 
100
    pointer to storage engine plugin handle
 
101
*/
 
102
plugin_ref ha_resolve_by_name(Session *session, const LEX_STRING *name)
 
103
{
 
104
  const LEX_STRING *table_alias;
 
105
  plugin_ref plugin;
 
106
 
 
107
redo:
 
108
  /* my_strnncoll is a macro and gcc doesn't do early expansion of macro */
 
109
  if (session && !my_charset_utf8_general_ci.coll->strnncoll(&my_charset_utf8_general_ci,
 
110
                           (const unsigned char *)name->str, name->length,
 
111
                           (const unsigned char *)STRING_WITH_LEN("DEFAULT"), 0))
 
112
    return ha_default_plugin(session);
 
113
 
 
114
  if ((plugin= my_plugin_lock_by_name(session, name, DRIZZLE_STORAGE_ENGINE_PLUGIN)))
 
115
  {
 
116
    handlerton *hton= plugin_data(plugin, handlerton *);
 
117
    if (!(hton->flags.test(HTON_BIT_NOT_USER_SELECTABLE)))
 
118
      return plugin;
 
119
 
 
120
    /*
 
121
      unlocking plugin immediately after locking is relatively low cost.
 
122
    */
 
123
    plugin_unlock(session, plugin);
 
124
  }
 
125
 
 
126
  /*
 
127
    We check for the historical aliases.
 
128
  */
 
129
  for (table_alias= sys_table_aliases; table_alias->str; table_alias+= 2)
 
130
  {
 
131
    if (!my_strnncoll(&my_charset_utf8_general_ci,
 
132
                      (const unsigned char *)name->str, name->length,
 
133
                      (const unsigned char *)table_alias->str,
 
134
                      table_alias->length))
 
135
    {
 
136
      name= table_alias + 1;
 
137
      goto redo;
 
138
    }
 
139
  }
 
140
 
 
141
  return NULL;
 
142
}
 
143
 
 
144
 
 
145
plugin_ref ha_lock_engine(Session *session, handlerton *hton)
 
146
{
 
147
  if (hton)
 
148
  {
 
149
    st_plugin_int **plugin= hton2plugin + hton->slot;
 
150
 
 
151
    return my_plugin_lock(session, &plugin);
 
152
  }
 
153
  return NULL;
 
154
}
 
155
 
 
156
 
 
157
/**
 
158
  Use other database handler if databasehandler is not compiled in.
 
159
*/
 
160
handlerton *ha_checktype(Session *session, enum legacy_db_type database_type,
 
161
                          bool no_substitute, bool report_error)
 
162
{
 
163
  handlerton *hton= ha_resolve_by_legacy_type(session, database_type);
 
164
  if (ha_storage_engine_is_enabled(hton))
 
165
    return hton;
 
166
 
 
167
  if (no_substitute)
 
168
  {
 
169
    if (report_error)
 
170
    {
 
171
      const char *engine_name= ha_resolve_storage_engine_name(hton);
 
172
      my_error(ER_FEATURE_DISABLED,MYF(0),engine_name,engine_name);
 
173
    }
 
174
    return NULL;
 
175
  }
 
176
 
 
177
  return ha_default_handlerton(session);
 
178
} /* ha_checktype */
 
179
 
 
180
 
 
181
handler *get_new_handler(TABLE_SHARE *share, MEM_ROOT *alloc,
 
182
                         handlerton *db_type)
 
183
{
 
184
  handler *file;
 
185
 
 
186
  if (db_type && db_type->state == SHOW_OPTION_YES && db_type->create)
 
187
  {
 
188
    if ((file= db_type->create(db_type, share, alloc)))
 
189
      file->init();
 
190
    return(file);
 
191
  }
 
192
  /*
 
193
    Try the default table type
 
194
    Here the call to current_session() is ok as we call this function a lot of
 
195
    times but we enter this branch very seldom.
 
196
  */
 
197
  return(get_new_handler(share, alloc, ha_default_handlerton(current_session)));
 
198
}
 
199
 
 
200
 
 
201
int ha_finalize_handlerton(st_plugin_int *plugin)
 
202
{
 
203
  handlerton *hton= (handlerton *)plugin->data;
 
204
 
 
205
  switch (hton->state)
 
206
  {
 
207
  case SHOW_OPTION_NO:
 
208
  case SHOW_OPTION_DISABLED:
 
209
    break;
 
210
  case SHOW_OPTION_YES:
 
211
    if (installed_htons[hton->db_type] == hton)
 
212
      installed_htons[hton->db_type]= NULL;
 
213
    break;
 
214
  };
 
215
 
 
216
  if (hton && plugin->plugin->deinit)
 
217
    (void)plugin->plugin->deinit(hton);
 
218
 
 
219
  free((unsigned char*)hton);
 
220
 
 
221
  return(0);
 
222
}
 
223
 
 
224
 
 
225
int ha_initialize_handlerton(st_plugin_int *plugin)
 
226
{
 
227
  handlerton *hton;
 
228
 
 
229
  hton= (handlerton *)malloc(sizeof(handlerton));
 
230
  memset(hton, 0, sizeof(handlerton));
 
231
 
 
232
  /* Historical Requirement */
 
233
  plugin->data= hton; // shortcut for the future
 
234
  if (plugin->plugin->init)
 
235
  {
 
236
    if (plugin->plugin->init(hton))
 
237
    {
 
238
      sql_print_error(_("Plugin '%s' init function returned error."),
 
239
                      plugin->name.str);
 
240
      goto err;
 
241
    }
 
242
  }
 
243
 
 
244
  hton->name= plugin->name.str;
 
245
 
 
246
  /*
 
247
    the switch below and hton->state should be removed when
 
248
    command-line options for plugins will be implemented
 
249
  */
 
250
  switch (hton->state) {
 
251
  case SHOW_OPTION_NO:
 
252
    break;
 
253
  case SHOW_OPTION_YES:
 
254
    {
 
255
      uint32_t tmp;
 
256
      /* now check the db_type for conflict */
 
257
      if (hton->db_type <= DB_TYPE_UNKNOWN ||
 
258
          hton->db_type >= DB_TYPE_DEFAULT ||
 
259
          installed_htons[hton->db_type])
1244
260
      {
1245
 
        return true;
 
261
        int idx= (int) DB_TYPE_FIRST_DYNAMIC;
 
262
 
 
263
        while (idx < (int) DB_TYPE_DEFAULT && installed_htons[idx])
 
264
          idx++;
 
265
 
 
266
        if (idx == (int) DB_TYPE_DEFAULT)
 
267
        {
 
268
          sql_print_warning(_("Too many storage engines!"));
 
269
          return(1);
 
270
        }
 
271
        if (hton->db_type != DB_TYPE_UNKNOWN)
 
272
          sql_print_warning(_("Storage engine '%s' has conflicting typecode. "
 
273
                            "Assigning value %d."), plugin->plugin->name, idx);
 
274
        hton->db_type= (enum legacy_db_type) idx;
1246
275
      }
1247
 
    }
1248
 
    catch (...)
1249
 
    {
1250
 
      my_error(ER_CORRUPT_TABLE_DEFINITION, MYF(0),
1251
 
               table_message.name().empty() ? path.c_str() : table_message.name().c_str(),
1252
 
               table_message.InitializationErrorString().empty() ? "": table_message.InitializationErrorString().c_str());
1253
 
    }
1254
 
  }
1255
 
  else
1256
 
  {
1257
 
    perror(path.c_str());
1258
 
  }
1259
 
 
1260
 
  return false;
1261
 
}
1262
 
 
1263
 
std::ostream& operator<<(std::ostream& output, const StorageEngine &engine)
1264
 
{
1265
 
  output << "StorageEngine:(";
1266
 
  output <<  engine.getName();
1267
 
  output << ")";
1268
 
 
1269
 
  return output;
1270
 
}
1271
 
 
1272
 
} /* namespace plugin */
1273
 
} /* namespace drizzled */
 
276
      installed_htons[hton->db_type]= hton;
 
277
      tmp= hton->savepoint_offset;
 
278
      hton->savepoint_offset= savepoint_alloc_size;
 
279
      savepoint_alloc_size+= tmp;
 
280
      hton->slot= total_ha++;
 
281
      hton2plugin[hton->slot]=plugin;
 
282
      if (hton->prepare)
 
283
        total_ha_2pc++;
 
284
      break;
 
285
    }
 
286
    /* fall through */
 
287
  default:
 
288
    hton->state= SHOW_OPTION_DISABLED;
 
289
    break;
 
290
  }
 
291
 
 
292
  /*
 
293
    This is entirely for legacy. We will create a new "disk based" hton and a
 
294
    "memory" hton which will be configurable longterm. We should be able to
 
295
    remove partition and myisammrg.
 
296
  */
 
297
  if (strcmp(plugin->plugin->name, "MEMORY") == 0)
 
298
    heap_hton= hton;
 
299
 
 
300
  if (strcmp(plugin->plugin->name, "MyISAM") == 0)
 
301
    myisam_hton= hton;
 
302
 
 
303
  return(0);
 
304
err:
 
305
  return(1);
 
306
}
 
307
 
 
308
enum legacy_db_type ha_legacy_type(const handlerton *db_type)
 
309
{
 
310
  return (db_type == NULL) ? DB_TYPE_UNKNOWN : db_type->db_type;
 
311
}
 
312
 
 
313
const char *ha_resolve_storage_engine_name(const handlerton *db_type)
 
314
{
 
315
  return db_type == NULL ? "UNKNOWN" : hton2plugin[db_type->slot]->name.str;
 
316
}
 
317
 
 
318
bool ha_check_storage_engine_flag(const handlerton *db_type, const hton_flag_bits flag)
 
319
{
 
320
  return db_type == NULL ? false : db_type->flags.test(static_cast<size_t>(flag));
 
321
}
 
322
 
 
323
bool ha_storage_engine_is_enabled(const handlerton *db_type)
 
324
{
 
325
  return (db_type && db_type->create) ?
 
326
         (db_type->state == SHOW_OPTION_YES) : false;
 
327
}
 
328
 
 
329
LEX_STRING *ha_storage_engine_name(const handlerton *hton)
 
330
{
 
331
  return &hton2plugin[hton->slot]->name;
 
332
}