| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183 |
- // Copyright (c) 2011-present, Facebook, Inc. All rights reserved.
- // This source code is licensed under both the GPLv2 (found in the
- // COPYING file in the root directory) and Apache 2.0 License
- // (found in the LICENSE.Apache file in the root directory).
- #include <string>
- #include <vector>
- #include <algorithm>
- #include <utility>
- #include "db/db_iter.h"
- #include "db/dbformat.h"
- #include "rocksdb/comparator.h"
- #include "rocksdb/options.h"
- #include "rocksdb/perf_context.h"
- #include "rocksdb/slice.h"
- #include "rocksdb/statistics.h"
- #include "table/iterator_wrapper.h"
- #include "table/merging_iterator.h"
- #include "test_util/sync_point.h"
- #include "test_util/testharness.h"
- #include "util/string_util.h"
- #include "utilities/merge_operators.h"
- namespace ROCKSDB_NAMESPACE {
- static uint64_t TestGetTickerCount(const Options& options,
- Tickers ticker_type) {
- return options.statistics->getTickerCount(ticker_type);
- }
- class TestIterator : public InternalIterator {
- public:
- explicit TestIterator(const Comparator* comparator)
- : initialized_(false),
- valid_(false),
- sequence_number_(0),
- iter_(0),
- cmp(comparator) {
- data_.reserve(16);
- }
- void AddPut(std::string argkey, std::string argvalue) {
- Add(argkey, kTypeValue, argvalue);
- }
- void AddDeletion(std::string argkey) {
- Add(argkey, kTypeDeletion, std::string());
- }
- void AddSingleDeletion(std::string argkey) {
- Add(argkey, kTypeSingleDeletion, std::string());
- }
- void AddMerge(std::string argkey, std::string argvalue) {
- Add(argkey, kTypeMerge, argvalue);
- }
- void Add(std::string argkey, ValueType type, std::string argvalue) {
- Add(argkey, type, argvalue, sequence_number_++);
- }
- void Add(std::string argkey, ValueType type, std::string argvalue,
- size_t seq_num, bool update_iter = false) {
- valid_ = true;
- ParsedInternalKey internal_key(argkey, seq_num, type);
- data_.push_back(
- std::pair<std::string, std::string>(std::string(), argvalue));
- AppendInternalKey(&data_.back().first, internal_key);
- if (update_iter && valid_ && cmp.Compare(data_.back().first, key()) < 0) {
- // insert a key smaller than current key
- Finish();
- // data_[iter_] is not anymore the current element of the iterator.
- // Increment it to reposition it to the right position.
- iter_++;
- }
- }
- // should be called before operations with iterator
- void Finish() {
- initialized_ = true;
- std::sort(data_.begin(), data_.end(),
- [this](std::pair<std::string, std::string> a,
- std::pair<std::string, std::string> b) {
- return (cmp.Compare(a.first, b.first) < 0);
- });
- }
- // Removes the key from the set of keys over which this iterator iterates.
- // Not to be confused with AddDeletion().
- // If the iterator is currently positioned on this key, the deletion will
- // apply next time the iterator moves.
- // Used for simulating ForwardIterator updating to a new version that doesn't
- // have some of the keys (e.g. after compaction with a filter).
- void Vanish(std::string _key) {
- if (valid_ && data_[iter_].first == _key) {
- delete_current_ = true;
- return;
- }
- for (auto it = data_.begin(); it != data_.end(); ++it) {
- ParsedInternalKey ikey;
- bool ok __attribute__((__unused__)) = ParseInternalKey(it->first, &ikey);
- assert(ok);
- if (ikey.user_key != _key) {
- continue;
- }
- if (valid_ && data_.begin() + iter_ > it) {
- --iter_;
- }
- data_.erase(it);
- return;
- }
- assert(false);
- }
- // Number of operations done on this iterator since construction.
- size_t steps() const { return steps_; }
- bool Valid() const override {
- assert(initialized_);
- return valid_;
- }
- void SeekToFirst() override {
- assert(initialized_);
- ++steps_;
- DeleteCurrentIfNeeded();
- valid_ = (data_.size() > 0);
- iter_ = 0;
- }
- void SeekToLast() override {
- assert(initialized_);
- ++steps_;
- DeleteCurrentIfNeeded();
- valid_ = (data_.size() > 0);
- iter_ = data_.size() - 1;
- }
- void Seek(const Slice& target) override {
- assert(initialized_);
- SeekToFirst();
- ++steps_;
- if (!valid_) {
- return;
- }
- while (iter_ < data_.size() &&
- (cmp.Compare(data_[iter_].first, target) < 0)) {
- ++iter_;
- }
- if (iter_ == data_.size()) {
- valid_ = false;
- }
- }
- void SeekForPrev(const Slice& target) override {
- assert(initialized_);
- DeleteCurrentIfNeeded();
- SeekForPrevImpl(target, &cmp);
- }
- void Next() override {
- assert(initialized_);
- assert(valid_);
- assert(iter_ < data_.size());
- ++steps_;
- if (delete_current_) {
- DeleteCurrentIfNeeded();
- } else {
- ++iter_;
- }
- valid_ = iter_ < data_.size();
- }
- void Prev() override {
- assert(initialized_);
- assert(valid_);
- assert(iter_ < data_.size());
- ++steps_;
- DeleteCurrentIfNeeded();
- if (iter_ == 0) {
- valid_ = false;
- } else {
- --iter_;
- }
- }
- Slice key() const override {
- assert(initialized_);
- return data_[iter_].first;
- }
- Slice value() const override {
- assert(initialized_);
- return data_[iter_].second;
- }
- Status status() const override {
- assert(initialized_);
- return Status::OK();
- }
- bool IsKeyPinned() const override { return true; }
- bool IsValuePinned() const override { return true; }
- private:
- bool initialized_;
- bool valid_;
- size_t sequence_number_;
- size_t iter_;
- size_t steps_ = 0;
- InternalKeyComparator cmp;
- std::vector<std::pair<std::string, std::string>> data_;
- bool delete_current_ = false;
- void DeleteCurrentIfNeeded() {
- if (!delete_current_) {
- return;
- }
- data_.erase(data_.begin() + iter_);
- delete_current_ = false;
- }
- };
- class DBIteratorTest : public testing::Test {
- public:
- Env* env_;
- DBIteratorTest() : env_(Env::Default()) {}
- };
- TEST_F(DBIteratorTest, DBIteratorPrevNext) {
- Options options;
- ImmutableCFOptions cf_options = ImmutableCFOptions(options);
- MutableCFOptions mutable_cf_options = MutableCFOptions(options);
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddDeletion("a");
- internal_iter->AddDeletion("a");
- internal_iter->AddDeletion("a");
- internal_iter->AddDeletion("a");
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddPut("b", "val_b");
- internal_iter->Finish();
- ReadOptions ro;
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 10, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "val_b");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "val_a");
- db_iter->Next();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "val_b");
- db_iter->Next();
- ASSERT_TRUE(!db_iter->Valid());
- }
- // Test to check the SeekToLast() with iterate_upper_bound not set
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddPut("b", "val_b");
- internal_iter->AddPut("b", "val_b");
- internal_iter->AddPut("c", "val_c");
- internal_iter->Finish();
- ReadOptions ro;
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 10, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- }
- // Test to check the SeekToLast() with iterate_upper_bound set
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddPut("b", "val_b");
- internal_iter->AddPut("c", "val_c");
- internal_iter->AddPut("d", "val_d");
- internal_iter->AddPut("e", "val_e");
- internal_iter->AddPut("f", "val_f");
- internal_iter->Finish();
- Slice prefix("d");
- ReadOptions ro;
- ro.iterate_upper_bound = &prefix;
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 10, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- db_iter->Next();
- ASSERT_TRUE(!db_iter->Valid());
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- }
- // Test to check the SeekToLast() iterate_upper_bound set to a key that
- // is not Put yet
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddPut("b", "val_b");
- internal_iter->AddPut("c", "val_c");
- internal_iter->AddPut("d", "val_d");
- internal_iter->Finish();
- Slice prefix("z");
- ReadOptions ro;
- ro.iterate_upper_bound = &prefix;
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 10, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "d");
- db_iter->Next();
- ASSERT_TRUE(!db_iter->Valid());
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "d");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- }
- // Test to check the SeekToLast() with iterate_upper_bound set to the
- // first key
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddPut("b", "val_b");
- internal_iter->AddPut("b", "val_b");
- internal_iter->Finish();
- Slice prefix("a");
- ReadOptions ro;
- ro.iterate_upper_bound = &prefix;
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 10, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(!db_iter->Valid());
- }
- // Test case to check SeekToLast with iterate_upper_bound set
- // (same key put may times - SeekToLast should start with the
- // maximum sequence id of the upper bound)
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddPut("b", "val_b");
- internal_iter->AddPut("c", "val_c");
- internal_iter->AddPut("c", "val_c");
- internal_iter->AddPut("c", "val_c");
- internal_iter->AddPut("c", "val_c");
- internal_iter->AddPut("c", "val_c");
- internal_iter->AddPut("c", "val_c");
- internal_iter->AddPut("c", "val_c");
- internal_iter->Finish();
- Slice prefix("c");
- ReadOptions ro;
- ro.iterate_upper_bound = &prefix;
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 7, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- SetPerfLevel(kEnableCount);
- ASSERT_TRUE(GetPerfLevel() == kEnableCount);
- get_perf_context()->Reset();
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(static_cast<int>(get_perf_context()->internal_key_skipped_count), 1);
- ASSERT_EQ(db_iter->key().ToString(), "b");
- SetPerfLevel(kDisable);
- }
- // Test to check the SeekToLast() with the iterate_upper_bound set
- // (Checking the value of the key which has sequence ids greater than
- // and less that the iterator's sequence id)
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "val_a1");
- internal_iter->AddPut("a", "val_a2");
- internal_iter->AddPut("b", "val_b1");
- internal_iter->AddPut("c", "val_c1");
- internal_iter->AddPut("c", "val_c2");
- internal_iter->AddPut("c", "val_c3");
- internal_iter->AddPut("b", "val_b2");
- internal_iter->AddPut("d", "val_d1");
- internal_iter->Finish();
- Slice prefix("c");
- ReadOptions ro;
- ro.iterate_upper_bound = &prefix;
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 4, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "val_b1");
- }
- // Test to check the SeekToLast() with the iterate_upper_bound set to the
- // key that is deleted
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddDeletion("a");
- internal_iter->AddPut("b", "val_b");
- internal_iter->AddPut("c", "val_c");
- internal_iter->Finish();
- Slice prefix("a");
- ReadOptions ro;
- ro.iterate_upper_bound = &prefix;
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 10, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(!db_iter->Valid());
- }
- // Test to check the SeekToLast() with the iterate_upper_bound set
- // (Deletion cases)
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddPut("b", "val_b");
- internal_iter->AddDeletion("b");
- internal_iter->AddPut("c", "val_c");
- internal_iter->Finish();
- Slice prefix("c");
- ReadOptions ro;
- ro.iterate_upper_bound = &prefix;
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 10, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- db_iter->Next();
- ASSERT_TRUE(!db_iter->Valid());
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- }
- // Test to check the SeekToLast() with iterate_upper_bound set
- // (Deletion cases - Lot of internal keys after the upper_bound
- // is deleted)
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddPut("b", "val_b");
- internal_iter->AddDeletion("c");
- internal_iter->AddDeletion("d");
- internal_iter->AddDeletion("e");
- internal_iter->AddDeletion("f");
- internal_iter->AddDeletion("g");
- internal_iter->AddDeletion("h");
- internal_iter->Finish();
- Slice prefix("c");
- ReadOptions ro;
- ro.iterate_upper_bound = &prefix;
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 7, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- SetPerfLevel(kEnableCount);
- ASSERT_TRUE(GetPerfLevel() == kEnableCount);
- get_perf_context()->Reset();
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(static_cast<int>(get_perf_context()->internal_delete_skipped_count), 0);
- ASSERT_EQ(db_iter->key().ToString(), "b");
- SetPerfLevel(kDisable);
- }
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddDeletion("a");
- internal_iter->AddDeletion("a");
- internal_iter->AddDeletion("a");
- internal_iter->AddDeletion("a");
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddPut("b", "val_b");
- internal_iter->Finish();
- ReadOptions ro;
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 10, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToFirst();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "val_a");
- db_iter->Next();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "val_b");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "val_a");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddPut("b", "val_b");
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddPut("b", "val_b");
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddPut("b", "val_b");
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddPut("b", "val_b");
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddPut("b", "val_b");
- internal_iter->Finish();
- ReadOptions ro;
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 2, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "val_b");
- db_iter->Next();
- ASSERT_TRUE(!db_iter->Valid());
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "val_b");
- }
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddPut("b", "val_b");
- internal_iter->AddPut("c", "val_c");
- internal_iter->Finish();
- ReadOptions ro;
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 10, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- ASSERT_EQ(db_iter->value().ToString(), "val_c");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "val_b");
- db_iter->Next();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- ASSERT_EQ(db_iter->value().ToString(), "val_c");
- }
- }
- TEST_F(DBIteratorTest, DBIteratorEmpty) {
- Options options;
- ImmutableCFOptions cf_options = ImmutableCFOptions(options);
- MutableCFOptions mutable_cf_options = MutableCFOptions(options);
- ReadOptions ro;
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 0, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(!db_iter->Valid());
- }
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 0, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToFirst();
- ASSERT_TRUE(!db_iter->Valid());
- }
- }
- TEST_F(DBIteratorTest, DBIteratorUseSkipCountSkips) {
- ReadOptions ro;
- Options options;
- options.statistics = ROCKSDB_NAMESPACE::CreateDBStatistics();
- options.merge_operator = MergeOperators::CreateFromStringId("stringappend");
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- for (size_t i = 0; i < 200; ++i) {
- internal_iter->AddPut("a", "a");
- internal_iter->AddPut("b", "b");
- internal_iter->AddPut("c", "c");
- }
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, ImmutableCFOptions(options), MutableCFOptions(options),
- BytewiseComparator(), internal_iter, 2,
- options.max_sequential_skip_in_iterations, nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- ASSERT_EQ(db_iter->value().ToString(), "c");
- ASSERT_EQ(TestGetTickerCount(options, NUMBER_OF_RESEEKS_IN_ITERATION), 1u);
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "b");
- ASSERT_EQ(TestGetTickerCount(options, NUMBER_OF_RESEEKS_IN_ITERATION), 2u);
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "a");
- ASSERT_EQ(TestGetTickerCount(options, NUMBER_OF_RESEEKS_IN_ITERATION), 3u);
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- ASSERT_EQ(TestGetTickerCount(options, NUMBER_OF_RESEEKS_IN_ITERATION), 3u);
- }
- TEST_F(DBIteratorTest, DBIteratorUseSkip) {
- ReadOptions ro;
- Options options;
- options.merge_operator = MergeOperators::CreateFromStringId("stringappend");
- ImmutableCFOptions cf_options = ImmutableCFOptions(options);
- MutableCFOptions mutable_cf_options = MutableCFOptions(options);
- {
- for (size_t i = 0; i < 200; ++i) {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("b", "merge_1");
- internal_iter->AddMerge("a", "merge_2");
- for (size_t k = 0; k < 200; ++k) {
- internal_iter->AddPut("c", ToString(k));
- }
- internal_iter->Finish();
- options.statistics = ROCKSDB_NAMESPACE::CreateDBStatistics();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, i + 2, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- ASSERT_EQ(db_iter->value().ToString(), ToString(i));
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "merge_1");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "merge_2");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- }
- {
- for (size_t i = 0; i < 200; ++i) {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("b", "merge_1");
- internal_iter->AddMerge("a", "merge_2");
- for (size_t k = 0; k < 200; ++k) {
- internal_iter->AddDeletion("c");
- }
- internal_iter->AddPut("c", "200");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, i + 2, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "merge_1");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "merge_2");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("b", "merge_1");
- internal_iter->AddMerge("a", "merge_2");
- for (size_t i = 0; i < 200; ++i) {
- internal_iter->AddDeletion("c");
- }
- internal_iter->AddPut("c", "200");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 202, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- ASSERT_EQ(db_iter->value().ToString(), "200");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "merge_1");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "merge_2");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- }
- {
- for (size_t i = 0; i < 200; ++i) {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- for (size_t k = 0; k < 200; ++k) {
- internal_iter->AddDeletion("c");
- }
- internal_iter->AddPut("c", "200");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, i, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(!db_iter->Valid());
- db_iter->SeekToFirst();
- ASSERT_TRUE(!db_iter->Valid());
- }
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- for (size_t i = 0; i < 200; ++i) {
- internal_iter->AddDeletion("c");
- }
- internal_iter->AddPut("c", "200");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 200, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- ASSERT_EQ(db_iter->value().ToString(), "200");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- db_iter->SeekToFirst();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- ASSERT_EQ(db_iter->value().ToString(), "200");
- db_iter->Next();
- ASSERT_TRUE(!db_iter->Valid());
- }
- {
- for (size_t i = 0; i < 200; ++i) {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("b", "merge_1");
- internal_iter->AddMerge("a", "merge_2");
- for (size_t k = 0; k < 200; ++k) {
- internal_iter->AddPut("d", ToString(k));
- }
- for (size_t k = 0; k < 200; ++k) {
- internal_iter->AddPut("c", ToString(k));
- }
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, i + 2, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "d");
- ASSERT_EQ(db_iter->value().ToString(), ToString(i));
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "merge_1");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "merge_2");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- }
- {
- for (size_t i = 0; i < 200; ++i) {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("b", "b");
- internal_iter->AddMerge("a", "a");
- for (size_t k = 0; k < 200; ++k) {
- internal_iter->AddMerge("c", ToString(k));
- }
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, i + 2, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- std::string merge_result = "0";
- for (size_t j = 1; j <= i; ++j) {
- merge_result += "," + ToString(j);
- }
- ASSERT_EQ(db_iter->value().ToString(), merge_result);
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "b");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "a");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- }
- }
- TEST_F(DBIteratorTest, DBIteratorSkipInternalKeys) {
- Options options;
- ImmutableCFOptions cf_options = ImmutableCFOptions(options);
- MutableCFOptions mutable_cf_options = MutableCFOptions(options);
- ReadOptions ro;
- // Basic test case ... Make sure explicityly passing the default value works.
- // Skipping internal keys is disabled by default, when the value is 0.
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddDeletion("b");
- internal_iter->AddDeletion("b");
- internal_iter->AddPut("c", "val_c");
- internal_iter->AddPut("c", "val_c");
- internal_iter->AddDeletion("c");
- internal_iter->AddPut("d", "val_d");
- internal_iter->Finish();
- ro.max_skippable_internal_keys = 0;
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 10, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToFirst();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "val_a");
- db_iter->Next();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "d");
- ASSERT_EQ(db_iter->value().ToString(), "val_d");
- db_iter->Next();
- ASSERT_TRUE(!db_iter->Valid());
- ASSERT_TRUE(db_iter->status().ok());
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "d");
- ASSERT_EQ(db_iter->value().ToString(), "val_d");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "val_a");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- ASSERT_TRUE(db_iter->status().ok());
- }
- // Test to make sure that the request will *not* fail as incomplete if
- // num_internal_keys_skipped is *equal* to max_skippable_internal_keys
- // threshold. (It will fail as incomplete only when the threshold is
- // exceeded.)
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddDeletion("b");
- internal_iter->AddDeletion("b");
- internal_iter->AddPut("c", "val_c");
- internal_iter->Finish();
- ro.max_skippable_internal_keys = 2;
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 10, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToFirst();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "val_a");
- db_iter->Next();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- ASSERT_EQ(db_iter->value().ToString(), "val_c");
- db_iter->Next();
- ASSERT_TRUE(!db_iter->Valid());
- ASSERT_TRUE(db_iter->status().ok());
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- ASSERT_EQ(db_iter->value().ToString(), "val_c");
- db_iter->Prev();
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "val_a");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- ASSERT_TRUE(db_iter->status().ok());
- }
- // Fail the request as incomplete when num_internal_keys_skipped >
- // max_skippable_internal_keys
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddDeletion("b");
- internal_iter->AddDeletion("b");
- internal_iter->AddDeletion("b");
- internal_iter->AddPut("c", "val_c");
- internal_iter->Finish();
- ro.max_skippable_internal_keys = 2;
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 10, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToFirst();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "val_a");
- db_iter->Next();
- ASSERT_TRUE(!db_iter->Valid());
- ASSERT_TRUE(db_iter->status().IsIncomplete());
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- ASSERT_EQ(db_iter->value().ToString(), "val_c");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- ASSERT_TRUE(db_iter->status().IsIncomplete());
- }
- // Test that the num_internal_keys_skipped counter resets after a successful
- // read.
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddDeletion("b");
- internal_iter->AddDeletion("b");
- internal_iter->AddPut("c", "val_c");
- internal_iter->AddDeletion("d");
- internal_iter->AddDeletion("d");
- internal_iter->AddDeletion("d");
- internal_iter->AddPut("e", "val_e");
- internal_iter->Finish();
- ro.max_skippable_internal_keys = 2;
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 10, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToFirst();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "val_a");
- db_iter->Next();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- ASSERT_EQ(db_iter->value().ToString(), "val_c");
- db_iter->Next(); // num_internal_keys_skipped counter resets here.
- ASSERT_TRUE(!db_iter->Valid());
- ASSERT_TRUE(db_iter->status().IsIncomplete());
- }
- // Test that the num_internal_keys_skipped counter resets after a successful
- // read.
- // Reverse direction
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddDeletion("b");
- internal_iter->AddDeletion("b");
- internal_iter->AddDeletion("b");
- internal_iter->AddPut("c", "val_c");
- internal_iter->AddDeletion("d");
- internal_iter->AddDeletion("d");
- internal_iter->AddPut("e", "val_e");
- internal_iter->Finish();
- ro.max_skippable_internal_keys = 2;
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 10, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "e");
- ASSERT_EQ(db_iter->value().ToString(), "val_e");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- ASSERT_EQ(db_iter->value().ToString(), "val_c");
- db_iter->Prev(); // num_internal_keys_skipped counter resets here.
- ASSERT_TRUE(!db_iter->Valid());
- ASSERT_TRUE(db_iter->status().IsIncomplete());
- }
- // Test that skipping separate keys is handled
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddDeletion("b");
- internal_iter->AddDeletion("c");
- internal_iter->AddDeletion("d");
- internal_iter->AddPut("e", "val_e");
- internal_iter->Finish();
- ro.max_skippable_internal_keys = 2;
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 10, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToFirst();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "val_a");
- db_iter->Next();
- ASSERT_TRUE(!db_iter->Valid());
- ASSERT_TRUE(db_iter->status().IsIncomplete());
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "e");
- ASSERT_EQ(db_iter->value().ToString(), "val_e");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- ASSERT_TRUE(db_iter->status().IsIncomplete());
- }
- // Test if alternating puts and deletes of the same key are handled correctly.
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddPut("b", "val_b");
- internal_iter->AddDeletion("b");
- internal_iter->AddPut("c", "val_c");
- internal_iter->AddDeletion("c");
- internal_iter->AddPut("d", "val_d");
- internal_iter->AddDeletion("d");
- internal_iter->AddPut("e", "val_e");
- internal_iter->Finish();
- ro.max_skippable_internal_keys = 2;
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 10, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToFirst();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "val_a");
- db_iter->Next();
- ASSERT_TRUE(!db_iter->Valid());
- ASSERT_TRUE(db_iter->status().IsIncomplete());
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "e");
- ASSERT_EQ(db_iter->value().ToString(), "val_e");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- ASSERT_TRUE(db_iter->status().IsIncomplete());
- }
- // Test for large number of skippable internal keys with *default*
- // max_sequential_skip_in_iterations.
- {
- for (size_t i = 1; i <= 200; ++i) {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "val_a");
- for (size_t j = 1; j <= i; ++j) {
- internal_iter->AddPut("b", "val_b");
- internal_iter->AddDeletion("b");
- }
- internal_iter->AddPut("c", "val_c");
- internal_iter->Finish();
- ro.max_skippable_internal_keys = i;
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 2 * i + 1, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToFirst();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "val_a");
- db_iter->Next();
- if ((options.max_sequential_skip_in_iterations + 1) >=
- ro.max_skippable_internal_keys) {
- ASSERT_TRUE(!db_iter->Valid());
- ASSERT_TRUE(db_iter->status().IsIncomplete());
- } else {
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- ASSERT_EQ(db_iter->value().ToString(), "val_c");
- }
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- ASSERT_EQ(db_iter->value().ToString(), "val_c");
- db_iter->Prev();
- if ((options.max_sequential_skip_in_iterations + 1) >=
- ro.max_skippable_internal_keys) {
- ASSERT_TRUE(!db_iter->Valid());
- ASSERT_TRUE(db_iter->status().IsIncomplete());
- } else {
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "val_a");
- }
- }
- }
- // Test for large number of skippable internal keys with a *non-default*
- // max_sequential_skip_in_iterations.
- {
- for (size_t i = 1; i <= 200; ++i) {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "val_a");
- for (size_t j = 1; j <= i; ++j) {
- internal_iter->AddPut("b", "val_b");
- internal_iter->AddDeletion("b");
- }
- internal_iter->AddPut("c", "val_c");
- internal_iter->Finish();
- options.max_sequential_skip_in_iterations = 1000;
- ro.max_skippable_internal_keys = i;
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 2 * i + 1, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToFirst();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "val_a");
- db_iter->Next();
- ASSERT_TRUE(!db_iter->Valid());
- ASSERT_TRUE(db_iter->status().IsIncomplete());
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- ASSERT_EQ(db_iter->value().ToString(), "val_c");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- ASSERT_TRUE(db_iter->status().IsIncomplete());
- }
- }
- }
- TEST_F(DBIteratorTest, DBIterator1) {
- ReadOptions ro;
- Options options;
- options.merge_operator = MergeOperators::CreateFromStringId("stringappend");
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "0");
- internal_iter->AddPut("b", "0");
- internal_iter->AddDeletion("b");
- internal_iter->AddMerge("a", "1");
- internal_iter->AddMerge("b", "2");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, ImmutableCFOptions(options), MutableCFOptions(options),
- BytewiseComparator(), internal_iter, 1,
- options.max_sequential_skip_in_iterations, nullptr /*read_callback*/));
- db_iter->SeekToFirst();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "0");
- db_iter->Next();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- db_iter->Next();
- ASSERT_FALSE(db_iter->Valid());
- }
- TEST_F(DBIteratorTest, DBIterator2) {
- ReadOptions ro;
- Options options;
- options.merge_operator = MergeOperators::CreateFromStringId("stringappend");
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "0");
- internal_iter->AddPut("b", "0");
- internal_iter->AddDeletion("b");
- internal_iter->AddMerge("a", "1");
- internal_iter->AddMerge("b", "2");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, ImmutableCFOptions(options), MutableCFOptions(options),
- BytewiseComparator(), internal_iter, 0,
- options.max_sequential_skip_in_iterations, nullptr /*read_callback*/));
- db_iter->SeekToFirst();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "0");
- db_iter->Next();
- ASSERT_TRUE(!db_iter->Valid());
- }
- TEST_F(DBIteratorTest, DBIterator3) {
- ReadOptions ro;
- Options options;
- options.merge_operator = MergeOperators::CreateFromStringId("stringappend");
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "0");
- internal_iter->AddPut("b", "0");
- internal_iter->AddDeletion("b");
- internal_iter->AddMerge("a", "1");
- internal_iter->AddMerge("b", "2");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, ImmutableCFOptions(options), MutableCFOptions(options),
- BytewiseComparator(), internal_iter, 2,
- options.max_sequential_skip_in_iterations, nullptr /*read_callback*/));
- db_iter->SeekToFirst();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "0");
- db_iter->Next();
- ASSERT_TRUE(!db_iter->Valid());
- }
- TEST_F(DBIteratorTest, DBIterator4) {
- ReadOptions ro;
- Options options;
- options.merge_operator = MergeOperators::CreateFromStringId("stringappend");
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "0");
- internal_iter->AddPut("b", "0");
- internal_iter->AddDeletion("b");
- internal_iter->AddMerge("a", "1");
- internal_iter->AddMerge("b", "2");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, ImmutableCFOptions(options), MutableCFOptions(options),
- BytewiseComparator(), internal_iter, 4,
- options.max_sequential_skip_in_iterations, nullptr /*read_callback*/));
- db_iter->SeekToFirst();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "0,1");
- db_iter->Next();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "2");
- db_iter->Next();
- ASSERT_TRUE(!db_iter->Valid());
- }
- TEST_F(DBIteratorTest, DBIterator5) {
- ReadOptions ro;
- Options options;
- options.merge_operator = MergeOperators::CreateFromStringId("stringappend");
- ImmutableCFOptions cf_options = ImmutableCFOptions(options);
- MutableCFOptions mutable_cf_options = MutableCFOptions(options);
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("a", "merge_1");
- internal_iter->AddMerge("a", "merge_2");
- internal_iter->AddMerge("a", "merge_3");
- internal_iter->AddPut("a", "put_1");
- internal_iter->AddMerge("a", "merge_4");
- internal_iter->AddMerge("a", "merge_5");
- internal_iter->AddMerge("a", "merge_6");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 0, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "merge_1");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("a", "merge_1");
- internal_iter->AddMerge("a", "merge_2");
- internal_iter->AddMerge("a", "merge_3");
- internal_iter->AddPut("a", "put_1");
- internal_iter->AddMerge("a", "merge_4");
- internal_iter->AddMerge("a", "merge_5");
- internal_iter->AddMerge("a", "merge_6");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 1, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "merge_1,merge_2");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("a", "merge_1");
- internal_iter->AddMerge("a", "merge_2");
- internal_iter->AddMerge("a", "merge_3");
- internal_iter->AddPut("a", "put_1");
- internal_iter->AddMerge("a", "merge_4");
- internal_iter->AddMerge("a", "merge_5");
- internal_iter->AddMerge("a", "merge_6");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 2, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "merge_1,merge_2,merge_3");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("a", "merge_1");
- internal_iter->AddMerge("a", "merge_2");
- internal_iter->AddMerge("a", "merge_3");
- internal_iter->AddPut("a", "put_1");
- internal_iter->AddMerge("a", "merge_4");
- internal_iter->AddMerge("a", "merge_5");
- internal_iter->AddMerge("a", "merge_6");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 3, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "put_1");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("a", "merge_1");
- internal_iter->AddMerge("a", "merge_2");
- internal_iter->AddMerge("a", "merge_3");
- internal_iter->AddPut("a", "put_1");
- internal_iter->AddMerge("a", "merge_4");
- internal_iter->AddMerge("a", "merge_5");
- internal_iter->AddMerge("a", "merge_6");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 4, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "put_1,merge_4");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("a", "merge_1");
- internal_iter->AddMerge("a", "merge_2");
- internal_iter->AddMerge("a", "merge_3");
- internal_iter->AddPut("a", "put_1");
- internal_iter->AddMerge("a", "merge_4");
- internal_iter->AddMerge("a", "merge_5");
- internal_iter->AddMerge("a", "merge_6");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 5, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "put_1,merge_4,merge_5");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("a", "merge_1");
- internal_iter->AddMerge("a", "merge_2");
- internal_iter->AddMerge("a", "merge_3");
- internal_iter->AddPut("a", "put_1");
- internal_iter->AddMerge("a", "merge_4");
- internal_iter->AddMerge("a", "merge_5");
- internal_iter->AddMerge("a", "merge_6");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 6, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "put_1,merge_4,merge_5,merge_6");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- {
- // put, singledelete, merge
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "val_a");
- internal_iter->AddSingleDeletion("a");
- internal_iter->AddMerge("a", "merge_1");
- internal_iter->AddMerge("a", "merge_2");
- internal_iter->AddPut("b", "val_b");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 10, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->Seek("b");
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- }
- }
- TEST_F(DBIteratorTest, DBIterator6) {
- ReadOptions ro;
- Options options;
- options.merge_operator = MergeOperators::CreateFromStringId("stringappend");
- ImmutableCFOptions cf_options = ImmutableCFOptions(options);
- MutableCFOptions mutable_cf_options = MutableCFOptions(options);
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("a", "merge_1");
- internal_iter->AddMerge("a", "merge_2");
- internal_iter->AddMerge("a", "merge_3");
- internal_iter->AddDeletion("a");
- internal_iter->AddMerge("a", "merge_4");
- internal_iter->AddMerge("a", "merge_5");
- internal_iter->AddMerge("a", "merge_6");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 0, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "merge_1");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("a", "merge_1");
- internal_iter->AddMerge("a", "merge_2");
- internal_iter->AddMerge("a", "merge_3");
- internal_iter->AddDeletion("a");
- internal_iter->AddMerge("a", "merge_4");
- internal_iter->AddMerge("a", "merge_5");
- internal_iter->AddMerge("a", "merge_6");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 1, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "merge_1,merge_2");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("a", "merge_1");
- internal_iter->AddMerge("a", "merge_2");
- internal_iter->AddMerge("a", "merge_3");
- internal_iter->AddDeletion("a");
- internal_iter->AddMerge("a", "merge_4");
- internal_iter->AddMerge("a", "merge_5");
- internal_iter->AddMerge("a", "merge_6");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 2, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "merge_1,merge_2,merge_3");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("a", "merge_1");
- internal_iter->AddMerge("a", "merge_2");
- internal_iter->AddMerge("a", "merge_3");
- internal_iter->AddDeletion("a");
- internal_iter->AddMerge("a", "merge_4");
- internal_iter->AddMerge("a", "merge_5");
- internal_iter->AddMerge("a", "merge_6");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 3, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(!db_iter->Valid());
- }
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("a", "merge_1");
- internal_iter->AddMerge("a", "merge_2");
- internal_iter->AddMerge("a", "merge_3");
- internal_iter->AddDeletion("a");
- internal_iter->AddMerge("a", "merge_4");
- internal_iter->AddMerge("a", "merge_5");
- internal_iter->AddMerge("a", "merge_6");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 4, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "merge_4");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("a", "merge_1");
- internal_iter->AddMerge("a", "merge_2");
- internal_iter->AddMerge("a", "merge_3");
- internal_iter->AddDeletion("a");
- internal_iter->AddMerge("a", "merge_4");
- internal_iter->AddMerge("a", "merge_5");
- internal_iter->AddMerge("a", "merge_6");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 5, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "merge_4,merge_5");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("a", "merge_1");
- internal_iter->AddMerge("a", "merge_2");
- internal_iter->AddMerge("a", "merge_3");
- internal_iter->AddDeletion("a");
- internal_iter->AddMerge("a", "merge_4");
- internal_iter->AddMerge("a", "merge_5");
- internal_iter->AddMerge("a", "merge_6");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 6, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "merge_4,merge_5,merge_6");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- }
- TEST_F(DBIteratorTest, DBIterator7) {
- ReadOptions ro;
- Options options;
- options.merge_operator = MergeOperators::CreateFromStringId("stringappend");
- ImmutableCFOptions cf_options = ImmutableCFOptions(options);
- MutableCFOptions mutable_cf_options = MutableCFOptions(options);
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("a", "merge_1");
- internal_iter->AddPut("b", "val");
- internal_iter->AddMerge("b", "merge_2");
- internal_iter->AddDeletion("b");
- internal_iter->AddMerge("b", "merge_3");
- internal_iter->AddMerge("c", "merge_4");
- internal_iter->AddMerge("c", "merge_5");
- internal_iter->AddDeletion("b");
- internal_iter->AddMerge("b", "merge_6");
- internal_iter->AddMerge("b", "merge_7");
- internal_iter->AddMerge("b", "merge_8");
- internal_iter->AddMerge("b", "merge_9");
- internal_iter->AddMerge("b", "merge_10");
- internal_iter->AddMerge("b", "merge_11");
- internal_iter->AddDeletion("c");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 0, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "merge_1");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("a", "merge_1");
- internal_iter->AddPut("b", "val");
- internal_iter->AddMerge("b", "merge_2");
- internal_iter->AddDeletion("b");
- internal_iter->AddMerge("b", "merge_3");
- internal_iter->AddMerge("c", "merge_4");
- internal_iter->AddMerge("c", "merge_5");
- internal_iter->AddDeletion("b");
- internal_iter->AddMerge("b", "merge_6");
- internal_iter->AddMerge("b", "merge_7");
- internal_iter->AddMerge("b", "merge_8");
- internal_iter->AddMerge("b", "merge_9");
- internal_iter->AddMerge("b", "merge_10");
- internal_iter->AddMerge("b", "merge_11");
- internal_iter->AddDeletion("c");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 2, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "val,merge_2");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "merge_1");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("a", "merge_1");
- internal_iter->AddPut("b", "val");
- internal_iter->AddMerge("b", "merge_2");
- internal_iter->AddDeletion("b");
- internal_iter->AddMerge("b", "merge_3");
- internal_iter->AddMerge("c", "merge_4");
- internal_iter->AddMerge("c", "merge_5");
- internal_iter->AddDeletion("b");
- internal_iter->AddMerge("b", "merge_6");
- internal_iter->AddMerge("b", "merge_7");
- internal_iter->AddMerge("b", "merge_8");
- internal_iter->AddMerge("b", "merge_9");
- internal_iter->AddMerge("b", "merge_10");
- internal_iter->AddMerge("b", "merge_11");
- internal_iter->AddDeletion("c");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 4, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "merge_3");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "merge_1");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("a", "merge_1");
- internal_iter->AddPut("b", "val");
- internal_iter->AddMerge("b", "merge_2");
- internal_iter->AddDeletion("b");
- internal_iter->AddMerge("b", "merge_3");
- internal_iter->AddMerge("c", "merge_4");
- internal_iter->AddMerge("c", "merge_5");
- internal_iter->AddDeletion("b");
- internal_iter->AddMerge("b", "merge_6");
- internal_iter->AddMerge("b", "merge_7");
- internal_iter->AddMerge("b", "merge_8");
- internal_iter->AddMerge("b", "merge_9");
- internal_iter->AddMerge("b", "merge_10");
- internal_iter->AddMerge("b", "merge_11");
- internal_iter->AddDeletion("c");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 5, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- ASSERT_EQ(db_iter->value().ToString(), "merge_4");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "merge_3");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "merge_1");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("a", "merge_1");
- internal_iter->AddPut("b", "val");
- internal_iter->AddMerge("b", "merge_2");
- internal_iter->AddDeletion("b");
- internal_iter->AddMerge("b", "merge_3");
- internal_iter->AddMerge("c", "merge_4");
- internal_iter->AddMerge("c", "merge_5");
- internal_iter->AddDeletion("b");
- internal_iter->AddMerge("b", "merge_6");
- internal_iter->AddMerge("b", "merge_7");
- internal_iter->AddMerge("b", "merge_8");
- internal_iter->AddMerge("b", "merge_9");
- internal_iter->AddMerge("b", "merge_10");
- internal_iter->AddMerge("b", "merge_11");
- internal_iter->AddDeletion("c");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 6, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- ASSERT_EQ(db_iter->value().ToString(), "merge_4,merge_5");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "merge_3");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "merge_1");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("a", "merge_1");
- internal_iter->AddPut("b", "val");
- internal_iter->AddMerge("b", "merge_2");
- internal_iter->AddDeletion("b");
- internal_iter->AddMerge("b", "merge_3");
- internal_iter->AddMerge("c", "merge_4");
- internal_iter->AddMerge("c", "merge_5");
- internal_iter->AddDeletion("b");
- internal_iter->AddMerge("b", "merge_6");
- internal_iter->AddMerge("b", "merge_7");
- internal_iter->AddMerge("b", "merge_8");
- internal_iter->AddMerge("b", "merge_9");
- internal_iter->AddMerge("b", "merge_10");
- internal_iter->AddMerge("b", "merge_11");
- internal_iter->AddDeletion("c");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 7, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- ASSERT_EQ(db_iter->value().ToString(), "merge_4,merge_5");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "merge_1");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("a", "merge_1");
- internal_iter->AddPut("b", "val");
- internal_iter->AddMerge("b", "merge_2");
- internal_iter->AddDeletion("b");
- internal_iter->AddMerge("b", "merge_3");
- internal_iter->AddMerge("c", "merge_4");
- internal_iter->AddMerge("c", "merge_5");
- internal_iter->AddDeletion("b");
- internal_iter->AddMerge("b", "merge_6");
- internal_iter->AddMerge("b", "merge_7");
- internal_iter->AddMerge("b", "merge_8");
- internal_iter->AddMerge("b", "merge_9");
- internal_iter->AddMerge("b", "merge_10");
- internal_iter->AddMerge("b", "merge_11");
- internal_iter->AddDeletion("c");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 9, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- ASSERT_EQ(db_iter->value().ToString(), "merge_4,merge_5");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "merge_6,merge_7");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "merge_1");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("a", "merge_1");
- internal_iter->AddPut("b", "val");
- internal_iter->AddMerge("b", "merge_2");
- internal_iter->AddDeletion("b");
- internal_iter->AddMerge("b", "merge_3");
- internal_iter->AddMerge("c", "merge_4");
- internal_iter->AddMerge("c", "merge_5");
- internal_iter->AddDeletion("b");
- internal_iter->AddMerge("b", "merge_6");
- internal_iter->AddMerge("b", "merge_7");
- internal_iter->AddMerge("b", "merge_8");
- internal_iter->AddMerge("b", "merge_9");
- internal_iter->AddMerge("b", "merge_10");
- internal_iter->AddMerge("b", "merge_11");
- internal_iter->AddDeletion("c");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 13, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- ASSERT_EQ(db_iter->value().ToString(), "merge_4,merge_5");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(),
- "merge_6,merge_7,merge_8,merge_9,merge_10,merge_11");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "merge_1");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("a", "merge_1");
- internal_iter->AddPut("b", "val");
- internal_iter->AddMerge("b", "merge_2");
- internal_iter->AddDeletion("b");
- internal_iter->AddMerge("b", "merge_3");
- internal_iter->AddMerge("c", "merge_4");
- internal_iter->AddMerge("c", "merge_5");
- internal_iter->AddDeletion("b");
- internal_iter->AddMerge("b", "merge_6");
- internal_iter->AddMerge("b", "merge_7");
- internal_iter->AddMerge("b", "merge_8");
- internal_iter->AddMerge("b", "merge_9");
- internal_iter->AddMerge("b", "merge_10");
- internal_iter->AddMerge("b", "merge_11");
- internal_iter->AddDeletion("c");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, cf_options, mutable_cf_options, BytewiseComparator(),
- internal_iter, 14, options.max_sequential_skip_in_iterations,
- nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(),
- "merge_6,merge_7,merge_8,merge_9,merge_10,merge_11");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "merge_1");
- db_iter->Prev();
- ASSERT_TRUE(!db_iter->Valid());
- }
- }
- TEST_F(DBIteratorTest, DBIterator8) {
- ReadOptions ro;
- Options options;
- options.merge_operator = MergeOperators::CreateFromStringId("stringappend");
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddDeletion("a");
- internal_iter->AddPut("a", "0");
- internal_iter->AddPut("b", "0");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, ImmutableCFOptions(options), MutableCFOptions(options),
- BytewiseComparator(), internal_iter, 10,
- options.max_sequential_skip_in_iterations, nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "0");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "0");
- }
- // TODO(3.13): fix the issue of Seek() then Prev() which might not necessary
- // return the biggest element smaller than the seek key.
- TEST_F(DBIteratorTest, DBIterator9) {
- ReadOptions ro;
- Options options;
- options.merge_operator = MergeOperators::CreateFromStringId("stringappend");
- {
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddMerge("a", "merge_1");
- internal_iter->AddMerge("a", "merge_2");
- internal_iter->AddMerge("b", "merge_3");
- internal_iter->AddMerge("b", "merge_4");
- internal_iter->AddMerge("d", "merge_5");
- internal_iter->AddMerge("d", "merge_6");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, ImmutableCFOptions(options), MutableCFOptions(options),
- BytewiseComparator(), internal_iter, 10,
- options.max_sequential_skip_in_iterations, nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "merge_3,merge_4");
- db_iter->Next();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "d");
- ASSERT_EQ(db_iter->value().ToString(), "merge_5,merge_6");
- db_iter->Seek("b");
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "merge_3,merge_4");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "merge_1,merge_2");
- db_iter->SeekForPrev("b");
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "merge_3,merge_4");
- db_iter->Next();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "d");
- ASSERT_EQ(db_iter->value().ToString(), "merge_5,merge_6");
- db_iter->Seek("c");
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "d");
- ASSERT_EQ(db_iter->value().ToString(), "merge_5,merge_6");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "merge_3,merge_4");
- db_iter->SeekForPrev("c");
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "merge_3,merge_4");
- db_iter->Next();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "d");
- ASSERT_EQ(db_iter->value().ToString(), "merge_5,merge_6");
- }
- }
- // TODO(3.13): fix the issue of Seek() then Prev() which might not necessary
- // return the biggest element smaller than the seek key.
- TEST_F(DBIteratorTest, DBIterator10) {
- ReadOptions ro;
- Options options;
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "1");
- internal_iter->AddPut("b", "2");
- internal_iter->AddPut("c", "3");
- internal_iter->AddPut("d", "4");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, ImmutableCFOptions(options), MutableCFOptions(options),
- BytewiseComparator(), internal_iter, 10,
- options.max_sequential_skip_in_iterations, nullptr /*read_callback*/));
- db_iter->Seek("c");
- ASSERT_TRUE(db_iter->Valid());
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "2");
- db_iter->Next();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- ASSERT_EQ(db_iter->value().ToString(), "3");
- db_iter->SeekForPrev("c");
- ASSERT_TRUE(db_iter->Valid());
- db_iter->Next();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "d");
- ASSERT_EQ(db_iter->value().ToString(), "4");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- ASSERT_EQ(db_iter->value().ToString(), "3");
- }
- TEST_F(DBIteratorTest, SeekToLastOccurrenceSeq0) {
- ReadOptions ro;
- Options options;
- options.merge_operator = nullptr;
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "1");
- internal_iter->AddPut("b", "2");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, ImmutableCFOptions(options), MutableCFOptions(options),
- BytewiseComparator(), internal_iter, 10, 0 /* force seek */,
- nullptr /*read_callback*/));
- db_iter->SeekToFirst();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "1");
- db_iter->Next();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "2");
- db_iter->Next();
- ASSERT_FALSE(db_iter->Valid());
- }
- TEST_F(DBIteratorTest, DBIterator11) {
- ReadOptions ro;
- Options options;
- options.merge_operator = MergeOperators::CreateFromStringId("stringappend");
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "0");
- internal_iter->AddPut("b", "0");
- internal_iter->AddSingleDeletion("b");
- internal_iter->AddMerge("a", "1");
- internal_iter->AddMerge("b", "2");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, ImmutableCFOptions(options), MutableCFOptions(options),
- BytewiseComparator(), internal_iter, 1,
- options.max_sequential_skip_in_iterations, nullptr /*read_callback*/));
- db_iter->SeekToFirst();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "0");
- db_iter->Next();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- db_iter->Next();
- ASSERT_FALSE(db_iter->Valid());
- }
- TEST_F(DBIteratorTest, DBIterator12) {
- ReadOptions ro;
- Options options;
- options.merge_operator = nullptr;
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "1");
- internal_iter->AddPut("b", "2");
- internal_iter->AddPut("c", "3");
- internal_iter->AddSingleDeletion("b");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, ImmutableCFOptions(options), MutableCFOptions(options),
- BytewiseComparator(), internal_iter, 10, 0, nullptr /*read_callback*/));
- db_iter->SeekToLast();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "c");
- ASSERT_EQ(db_iter->value().ToString(), "3");
- db_iter->Prev();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "1");
- db_iter->Prev();
- ASSERT_FALSE(db_iter->Valid());
- }
- TEST_F(DBIteratorTest, DBIterator13) {
- ReadOptions ro;
- Options options;
- options.merge_operator = nullptr;
- std::string key;
- key.resize(9);
- key.assign(9, static_cast<char>(0));
- key[0] = 'b';
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut(key, "0");
- internal_iter->AddPut(key, "1");
- internal_iter->AddPut(key, "2");
- internal_iter->AddPut(key, "3");
- internal_iter->AddPut(key, "4");
- internal_iter->AddPut(key, "5");
- internal_iter->AddPut(key, "6");
- internal_iter->AddPut(key, "7");
- internal_iter->AddPut(key, "8");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, ImmutableCFOptions(options), MutableCFOptions(options),
- BytewiseComparator(), internal_iter, 2, 3, nullptr /*read_callback*/));
- db_iter->Seek("b");
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), key);
- ASSERT_EQ(db_iter->value().ToString(), "2");
- }
- TEST_F(DBIteratorTest, DBIterator14) {
- ReadOptions ro;
- Options options;
- options.merge_operator = nullptr;
- std::string key("b");
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("b", "0");
- internal_iter->AddPut("b", "1");
- internal_iter->AddPut("b", "2");
- internal_iter->AddPut("b", "3");
- internal_iter->AddPut("a", "4");
- internal_iter->AddPut("a", "5");
- internal_iter->AddPut("a", "6");
- internal_iter->AddPut("c", "7");
- internal_iter->AddPut("c", "8");
- internal_iter->AddPut("c", "9");
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, ImmutableCFOptions(options), MutableCFOptions(options),
- BytewiseComparator(), internal_iter, 4, 1, nullptr /*read_callback*/));
- db_iter->Seek("b");
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(db_iter->key().ToString(), "b");
- ASSERT_EQ(db_iter->value().ToString(), "3");
- db_iter->SeekToFirst();
- ASSERT_EQ(db_iter->key().ToString(), "a");
- ASSERT_EQ(db_iter->value().ToString(), "4");
- }
- TEST_F(DBIteratorTest, DBIteratorTestDifferentialSnapshots) {
- { // test that KVs earlier that iter_start_seqnum are filtered out
- ReadOptions ro;
- ro.iter_start_seqnum=5;
- Options options;
- options.statistics = ROCKSDB_NAMESPACE::CreateDBStatistics();
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- for (size_t i = 0; i < 10; ++i) {
- internal_iter->AddPut(std::to_string(i), std::to_string(i) + "a");
- internal_iter->AddPut(std::to_string(i), std::to_string(i) + "b");
- internal_iter->AddPut(std::to_string(i), std::to_string(i) + "c");
- }
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, ImmutableCFOptions(options), MutableCFOptions(options),
- BytewiseComparator(), internal_iter, 13,
- options.max_sequential_skip_in_iterations, nullptr));
- // Expecting InternalKeys in [5,8] range with correct type
- int seqnums[4] = {5,8,11,13};
- std::string user_keys[4] = {"1","2","3","4"};
- std::string values[4] = {"1c", "2c", "3c", "4b"};
- int i = 0;
- for (db_iter->SeekToFirst(); db_iter->Valid(); db_iter->Next()) {
- FullKey fkey;
- ParseFullKey(db_iter->key(), &fkey);
- ASSERT_EQ(user_keys[i], fkey.user_key.ToString());
- ASSERT_EQ(EntryType::kEntryPut, fkey.type);
- ASSERT_EQ(seqnums[i], fkey.sequence);
- ASSERT_EQ(values[i], db_iter->value().ToString());
- i++;
- }
- ASSERT_EQ(i, 4);
- }
- { // Test that deletes are returned correctly as internal KVs
- ReadOptions ro;
- ro.iter_start_seqnum=5;
- Options options;
- options.statistics = ROCKSDB_NAMESPACE::CreateDBStatistics();
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- for (size_t i = 0; i < 10; ++i) {
- internal_iter->AddPut(std::to_string(i), std::to_string(i) + "a");
- internal_iter->AddPut(std::to_string(i), std::to_string(i) + "b");
- internal_iter->AddDeletion(std::to_string(i));
- }
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, ImmutableCFOptions(options), MutableCFOptions(options),
- BytewiseComparator(), internal_iter, 13,
- options.max_sequential_skip_in_iterations, nullptr));
- // Expecting InternalKeys in [5,8] range with correct type
- int seqnums[4] = {5,8,11,13};
- EntryType key_types[4] = {EntryType::kEntryDelete,EntryType::kEntryDelete,
- EntryType::kEntryDelete,EntryType::kEntryPut};
- std::string user_keys[4] = {"1","2","3","4"};
- std::string values[4] = {"", "", "", "4b"};
- int i = 0;
- for (db_iter->SeekToFirst(); db_iter->Valid(); db_iter->Next()) {
- FullKey fkey;
- ParseFullKey(db_iter->key(), &fkey);
- ASSERT_EQ(user_keys[i], fkey.user_key.ToString());
- ASSERT_EQ(key_types[i], fkey.type);
- ASSERT_EQ(seqnums[i], fkey.sequence);
- ASSERT_EQ(values[i], db_iter->value().ToString());
- i++;
- }
- ASSERT_EQ(i, 4);
- }
- }
- class DBIterWithMergeIterTest : public testing::Test {
- public:
- DBIterWithMergeIterTest()
- : env_(Env::Default()), icomp_(BytewiseComparator()) {
- options_.merge_operator = nullptr;
- internal_iter1_ = new TestIterator(BytewiseComparator());
- internal_iter1_->Add("a", kTypeValue, "1", 3u);
- internal_iter1_->Add("f", kTypeValue, "2", 5u);
- internal_iter1_->Add("g", kTypeValue, "3", 7u);
- internal_iter1_->Finish();
- internal_iter2_ = new TestIterator(BytewiseComparator());
- internal_iter2_->Add("a", kTypeValue, "4", 6u);
- internal_iter2_->Add("b", kTypeValue, "5", 1u);
- internal_iter2_->Add("c", kTypeValue, "6", 2u);
- internal_iter2_->Add("d", kTypeValue, "7", 3u);
- internal_iter2_->Finish();
- std::vector<InternalIterator*> child_iters;
- child_iters.push_back(internal_iter1_);
- child_iters.push_back(internal_iter2_);
- InternalKeyComparator icomp(BytewiseComparator());
- merge_iter =
- NewMergingIterator(&icomp_, &child_iters[0], 2u);
- db_iter_.reset(NewDBIterator(
- env_, ro_, ImmutableCFOptions(options_), MutableCFOptions(options_),
- BytewiseComparator(), merge_iter,
- 8 /* read data earlier than seqId 8 */,
- 3 /* max iterators before reseek */, nullptr /*read_callback*/));
- }
- Env* env_;
- ReadOptions ro_;
- Options options_;
- TestIterator* internal_iter1_;
- TestIterator* internal_iter2_;
- InternalKeyComparator icomp_;
- Iterator* merge_iter_;
- InternalIterator* merge_iter;
- std::unique_ptr<Iterator> db_iter_;
- };
- #include <iostream>
- TEST_F(DBIterWithMergeIterTest, InnerMergeIterator1) {
- merge_iter->SeekToFirst();
- std::cout<<" merge_iter key value :" << std::endl;
- while(merge_iter->Valid()) {
- std::cout<< merge_iter->key().ToString() << " " << merge_iter->value().ToString() <<std::endl;
- merge_iter->Next();
- }
-
- db_iter_->SeekToFirst();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "a");
- ASSERT_EQ(db_iter_->value().ToString(), "4");
- db_iter_->Next();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "b");
- ASSERT_EQ(db_iter_->value().ToString(), "5");
- db_iter_->Next();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "c");
- ASSERT_EQ(db_iter_->value().ToString(), "6");
- db_iter_->Next();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "d");
- ASSERT_EQ(db_iter_->value().ToString(), "7");
- db_iter_->Next();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "f");
- ASSERT_EQ(db_iter_->value().ToString(), "2");
- db_iter_->Next();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "g");
- ASSERT_EQ(db_iter_->value().ToString(), "3");
- db_iter_->Next();
- ASSERT_FALSE(db_iter_->Valid());
- }
- TEST_F(DBIterWithMergeIterTest, InnerMergeIterator2) {
- // Test Prev() when one child iterator is at its end.
- db_iter_->SeekForPrev("g");
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "g");
- ASSERT_EQ(db_iter_->value().ToString(), "3");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "f");
- ASSERT_EQ(db_iter_->value().ToString(), "2");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "d");
- ASSERT_EQ(db_iter_->value().ToString(), "7");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "c");
- ASSERT_EQ(db_iter_->value().ToString(), "6");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "b");
- ASSERT_EQ(db_iter_->value().ToString(), "5");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "a");
- ASSERT_EQ(db_iter_->value().ToString(), "4");
- }
- TEST_F(DBIterWithMergeIterTest, InnerMergeIteratorDataRace1) {
- // Test Prev() when one child iterator is at its end but more rows
- // are added.
- db_iter_->Seek("f");
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "f");
- ASSERT_EQ(db_iter_->value().ToString(), "2");
- // Test call back inserts a key in the end of the mem table after
- // MergeIterator::Prev() realized the mem table iterator is at its end
- // and before an SeekToLast() is called.
- ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->SetCallBack(
- "MergeIterator::Prev:BeforePrev",
- [&](void* /*arg*/) { internal_iter2_->Add("z", kTypeValue, "7", 12u); });
- ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->EnableProcessing();
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "d");
- ASSERT_EQ(db_iter_->value().ToString(), "7");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "c");
- ASSERT_EQ(db_iter_->value().ToString(), "6");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "b");
- ASSERT_EQ(db_iter_->value().ToString(), "5");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "a");
- ASSERT_EQ(db_iter_->value().ToString(), "4");
- ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->DisableProcessing();
- }
- TEST_F(DBIterWithMergeIterTest, InnerMergeIteratorDataRace2) {
- // Test Prev() when one child iterator is at its end but more rows
- // are added.
- db_iter_->Seek("f");
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "f");
- ASSERT_EQ(db_iter_->value().ToString(), "2");
- // Test call back inserts entries for update a key in the end of the
- // mem table after MergeIterator::Prev() realized the mem tableiterator is at
- // its end and before an SeekToLast() is called.
- ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->SetCallBack(
- "MergeIterator::Prev:BeforePrev", [&](void* /*arg*/) {
- internal_iter2_->Add("z", kTypeValue, "7", 12u);
- internal_iter2_->Add("z", kTypeValue, "7", 11u);
- });
- ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->EnableProcessing();
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "d");
- ASSERT_EQ(db_iter_->value().ToString(), "7");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "c");
- ASSERT_EQ(db_iter_->value().ToString(), "6");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "b");
- ASSERT_EQ(db_iter_->value().ToString(), "5");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "a");
- ASSERT_EQ(db_iter_->value().ToString(), "4");
- ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->DisableProcessing();
- }
- TEST_F(DBIterWithMergeIterTest, InnerMergeIteratorDataRace3) {
- // Test Prev() when one child iterator is at its end but more rows
- // are added and max_skipped is triggered.
- db_iter_->Seek("f");
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "f");
- ASSERT_EQ(db_iter_->value().ToString(), "2");
- // Test call back inserts entries for update a key in the end of the
- // mem table after MergeIterator::Prev() realized the mem table iterator is at
- // its end and before an SeekToLast() is called.
- ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->SetCallBack(
- "MergeIterator::Prev:BeforePrev", [&](void* /*arg*/) {
- internal_iter2_->Add("z", kTypeValue, "7", 16u, true);
- internal_iter2_->Add("z", kTypeValue, "7", 15u, true);
- internal_iter2_->Add("z", kTypeValue, "7", 14u, true);
- internal_iter2_->Add("z", kTypeValue, "7", 13u, true);
- internal_iter2_->Add("z", kTypeValue, "7", 12u, true);
- internal_iter2_->Add("z", kTypeValue, "7", 11u, true);
- });
- ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->EnableProcessing();
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "d");
- ASSERT_EQ(db_iter_->value().ToString(), "7");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "c");
- ASSERT_EQ(db_iter_->value().ToString(), "6");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "b");
- ASSERT_EQ(db_iter_->value().ToString(), "5");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "a");
- ASSERT_EQ(db_iter_->value().ToString(), "4");
- ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->DisableProcessing();
- }
- TEST_F(DBIterWithMergeIterTest, InnerMergeIteratorDataRace4) {
- // Test Prev() when one child iterator has more rows inserted
- // between Seek() and Prev() when changing directions.
- internal_iter2_->Add("z", kTypeValue, "9", 4u);
- db_iter_->Seek("g");
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "g");
- ASSERT_EQ(db_iter_->value().ToString(), "3");
- // Test call back inserts entries for update a key before "z" in
- // mem table after MergeIterator::Prev() calls mem table iterator's
- // Seek() and before calling Prev()
- ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->SetCallBack(
- "MergeIterator::Prev:BeforePrev", [&](void* arg) {
- IteratorWrapper* it = reinterpret_cast<IteratorWrapper*>(arg);
- if (it->key().starts_with("z")) {
- internal_iter2_->Add("x", kTypeValue, "7", 16u, true);
- internal_iter2_->Add("x", kTypeValue, "7", 15u, true);
- internal_iter2_->Add("x", kTypeValue, "7", 14u, true);
- internal_iter2_->Add("x", kTypeValue, "7", 13u, true);
- internal_iter2_->Add("x", kTypeValue, "7", 12u, true);
- internal_iter2_->Add("x", kTypeValue, "7", 11u, true);
- }
- });
- ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->EnableProcessing();
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "f");
- ASSERT_EQ(db_iter_->value().ToString(), "2");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "d");
- ASSERT_EQ(db_iter_->value().ToString(), "7");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "c");
- ASSERT_EQ(db_iter_->value().ToString(), "6");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "b");
- ASSERT_EQ(db_iter_->value().ToString(), "5");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "a");
- ASSERT_EQ(db_iter_->value().ToString(), "4");
- ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->DisableProcessing();
- }
- TEST_F(DBIterWithMergeIterTest, InnerMergeIteratorDataRace5) {
- internal_iter2_->Add("z", kTypeValue, "9", 4u);
- // Test Prev() when one child iterator has more rows inserted
- // between Seek() and Prev() when changing directions.
- db_iter_->Seek("g");
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "g");
- ASSERT_EQ(db_iter_->value().ToString(), "3");
- // Test call back inserts entries for update a key before "z" in
- // mem table after MergeIterator::Prev() calls mem table iterator's
- // Seek() and before calling Prev()
- ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->SetCallBack(
- "MergeIterator::Prev:BeforePrev", [&](void* arg) {
- IteratorWrapper* it = reinterpret_cast<IteratorWrapper*>(arg);
- if (it->key().starts_with("z")) {
- internal_iter2_->Add("x", kTypeValue, "7", 16u, true);
- internal_iter2_->Add("x", kTypeValue, "7", 15u, true);
- }
- });
- ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->EnableProcessing();
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "f");
- ASSERT_EQ(db_iter_->value().ToString(), "2");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "d");
- ASSERT_EQ(db_iter_->value().ToString(), "7");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "c");
- ASSERT_EQ(db_iter_->value().ToString(), "6");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "b");
- ASSERT_EQ(db_iter_->value().ToString(), "5");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "a");
- ASSERT_EQ(db_iter_->value().ToString(), "4");
- ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->DisableProcessing();
- }
- TEST_F(DBIterWithMergeIterTest, InnerMergeIteratorDataRace6) {
- internal_iter2_->Add("z", kTypeValue, "9", 4u);
- // Test Prev() when one child iterator has more rows inserted
- // between Seek() and Prev() when changing directions.
- db_iter_->Seek("g");
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "g");
- ASSERT_EQ(db_iter_->value().ToString(), "3");
- // Test call back inserts an entry for update a key before "z" in
- // mem table after MergeIterator::Prev() calls mem table iterator's
- // Seek() and before calling Prev()
- ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->SetCallBack(
- "MergeIterator::Prev:BeforePrev", [&](void* arg) {
- IteratorWrapper* it = reinterpret_cast<IteratorWrapper*>(arg);
- if (it->key().starts_with("z")) {
- internal_iter2_->Add("x", kTypeValue, "7", 16u, true);
- }
- });
- ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->EnableProcessing();
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "f");
- ASSERT_EQ(db_iter_->value().ToString(), "2");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "d");
- ASSERT_EQ(db_iter_->value().ToString(), "7");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "c");
- ASSERT_EQ(db_iter_->value().ToString(), "6");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "b");
- ASSERT_EQ(db_iter_->value().ToString(), "5");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "a");
- ASSERT_EQ(db_iter_->value().ToString(), "4");
- ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->DisableProcessing();
- }
- TEST_F(DBIterWithMergeIterTest, InnerMergeIteratorDataRace7) {
- internal_iter1_->Add("u", kTypeValue, "10", 4u);
- internal_iter1_->Add("v", kTypeValue, "11", 4u);
- internal_iter1_->Add("w", kTypeValue, "12", 4u);
- internal_iter2_->Add("z", kTypeValue, "9", 4u);
- // Test Prev() when one child iterator has more rows inserted
- // between Seek() and Prev() when changing directions.
- db_iter_->Seek("g");
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "g");
- ASSERT_EQ(db_iter_->value().ToString(), "3");
- // Test call back inserts entries for update a key before "z" in
- // mem table after MergeIterator::Prev() calls mem table iterator's
- // Seek() and before calling Prev()
- ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->SetCallBack(
- "MergeIterator::Prev:BeforePrev", [&](void* arg) {
- IteratorWrapper* it = reinterpret_cast<IteratorWrapper*>(arg);
- if (it->key().starts_with("z")) {
- internal_iter2_->Add("x", kTypeValue, "7", 16u, true);
- internal_iter2_->Add("x", kTypeValue, "7", 15u, true);
- internal_iter2_->Add("x", kTypeValue, "7", 14u, true);
- internal_iter2_->Add("x", kTypeValue, "7", 13u, true);
- internal_iter2_->Add("x", kTypeValue, "7", 12u, true);
- internal_iter2_->Add("x", kTypeValue, "7", 11u, true);
- }
- });
- ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->EnableProcessing();
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "f");
- ASSERT_EQ(db_iter_->value().ToString(), "2");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "d");
- ASSERT_EQ(db_iter_->value().ToString(), "7");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "c");
- ASSERT_EQ(db_iter_->value().ToString(), "6");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "b");
- ASSERT_EQ(db_iter_->value().ToString(), "5");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "a");
- ASSERT_EQ(db_iter_->value().ToString(), "4");
- ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->DisableProcessing();
- }
- TEST_F(DBIterWithMergeIterTest, InnerMergeIteratorDataRace8) {
- // internal_iter1_: a, f, g
- // internal_iter2_: a, b, c, d, adding (z)
- internal_iter2_->Add("z", kTypeValue, "9", 4u);
- // Test Prev() when one child iterator has more rows inserted
- // between Seek() and Prev() when changing directions.
- db_iter_->Seek("g");
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "g");
- ASSERT_EQ(db_iter_->value().ToString(), "3");
- // Test call back inserts two keys before "z" in mem table after
- // MergeIterator::Prev() calls mem table iterator's Seek() and
- // before calling Prev()
- ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->SetCallBack(
- "MergeIterator::Prev:BeforePrev", [&](void* arg) {
- IteratorWrapper* it = reinterpret_cast<IteratorWrapper*>(arg);
- if (it->key().starts_with("z")) {
- internal_iter2_->Add("x", kTypeValue, "7", 16u, true);
- internal_iter2_->Add("y", kTypeValue, "7", 17u, true);
- }
- });
- ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->EnableProcessing();
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "f");
- ASSERT_EQ(db_iter_->value().ToString(), "2");
- db_iter_->Prev();
- ASSERT_TRUE(db_iter_->Valid());
- ASSERT_EQ(db_iter_->key().ToString(), "d");
- ASSERT_EQ(db_iter_->value().ToString(), "7");
- ROCKSDB_NAMESPACE::SyncPoint::GetInstance()->DisableProcessing();
- }
- TEST_F(DBIteratorTest, SeekPrefixTombstones) {
- ReadOptions ro;
- Options options;
- options.prefix_extractor.reset(NewNoopTransform());
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddDeletion("b");
- internal_iter->AddDeletion("c");
- internal_iter->AddDeletion("d");
- internal_iter->AddDeletion("e");
- internal_iter->AddDeletion("f");
- internal_iter->AddDeletion("g");
- internal_iter->Finish();
- ro.prefix_same_as_start = true;
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, ImmutableCFOptions(options), MutableCFOptions(options),
- BytewiseComparator(), internal_iter, 10,
- options.max_sequential_skip_in_iterations, nullptr /*read_callback*/));
- int skipped_keys = 0;
- get_perf_context()->Reset();
- db_iter->SeekForPrev("z");
- skipped_keys =
- static_cast<int>(get_perf_context()->internal_key_skipped_count);
- ASSERT_EQ(skipped_keys, 0);
- get_perf_context()->Reset();
- db_iter->Seek("a");
- skipped_keys =
- static_cast<int>(get_perf_context()->internal_key_skipped_count);
- ASSERT_EQ(skipped_keys, 0);
- }
- TEST_F(DBIteratorTest, SeekToFirstLowerBound) {
- const int kNumKeys = 3;
- for (int i = 0; i < kNumKeys + 2; ++i) {
- // + 2 for two special cases: lower bound before and lower bound after the
- // internal iterator's keys
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- for (int j = 1; j <= kNumKeys; ++j) {
- internal_iter->AddPut(std::to_string(j), "val");
- }
- internal_iter->Finish();
- ReadOptions ro;
- auto lower_bound_str = std::to_string(i);
- Slice lower_bound(lower_bound_str);
- ro.iterate_lower_bound = &lower_bound;
- Options options;
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, ImmutableCFOptions(options), MutableCFOptions(options),
- BytewiseComparator(), internal_iter, 10 /* sequence */,
- options.max_sequential_skip_in_iterations,
- nullptr /* read_callback */));
- db_iter->SeekToFirst();
- if (i == kNumKeys + 1) {
- // lower bound was beyond the last key
- ASSERT_FALSE(db_iter->Valid());
- } else {
- ASSERT_TRUE(db_iter->Valid());
- int expected;
- if (i == 0) {
- // lower bound was before the first key
- expected = 1;
- } else {
- // lower bound was at the ith key
- expected = i;
- }
- ASSERT_EQ(std::to_string(expected), db_iter->key().ToString());
- }
- }
- }
- TEST_F(DBIteratorTest, PrevLowerBound) {
- const int kNumKeys = 3;
- const int kLowerBound = 2;
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- for (int j = 1; j <= kNumKeys; ++j) {
- internal_iter->AddPut(std::to_string(j), "val");
- }
- internal_iter->Finish();
- ReadOptions ro;
- auto lower_bound_str = std::to_string(kLowerBound);
- Slice lower_bound(lower_bound_str);
- ro.iterate_lower_bound = &lower_bound;
- Options options;
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, ImmutableCFOptions(options), MutableCFOptions(options),
- BytewiseComparator(), internal_iter, 10 /* sequence */,
- options.max_sequential_skip_in_iterations, nullptr /* read_callback */));
- db_iter->SeekToLast();
- for (int i = kNumKeys; i >= kLowerBound; --i) {
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(std::to_string(i), db_iter->key().ToString());
- db_iter->Prev();
- }
- ASSERT_FALSE(db_iter->Valid());
- }
- TEST_F(DBIteratorTest, SeekLessLowerBound) {
- const int kNumKeys = 3;
- const int kLowerBound = 2;
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- for (int j = 1; j <= kNumKeys; ++j) {
- internal_iter->AddPut(std::to_string(j), "val");
- }
- internal_iter->Finish();
- ReadOptions ro;
- auto lower_bound_str = std::to_string(kLowerBound);
- Slice lower_bound(lower_bound_str);
- ro.iterate_lower_bound = &lower_bound;
- Options options;
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ro, ImmutableCFOptions(options), MutableCFOptions(options),
- BytewiseComparator(), internal_iter, 10 /* sequence */,
- options.max_sequential_skip_in_iterations, nullptr /* read_callback */));
- auto before_lower_bound_str = std::to_string(kLowerBound - 1);
- Slice before_lower_bound(lower_bound_str);
- db_iter->Seek(before_lower_bound);
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_EQ(lower_bound_str, db_iter->key().ToString());
- }
- TEST_F(DBIteratorTest, ReverseToForwardWithDisappearingKeys) {
- Options options;
- options.prefix_extractor.reset(NewCappedPrefixTransform(0));
- TestIterator* internal_iter = new TestIterator(BytewiseComparator());
- internal_iter->AddPut("a", "A");
- internal_iter->AddPut("b", "B");
- for (int i = 0; i < 100; ++i) {
- internal_iter->AddPut("c" + ToString(i), "");
- }
- internal_iter->Finish();
- std::unique_ptr<Iterator> db_iter(NewDBIterator(
- env_, ReadOptions(), ImmutableCFOptions(options),
- MutableCFOptions(options), BytewiseComparator(), internal_iter, 10,
- options.max_sequential_skip_in_iterations, nullptr /*read_callback*/));
- db_iter->SeekForPrev("a");
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_OK(db_iter->status());
- ASSERT_EQ("a", db_iter->key().ToString());
- internal_iter->Vanish("a");
- db_iter->Next();
- ASSERT_TRUE(db_iter->Valid());
- ASSERT_OK(db_iter->status());
- ASSERT_EQ("b", db_iter->key().ToString());
- // A (sort of) bug used to cause DBIter to pointlessly drag the internal
- // iterator all the way to the end. But this doesn't really matter at the time
- // of writing because the only iterator that can see disappearing keys is
- // ForwardIterator, which doesn't support SeekForPrev().
- EXPECT_LT(internal_iter->steps(), 20);
- }
- } // namespace ROCKSDB_NAMESPACE
- int main(int argc, char** argv) {
- ::testing::InitGoogleTest(&argc, argv);
- return RUN_ALL_TESTS();
- }
|