//主函数-各个函数的测试
int main(int argc, char ** argv)
{
int i = 0;
List_head head = NULL;
head = init_list();
List_node *node = create_node();
printf("\n头部插入:\n");
for(i = 0 ; i < seed ; i++ )
{
push_front(head,rand()%100);
}
show_list(head);
printf("\n尾部插入:\n");
for(i = 0 ; i < seed ; i++ )
{
push_back(head,rand()%100);
}
show_list(head);
printf("\n头部删除:\n");
pop_front(head);
show_list(head);
printf("\n尾部删除:\n");
pop_back(head);
show_list(head);
printf("\n插入节点:\n");
insert_node(head, 3, 100);
show_list(head);
find_node(head, 77,&node);
printf("\n升序排序:\n");
sort_list_ascend(head);
show_list(head);
printf("\n降序排序:\n");
sort_list_descend(head);
show_list(head);
printf("\n修改节点数据：\n");
node = head->next->next;
modify_node(node, 777);
show_list(head);
printf("\nthe length of this list :%d\n",get_list_length(head));
destroy_list(&head);
return 0;
}
couchfile_pointer_info* modify_btree(couchfile_modify_request *rq,
couchfile_pointer_info *root, int *errcode)
{
couchfile_pointer_info* ret_ptr = root;
couchfile_modify_result* root_result = make_modres(rq);
root_result->node_type = KP_NODE;
*errcode = modify_node(rq, root, 0, rq->num_actions, root_result);
if(*errcode < 0)
{
free_modres(root_result);
return NULL;
}
if(root_result->values_end->value.pointer == root)
{
//If we got the root pointer back, remove it from the list
//so we don't try to free it.
root_result->values_end->value.mem = NULL;
}
if(!root_result->modified)
{
free_modres(root_result);
}
else
{
if(root_result->count > 1 || root_result->pointers != root_result->pointers_end)
{
//The root was split
//Write it to disk and return the pointer to it.
ret_ptr = finish_root(rq, root_result, errcode);
if(*errcode < 0)
{
ret_ptr = NULL;
}
}
else
{
ret_ptr = root_result->values_end->value.pointer;
root_result->values_end->value.mem = NULL;
free_modres(root_result);
}
}
return ret_ptr;
}
node_pointer *modify_btree(couchfile_modify_request *rq,
node_pointer *root,
couchstore_error_t *errcode)
{
arena* a = new_arena(0);
node_pointer *ret_ptr = root;
couchfile_modify_result *root_result = make_modres(a, rq);
if (!root_result) {
delete_arena(a);
*errcode = COUCHSTORE_ERROR_ALLOC_FAIL;
return root;
}
root_result->node_type = KP_NODE;
*errcode = modify_node(rq, root, 0, rq->num_actions, root_result);
if (*errcode < 0) {
delete_arena(a);
return NULL;
}
if (root_result->values_end->pointer == root) {
//If we got the root pointer back, remove it from the list
//so we don't try to free it.
root_result->values_end->pointer = NULL;
}
if (root_result->modified) {
if (root_result->count > 1 || root_result->pointers != root_result->pointers_end) {
//The root was split
//Write it to disk and return the pointer to it.
ret_ptr = finish_root(rq, root_result, errcode);
if (*errcode < 0) {
ret_ptr = NULL;
}
} else {
ret_ptr = root_result->values_end->pointer;
}
}
if (ret_ptr != root) {
ret_ptr = copy_node_pointer(ret_ptr);
}
delete_arena(a);
return ret_ptr;
}
static couchstore_error_t modify_node(couchfile_modify_request *rq,
node_pointer *nptr,
int start, int end,
couchfile_modify_result *dst)
{
char *nodebuf = NULL; // FYI, nodebuf is a malloced block, not in the arena
int bufpos = 1;
int nodebuflen = 0;
int errcode = 0;
couchfile_modify_result *local_result = NULL;
if (start == end) {
return 0;
}
if (nptr) {
if ((nodebuflen = pread_compressed(rq->db, nptr->pointer, (char **) &nodebuf)) < 0) {
error_pass(COUCHSTORE_ERROR_READ);
}
}
local_result = make_modres(dst->arena, rq);
error_unless(local_result, COUCHSTORE_ERROR_ALLOC_FAIL);
if (nptr == NULL || nodebuf[0] == 1) { //KV Node
local_result->node_type = KV_NODE;
while (bufpos < nodebuflen) {
sized_buf cmp_key, val_buf;
bufpos += read_kv(nodebuf + bufpos, &cmp_key, &val_buf);
int advance = 0;
while (!advance && start < end) {
advance = 1;
int cmp_val = rq->cmp.compare(&cmp_key, rq->actions[start].key);
if (cmp_val < 0) { //Key less than action key
mr_push_item(&cmp_key, &val_buf, local_result);
} else if (cmp_val > 0) { //Key greater than action key
switch (rq->actions[start].type) {
case ACTION_INSERT:
local_result->modified = 1;
mr_push_item(rq->actions[start].key, rq->actions[start].value.data, local_result);
break;
case ACTION_REMOVE:
local_result->modified = 1;
break;
case ACTION_FETCH:
if (rq->fetch_callback) {
//not found
(*rq->fetch_callback)(rq, rq->actions[start].key, NULL, rq->actions[start].value.arg);
}
}
start++;
//Do next action on same item in the node, as our action was
//not >= it.
advance = 0;
} else if (cmp_val == 0) { //Node key is equal to action key
switch (rq->actions[start].type) {
case ACTION_INSERT:
local_result->modified = 1;
mr_push_item(rq->actions[start].key, rq->actions[start].value.data, local_result);
break;
case ACTION_REMOVE:
local_result->modified = 1;
break;
case ACTION_FETCH:
if (rq->fetch_callback) {
(*rq->fetch_callback)(rq, rq->actions[start].key, &val_buf, rq->actions[start].value.arg);
}
//Do next action on same item in the node, as our action was a fetch
//and there may be an equivalent insert or remove
//following.
advance = 0;
}
start++;
}
}
if (start == end && !advance) {
//If we've exhausted actions then just keep this key
mr_push_item(&cmp_key, &val_buf, local_result);
}
}
while (start < end) {
//We're at the end of a leaf node.
switch (rq->actions[start].type) {
case ACTION_INSERT:
local_result->modified = 1;
mr_push_item(rq->actions[start].key, rq->actions[start].value.data, local_result);
break;
case ACTION_REMOVE:
local_result->modified = 1;
break;
case ACTION_FETCH:
if (rq->fetch_callback) {
//not found
(*rq->fetch_callback)(rq, rq->actions[start].key, NULL, rq->actions[start].value.arg);
}
break;
}
start++;
}
} else if (nodebuf[0] == 0) { //KP Node
local_result->node_type = KP_NODE;
while (bufpos < nodebuflen && start < end) {
sized_buf cmp_key, val_buf;
bufpos += read_kv(nodebuf + bufpos, &cmp_key, &val_buf);
int cmp_val = rq->cmp.compare(&cmp_key, rq->actions[start].key);
if (bufpos == nodebuflen) {
//We're at the last item in the kpnode, must apply all our
//actions here.
node_pointer *desc = read_pointer(dst->arena, &cmp_key, val_buf.buf);
if (!desc) {
errcode = COUCHSTORE_ERROR_ALLOC_FAIL;
goto cleanup;
}
errcode = modify_node(rq, desc, start, end, local_result);
if (errcode != COUCHSTORE_SUCCESS) {
goto cleanup;
}
break;
}
if (cmp_val < 0) {
//Key in node item less than action item and not at end
//position, so just add it and continue.
node_pointer *add = read_pointer(dst->arena, &cmp_key, val_buf.buf);
if (!add) {
errcode = COUCHSTORE_ERROR_ALLOC_FAIL;
goto cleanup;
}
errcode = mr_push_pointerinfo(add, local_result);
if (errcode != COUCHSTORE_SUCCESS) {
goto cleanup;
}
} else if (cmp_val >= 0) {
//Found a key in the node greater than the one in the current
//action. Descend into the pointed node with as many actions as
//are less than the key here.
int range_end = start;
while (range_end < end &&
rq->cmp.compare(rq->actions[range_end].key, &cmp_key) <= 0) {
range_end++;
}
node_pointer *desc = read_pointer(dst->arena, &cmp_key, val_buf.buf);
if (!desc) {
errcode = COUCHSTORE_ERROR_ALLOC_FAIL;
goto cleanup;
}
errcode = modify_node(rq, desc, start, range_end, local_result);
start = range_end;
if (errcode != COUCHSTORE_SUCCESS) {
goto cleanup;
}
}
}
while (bufpos < nodebuflen) {
sized_buf cmp_key, val_buf;
bufpos += read_kv(nodebuf + bufpos, &cmp_key, &val_buf);
node_pointer *add = read_pointer(dst->arena, &cmp_key, val_buf.buf);
if (!add) {
errcode = COUCHSTORE_ERROR_ALLOC_FAIL;
goto cleanup;
}
errcode = mr_push_pointerinfo(add, local_result);
if (errcode != COUCHSTORE_SUCCESS) {
goto cleanup;
}
}
} else {
errcode = COUCHSTORE_ERROR_CORRUPT;
goto cleanup;
}
//If we've done modifications, write out the last leaf node.
error_pass(flush_mr(local_result));
if (!local_result->modified && nptr != NULL) {
//If we didn't do anything, give back the pointer to the original
mr_push_pointerinfo(nptr, dst);
} else {
//Otherwise, give back the pointers to the nodes we've created.
dst->modified = 1;
error_pass(mr_move_pointers(local_result, dst));
}
cleanup:
if (nodebuf) {
free(nodebuf);
}
return errcode;
}
int main(void)
{
/* One structure used in the program */
struct node llist;
int rc,i;
printf("\nAdding at the start of the list...");
rc = add_at_start(&llist,53);
if(rc)
{
printf("\nAdding at the start of the list not successful...");
exit(0);
}
printf("\nAdding at the start of the list successful...");
printf("\nDisplaying the list...");
rc = display_ll(&llist);
printf("\nAdding at the end...");
rc = add_at_end(&llist,103);
printf("\nDisplaying the list...");
rc = display_ll(&llist);
printf("\nModifying the start of the list...");
rc = modify_at_start(&llist,43);
printf("\nDisplaying the list...");
rc = display_ll(&llist);
printf("\nModifying the end of the list...");
rc = modify_at_end(&llist,203);
for(i=0;i<5;i++)
{
rc = add_at_end(&llist,(i+99));
}
printf("\nDisplaying the list...");
rc = display_ll(&llist);
for(i=0;i<3;i++)
{
rc = add_at_end(&llist,(i+150));
}
printf("\nDisplaying the list...");
rc = display_ll(&llist);
printf("\nInserting after node 3...");
rc = add_after(&llist,3,504);
printf("\nDisplaying the list...");
rc = display_ll(&llist);
printf("\nModifying node 4...");
rc = modify_node(&llist,4,555);
printf("\nDisplaying the list...");
rc = display_ll(&llist);
printf("\nDeleting Node 5...");
rc = delete_node(&llist,5);
printf("\nDisplaying the list...");
rc = display_ll(&llist);
for(i=0;i<3;i++)
{
rc = add_at_end(&llist,(i+342));
}
printf("\nDisplaying the list...");
rc = display_ll(&llist);
printf("\nInserting after node 1...");
rc = add_after(&llist,1,35);
printf("\nDisplaying the list...");
rc = display_ll(&llist);
printf("\n\n");
return 0;
}
int modify_node(couchfile_modify_request *rq, couchfile_pointer_info *nptr,
int start, int end, couchfile_modify_result *dst)
{
eterm_buf current_node;
int curnode_pos = 0;
int read_size = 0;
int list_start_pos = 0;
int node_len = 0;
int node_bound = 0;
int errcode = 0;
int kpos = 0;
char node_type[MAXATOMLEN + 1];
node_type[0] = 0;
DBG("Enter modify_node. %d - %d\r\n", start, end);
if(start == end)
{
return 0;
}
if(nptr == NULL)
{
current_node = empty_root;
}
else
{
if((read_size = pread_bin(rq->fd, nptr->pointer, ¤t_node.buf)) < 0)
{
return ERROR_READ_FILE;
}
current_node.size = read_size;
DBG("... read node from %d\r\n", nptr->pointer);
curnode_pos++; //Skip over 131.
}
couchfile_modify_result *local_result = make_modres(rq);
ei_decode_tuple_header(current_node.buf, &curnode_pos, NULL);
if(ei_decode_atom(current_node.buf, &curnode_pos, node_type) < 0)
{
errcode = ERROR_PARSE;
goto cleanup;
}
list_start_pos = curnode_pos;
if(ei_decode_list_header(current_node.buf, &curnode_pos, &node_len) < 0)
{
errcode = ERROR_PARSE;
goto cleanup;
}
if(strcmp("kv_node", node_type) == 0)
{
local_result->node_type = KV_NODE;
while(start < end)
{
DBG("act on kvnode item\r\n");
if(node_bound >= node_len)
{ //We're at the end of a leaf node.
DBG(" ... exec action at end!\r\n");
switch(rq->actions[start].type)
{
case ACTION_INSERT:
local_result->modified = 1;
mr_push_action(&rq->actions[start], local_result);
break;
case ACTION_REMOVE:
local_result->modified = 1;
break;
case ACTION_FETCH:
if(rq->fetch_callback)
{
//not found
(*rq->fetch_callback)(rq, rq->actions[start].key, NULL);
}
break;
}
start++;
}
else
{
kpos = find_first_gteq(current_node.buf, list_start_pos,
rq->actions[start].cmp_key,
&rq->cmp, node_bound);
if(kpos < 0)
{
errcode = ERROR_PARSE;
goto cleanup;
}
//Add items from node_bound up to but not including the current
mr_push_kv_range(current_node.buf, list_start_pos, node_bound,
rq->cmp.list_pos, local_result);
if(rq->cmp.last_cmp_val > 0) // Node key > action key
{
DBG(" Inserting action before\r\n");
switch(rq->actions[start].type)
{
case ACTION_INSERT:
local_result->modified = 1;
mr_push_action(&rq->actions[start], local_result);
break;
case ACTION_REMOVE:
local_result->modified = 1;
break;
case ACTION_FETCH:
if(rq->fetch_callback)
{
//not found
(*rq->fetch_callback)(rq, rq->actions[start].key, NULL);
}
break;
}
start++;
node_bound = rq->cmp.list_pos;
}
else if(rq->cmp.last_cmp_val < 0) // Node key < action key
{
DBG(" -- Continue with this action\r\n");
node_bound = rq->cmp.list_pos + 1;
mr_push_kv_range(current_node.buf, list_start_pos, node_bound - 1,
node_bound, local_result);
}
else //Node key == action key
{
DBG(" Replacing value with action\r\n");
switch(rq->actions[start].type)
{
case ACTION_INSERT:
local_result->modified = 1;
mr_push_action(&rq->actions[start], local_result);
node_bound = rq->cmp.list_pos + 1;
break;
case ACTION_REMOVE:
local_result->modified = 1;
node_bound = rq->cmp.list_pos + 1;
break;
case ACTION_FETCH:
if(rq->fetch_callback)
{
eterm_buf cb_tmp;
int cb_vpos = kpos;
ei_decode_tuple_header(current_node.buf, &cb_vpos, NULL);
ei_skip_term(current_node.buf, &cb_vpos);
cb_tmp.buf = current_node.buf + cb_vpos;
cb_tmp.size = cb_vpos;
ei_skip_term(current_node.buf, &cb_vpos);
cb_tmp.size = cb_vpos - cb_tmp.size;
(*rq->fetch_callback)(rq, rq->actions[start].key, &cb_tmp);
}
node_bound = rq->cmp.list_pos;
break;
}
start++;
}
}
}
//Push any items past the end of what we dealt with onto result.
if(node_bound < node_len)
{
mr_push_kv_range(current_node.buf, list_start_pos, node_bound,
node_len, local_result);
}
}
else if(strcmp("kp_node", node_type) == 0)
{
local_result->node_type = KP_NODE;
while(start < end)
{
kpos = find_first_gteq(current_node.buf, list_start_pos,
rq->actions[start].cmp_key,
&rq->cmp, node_bound);
if(kpos < 0)
{
errcode = ERROR_PARSE;
goto cleanup;
}
if(rq->cmp.list_pos == (node_len - 1)) //got last item in kp_node
{
//Push all items in node onto mr
mr_push_kp_range(current_node.buf, list_start_pos, node_bound,
rq->cmp.list_pos, local_result);
DBG(" ...descending into final item of kpnode\r\n");
couchfile_pointer_info *desc = read_pointer(current_node.buf, kpos);
errcode = modify_node(rq, desc, start, end, local_result);
if(local_result->values_end->value.pointer != desc)
{
free(desc);
}
if(errcode < 0)
{
goto cleanup;
}
node_bound = node_len;
break;
}
else
{
//Get all actions with key <= the key of the current item in the
//kp_node
//Push items in node up to but not including current onto mr
mr_push_kp_range(current_node.buf, list_start_pos, node_bound,
rq->cmp.list_pos - 1, local_result);
int range_end = start;
while(range_end < end &&
((*rq->cmp.compare)(rq->actions[range_end].cmp_key, rq->cmp.last_cmp_key) <= 0))
{
range_end++;
}
DBG(" ...descending into item %d of kpnode\r\n", rq->cmp.list_pos);
node_bound = rq->cmp.list_pos + 1;
couchfile_pointer_info *desc = read_pointer(current_node.buf, kpos);
errcode = modify_node(rq, desc, start, range_end, local_result);
if(local_result->values_end->value.pointer != desc)
{
free(desc);
}
if(errcode < 0)
{
goto cleanup;
}
start = range_end;
}
}
DBG(".. Finished kp node, up to %d\r\n", node_bound);
if(node_bound < node_len)
{
//Processed all the actions but haven't exhausted this kpnode.
//push the rest of it onto the mr.
mr_push_kp_range(current_node.buf, list_start_pos, node_bound, node_len,
local_result);
}
}
else
{
errcode = ERROR_PARSE;
goto cleanup;
}
//If we've done modifications, write out the last leaf node.
errcode = flush_mr(local_result);
if(errcode == 0)
{
if(!local_result->modified && nptr != NULL)
{
//If we didn't do anything, give back the pointer to the original
mr_push_pointerinfo(nptr, dst);
}
else
{
//Otherwise, give back the pointers to the nodes we've created.
dst->modified = 1;
errcode = mr_move_pointers(local_result, dst);
}
}
cleanup:
free_modres(local_result);
if(current_node.buf != empty_root.buf)
{
free(current_node.buf);
}
return errcode;
}