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 */
14
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */
16
16
#include "heap_priv.h"
17
17
#include <drizzled/error.h>
53
51
HTON_SKIP_STORE_LOCK |
54
52
HTON_TEMPORARY_ONLY)
54
pthread_mutex_init(&THR_LOCK_heap, MY_MUTEX_INIT_FAST);
58
57
virtual ~HeapEngine()
60
59
hp_panic(HA_PANIC_CLOSE);
61
pthread_mutex_destroy(&THR_LOCK_heap);
63
virtual Cursor *create(Table &table)
64
virtual Cursor *create(TableShare &table,
65
memory::Root *mem_root)
65
return new ha_heap(*this, table);
67
return new (mem_root) ha_heap(*this, table);
68
70
const char **bas_ext() const {
93
95
const TableIdentifier &identifier,
94
96
message::Table &table_message);
98
/* Temp only engine, so do not return values. */
99
void doGetTableNames(CachedDirectory &, SchemaIdentifier& , set<string>&) { };
96
101
uint32_t max_supported_keys() const { return MAX_KEY; }
97
102
uint32_t max_supported_key_part_length() const { return MAX_KEY_LENGTH; }
99
uint32_t index_flags(enum ha_key_alg ) const
104
uint32_t index_flags(enum ha_key_alg algorithm) const
101
return ( HA_ONLY_WHOLE_INDEX | HA_KEY_SCAN_NOT_ROR);
106
return ((algorithm == HA_KEY_ALG_BTREE) ?
111
HA_ONLY_WHOLE_INDEX |
112
HA_KEY_SCAN_NOT_ROR);
104
115
bool doDoesTableExist(Session& session, const TableIdentifier &identifier);
105
116
void doGetTableIdentifiers(CachedDirectory &directory,
106
const SchemaIdentifier &schema_identifier,
107
TableIdentifier::vector &set_of_identifiers);
117
SchemaIdentifier &schema_identifier,
118
TableIdentifiers &set_of_identifiers);
110
121
void HeapEngine::doGetTableIdentifiers(CachedDirectory&,
111
const SchemaIdentifier&,
112
TableIdentifier::vector&)
116
127
bool HeapEngine::doDoesTableExist(Session& session, const TableIdentifier &identifier)
118
return session.getMessageCache().doesTableMessageExist(identifier);
129
return session.doesTableMessageExist(identifier);
121
132
int HeapEngine::doGetTableDefinition(Session &session,
122
133
const TableIdentifier &identifier,
123
134
message::Table &table_proto)
125
if (session.getMessageCache().getTableMessage(identifier, table_proto))
136
if (session.getTableMessage(identifier, table_proto))
134
145
int HeapEngine::doDropTable(Session &session, const TableIdentifier &identifier)
136
session.getMessageCache().removeTableMessage(identifier);
147
session.removeTableMessage(identifier);
138
149
int error= heap_delete_table(identifier.getPath().c_str());
158
169
*****************************************************************************/
160
171
ha_heap::ha_heap(plugin::StorageEngine &engine_arg,
172
TableShare &table_arg)
162
173
:Cursor(engine_arg, table_arg), file(0), records_changed(0), key_stat_version(0),
163
174
internal_table(0)
177
188
#define MEMORY_STATS_UPDATE_THRESHOLD 10
179
int ha_heap::doOpen(const drizzled::TableIdentifier &identifier, int mode, uint32_t test_if_locked)
190
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))
192
if ((test_if_locked & HA_OPEN_INTERNAL_TABLE) || (!(file= heap_open(name, mode)) && errno == ENOENT))
194
HA_CREATE_INFO create_info;
183
195
internal_table= test(test_if_locked & HA_OPEN_INTERNAL_TABLE);
196
memset(&create_info, 0, sizeof(create_info));
185
198
HP_SHARE *internal_share= NULL;
186
199
message::Table create_proto;
188
if (not heap_storage_engine->heap_create_table(getTable()->in_use,
189
identifier.getPath().c_str(),
201
if (!heap_storage_engine->heap_create_table(table->in_use, name, table,
195
206
file= internal_table ?
196
207
heap_open_from_share(internal_share, mode) :
238
249
with '\'-delimited path.
241
Cursor *ha_heap::clone(memory::Root *)
252
Cursor *ha_heap::clone(memory::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());
254
Cursor *new_handler= table->getMutableShare()->db_type()->getCursor(*(table->getMutableShare()), mem_root);
255
TableIdentifier identifier(table->getShare()->getSchemaName(),
256
table->getShare()->getTableName(),
257
table->getShare()->getPath());
248
if (new_handler && !new_handler->ha_open(identifier, getTable()->db_stat,
259
if (new_handler && !new_handler->ha_open(identifier, table, file->s->name, table->db_stat,
249
260
HA_OPEN_IGNORE_IF_LOCKED))
250
261
return new_handler;
255
const char *ha_heap::index_type(uint32_t )
266
const char *ha_heap::index_type(uint32_t inx)
268
return ((table_share->getKeyInfo(inx).algorithm == HA_KEY_ALG_BTREE) ?
277
289
void ha_heap::set_keys_for_scanning(void)
292
for (uint32_t i= 0 ; i < table->getShare()->sizeKeys() ; i++)
294
if (table->key_info[i].algorithm == HA_KEY_ALG_BTREE)
282
300
void ha_heap::update_key_stats()
284
for (uint32_t i= 0; i < getTable()->getShare()->sizeKeys(); i++)
302
for (uint32_t i= 0; i < table->getShare()->sizeKeys(); i++)
286
KeyInfo *key= &getTable()->key_info[i];
304
KeyInfo *key= &table->key_info[i];
288
305
if (!key->rec_per_key)
307
if (key->algorithm != HA_KEY_ALG_BTREE)
292
309
if (key->flags & HA_NOSAME)
293
310
key->rec_per_key[key->key_parts-1]= 1;
296
ha_rows hash_buckets= file->getShare()->keydef[i].hash_buckets;
297
uint32_t no_records= hash_buckets ? (uint) (file->getShare()->records/hash_buckets) : 2;
313
ha_rows hash_buckets= file->s->keydef[i].hash_buckets;
314
uint32_t no_records= hash_buckets ? (uint) (file->s->records/hash_buckets) : 2;
298
315
if (no_records < 2)
300
317
key->rec_per_key[key->key_parts-1]= no_records;
304
321
records_changed= 0;
305
322
/* At the end of update_key_stats() we can proudly claim they are OK. */
306
key_stat_version= file->getShare()->key_stat_version;
323
key_stat_version= file->s->key_stat_version;
310
327
int ha_heap::doInsertRecord(unsigned char * buf)
313
if (getTable()->next_number_field && buf == getTable()->getInsertRecord())
330
if (table->next_number_field && buf == table->record[0])
315
332
if ((res= update_auto_increment()))
318
335
res= heap_write(file,buf);
319
336
if (!res && (++records_changed*MEMORY_STATS_UPDATE_THRESHOLD >
320
file->getShare()->records))
323
340
We can perform this safely since only one writer at the time is
324
341
allowed on the table.
326
file->getShare()->key_stat_version++;
343
file->s->key_stat_version++;
335
352
res= heap_update(file,old_data,new_data);
336
353
if (!res && ++records_changed*MEMORY_STATS_UPDATE_THRESHOLD >
337
file->getShare()->records)
340
357
We can perform this safely since only one writer at the time is
341
358
allowed on the table.
343
file->getShare()->key_stat_version++;
360
file->s->key_stat_version++;
352
369
res= heap_delete(file,buf);
353
if (!res && getTable()->getShare()->getType() == message::Table::STANDARD &&
354
++records_changed*MEMORY_STATS_UPDATE_THRESHOLD > file->getShare()->records)
370
if (!res && table->getShare()->getType() == message::Table::STANDARD &&
371
++records_changed*MEMORY_STATS_UPDATE_THRESHOLD > file->s->records)
357
374
We can perform this safely since only one writer at the time is
358
375
allowed on the table.
360
file->getShare()->key_stat_version++;
377
file->s->key_stat_version++;
369
386
assert(inited==INDEX);
370
387
ha_statistic_increment(&system_status_var::ha_read_key_count);
371
388
int error = heap_rkey(file,buf,active_index, key, keypart_map, find_flag);
372
getTable()->status = error ? STATUS_NOT_FOUND : 0;
389
table->status = error ? STATUS_NOT_FOUND : 0;
380
397
ha_statistic_increment(&system_status_var::ha_read_key_count);
381
398
int error= heap_rkey(file, buf, active_index, key, keypart_map,
382
399
HA_READ_PREFIX_LAST);
383
getTable()->status= error ? STATUS_NOT_FOUND : 0;
400
table->status= error ? STATUS_NOT_FOUND : 0;
391
408
ha_statistic_increment(&system_status_var::ha_read_key_count);
392
409
int error = heap_rkey(file, buf, index, key, keypart_map, find_flag);
393
getTable()->status = error ? STATUS_NOT_FOUND : 0;
410
table->status = error ? STATUS_NOT_FOUND : 0;
399
416
assert(inited==INDEX);
400
417
ha_statistic_increment(&system_status_var::ha_read_next_count);
401
418
int error=heap_rnext(file,buf);
402
getTable()->status=error ? STATUS_NOT_FOUND: 0;
419
table->status=error ? STATUS_NOT_FOUND: 0;
408
425
assert(inited==INDEX);
409
426
ha_statistic_increment(&system_status_var::ha_read_prev_count);
410
427
int error=heap_rprev(file,buf);
411
getTable()->status=error ? STATUS_NOT_FOUND: 0;
428
table->status=error ? STATUS_NOT_FOUND: 0;
417
434
assert(inited==INDEX);
418
435
ha_statistic_increment(&system_status_var::ha_read_first_count);
419
436
int error=heap_rfirst(file, buf, active_index);
420
getTable()->status=error ? STATUS_NOT_FOUND: 0;
437
table->status=error ? STATUS_NOT_FOUND: 0;
426
443
assert(inited==INDEX);
427
444
ha_statistic_increment(&system_status_var::ha_read_last_count);
428
445
int error=heap_rlast(file, buf, active_index);
429
getTable()->status=error ? STATUS_NOT_FOUND: 0;
446
table->status=error ? STATUS_NOT_FOUND: 0;
440
457
ha_statistic_increment(&system_status_var::ha_read_rnd_next_count);
441
458
int error=heap_scan(file, buf);
442
getTable()->status=error ? STATUS_NOT_FOUND: 0;
459
table->status=error ? STATUS_NOT_FOUND: 0;
450
467
ha_statistic_increment(&system_status_var::ha_read_rnd_count);
451
468
memcpy(&heap_position, pos, sizeof(HEAP_PTR));
452
469
error=heap_rrnd(file, buf, heap_position);
453
getTable()->status=error ? STATUS_NOT_FOUND: 0;
470
table->status=error ? STATUS_NOT_FOUND: 0;
479
496
have to update the key statistics. Hoping that a table lock is now
482
if (key_stat_version != file->getShare()->key_stat_version)
499
if (key_stat_version != file->s->key_stat_version)
483
500
update_key_stats();
505
enum row_type ha_heap::get_row_type() const
507
if (file->s->recordspace.is_variable_size)
508
return ROW_TYPE_DYNAMIC;
510
return ROW_TYPE_FIXED;
487
513
int ha_heap::extra(enum ha_extra_function operation)
489
515
return heap_extra(file,operation);
499
525
int ha_heap::delete_all_rows()
501
527
heap_clear(file);
502
if (getTable()->getShare()->getType() == message::Table::STANDARD)
528
if (table->getShare()->getType() == message::Table::STANDARD)
505
531
We can perform this safely since only one writer at the time is
506
532
allowed on the table.
508
file->getShare()->key_stat_version++;
534
file->s->key_stat_version++;
620
646
void ha_heap::drop_table(const char *)
622
file->getShare()->delete_on_close= 1;
648
file->s->delete_on_close= 1;
627
653
int HeapEngine::doRenameTable(Session &session, const TableIdentifier &from, const TableIdentifier &to)
629
session.getMessageCache().renameTableMessage(from, to);
655
session.renameTableMessage(from, to);
630
656
return heap_rename(from.getPath().c_str(), to.getPath().c_str());
634
660
ha_rows ha_heap::records_in_range(uint32_t inx, key_range *min_key,
635
661
key_range *max_key)
637
KeyInfo *key= &getTable()->key_info[inx];
663
KeyInfo *key= &table->key_info[inx];
664
if (key->algorithm == HA_KEY_ALG_BTREE)
665
return hp_rb_records_in_range(file, inx, min_key, max_key);
639
667
if (!min_key || !max_key ||
640
668
min_key->length != max_key->length ||
647
675
return stats.records;
649
677
/* Assert that info() did run. We need current statistics here. */
650
assert(key_stat_version == file->getShare()->key_stat_version);
678
assert(key_stat_version == file->s->key_stat_version);
651
679
return key->rec_per_key[key->key_parts-1];
680
708
message::Table &create_proto,
681
709
HP_SHARE **internal_share)
683
uint32_t key, parts, mem_per_row_keys= 0;
684
uint32_t keys= table_arg->getShare()->sizeKeys();
711
uint32_t key, parts, mem_per_row_keys= 0, keys= table_arg->getShare()->sizeKeys();
685
712
uint32_t auto_key= 0, auto_key_type= 0;
686
713
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;
714
uint32_t column_idx, column_count= table_arg->getShare()->sizeFields();
715
HP_COLUMNDEF *columndef;
718
char buff[FN_REFLEN];
690
720
bool found_real_auto_increment= 0;
699
729
if (num_rows > UINT32_MAX)
732
if (!(columndef= (HP_COLUMNDEF*) malloc(column_count * sizeof(HP_COLUMNDEF))))
735
for (column_idx= 0; column_idx < column_count; column_idx++)
737
Field* field= *(table_arg->getFields() + column_idx);
738
HP_COLUMNDEF* column= columndef + column_idx;
739
column->type= (uint16_t)field->type();
740
column->length= field->pack_length();
741
column->offset= field->offset(field->table->record[0]);
745
column->null_bit= field->null_bit;
746
column->null_pos= (uint) (field->null_ptr - (unsigned char*) table_arg->record[0]);
754
if (field->type() == DRIZZLE_TYPE_VARCHAR)
756
column->length_bytes= (uint8_t)(((Field_varstring*)field)->length_bytes);
760
column->length_bytes= 0;
702
764
for (key= parts= 0; key < keys; key++)
703
765
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];
767
if (!(keydef= (HP_KEYDEF*) malloc(keys * sizeof(HP_KEYDEF) +
768
parts * sizeof(HA_KEYSEG))))
770
free((void *) columndef);
774
seg= reinterpret_cast<HA_KEYSEG*> (keydef + keys);
710
775
for (key= 0; key < keys; key++)
712
777
KeyInfo *pos= &table_arg->key_info[key];
717
782
keydef[key].flag= (pos->flags & (HA_NOSAME | HA_NULL_ARE_EQUAL));
718
783
keydef[key].seg= seg;
720
mem_per_row_keys+= sizeof(char*) * 2; // = sizeof(HASH_INFO)
785
switch (pos->algorithm) {
786
case HA_KEY_ALG_UNDEF:
787
case HA_KEY_ALG_HASH:
788
keydef[key].algorithm= HA_KEY_ALG_HASH;
789
mem_per_row_keys+= sizeof(char*) * 2; // = sizeof(HASH_INFO)
791
case HA_KEY_ALG_BTREE:
792
keydef[key].algorithm= HA_KEY_ALG_BTREE;
793
mem_per_row_keys+=sizeof(TREE_ELEMENT)+pos->key_length+sizeof(char*);
796
assert(0); // cannot happen
722
799
for (; key_part != key_part_end; key_part++, seg++)
724
801
Field *field= key_part->field;
803
if (pos->algorithm == HA_KEY_ALG_BTREE)
804
seg->type= field->key_type();
727
807
if ((seg->type = field->key_type()) != (int) HA_KEYTYPE_TEXT &&
728
808
seg->type != HA_KEYTYPE_VARTEXT1 &&
738
818
next_field_pos= seg->start + seg->length;
739
819
if (field->type() == DRIZZLE_TYPE_VARCHAR)
741
next_field_pos+= (uint8_t)(((Field_varstring*)field)->pack_length_no_ptr());
821
next_field_pos+= (uint8_t)(((Field_varstring*)field)->length_bytes);
744
824
if (next_field_pos > key_part_size) {
770
850
auto_key= key+ 1;
771
851
auto_key_type= field->key_type();
773
if ((uint)field->position() + 1 > max_key_fieldnr)
853
if ((uint)field->field_index + 1 > max_key_fieldnr)
775
855
/* Do not use seg->fieldnr as it's not reliable in case of temp tables */
776
max_key_fieldnr= field->position() + 1;
856
max_key_fieldnr= field->field_index + 1;
800
880
hp_create_info.with_auto_increment= found_real_auto_increment;
801
881
hp_create_info.internal_table= internal_table;
802
882
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);
883
hp_create_info.is_dynamic= (table_arg->getShare()->row_type == ROW_TYPE_DYNAMIC);
885
error= heap_create(internal::fn_format(buff,table_name,"","",
886
MY_REPLACE_EXT|MY_UNPACK_FILENAME),
888
column_count, columndef,
889
max_key_fieldnr, key_part_size,
890
table_arg->getShare()->getRecordLength(), mem_per_row_keys,
891
static_cast<uint32_t>(num_rows), /* We check for overflow above, so cast is fine here. */
893
&hp_create_info, internal_share);
895
free((unsigned char*) keydef);
896
free((void *) columndef);