mirror of
https://github.com/MariaDB/server.git
synced 2025-03-27 17:38:45 +01:00
Removed unnecessary key search in the hp_rb_write_key() function
This commit is contained in:
parent
c1f3be5bb5
commit
c78685e0c2
4 changed files with 25 additions and 16 deletions
|
@ -99,8 +99,8 @@ HP_INFO *heap_open(const char *name, int mode, uint keys, HP_KEYDEF *keydef,
|
|||
|
||||
if (keydef[i].algorithm == HA_KEY_ALG_BTREE)
|
||||
{
|
||||
init_tree(&keyinfo->rb_tree, 0, 0, 0, (qsort_cmp2)keys_compare, 1,
|
||||
NULL, NULL);
|
||||
init_tree(&keyinfo->rb_tree, 0, 0, sizeof(byte*),
|
||||
(qsort_cmp2)keys_compare, 1, NULL, NULL);
|
||||
keyinfo->delete_key= hp_rb_delete_key;
|
||||
keyinfo->write_key= hp_rb_write_key;
|
||||
nsegs++;
|
||||
|
|
|
@ -96,22 +96,23 @@ int hp_rb_write_key(HP_INFO *info, HP_KEYDEF *keyinfo, const byte *record,
|
|||
info->last_pos= NULL; /* For heap_rnext/heap_rprev */
|
||||
custom_arg.keyseg= keyinfo->seg;
|
||||
custom_arg.key_length= hp_rb_make_key(keyinfo, info->recbuf, record, recpos);
|
||||
if ((keyinfo->flag & HA_NOSAME) &&
|
||||
(!(keyinfo->flag & HA_NULL_PART_KEY) ||
|
||||
!hp_if_null_in_key(keyinfo, record)))
|
||||
if (keyinfo->flag & HA_NOSAME)
|
||||
{
|
||||
custom_arg.search_flag= SEARCH_FIND | SEARCH_SAME;
|
||||
if (tree_search_key(&keyinfo->rb_tree, info->recbuf, info->parents,
|
||||
&info->last_pos, 0, &custom_arg))
|
||||
{
|
||||
my_errno= HA_ERR_FOUND_DUPP_KEY;
|
||||
return 1;
|
||||
}
|
||||
keyinfo->rb_tree.flag= TREE_NO_DUPS;
|
||||
}
|
||||
custom_arg.search_flag= SEARCH_SAME;
|
||||
return tree_insert(&keyinfo->rb_tree, (void*)info->recbuf,
|
||||
custom_arg.key_length + sizeof(byte*),
|
||||
&custom_arg) ? 0 : 1;
|
||||
else
|
||||
{
|
||||
custom_arg.search_flag= SEARCH_SAME;
|
||||
keyinfo->rb_tree.flag= 0;
|
||||
}
|
||||
if (!tree_insert(&keyinfo->rb_tree, (void*)info->recbuf,
|
||||
custom_arg.key_length, &custom_arg))
|
||||
{
|
||||
my_errno= HA_ERR_FOUND_DUPP_KEY;
|
||||
return 1;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
/* Find where to place new record */
|
||||
|
|
|
@ -27,6 +27,8 @@ extern "C" {
|
|||
|
||||
#define tree_set_pointer(element,ptr) *((byte **) (element+1))=((byte*) (ptr))
|
||||
|
||||
#define TREE_NO_DUPS 1
|
||||
|
||||
typedef enum { left_root_right, right_root_left } TREE_WALK;
|
||||
typedef uint32 element_count;
|
||||
typedef int (*tree_walk_action)(void *,element_count,void *);
|
||||
|
@ -55,10 +57,11 @@ typedef struct st_tree {
|
|||
TREE_ELEMENT **parents[MAX_TREE_HEIGHT];
|
||||
uint offset_to_key,elements_in_tree,size_of_element,memory_limit,allocated;
|
||||
qsort_cmp2 compare;
|
||||
void* custom_arg;
|
||||
void *custom_arg;
|
||||
MEM_ROOT mem_root;
|
||||
my_bool with_delete;
|
||||
tree_element_free free;
|
||||
uint flag;
|
||||
} TREE;
|
||||
|
||||
/* Functions on whole tree */
|
||||
|
|
|
@ -86,6 +86,7 @@ void init_tree(TREE *tree, uint default_alloc_size, uint memory_limit,
|
|||
tree->custom_arg = custom_arg;
|
||||
tree->null_element.colour=BLACK;
|
||||
tree->null_element.left=tree->null_element.right=0;
|
||||
tree->flag= 0;
|
||||
if (!free_element && size >= 0 &&
|
||||
((uint) size <= sizeof(void*) || ((uint) size & (sizeof(void*)-1))))
|
||||
{
|
||||
|
@ -231,7 +232,11 @@ TREE_ELEMENT *tree_insert(TREE *tree, void *key, uint key_size,
|
|||
rb_insert(tree,parent,element); /* rebalance tree */
|
||||
}
|
||||
else
|
||||
{
|
||||
if (tree->flag & TREE_NO_DUPS)
|
||||
return(NULL);
|
||||
element->count++;
|
||||
}
|
||||
DBUG_EXECUTE("check_tree", test_rb_tree(tree->root););
|
||||
return element;
|
||||
}
|
||||
|
|
Loading…
Add table
Reference in a new issue