12
12
You should have received a copy of the GNU General Public License
13
13
along with this program; if not, write to the Free Software
14
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA */
16
#include "heap_priv.h"
14
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */
16
#include <drizzled/server_includes.h>
19
#include <storage/heap/ha_heap.h>
20
#include <storage/heap/heapdef.h>
17
21
#include <drizzled/error.h>
18
22
#include <drizzled/table.h>
19
23
#include <drizzled/session.h>
24
#include <drizzled/current_session.h>
20
25
#include <drizzled/field/timestamp.h>
21
26
#include <drizzled/field/varstring.h>
22
#include "drizzled/plugin/daemon.h"
24
#include <boost/thread/mutex.hpp>
32
using namespace drizzled;
35
static const string engine_name("MEMORY");
37
boost::mutex THR_LOCK_heap;
39
static const char *ha_heap_exts[] = {
43
class HeapEngine : public plugin::StorageEngine
46
explicit HeapEngine(string name_arg) :
47
plugin::StorageEngine(name_arg,
48
HTON_STATS_RECORDS_IS_EXACT |
53
HTON_SKIP_STORE_LOCK |
60
hp_panic(HA_PANIC_CLOSE);
63
virtual Cursor *create(Table &table)
65
return new ha_heap(*this, table);
68
const char **bas_ext() const {
72
int doCreateTable(Session &session,
74
const TableIdentifier &identifier,
75
message::Table &create_proto);
77
/* For whatever reason, internal tables can be created by Cursor::open()
79
Instead of diving down a rat hole, let's just cry ourselves to sleep
80
at night with this odd hackish workaround.
82
int heap_create_table(Session *session, const char *table_name,
85
message::Table &create_proto,
86
HP_SHARE **internal_share);
88
int doRenameTable(Session&, const TableIdentifier &from, const TableIdentifier &to);
90
int doDropTable(Session&, const TableIdentifier &identifier);
92
int doGetTableDefinition(Session& session,
93
const TableIdentifier &identifier,
94
message::Table &table_message);
96
uint32_t max_supported_keys() const { return MAX_KEY; }
97
uint32_t max_supported_key_part_length() const { return MAX_KEY_LENGTH; }
99
uint32_t index_flags(enum ha_key_alg ) const
101
return ( HA_ONLY_WHOLE_INDEX | HA_KEY_SCAN_NOT_ROR);
104
bool doDoesTableExist(Session& session, const TableIdentifier &identifier);
105
void doGetTableIdentifiers(CachedDirectory &directory,
106
const SchemaIdentifier &schema_identifier,
107
TableIdentifier::vector &set_of_identifiers);
110
void HeapEngine::doGetTableIdentifiers(CachedDirectory&,
111
const SchemaIdentifier&,
112
TableIdentifier::vector&)
116
bool HeapEngine::doDoesTableExist(Session& session, const TableIdentifier &identifier)
118
return session.getMessageCache().doesTableMessageExist(identifier);
121
int HeapEngine::doGetTableDefinition(Session &session,
122
const TableIdentifier &identifier,
123
message::Table &table_proto)
125
if (session.getMessageCache().getTableMessage(identifier, table_proto))
131
We have to ignore ENOENT entries as the MEMORY table is created on open and
132
not when doing a CREATE on the table.
134
int HeapEngine::doDropTable(Session &session, const TableIdentifier &identifier)
136
session.getMessageCache().removeTableMessage(identifier);
138
int error= heap_delete_table(identifier.getPath().c_str());
146
static HeapEngine *heap_storage_engine= NULL;
148
static int heap_init(module::Context &context)
150
heap_storage_engine= new HeapEngine(engine_name);
151
context.add(heap_storage_engine);
29
pthread_mutex_t THR_LOCK_heap= PTHREAD_MUTEX_INITIALIZER;
31
static handler *heap_create_handler(handlerton *hton,
35
int heap_deinit(void *)
37
pthread_mutex_init(&THR_LOCK_heap,MY_MUTEX_INIT_FAST);
39
return hp_panic(HA_PANIC_CLOSE);
43
int heap_init(void *p)
45
handlerton *heap_hton;
47
heap_hton= (handlerton *)p;
48
heap_hton->state= SHOW_OPTION_YES;
49
heap_hton->create= heap_create_handler;
50
heap_hton->flags= HTON_CAN_RECREATE;
55
static handler *heap_create_handler(handlerton *hton,
59
return new (mem_root) ha_heap(hton, table);
156
63
/*****************************************************************************
158
65
*****************************************************************************/
160
ha_heap::ha_heap(plugin::StorageEngine &engine_arg,
162
:Cursor(engine_arg, table_arg), file(0), records_changed(0), key_stat_version(0),
67
ha_heap::ha_heap(handlerton *hton, TABLE_SHARE *table_arg)
68
:handler(hton, table_arg), file(0), records_changed(0), key_stat_version(0),
73
static const char *ha_heap_exts[] = {
77
const char **ha_heap::bas_ext() const
167
83
Hash index statistics is updated (copied from HP_KEYDEF::hash_buckets to
168
rec_per_key) after 1/MEMORY_STATS_UPDATE_THRESHOLD fraction of table records
84
rec_per_key) after 1/HEAP_STATS_UPDATE_THRESHOLD fraction of table records
169
85
have been inserted/updated/deleted. delete_all_rows() and table flush cause
174
90
from 0 to non-zero value and vice versa. Otherwise records_in_range may
175
91
erroneously return 0 and 'range' may miss records.
177
#define MEMORY_STATS_UPDATE_THRESHOLD 10
93
#define HEAP_STATS_UPDATE_THRESHOLD 10
179
int ha_heap::doOpen(const drizzled::TableIdentifier &identifier, int mode, uint32_t test_if_locked)
95
int ha_heap::open(const char *name, int mode, uint32_t test_if_locked)
181
if ((test_if_locked & HA_OPEN_INTERNAL_TABLE) || (!(file= heap_open(identifier.getPath().c_str(), mode)) && errno == ENOENT))
97
if ((test_if_locked & HA_OPEN_INTERNAL_TABLE) || (!(file= heap_open(name, mode)) && my_errno == ENOENT))
99
HA_CREATE_INFO create_info;
183
100
internal_table= test(test_if_locked & HA_OPEN_INTERNAL_TABLE);
101
memset(&create_info, 0, sizeof(create_info));
185
HP_SHARE *internal_share= NULL;
186
message::Table create_proto;
188
if (not heap_storage_engine->heap_create_table(getTable()->in_use,
189
identifier.getPath().c_str(),
103
if (!create(name, table, &create_info))
195
105
file= internal_table ?
196
106
heap_open_from_share(internal_share, mode) :
235
146
Do same as default implementation but use file->s->name instead of
236
table->getShare()->path. This is needed by Windows where the clone() call sees
237
'/'-delimited path in table->getShare()->path, while ha_peap::open() was called
147
table->s->path. This is needed by Windows where the clone() call sees
148
'/'-delimited path in table->s->path, while ha_peap::open() was called
238
149
with '\'-delimited path.
241
Cursor *ha_heap::clone(memory::Root *)
152
handler *ha_heap::clone(MEM_ROOT *mem_root)
243
Cursor *new_handler= getTable()->getMutableShare()->db_type()->getCursor(*getTable());
244
TableIdentifier identifier(getTable()->getShare()->getSchemaName(),
245
getTable()->getShare()->getTableName(),
246
getTable()->getShare()->getPath());
248
if (new_handler && !new_handler->ha_open(identifier, getTable()->db_stat,
154
handler *new_handler= get_new_handler(table->s, mem_root, table->s->db_type());
155
if (new_handler && !new_handler->ha_open(table, file->s->name, table->db_stat,
249
156
HA_OPEN_IGNORE_IF_LOCKED))
250
157
return new_handler;
255
const char *ha_heap::index_type(uint32_t )
158
return NULL; /* purecov: inspected */
162
const char *ha_heap::index_type(uint32_t inx)
164
return ((table_share->key_info[inx].algorithm == HA_KEY_ALG_BTREE) ?
169
uint32_t ha_heap::index_flags(uint32_t inx, uint32_t, bool) const
171
return ((table_share->key_info[inx].algorithm == HA_KEY_ALG_BTREE) ?
172
HA_READ_NEXT | HA_READ_PREV | HA_READ_ORDER | HA_READ_RANGE :
173
HA_ONLY_WHOLE_INDEX | HA_KEY_SCAN_NOT_ROR);
304
225
records_changed= 0;
305
226
/* At the end of update_key_stats() we can proudly claim they are OK. */
306
key_stat_version= file->getShare()->key_stat_version;
227
key_stat_version= file->s->key_stat_version;
310
int ha_heap::doInsertRecord(unsigned char * buf)
231
int ha_heap::write_row(unsigned char * buf)
313
if (getTable()->next_number_field && buf == getTable()->getInsertRecord())
234
ha_statistic_increment(&SSV::ha_write_count);
235
if (table->timestamp_field_type & TIMESTAMP_AUTO_SET_ON_INSERT)
236
table->timestamp_field->set_time();
237
if (table->next_number_field && buf == table->record[0])
315
239
if ((res= update_auto_increment()))
318
242
res= heap_write(file,buf);
319
if (!res && (++records_changed*MEMORY_STATS_UPDATE_THRESHOLD >
320
file->getShare()->records))
243
if (!res && (++records_changed*HEAP_STATS_UPDATE_THRESHOLD >
323
247
We can perform this safely since only one writer at the time is
324
248
allowed on the table.
326
file->getShare()->key_stat_version++;
250
file->s->key_stat_version++;
331
int ha_heap::doUpdateRecord(const unsigned char * old_data, unsigned char * new_data)
255
int ha_heap::update_row(const unsigned char * old_data, unsigned char * new_data)
258
ha_statistic_increment(&SSV::ha_update_count);
259
if (table->timestamp_field_type & TIMESTAMP_AUTO_SET_ON_UPDATE)
260
table->timestamp_field->set_time();
335
261
res= heap_update(file,old_data,new_data);
336
if (!res && ++records_changed*MEMORY_STATS_UPDATE_THRESHOLD >
337
file->getShare()->records)
262
if (!res && ++records_changed*HEAP_STATS_UPDATE_THRESHOLD >
340
266
We can perform this safely since only one writer at the time is
341
267
allowed on the table.
343
file->getShare()->key_stat_version++;
269
file->s->key_stat_version++;
348
int ha_heap::doDeleteRecord(const unsigned char * buf)
274
int ha_heap::delete_row(const unsigned char * buf)
277
ha_statistic_increment(&SSV::ha_delete_count);
352
278
res= heap_delete(file,buf);
353
if (!res && getTable()->getShare()->getType() == message::Table::STANDARD &&
354
++records_changed*MEMORY_STATS_UPDATE_THRESHOLD > file->getShare()->records)
279
if (!res && table->s->tmp_table == NO_TMP_TABLE &&
280
++records_changed*HEAP_STATS_UPDATE_THRESHOLD > file->s->records)
357
283
We can perform this safely since only one writer at the time is
358
284
allowed on the table.
360
file->getShare()->key_stat_version++;
286
file->s->key_stat_version++;
388
314
key_part_map keypart_map,
389
315
enum ha_rkey_function find_flag)
391
ha_statistic_increment(&system_status_var::ha_read_key_count);
317
ha_statistic_increment(&SSV::ha_read_key_count);
392
318
int error = heap_rkey(file, buf, index, key, keypart_map, find_flag);
393
getTable()->status = error ? STATUS_NOT_FOUND : 0;
319
table->status = error ? STATUS_NOT_FOUND : 0;
397
323
int ha_heap::index_next(unsigned char * buf)
399
325
assert(inited==INDEX);
400
ha_statistic_increment(&system_status_var::ha_read_next_count);
326
ha_statistic_increment(&SSV::ha_read_next_count);
401
327
int error=heap_rnext(file,buf);
402
getTable()->status=error ? STATUS_NOT_FOUND: 0;
328
table->status=error ? STATUS_NOT_FOUND: 0;
406
332
int ha_heap::index_prev(unsigned char * buf)
408
334
assert(inited==INDEX);
409
ha_statistic_increment(&system_status_var::ha_read_prev_count);
335
ha_statistic_increment(&SSV::ha_read_prev_count);
410
336
int error=heap_rprev(file,buf);
411
getTable()->status=error ? STATUS_NOT_FOUND: 0;
337
table->status=error ? STATUS_NOT_FOUND: 0;
415
341
int ha_heap::index_first(unsigned char * buf)
417
343
assert(inited==INDEX);
418
ha_statistic_increment(&system_status_var::ha_read_first_count);
344
ha_statistic_increment(&SSV::ha_read_first_count);
419
345
int error=heap_rfirst(file, buf, active_index);
420
getTable()->status=error ? STATUS_NOT_FOUND: 0;
346
table->status=error ? STATUS_NOT_FOUND: 0;
424
350
int ha_heap::index_last(unsigned char * buf)
426
352
assert(inited==INDEX);
427
ha_statistic_increment(&system_status_var::ha_read_last_count);
353
ha_statistic_increment(&SSV::ha_read_last_count);
428
354
int error=heap_rlast(file, buf, active_index);
429
getTable()->status=error ? STATUS_NOT_FOUND: 0;
355
table->status=error ? STATUS_NOT_FOUND: 0;
433
int ha_heap::doStartTableScan(bool scan)
359
int ha_heap::rnd_init(bool scan)
435
361
return scan ? heap_scan_init(file) : 0;
438
364
int ha_heap::rnd_next(unsigned char *buf)
440
ha_statistic_increment(&system_status_var::ha_read_rnd_next_count);
366
ha_statistic_increment(&SSV::ha_read_rnd_next_count);
441
367
int error=heap_scan(file, buf);
442
getTable()->status=error ? STATUS_NOT_FOUND: 0;
368
table->status=error ? STATUS_NOT_FOUND: 0;
617
558
return heap_indexes_are_disabled(file);
561
THR_LOCK_DATA **ha_heap::store_lock(Session *,
563
enum thr_lock_type lock_type)
565
if (lock_type != TL_IGNORE && file->lock.type == TL_UNLOCK)
566
file->lock.type=lock_type;
572
We have to ignore ENOENT entries as the HEAP table is created on open and
573
not when doing a CREATE on the table.
576
int ha_heap::delete_table(const char *name)
578
return heap_delete_table(name);
620
582
void ha_heap::drop_table(const char *)
622
file->getShare()->delete_on_close= 1;
584
file->s->delete_on_close= 1;
627
int HeapEngine::doRenameTable(Session &session, const TableIdentifier &from, const TableIdentifier &to)
589
int ha_heap::rename_table(const char * from, const char * to)
629
session.getMessageCache().renameTableMessage(from, to);
630
return heap_rename(from.getPath().c_str(), to.getPath().c_str());
591
return heap_rename(from,to);
634
595
ha_rows ha_heap::records_in_range(uint32_t inx, key_range *min_key,
635
596
key_range *max_key)
637
KeyInfo *key= &getTable()->key_info[inx];
598
KEY *key=table->key_info+inx;
599
if (key->algorithm == HA_KEY_ALG_BTREE)
600
return hp_rb_records_in_range(file, inx, min_key, max_key);
639
602
if (!min_key || !max_key ||
640
603
min_key->length != max_key->length ||
647
610
return stats.records;
649
612
/* Assert that info() did run. We need current statistics here. */
650
assert(key_stat_version == file->getShare()->key_stat_version);
613
assert(key_stat_version == file->s->key_stat_version);
651
614
return key->rec_per_key[key->key_parts-1];
654
int HeapEngine::doCreateTable(Session &session,
656
const TableIdentifier &identifier,
657
message::Table& create_proto)
660
HP_SHARE *internal_share;
661
const char *table_name= identifier.getPath().c_str();
663
error= heap_create_table(&session, table_name, &table_arg,
670
session.getMessageCache().storeTableMessage(identifier, create_proto);
677
int HeapEngine::heap_create_table(Session *session, const char *table_name,
680
message::Table &create_proto,
681
HP_SHARE **internal_share)
683
uint32_t key, parts, mem_per_row_keys= 0;
684
uint32_t keys= table_arg->getShare()->sizeKeys();
618
int ha_heap::create(const char *name, Table *table_arg,
619
HA_CREATE_INFO *create_info)
621
uint32_t key, parts, mem_per_row_keys= 0, keys= table_arg->s->keys;
685
622
uint32_t auto_key= 0, auto_key_type= 0;
686
623
uint32_t max_key_fieldnr = 0, key_part_size = 0, next_field_pos = 0;
687
uint32_t column_count= table_arg->getShare()->sizeFields();
688
std::vector<HP_KEYDEF> keydef;
624
uint32_t column_idx, column_count= table_arg->s->fields;
625
HP_COLUMNDEF *columndef;
628
char buff[FN_REFLEN];
630
TABLE_SHARE *share= table_arg->s;
690
631
bool found_real_auto_increment= 0;
693
* We cannot create tables with more rows than UINT32_MAX. This is a
694
* limitation of the HEAP engine. Here, since TableShare::getMaxRows()
695
* can return a number more than that, we trap it here instead of casting
696
* to a truncated integer.
698
uint64_t num_rows= table_arg->getShare()->getMaxRows();
699
if (num_rows > UINT32_MAX)
633
if (!(columndef= (HP_COLUMNDEF*) malloc(column_count * sizeof(HP_COLUMNDEF))))
636
for (column_idx= 0; column_idx < column_count; column_idx++)
638
Field* field= *(table_arg->field + column_idx);
639
HP_COLUMNDEF* column= columndef + column_idx;
640
column->type= (uint16_t)field->type();
641
column->length= field->pack_length();
642
column->offset= field->offset(field->table->record[0]);
646
column->null_bit= field->null_bit;
647
column->null_pos= (uint) (field->null_ptr - (unsigned char*) table_arg->record[0]);
655
if (field->type() == DRIZZLE_TYPE_VARCHAR)
657
column->length_bytes= (uint8_t)(((Field_varstring*)field)->length_bytes);
661
column->length_bytes= 0;
702
665
for (key= parts= 0; key < keys; key++)
703
666
parts+= table_arg->key_info[key].key_parts;
706
std::vector<HA_KEYSEG> seg_buffer;
707
seg_buffer.resize(parts);
708
HA_KEYSEG *seg= &seg_buffer[0];
668
if (!(keydef= (HP_KEYDEF*) malloc(keys * sizeof(HP_KEYDEF) +
669
parts * sizeof(HA_KEYSEG))))
671
free((void *) columndef);
675
seg= reinterpret_cast<HA_KEYSEG*> (keydef + keys);
710
676
for (key= 0; key < keys; key++)
712
KeyInfo *pos= &table_arg->key_info[key];
713
KeyPartInfo *key_part= pos->key_part;
714
KeyPartInfo *key_part_end= key_part + pos->key_parts;
678
KEY *pos= table_arg->key_info+key;
679
KEY_PART_INFO *key_part= pos->key_part;
680
KEY_PART_INFO *key_part_end= key_part + pos->key_parts;
716
682
keydef[key].keysegs= (uint) pos->key_parts;
717
683
keydef[key].flag= (pos->flags & (HA_NOSAME | HA_NULL_ARE_EQUAL));
718
684
keydef[key].seg= seg;
720
mem_per_row_keys+= sizeof(char*) * 2; // = sizeof(HASH_INFO)
686
switch (pos->algorithm) {
687
case HA_KEY_ALG_UNDEF:
688
case HA_KEY_ALG_HASH:
689
keydef[key].algorithm= HA_KEY_ALG_HASH;
690
mem_per_row_keys+= sizeof(char*) * 2; // = sizeof(HASH_INFO)
692
case HA_KEY_ALG_BTREE:
693
keydef[key].algorithm= HA_KEY_ALG_BTREE;
694
mem_per_row_keys+=sizeof(TREE_ELEMENT)+pos->key_length+sizeof(char*);
697
assert(0); // cannot happen
722
700
for (; key_part != key_part_end; key_part++, seg++)
724
702
Field *field= key_part->field;
704
if (pos->algorithm == HA_KEY_ALG_BTREE)
705
seg->type= field->key_type();
727
708
if ((seg->type = field->key_type()) != (int) HA_KEYTYPE_TEXT &&
728
709
seg->type != HA_KEYTYPE_VARTEXT1 &&
770
751
auto_key= key+ 1;
771
752
auto_key_type= field->key_type();
773
if ((uint)field->position() + 1 > max_key_fieldnr)
754
if ((uint)field->field_index + 1 > max_key_fieldnr)
775
756
/* Do not use seg->fieldnr as it's not reliable in case of temp tables */
776
max_key_fieldnr= field->position() + 1;
757
max_key_fieldnr= field->field_index + 1;
781
if (key_part_size < table_arg->getShare()->null_bytes + ((table_arg->getShare()->last_null_bit_pos+7) >> 3))
762
if (key_part_size < share->null_bytes + ((share->last_null_bit_pos+7) >> 3))
783
764
/* Make sure to include null fields regardless of the presense of keys */
784
key_part_size = table_arg->getShare()->null_bytes + ((table_arg->getShare()->last_null_bit_pos+7) >> 3);
765
key_part_size = share->null_bytes + ((share->last_null_bit_pos+7) >> 3);
789
770
if (table_arg->found_next_number_field)
791
keydef[table_arg->getShare()->next_number_index].flag|= HA_AUTO_KEY;
792
found_real_auto_increment= table_arg->getShare()->next_number_key_offset == 0;
772
keydef[share->next_number_index].flag|= HA_AUTO_KEY;
773
found_real_auto_increment= share->next_number_key_offset == 0;
794
775
HP_CREATE_INFO hp_create_info;
795
776
hp_create_info.auto_key= auto_key;
796
777
hp_create_info.auto_key_type= auto_key_type;
797
hp_create_info.auto_increment= (create_proto.options().has_auto_increment_value() ?
798
create_proto.options().auto_increment_value() - 1 : 0);
799
hp_create_info.max_table_size=session->variables.max_heap_table_size;
778
hp_create_info.auto_increment= (create_info->auto_increment_value ?
779
create_info->auto_increment_value - 1 : 0);
780
hp_create_info.max_table_size=current_session->variables.max_heap_table_size;
800
781
hp_create_info.with_auto_increment= found_real_auto_increment;
801
782
hp_create_info.internal_table= internal_table;
802
hp_create_info.max_chunk_size= table_arg->getShare()->block_size;
804
error= heap_create(table_name,
808
table_arg->getShare()->getRecordLength(), mem_per_row_keys,
809
static_cast<uint32_t>(num_rows), /* We check for overflow above, so cast is fine here. */
811
&hp_create_info, internal_share);
783
hp_create_info.max_chunk_size= share->block_size;
784
hp_create_info.is_dynamic= (share->row_type == ROW_TYPE_DYNAMIC);
785
error= heap_create(fn_format(buff,name,"","",
786
MY_REPLACE_EXT|MY_UNPACK_FILENAME),
788
column_count, columndef,
789
max_key_fieldnr, key_part_size,
790
share->reclength, mem_per_row_keys,
791
(uint32_t) share->max_rows, (uint32_t) share->min_rows,
792
&hp_create_info, &internal_share);
794
free((unsigned char*) keydef);
795
free((void *) columndef);
801
void ha_heap::update_create_info(HA_CREATE_INFO *create_info)
803
table->file->info(HA_STATUS_AUTO);
804
if (!(create_info->used_fields & HA_CREATE_USED_AUTO))
805
create_info->auto_increment_value= stats.auto_increment_value;
806
if (!(create_info->used_fields & HA_CREATE_USED_BLOCK_SIZE))
808
if (file->s->recordspace.is_variable_size)
809
create_info->block_size= file->s->recordspace.chunk_length;
811
create_info->block_size= 0;
817
815
void ha_heap::get_auto_increment(uint64_t, uint64_t, uint64_t,
818
816
uint64_t *first_value,
819
817
uint64_t *nb_reserved_values)