/*- * Copyright (c) 2014-2015 MongoDB, Inc. * Copyright (c) 2008-2014 WiredTiger, Inc. * All rights reserved. * * See the file LICENSE for redistribution information. */ #include "wt_internal.h" /* * __check_leaf_key_range -- * Check the search key is in the leaf page's key range. */ static inline int __check_leaf_key_range(WT_SESSION_IMPL *session, uint64_t recno, WT_REF *leaf, WT_CURSOR_BTREE *cbt) { WT_PAGE_INDEX *pindex; uint32_t indx; /* * There are reasons we can't do the fast checks, and we continue with * the leaf page search in those cases, only skipping the complete leaf * page search if we know it's not going to work. */ cbt->compare = 0; /* * Check if the search key is smaller than the parent's starting key for * this page. */ if (recno < leaf->key.recno) { cbt->compare = 1; /* page keys > search key */ return (0); } /* * Check if the search key is greater than or equal to the starting key * for the parent's next page. * * !!! * Check that "indx + 1" is a valid page-index entry first, because it * also checks that "indx" is a valid page-index entry, and we have to * do that latter check before looking at the indx slot of the array * for a match to leaf (in other words, our page hint might be wrong). */ WT_INTL_INDEX_GET(session, leaf->home, pindex); indx = leaf->pindex_hint; if (indx + 1 < pindex->entries && pindex->index[indx] == leaf) if (recno >= pindex->index[indx + 1]->key.recno) { cbt->compare = -1; /* page keys < search key */ return (0); } return (0); } /* * __wt_col_search -- * Search a column-store tree for a specific record-based key. */ int __wt_col_search(WT_SESSION_IMPL *session, uint64_t search_recno, WT_REF *leaf, WT_CURSOR_BTREE *cbt) { WT_BTREE *btree; WT_COL *cip; WT_DECL_RET; WT_INSERT *ins; WT_INSERT_HEAD *ins_head; WT_PAGE *page; WT_PAGE_INDEX *pindex, *parent_pindex; WT_REF *current, *descent; uint64_t recno; uint32_t base, indx, limit; int depth; btree = S2BT(session); __cursor_pos_clear(cbt); /* * When appending a new record, the search record number will be an * out-of-band value, search for the largest key in the table instead. */ if ((recno = search_recno) == WT_RECNO_OOB) recno = UINT64_MAX; /* * We may be searching only a single leaf page, not the full tree. In * the normal case where the page links to a parent, check the page's * parent keys before doing the full search, it's faster when the * cursor is being re-positioned. (One case where the page doesn't * have a parent is if it is being re-instantiated in memory as part * of a split). */ if (leaf != NULL) { WT_ASSERT(session, search_recno != WT_RECNO_OOB); if (leaf->home != NULL) { WT_RET(__check_leaf_key_range( session, recno, leaf, cbt)); if (cbt->compare != 0) { /* * !!! * WT_CURSOR.search_near uses the slot value to * decide if there was an on-page match. */ cbt->slot = 0; return (0); } } current = leaf; goto leaf_only; } restart_root: /* Search the internal pages of the tree. */ current = &btree->root; for (depth = 2, pindex = NULL;; ++depth) { parent_pindex = pindex; restart_page: page = current->page; if (page->type != WT_PAGE_COL_INT) break; WT_ASSERT(session, current->key.recno == page->pg_intl_recno); WT_INTL_INDEX_GET(session, page, pindex); base = pindex->entries; descent = pindex->index[base - 1]; /* Fast path appends. */ if (recno >= descent->key.recno) { /* * If on the last slot (the key is larger than any key * on the page), check for an internal page split race. */ if (parent_pindex != NULL && __wt_split_intl_race( session, current->home, parent_pindex)) { WT_RET(__wt_page_release(session, current, 0)); goto restart_root; } goto descend; } /* Binary search of internal pages. */ for (base = 0, limit = pindex->entries - 1; limit != 0; limit >>= 1) { indx = base + (limit >> 1); descent = pindex->index[indx]; if (recno == descent->key.recno) break; if (recno < descent->key.recno) continue; base = indx + 1; --limit; } descend: /* * Reference the slot used for next step down the tree. * * Base is the smallest index greater than recno and may be the * (last + 1) index. The slot for descent is the one before * base. */ if (recno != descent->key.recno) { /* * We don't have to correct for base == 0 because the * only way for base to be 0 is if recno is the page's * starting recno. */ WT_ASSERT(session, base > 0); descent = pindex->index[base - 1]; } /* * Swap the current page for the child page. If the page splits * while we're retrieving it, restart the search in the current * page; otherwise return on error, the swap call ensures we're * holding nothing on failure. */ if ((ret = __wt_page_swap(session, current, descent, 0)) == 0) { current = descent; continue; } if (ret == WT_RESTART) goto restart_page; return (ret); } /* Track how deep the tree gets. */ if (depth > btree->maximum_depth) btree->maximum_depth = depth; leaf_only: page = current->page; cbt->ref = current; cbt->recno = recno; /* * Don't bother searching if the caller is appending a new record where * we'll allocate the record number; we're not going to find a match by * definition, and we figure out the record number and position when we * do the work. */ if (search_recno == WT_RECNO_OOB) { cbt->compare = -1; return (0); } /* * Set the on-page slot to an impossible value larger than any possible * slot (it's used to interpret the search function's return after the * search returns an insert list for a page that has no entries). */ cbt->slot = UINT32_MAX; /* * Search the leaf page. * * Search after a page is pinned does a search of the pinned page before * doing a full tree search, in which case we might be searching for a * record logically before the page. Return failure, and there's nothing * else to do, the record isn't going to be on this page. * * We don't check inside the search path for a record greater than the * maximum record in the tree; in that case, we get here with a record * that's impossibly large for the page. We do have additional setup to * do in that case, the record may be appended to the page. */ cbt->compare = 0; if (page->type == WT_PAGE_COL_FIX) { if (recno < page->pg_fix_recno) { cbt->compare = 1; return (0); } if (recno >= page->pg_fix_recno + page->pg_fix_entries) { cbt->recno = page->pg_fix_recno + page->pg_fix_entries; goto past_end; } else ins_head = WT_COL_UPDATE_SINGLE(page); } else { if (recno < page->pg_var_recno) { cbt->compare = 1; return (0); } if ((cip = __col_var_search(page, recno, NULL)) == NULL) { cbt->recno = __col_var_last_recno(page); goto past_end; } else { cbt->slot = WT_COL_SLOT(page, cip); ins_head = WT_COL_UPDATE_SLOT(page, cbt->slot); } } /* * We have a match on the page, check for an update. Check the page's * update list (fixed-length), or slot's update list (variable-length) * for a better match. The only better match we can find is an exact * match, otherwise the existing match on the page is the one we want. * For that reason, don't set the cursor's WT_INSERT_HEAD/WT_INSERT pair * until we know we have a useful entry. */ if ((ins = __col_insert_search( ins_head, cbt->ins_stack, cbt->next_stack, recno)) != NULL) if (recno == WT_INSERT_RECNO(ins)) { cbt->ins_head = ins_head; cbt->ins = ins; } return (0); past_end: /* * A record past the end of the page's standard information. Check the * append list; by definition, any record on the append list is closer * than the last record on the page, so it's a better choice for return. * This is a rarely used path: we normally find exact matches, because * column-store files are dense, but in this case the caller searched * past the end of the table. */ cbt->ins_head = WT_COL_APPEND(page); if ((cbt->ins = __col_insert_search( cbt->ins_head, cbt->ins_stack, cbt->next_stack, recno)) == NULL) cbt->compare = -1; else { cbt->recno = WT_INSERT_RECNO(cbt->ins); if (recno == cbt->recno) cbt->compare = 0; else if (recno < cbt->recno) cbt->compare = 1; else cbt->compare = -1; } return (0); }