1433 lines
51 KiB
C++
1433 lines
51 KiB
C++
/**
|
|
* Copyright (c) 2021 OceanBase
|
|
* OceanBase CE is licensed under Mulan PubL v2.
|
|
* You can use this software according to the terms and conditions of the Mulan PubL v2.
|
|
* You may obtain a copy of Mulan PubL v2 at:
|
|
* http://license.coscl.org.cn/MulanPubL-2.0
|
|
* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
|
|
* EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
|
|
* MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
|
|
* See the Mulan PubL v2 for more details.
|
|
*/
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#define private public
|
|
#define protected public
|
|
#include "src/storage/tx/ob_trans_ctx.h"
|
|
|
|
namespace oceanbase
|
|
{
|
|
|
|
namespace unittest
|
|
{
|
|
using namespace oceanbase::common;
|
|
using namespace oceanbase::memtable;
|
|
|
|
class ObMockTxCallback : public ObITransCallback
|
|
{
|
|
public:
|
|
ObMockTxCallback(ObMemtable *mt,
|
|
bool need_submit_log = true,
|
|
share::SCN scn = share::SCN::max_scn(),
|
|
transaction::ObTxSEQ seq_no = transaction::ObTxSEQ::MAX_VAL())
|
|
: ObITransCallback(need_submit_log),
|
|
mt_(mt), seq_no_(seq_no) { scn_ = scn; }
|
|
|
|
virtual ObIMemtable* get_memtable() const override { return mt_; }
|
|
virtual transaction::ObTxSEQ get_seq_no() const override { return seq_no_; }
|
|
virtual int checkpoint_callback() override;
|
|
virtual int rollback_callback() override;
|
|
virtual int calc_checksum(const share::SCN checksum_scn,
|
|
TxChecksum *checksumer) override;
|
|
|
|
ObMemtable *mt_;
|
|
transaction::ObTxSEQ seq_no_;
|
|
};
|
|
|
|
class ObMockBitSet {
|
|
public:
|
|
ObMockBitSet() { reset(); }
|
|
void reset()
|
|
{
|
|
memset(set_, 0, sizeof(char) * 100000);
|
|
}
|
|
|
|
void add_bit(int64_t bit)
|
|
{
|
|
if (bit >= 100000) {
|
|
ob_abort();
|
|
} else if (1 == set_[bit]) {
|
|
TRANS_LOG_RET(ERROR, OB_ERR_UNEXPECTED, "set in unexpected", K(bit));
|
|
ob_abort();
|
|
} else {
|
|
set_[bit] = 1;
|
|
TRANS_LOG(INFO, "add bit", K(bit), K(lbt()));
|
|
}
|
|
}
|
|
|
|
bool equal(ObMockBitSet &other)
|
|
{
|
|
bool bret = true;
|
|
for (int i = 0; i < 100000; i++) {
|
|
if (set_[i] != other.set_[i]) {
|
|
if (set_[i] == 1) {
|
|
TRANS_LOG_RET(ERROR, OB_ERR_UNEXPECTED, "different bit set", K(i),
|
|
"set_[i]", "1",
|
|
"other.set_[i]", "0");
|
|
} else {
|
|
TRANS_LOG_RET(ERROR, OB_ERR_UNEXPECTED, "different bit set", K(i),
|
|
"set_[i]", "0",
|
|
"other.set_[i]", "1");
|
|
}
|
|
bret = false;
|
|
}
|
|
}
|
|
|
|
return bret;
|
|
}
|
|
|
|
char set_[100000];
|
|
};
|
|
|
|
class TestTxCallbackList : public ::testing::Test
|
|
{
|
|
public:
|
|
TestTxCallbackList()
|
|
: seq_counter_(100, 0),
|
|
mt_counter_(0),
|
|
mt_ctx_(),
|
|
cb_allocator_(),
|
|
mgr_(mt_ctx_, cb_allocator_, mt_ctx_.mem_ctx_obj_pool_),
|
|
callback_list_(mgr_, 101) { }
|
|
virtual void SetUp() override
|
|
{
|
|
mt_counter_ = 0;
|
|
fast_commit_reserve_cnt_ = 0;
|
|
checkpoint_cnt_ = 0;
|
|
rollback_cnt_ = 0;
|
|
checksum_.reset();
|
|
callback_list_.reset();
|
|
mgr_.reset();
|
|
TRANS_LOG(INFO, "setup success");
|
|
}
|
|
|
|
virtual void TearDown() override
|
|
{
|
|
mt_counter_ = 0;
|
|
fast_commit_reserve_cnt_ = 0;
|
|
callback_list_.appended_ = 0;
|
|
callback_list_.removed_ = 0;
|
|
callback_list_.length_ = 0;
|
|
callback_list_.synced_ = 0;
|
|
callback_list_.logged_ = 0;
|
|
callback_list_.unlog_removed_ = 0;
|
|
callback_list_.reset();
|
|
mgr_.reset();
|
|
TRANS_LOG(INFO, "teardown success");
|
|
}
|
|
|
|
static void SetUpTestCase()
|
|
{
|
|
TRANS_LOG(INFO, "SetUpTestCase");
|
|
}
|
|
static void TearDownTestCase()
|
|
{
|
|
TRANS_LOG(INFO, "TearDownTestCase");
|
|
}
|
|
|
|
ObMockTxCallback *create_callback(ObMemtable *mt,
|
|
bool need_submit_log = true,
|
|
share::SCN scn = share::SCN::max_scn())
|
|
{
|
|
auto seq_no = ++seq_counter_;
|
|
ObMockTxCallback *cb = new ObMockTxCallback(mt,
|
|
need_submit_log,
|
|
scn,
|
|
seq_no);
|
|
return cb;
|
|
}
|
|
|
|
ObITransCallback *create_and_append_callback(ObMemtable *mt,
|
|
bool need_submit_log = true,
|
|
share::SCN scn = share::SCN::max_scn())
|
|
{
|
|
ObMockTxCallback *cb = create_callback(mt,
|
|
need_submit_log,
|
|
scn);
|
|
EXPECT_NE(NULL, (long)cb);
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.append_callback(cb, false/*for_replay*/));
|
|
cb->need_submit_log_ = need_submit_log;
|
|
return cb;
|
|
}
|
|
ObMockTxCallback * replay_callback(ObMemtable *mt,
|
|
transaction::ObTxSEQ seq_no,
|
|
int64_t scn_v,
|
|
bool parallel_replay,
|
|
bool serial_final)
|
|
{
|
|
share::SCN scn;
|
|
EXPECT_EQ(OB_SUCCESS, scn.convert_for_tx(scn_v));
|
|
ObMockTxCallback *cb = new ObMockTxCallback(mt,
|
|
false,
|
|
scn,
|
|
seq_no);
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.append_callback(cb, true, parallel_replay, serial_final));
|
|
return cb;
|
|
}
|
|
ObMockTxCallback * parallel_replay_callback(ObMemtable *mt,
|
|
transaction::ObTxSEQ seq_no,
|
|
int64_t scn_v,
|
|
bool serial_final)
|
|
{
|
|
return replay_callback(mt, seq_no, scn_v, true, serial_final);
|
|
}
|
|
|
|
ObMockTxCallback * serial_replay_callback(ObMemtable *mt,
|
|
transaction::ObTxSEQ seq_no,
|
|
int64_t scn_v,
|
|
bool serial_final)
|
|
{
|
|
return replay_callback(mt, seq_no, scn_v, false, serial_final);
|
|
}
|
|
|
|
ObMemtable *create_memtable()
|
|
{
|
|
mt_counter_++;
|
|
return (ObMemtable *)(mt_counter_);
|
|
}
|
|
|
|
transaction::ObTxSEQ get_seq_no() const
|
|
{
|
|
return seq_counter_;
|
|
}
|
|
|
|
bool is_checksum_equal(int64_t no, ObMockBitSet &res)
|
|
{
|
|
ObMockBitSet other;
|
|
other.reset();
|
|
for (int64_t i = 1; i <= no; i++)
|
|
{
|
|
other.add_bit(i + 100 /*seq's base is 100*/);
|
|
}
|
|
|
|
return res.equal(other);
|
|
}
|
|
|
|
static int64_t fast_commit_reserve_cnt_;
|
|
static int64_t checkpoint_cnt_;
|
|
static int64_t rollback_cnt_;
|
|
static ObMockBitSet checksum_;
|
|
|
|
transaction::ObTxSEQ seq_counter_;
|
|
int64_t mt_counter_;
|
|
ObMemtableCtx mt_ctx_;
|
|
ObMemtableCtxCbAllocator cb_allocator_;
|
|
ObTransCallbackMgr mgr_;
|
|
ObTxCallbackList callback_list_;
|
|
};
|
|
|
|
int64_t TestTxCallbackList::fast_commit_reserve_cnt_;
|
|
int64_t TestTxCallbackList::checkpoint_cnt_;
|
|
int64_t TestTxCallbackList::rollback_cnt_;
|
|
ObMockBitSet TestTxCallbackList::checksum_;
|
|
|
|
static bool has_remove = false;
|
|
|
|
int ObMockTxCallback::checkpoint_callback()
|
|
{
|
|
TestTxCallbackList::checkpoint_cnt_++;
|
|
return OB_SUCCESS;
|
|
}
|
|
|
|
int ObMockTxCallback::rollback_callback()
|
|
{
|
|
TestTxCallbackList::rollback_cnt_++;
|
|
return OB_SUCCESS;
|
|
}
|
|
|
|
int ObMockTxCallback::calc_checksum(const share::SCN checksum_scn,
|
|
TxChecksum *)
|
|
{
|
|
if (checksum_scn <= scn_) {
|
|
TestTxCallbackList::checksum_.add_bit(seq_no_.get_seq());
|
|
TRANS_LOG(INFO, "need to calc checksum", K(checksum_scn), K(scn_), K(seq_no_));
|
|
} else {
|
|
TRANS_LOG(INFO, "no need to calc checksum", K(checksum_scn), K(scn_), K(seq_no_));
|
|
}
|
|
return OB_SUCCESS;
|
|
}
|
|
|
|
TEST_F(TestTxCallbackList, remove_callback_on_failure)
|
|
{
|
|
ObMemtable *memtable = create_memtable();
|
|
share::SCN scn_1;
|
|
scn_1.convert_for_logservice(1);
|
|
|
|
create_and_append_callback(memtable,
|
|
false, /*need_submit_log*/
|
|
scn_1);
|
|
auto cb1 = create_and_append_callback(memtable,
|
|
false /*need_submit_log*/);
|
|
auto cb2 = create_and_append_callback(memtable,
|
|
false /*need_submit_log*/);
|
|
create_and_append_callback(memtable,
|
|
false /*need_submit_log*/);
|
|
|
|
ObCallbackScope scope;
|
|
int64_t removed_cnt = 0;
|
|
scope.start_ = ObITransCallbackIterator(cb1);
|
|
scope.end_ = ObITransCallbackIterator(cb2);
|
|
|
|
EXPECT_EQ(false, scope.is_empty());
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.sync_log_fail(scope, scn_1, removed_cnt));
|
|
|
|
EXPECT_EQ(2, removed_cnt);
|
|
EXPECT_EQ(2, callback_list_.get_length());
|
|
}
|
|
|
|
TEST_F(TestTxCallbackList, remove_callback_by_tx_commit)
|
|
{
|
|
ObMemtable *memtable = create_memtable();
|
|
share::SCN scn_1;
|
|
scn_1.convert_for_logservice(1);
|
|
|
|
create_and_append_callback(memtable,
|
|
false, /*need_submit_log*/
|
|
scn_1/*scn*/);
|
|
create_and_append_callback(memtable,
|
|
false, /*need_submit_log*/
|
|
scn_1/*scn*/);
|
|
create_and_append_callback(memtable,
|
|
false, /*need_submit_log*/
|
|
scn_1/*scn*/);
|
|
|
|
EXPECT_EQ(3, callback_list_.get_length());
|
|
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.tx_commit());
|
|
|
|
EXPECT_EQ(true, callback_list_.empty());
|
|
EXPECT_EQ(3, mgr_.get_callback_remove_for_trans_end_count());
|
|
}
|
|
|
|
TEST_F(TestTxCallbackList, remove_callback_by_tx_abort)
|
|
{
|
|
ObMemtable *memtable = create_memtable();
|
|
share::SCN scn_1;
|
|
scn_1.convert_for_logservice(1);
|
|
create_and_append_callback(memtable,
|
|
false, /*need_submit_log*/
|
|
scn_1/*scn*/);
|
|
create_and_append_callback(memtable,
|
|
false, /*need_submit_log*/
|
|
scn_1/*scn*/);
|
|
create_and_append_callback(memtable,
|
|
false /*need_submit_log*/);
|
|
|
|
EXPECT_EQ(3, callback_list_.get_length());
|
|
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.tx_abort());
|
|
|
|
EXPECT_EQ(true, callback_list_.empty());
|
|
EXPECT_EQ(3, mgr_.get_callback_remove_for_trans_end_count());
|
|
}
|
|
|
|
TEST_F(TestTxCallbackList, remove_callback_by_release_memtable)
|
|
{
|
|
ObMemtable *memtable1 = create_memtable();
|
|
ObMemtable *memtable2 = create_memtable();
|
|
ObMemtable *memtable3 = create_memtable();
|
|
share::SCN scn_1;
|
|
share::SCN scn_2;
|
|
share::SCN scn_100;
|
|
scn_1.convert_for_logservice(1);
|
|
scn_2.convert_for_logservice(2);
|
|
scn_100.convert_for_logservice(100);
|
|
create_and_append_callback(memtable1,
|
|
false, /*need_submit_log*/
|
|
scn_1/*scn*/);
|
|
create_and_append_callback(memtable2,
|
|
false, /*need_submit_log*/
|
|
scn_1/*scn*/);
|
|
create_and_append_callback(memtable1,
|
|
false, /*need_submit_log*/
|
|
scn_1/*scn*/);
|
|
create_and_append_callback(memtable3,
|
|
false, /*need_submit_log*/
|
|
scn_1/*scn*/);
|
|
create_and_append_callback(memtable2,
|
|
false, /*need_submit_log*/
|
|
scn_2/*scn*/);
|
|
create_and_append_callback(memtable3,
|
|
false /*need_submit_log*/);
|
|
create_and_append_callback(memtable1,
|
|
false /*need_submit_log*/);
|
|
create_and_append_callback(memtable3,
|
|
true /*need_submit_log*/);
|
|
create_and_append_callback(memtable1,
|
|
false, /*need_submit_log*/
|
|
scn_100 /*scn*/);
|
|
|
|
EXPECT_EQ(9, callback_list_.get_length());
|
|
|
|
memtable::ObMemtableSet memtable_set;
|
|
EXPECT_EQ(OB_SUCCESS, memtable_set.create(24));
|
|
EXPECT_EQ(OB_SUCCESS, memtable_set.set_refactored((uint64_t)(memtable2)));
|
|
EXPECT_EQ(OB_SUCCESS, memtable_set.set_refactored((uint64_t)(memtable1)));
|
|
EXPECT_EQ(OB_SUCCESS, memtable_set.set_refactored((uint64_t)(memtable3)));
|
|
|
|
EXPECT_EQ(OB_SUCCESS,
|
|
callback_list_.remove_callbacks_for_remove_memtable(&memtable_set, scn_1/*not used*/));
|
|
|
|
EXPECT_EQ(4, callback_list_.get_length());
|
|
EXPECT_EQ(5, mgr_.get_callback_remove_for_remove_memtable_count());
|
|
|
|
EXPECT_EQ(5, checkpoint_cnt_);
|
|
}
|
|
|
|
TEST_F(TestTxCallbackList, remove_callback_by_fast_commit)
|
|
{
|
|
ObMemtable *memtable1 = create_memtable();
|
|
ObMemtable *memtable2 = create_memtable();
|
|
ObMemtable *memtable3 = create_memtable();
|
|
share::SCN scn_1;
|
|
share::SCN scn_2;
|
|
share::SCN scn_3;
|
|
share::SCN scn_100;
|
|
scn_1.convert_for_logservice(1);
|
|
scn_2.convert_for_logservice(2);
|
|
scn_3.convert_for_logservice(3);
|
|
scn_100.convert_for_logservice(100);
|
|
|
|
create_and_append_callback(memtable1,
|
|
false, /*need_submit_log*/
|
|
scn_1/*scn*/);
|
|
create_and_append_callback(memtable2,
|
|
false, /*need_submit_log*/
|
|
scn_2/*scn*/);
|
|
create_and_append_callback(memtable1,
|
|
false, /*need_submit_log*/
|
|
scn_2/*scn*/);
|
|
create_and_append_callback(memtable3,
|
|
false, /*need_submit_log*/
|
|
scn_3/*scn*/);
|
|
create_and_append_callback(memtable2,
|
|
false, /*need_submit_log*/
|
|
scn_3/*scn*/);
|
|
create_and_append_callback(memtable3,
|
|
false /*need_submit_log*/);
|
|
create_and_append_callback(memtable1,
|
|
false /*need_submit_log*/);
|
|
create_and_append_callback(memtable3,
|
|
true /*need_submit_log*/);
|
|
create_and_append_callback(memtable1,
|
|
false, /*need_submit_log*/
|
|
scn_100 /*scn*/);
|
|
int sync_cnt = 9;
|
|
callback_list_.sync_log_succ(scn_100, sync_cnt);
|
|
EXPECT_EQ(9, callback_list_.get_length());
|
|
|
|
fast_commit_reserve_cnt_ = 16;
|
|
has_remove = false;
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.remove_callbacks_for_fast_commit());
|
|
EXPECT_EQ(8, callback_list_.get_length());
|
|
EXPECT_EQ(1, mgr_.get_callback_remove_for_fast_commit_count());
|
|
EXPECT_EQ(true, has_remove);
|
|
|
|
fast_commit_reserve_cnt_ = 14;
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.remove_callbacks_for_fast_commit());
|
|
EXPECT_EQ(6, callback_list_.get_length());
|
|
EXPECT_EQ(3, mgr_.get_callback_remove_for_fast_commit_count());
|
|
EXPECT_EQ(true, has_remove);
|
|
|
|
fast_commit_reserve_cnt_ = 1;
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.remove_callbacks_for_fast_commit());
|
|
EXPECT_EQ(4, callback_list_.get_length());
|
|
EXPECT_EQ(5, mgr_.get_callback_remove_for_fast_commit_count());
|
|
EXPECT_EQ(true, has_remove);
|
|
|
|
fast_commit_reserve_cnt_ = 1;
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.remove_callbacks_for_fast_commit());
|
|
EXPECT_EQ(4, callback_list_.get_length());
|
|
EXPECT_EQ(5, mgr_.get_callback_remove_for_fast_commit_count());
|
|
EXPECT_EQ(false, has_remove);
|
|
|
|
EXPECT_EQ(5, checkpoint_cnt_);
|
|
}
|
|
|
|
TEST_F(TestTxCallbackList, remove_callback_by_rollback_to)
|
|
{
|
|
ObMemtable *memtable1 = create_memtable();
|
|
ObMemtable *memtable2 = create_memtable();
|
|
ObMemtable *memtable3 = create_memtable();
|
|
share::SCN scn_1;
|
|
share::SCN scn_2;
|
|
share::SCN scn_3;
|
|
scn_1.convert_for_logservice(1);
|
|
scn_2.convert_for_logservice(2);
|
|
scn_3.convert_for_logservice(3);
|
|
|
|
auto savepoint0 = get_seq_no();
|
|
create_and_append_callback(memtable1,
|
|
false, /*need_submit_log*/
|
|
scn_1/*scn*/);
|
|
create_and_append_callback(memtable2,
|
|
false, /*need_submit_log*/
|
|
scn_2/*scn*/);
|
|
create_and_append_callback(memtable1,
|
|
false, /*need_submit_log*/
|
|
scn_2/*scn*/);
|
|
auto savepoint1 = get_seq_no();
|
|
create_and_append_callback(memtable3,
|
|
false, /*need_submit_log*/
|
|
scn_3/*scn*/);
|
|
create_and_append_callback(memtable2,
|
|
false, /*need_submit_log*/
|
|
scn_3/*scn*/);
|
|
create_and_append_callback(memtable3,
|
|
true /*need_submit_log*/);
|
|
create_and_append_callback(memtable1,
|
|
true /*need_submit_log*/);
|
|
auto savepoint2 = get_seq_no();
|
|
create_and_append_callback(memtable3,
|
|
true /*need_submit_log*/);
|
|
auto savepoint3 = get_seq_no();
|
|
create_and_append_callback(memtable1,
|
|
true /*need_submit_log*/);
|
|
EXPECT_EQ(9, callback_list_.get_length());
|
|
auto from = get_seq_no() + 1;
|
|
|
|
from.set_branch(savepoint3.get_branch());
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.remove_callbacks_for_rollback_to(savepoint3, from, share::SCN::invalid_scn()));
|
|
EXPECT_EQ(8, callback_list_.get_length());
|
|
EXPECT_EQ(1, mgr_.get_callback_remove_for_rollback_to_count());
|
|
|
|
from.set_branch(savepoint2.get_branch());
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.remove_callbacks_for_rollback_to(savepoint2, from, share::SCN::invalid_scn()));
|
|
EXPECT_EQ(7, callback_list_.get_length());
|
|
EXPECT_EQ(2, mgr_.get_callback_remove_for_rollback_to_count());
|
|
|
|
from.set_branch(savepoint1.get_branch());
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.remove_callbacks_for_rollback_to(savepoint1, from, share::SCN::invalid_scn()));
|
|
EXPECT_EQ(3, callback_list_.get_length());
|
|
EXPECT_EQ(6, mgr_.get_callback_remove_for_rollback_to_count());
|
|
|
|
from.set_branch(savepoint0.get_branch());
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.remove_callbacks_for_rollback_to(savepoint0, from, share::SCN::invalid_scn()));
|
|
EXPECT_EQ(0, callback_list_.get_length());
|
|
EXPECT_EQ(9, mgr_.get_callback_remove_for_rollback_to_count());
|
|
|
|
EXPECT_EQ(9, rollback_cnt_);
|
|
}
|
|
|
|
TEST_F(TestTxCallbackList, remove_callback_by_branch_rollback_to)
|
|
{
|
|
ObMemtable *memtable1 = create_memtable();
|
|
share::SCN scn;
|
|
scn.convert_for_logservice(1);
|
|
#define SP(branch, i) auto sp_##i = get_seq_no(); sp_##i.set_branch(branch);
|
|
#define APPEND_CB(branch, i) \
|
|
seq_counter_.set_branch(branch); \
|
|
auto cb_##i = create_and_append_callback(memtable1, \
|
|
false, /*need_submit_log*/ \
|
|
share::SCN::plus(scn, i)/*scn*/);
|
|
SP(0, 0);
|
|
APPEND_CB(1,1);
|
|
SP(1, 1);
|
|
APPEND_CB(2,2);
|
|
SP(2, 2);
|
|
APPEND_CB(0,3);
|
|
APPEND_CB(1,4);
|
|
APPEND_CB(1,5);
|
|
APPEND_CB(2,6);
|
|
seq_counter_.set_branch(0);
|
|
#undef APPEND_CB
|
|
#undef SP
|
|
#define RB_TO(i) { \
|
|
auto from = (seq_counter_ + 1); \
|
|
from.set_branch(sp_##i.get_branch()); \
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.remove_callbacks_for_rollback_to(sp_##i, from, share::SCN::invalid_scn())); \
|
|
}
|
|
EXPECT_EQ(6, callback_list_.get_length());
|
|
|
|
RB_TO(2); // rollback branch 2
|
|
EXPECT_EQ(5, callback_list_.get_length());
|
|
EXPECT_EQ(1, mgr_.get_callback_remove_for_rollback_to_count());
|
|
|
|
RB_TO(1); // rollback branch 1
|
|
EXPECT_EQ(3, callback_list_.get_length());
|
|
EXPECT_EQ(3, mgr_.get_callback_remove_for_rollback_to_count());
|
|
|
|
RB_TO(0); // rollback to head
|
|
EXPECT_EQ(0, callback_list_.get_length());
|
|
EXPECT_EQ(6, mgr_.get_callback_remove_for_rollback_to_count());
|
|
|
|
EXPECT_EQ(6, rollback_cnt_);
|
|
}
|
|
|
|
TEST_F(TestTxCallbackList, remove_callback_by_clean_unlog_callbacks)
|
|
{
|
|
ObMemtable *memtable1 = create_memtable();
|
|
ObMemtable *memtable2 = create_memtable();
|
|
ObMemtable *memtable3 = create_memtable();
|
|
share::SCN scn_1;
|
|
share::SCN scn_2;
|
|
share::SCN scn_3;
|
|
scn_1.convert_for_logservice(1);
|
|
scn_2.convert_for_logservice(2);
|
|
scn_3.convert_for_logservice(3);
|
|
|
|
create_and_append_callback(memtable1,
|
|
false, /*need_submit_log*/
|
|
scn_1/*scn*/);
|
|
create_and_append_callback(memtable2,
|
|
false, /*need_submit_log*/
|
|
scn_2/*scn*/);
|
|
create_and_append_callback(memtable1,
|
|
false, /*need_submit_log*/
|
|
scn_2/*scn*/);
|
|
create_and_append_callback(memtable3,
|
|
false, /*need_submit_log*/
|
|
scn_3/*scn*/);
|
|
create_and_append_callback(memtable2,
|
|
false, /*need_submit_log*/
|
|
scn_3/*scn*/);
|
|
create_and_append_callback(memtable3,
|
|
true /*need_submit_log*/);
|
|
create_and_append_callback(memtable1,
|
|
true /*need_submit_log*/);
|
|
create_and_append_callback(memtable3,
|
|
true /*need_submit_log*/);
|
|
create_and_append_callback(memtable1,
|
|
true /*need_submit_log*/);
|
|
|
|
EXPECT_EQ(9, callback_list_.get_length());
|
|
int64_t removed_cnt = 0;
|
|
ObFunction<void()> before_remove = []{};
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.clean_unlog_callbacks(removed_cnt, before_remove));
|
|
EXPECT_EQ(5, callback_list_.get_length());
|
|
|
|
EXPECT_EQ(4, rollback_cnt_);
|
|
}
|
|
|
|
TEST_F(TestTxCallbackList, remove_callback_by_replay_fail)
|
|
{
|
|
TRANS_LOG(INFO, "CASE: remove_callback_by_replay_fail");
|
|
ObMemtable *memtable1 = create_memtable();
|
|
ObMemtable *memtable2 = create_memtable();
|
|
ObMemtable *memtable3 = create_memtable();
|
|
share::SCN scn_1;
|
|
share::SCN scn_2;
|
|
share::SCN scn_3;
|
|
share::SCN scn_4;
|
|
scn_1.convert_for_logservice(1);
|
|
scn_2.convert_for_logservice(2);
|
|
scn_3.convert_for_logservice(3);
|
|
scn_4.convert_for_logservice(4);
|
|
|
|
create_and_append_callback(memtable1,
|
|
false, /*need_submit_log*/
|
|
scn_1/*scn*/);
|
|
create_and_append_callback(memtable2,
|
|
false, /*need_submit_log*/
|
|
scn_2/*scn*/);
|
|
create_and_append_callback(memtable1,
|
|
false, /*need_submit_log*/
|
|
scn_2/*scn*/);
|
|
create_and_append_callback(memtable3,
|
|
false, /*need_submit_log*/
|
|
scn_3/*scn*/);
|
|
create_and_append_callback(memtable2,
|
|
false, /*need_submit_log*/
|
|
scn_3/*scn*/);
|
|
create_and_append_callback(memtable3,
|
|
false, /*need_submit_log*/
|
|
scn_3/*scn*/);
|
|
create_and_append_callback(memtable1,
|
|
false, /*need_submit_log*/
|
|
scn_4/*scn*/);
|
|
create_and_append_callback(memtable3,
|
|
false, /*need_submit_log*/
|
|
scn_4/*scn*/);
|
|
create_and_append_callback(memtable1,
|
|
false, /*need_submit_log*/
|
|
scn_4/*scn*/);
|
|
|
|
EXPECT_EQ(9, callback_list_.get_length());
|
|
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.replay_fail(scn_4 /*log_timestamp*/, true));
|
|
EXPECT_EQ(6, callback_list_.get_length());
|
|
|
|
EXPECT_EQ(3, rollback_cnt_);
|
|
}
|
|
|
|
TEST_F(TestTxCallbackList, checksum_leader_tx_end_basic)
|
|
{
|
|
TRANS_LOG(INFO, "CASE: checksum_leader_tx_end_basic");
|
|
ObMemtable *memtable = create_memtable();
|
|
|
|
create_and_append_callback(memtable,
|
|
true /*need_submit_log*/);
|
|
create_and_append_callback(memtable,
|
|
true /*need_submit_log*/);
|
|
create_and_append_callback(memtable,
|
|
true /*need_submit_log*/);
|
|
|
|
EXPECT_EQ(3, callback_list_.get_length());
|
|
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.tx_calc_checksum_all());
|
|
|
|
EXPECT_EQ(true, is_checksum_equal(3, checksum_));
|
|
EXPECT_EQ(share::SCN::max_scn(), callback_list_.checksum_scn_);
|
|
}
|
|
|
|
TEST_F(TestTxCallbackList, checksum_follower_tx_end)
|
|
{
|
|
ObMemtable *memtable = create_memtable();
|
|
share::SCN scn_1;
|
|
share::SCN scn_2;
|
|
share::SCN scn_3;
|
|
scn_1.convert_for_logservice(1);
|
|
scn_2.convert_for_logservice(2);
|
|
scn_3.convert_for_logservice(3);
|
|
|
|
create_and_append_callback(memtable,
|
|
false, /*need_submit_log*/
|
|
scn_1 /*scn*/);
|
|
create_and_append_callback(memtable,
|
|
false, /*need_submit_log*/
|
|
scn_2 /*scn*/);
|
|
create_and_append_callback(memtable,
|
|
false, /*need_submit_log*/
|
|
scn_3 /*scn*/);
|
|
|
|
EXPECT_EQ(3, callback_list_.get_length());
|
|
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.tx_calc_checksum_all());
|
|
|
|
EXPECT_EQ(true, is_checksum_equal(3, checksum_));
|
|
EXPECT_EQ(share::SCN::max_scn(), callback_list_.checksum_scn_);
|
|
}
|
|
|
|
TEST_F(TestTxCallbackList, checksum_leader_tx_end_harder)
|
|
{
|
|
TRANS_LOG(INFO, "CASE: checksum_leader_tx_end_harder");
|
|
ObMemtable *memtable = create_memtable();
|
|
share::SCN scn_1;
|
|
share::SCN scn_2;
|
|
scn_1.convert_for_logservice(1);
|
|
scn_2.convert_for_logservice(2);
|
|
|
|
create_and_append_callback(memtable,
|
|
false, /*need_submit_log*/
|
|
scn_1 /*scn*/);
|
|
create_and_append_callback(memtable,
|
|
false, /*need_submit_log*/
|
|
scn_1 /*scn*/);
|
|
create_and_append_callback(memtable,
|
|
false, /*need_submit_log*/
|
|
scn_2 /*scn*/);
|
|
create_and_append_callback(memtable,
|
|
false /*need_submit_log*/);
|
|
create_and_append_callback(memtable,
|
|
false /*need_submit_log*/);
|
|
|
|
EXPECT_EQ(5, callback_list_.get_length());
|
|
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.tx_calc_checksum_all());
|
|
|
|
EXPECT_EQ(true, is_checksum_equal(5, checksum_));
|
|
EXPECT_EQ(share::SCN::max_scn(), callback_list_.checksum_scn_);
|
|
}
|
|
|
|
TEST_F(TestTxCallbackList, checksum_leader_tx_end_harderer)
|
|
{
|
|
TRANS_LOG(INFO, "CASE: checksum_leader_tx_end_harderer");
|
|
ObMemtable *memtable = create_memtable();
|
|
share::SCN scn_1;
|
|
share::SCN scn_2;
|
|
scn_1.convert_for_logservice(1);
|
|
scn_2.convert_for_logservice(2);
|
|
|
|
create_and_append_callback(memtable,
|
|
false, /*need_submit_log*/
|
|
scn_1 /*scn*/);
|
|
create_and_append_callback(memtable,
|
|
false, /*need_submit_log*/
|
|
scn_1 /*scn*/);
|
|
create_and_append_callback(memtable,
|
|
false, /*need_submit_log*/
|
|
scn_2 /*scn*/);
|
|
create_and_append_callback(memtable,
|
|
false, /*need_submit_log*/
|
|
scn_1 /*scn*/);
|
|
create_and_append_callback(memtable,
|
|
false /*need_submit_log*/);
|
|
create_and_append_callback(memtable,
|
|
false /*need_submit_log*/);
|
|
|
|
EXPECT_EQ(6, callback_list_.get_length());
|
|
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.tx_calc_checksum_all());
|
|
|
|
EXPECT_EQ(true, is_checksum_equal(6, checksum_));
|
|
EXPECT_EQ(share::SCN::max_scn(), callback_list_.checksum_scn_);
|
|
}
|
|
|
|
TEST_F(TestTxCallbackList, checksum_remove_memtable_and_tx_end)
|
|
{
|
|
TRANS_LOG(INFO, "CASE: checksum_remove_memtable_and_tx_end");
|
|
ObMemtable *memtable1 = create_memtable();
|
|
ObMemtable *memtable2 = create_memtable();
|
|
ObMemtable *memtable3 = create_memtable();
|
|
share::SCN scn_1;
|
|
share::SCN scn_2;
|
|
share::SCN scn_3;
|
|
scn_1.convert_for_logservice(1);
|
|
scn_2.convert_for_logservice(2);
|
|
scn_3.convert_for_logservice(3);
|
|
|
|
create_and_append_callback(memtable1,
|
|
false, /*need_submit_log*/
|
|
scn_1/*scn*/);
|
|
create_and_append_callback(memtable2,
|
|
false, /*need_submit_log*/
|
|
scn_1/*scn*/);
|
|
create_and_append_callback(memtable1,
|
|
false, /*need_submit_log*/
|
|
scn_1/*scn*/);
|
|
create_and_append_callback(memtable3,
|
|
false, /*need_submit_log*/
|
|
scn_2/*scn*/);
|
|
create_and_append_callback(memtable2,
|
|
false, /*need_submit_log*/
|
|
scn_2/*scn*/);
|
|
create_and_append_callback(memtable3,
|
|
false /*need_submit_log*/);
|
|
create_and_append_callback(memtable1,
|
|
false /*need_submit_log*/);
|
|
create_and_append_callback(memtable3,
|
|
true /*need_submit_log*/);
|
|
create_and_append_callback(memtable1,
|
|
true /*need_submit_log*/);
|
|
|
|
|
|
EXPECT_EQ(9, callback_list_.get_length());
|
|
|
|
memtable::ObMemtableSet memtable_set;
|
|
EXPECT_EQ(OB_SUCCESS, memtable_set.create(24));
|
|
EXPECT_EQ(OB_SUCCESS, memtable_set.set_refactored((uint64_t)(memtable2)));
|
|
EXPECT_EQ(OB_SUCCESS, memtable_set.set_refactored((uint64_t)(memtable1)));
|
|
EXPECT_EQ(OB_SUCCESS, memtable_set.set_refactored((uint64_t)(memtable3)));
|
|
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.remove_callbacks_for_remove_memtable(&memtable_set, scn_1/*not used*/));
|
|
EXPECT_EQ(true, is_checksum_equal(5, checksum_));
|
|
EXPECT_EQ(scn_3, callback_list_.checksum_scn_);
|
|
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.tx_calc_checksum_all());
|
|
EXPECT_EQ(true, is_checksum_equal(9, checksum_));
|
|
EXPECT_EQ(share::SCN::max_scn(), callback_list_.checksum_scn_);
|
|
}
|
|
|
|
|
|
TEST_F(TestTxCallbackList, checksum_fast_commit_and_tx_end)
|
|
{
|
|
TRANS_LOG(INFO, "CASE: checksum_fast_commit_and_tx_end");
|
|
ObMemtable *memtable1 = create_memtable();
|
|
ObMemtable *memtable2 = create_memtable();
|
|
ObMemtable *memtable3 = create_memtable();
|
|
share::SCN scn_1;
|
|
share::SCN scn_2;
|
|
share::SCN scn_3;
|
|
share::SCN scn_4;
|
|
scn_1.convert_for_logservice(1);
|
|
scn_2.convert_for_logservice(2);
|
|
scn_3.convert_for_logservice(3);
|
|
scn_4.convert_for_logservice(4);
|
|
|
|
create_and_append_callback(memtable1,
|
|
false, /*need_submit_log*/
|
|
scn_1/*scn*/);
|
|
create_and_append_callback(memtable2,
|
|
false, /*need_submit_log*/
|
|
scn_2/*scn*/);
|
|
create_and_append_callback(memtable1,
|
|
false, /*need_submit_log*/
|
|
scn_2/*scn*/);
|
|
create_and_append_callback(memtable3,
|
|
false, /*need_submit_log*/
|
|
scn_3/*scn*/);
|
|
create_and_append_callback(memtable2,
|
|
false, /*need_submit_log*/
|
|
scn_3/*scn*/);
|
|
create_and_append_callback(memtable3,
|
|
false /*need_submit_log*/);
|
|
create_and_append_callback(memtable1,
|
|
false /*need_submit_log*/);
|
|
create_and_append_callback(memtable3,
|
|
true /*need_submit_log*/);
|
|
create_and_append_callback(memtable1,
|
|
true /*need_submit_log*/);
|
|
int sync_cnt = 0;
|
|
callback_list_.sync_log_succ(share::SCN::plus(scn_3, 100), sync_cnt);
|
|
EXPECT_EQ(9, callback_list_.get_length());
|
|
|
|
fast_commit_reserve_cnt_ = 16;
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.remove_callbacks_for_fast_commit());
|
|
EXPECT_EQ(true, is_checksum_equal(1, checksum_));
|
|
EXPECT_EQ(scn_2, callback_list_.checksum_scn_);
|
|
|
|
fast_commit_reserve_cnt_ = 14;
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.remove_callbacks_for_fast_commit());
|
|
EXPECT_EQ(true, is_checksum_equal(3, checksum_));
|
|
EXPECT_EQ(scn_3, callback_list_.checksum_scn_);
|
|
|
|
fast_commit_reserve_cnt_ = 1;
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.remove_callbacks_for_fast_commit());
|
|
EXPECT_EQ(true, is_checksum_equal(5, checksum_));
|
|
EXPECT_EQ(scn_4, callback_list_.checksum_scn_);
|
|
|
|
fast_commit_reserve_cnt_ = 1;
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.remove_callbacks_for_fast_commit());
|
|
EXPECT_EQ(true, is_checksum_equal(5, checksum_));
|
|
EXPECT_EQ(scn_4, callback_list_.checksum_scn_);
|
|
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.tx_calc_checksum_all());
|
|
EXPECT_EQ(true, is_checksum_equal(9, checksum_));
|
|
EXPECT_EQ(share::SCN::max_scn(), callback_list_.checksum_scn_);
|
|
}
|
|
|
|
TEST_F(TestTxCallbackList, checksum_rollback_to_and_tx_end)
|
|
{
|
|
TRANS_LOG(INFO, "CASE: checksum_rollback_to_and_tx_end");
|
|
ObMemtable *memtable1 = create_memtable();
|
|
ObMemtable *memtable2 = create_memtable();
|
|
ObMemtable *memtable3 = create_memtable();
|
|
share::SCN scn_1;
|
|
share::SCN scn_2;
|
|
share::SCN scn_3;
|
|
share::SCN scn_4;
|
|
scn_1.convert_for_logservice(1);
|
|
scn_2.convert_for_logservice(2);
|
|
scn_3.convert_for_logservice(3);
|
|
scn_4.convert_for_logservice(4);
|
|
|
|
auto savepoint0 = get_seq_no();
|
|
create_and_append_callback(memtable1,
|
|
false, /*need_submit_log*/
|
|
scn_1/*scn*/);
|
|
create_and_append_callback(memtable2,
|
|
false, /*need_submit_log*/
|
|
scn_2/*scn*/);
|
|
create_and_append_callback(memtable1,
|
|
false, /*need_submit_log*/
|
|
scn_2/*scn*/);
|
|
auto savepoint1 = get_seq_no();
|
|
create_and_append_callback(memtable3,
|
|
false, /*need_submit_log*/
|
|
scn_3/*scn*/);
|
|
create_and_append_callback(memtable2,
|
|
false, /*need_submit_log*/
|
|
scn_3/*scn*/);
|
|
create_and_append_callback(memtable3,
|
|
true /*need_submit_log*/);
|
|
create_and_append_callback(memtable1,
|
|
true /*need_submit_log*/);
|
|
auto savepoint2 = get_seq_no();
|
|
create_and_append_callback(memtable3,
|
|
true /*need_submit_log*/);
|
|
auto savepoint3 = get_seq_no();
|
|
create_and_append_callback(memtable1,
|
|
true /*need_submit_log*/);
|
|
|
|
EXPECT_EQ(9, callback_list_.get_length());
|
|
auto from = get_seq_no() + 1;
|
|
|
|
from.set_branch(savepoint3.get_branch());
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.remove_callbacks_for_rollback_to(savepoint3, from, share::SCN::invalid_scn()));
|
|
EXPECT_EQ(true, is_checksum_equal(5, checksum_));
|
|
EXPECT_EQ(scn_4, callback_list_.checksum_scn_);
|
|
|
|
from.set_branch(savepoint2.get_branch());
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.remove_callbacks_for_rollback_to(savepoint2, from, share::SCN::invalid_scn()));
|
|
EXPECT_EQ(true, is_checksum_equal(5, checksum_));
|
|
EXPECT_EQ(scn_4, callback_list_.checksum_scn_);
|
|
|
|
from.set_branch(savepoint1.get_branch());
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.remove_callbacks_for_rollback_to(savepoint1, from, share::SCN::invalid_scn()));
|
|
EXPECT_EQ(true, is_checksum_equal(5, checksum_));
|
|
EXPECT_EQ(scn_4, callback_list_.checksum_scn_);
|
|
|
|
from.set_branch(savepoint0.get_branch());
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.remove_callbacks_for_rollback_to(savepoint0, from, share::SCN::invalid_scn()));
|
|
EXPECT_EQ(true, is_checksum_equal(5, checksum_));
|
|
EXPECT_EQ(scn_4, callback_list_.checksum_scn_);
|
|
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.tx_calc_checksum_all());
|
|
EXPECT_EQ(true, is_checksum_equal(5, checksum_));
|
|
EXPECT_EQ(share::SCN::max_scn(), callback_list_.checksum_scn_);
|
|
}
|
|
|
|
TEST_F(TestTxCallbackList, checksum_all_and_tx_end_test) {
|
|
TRANS_LOG(INFO, "CASE: checksum_all_and_tx_end_test");
|
|
int64_t length = 0;
|
|
int64_t mt_cnt = 5;
|
|
int64_t cur_log = 0;
|
|
ObMemtable *mts[mt_cnt];
|
|
ObITransCallback *need_submit_head = &(callback_list_.head_);
|
|
ObMockBitSet my_calculate;
|
|
my_calculate.reset();
|
|
for (int i = 0; i < mt_cnt; i++) {
|
|
mts[i] = create_memtable();
|
|
}
|
|
|
|
ObFunction<bool()> append_callback_op =
|
|
[&]() -> bool {
|
|
for (int i = 0; i < 10; i++) {
|
|
ObMemtable *mt = mts[ObRandom::rand(0, mt_cnt - 1)];
|
|
create_and_append_callback(mt,
|
|
true /*need_submit_log*/);
|
|
|
|
}
|
|
return true;
|
|
};
|
|
|
|
ObFunction<bool()> on_success_op =
|
|
[&]() -> bool {
|
|
cur_log++;
|
|
int i = 0;
|
|
bool enable = false;
|
|
int64_t max_cnt = ObRandom::rand(1, 5);
|
|
for (ObITransCallback* it = need_submit_head->next_;
|
|
it != &(callback_list_.head_) && i < max_cnt;
|
|
it = it->next_) {
|
|
i++;
|
|
it->need_submit_log_ = false;
|
|
it->scn_.convert_for_logservice(cur_log);
|
|
enable = true;
|
|
need_submit_head = it;
|
|
my_calculate.add_bit(it->get_seq_no().get_seq());
|
|
int64_t c = 1;
|
|
callback_list_.sync_log_succ(it->scn_, c);
|
|
}
|
|
|
|
if (!enable) {
|
|
cur_log--;
|
|
}
|
|
|
|
return enable;
|
|
};
|
|
|
|
ObFunction<bool()> remove_memtable_op =
|
|
[&]() -> bool {
|
|
ObMemtable *mt = mts[ObRandom::rand(0, mt_cnt-1)];
|
|
share::SCN scn_1;
|
|
scn_1.convert_for_logservice(1);
|
|
bool enable = false;
|
|
for (ObITransCallback* it = need_submit_head->next_;
|
|
it != &(callback_list_.head_);
|
|
it = it->next_) {
|
|
if (it->need_submit_log_) {
|
|
break;
|
|
} else if (it->get_memtable() == mt) {
|
|
enable = true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (enable) {
|
|
memtable::ObMemtableSet memtable_set;
|
|
EXPECT_EQ(OB_SUCCESS, memtable_set.create(24));
|
|
EXPECT_EQ(OB_SUCCESS, memtable_set.set_refactored((uint64_t)(mt)));
|
|
EXPECT_EQ(OB_SUCCESS,
|
|
callback_list_.remove_callbacks_for_remove_memtable(&memtable_set, scn_1/*not used*/));
|
|
}
|
|
|
|
return enable;
|
|
};
|
|
|
|
ObFunction<bool()> fast_commit_op =
|
|
[&]() -> bool{
|
|
bool enable = false;
|
|
|
|
fast_commit_reserve_cnt_ = 30;
|
|
if (callback_list_.length_ > fast_commit_reserve_cnt_ / 2
|
|
&& !callback_list_.head_.next_->need_submit_log_) {
|
|
enable = true;
|
|
}
|
|
|
|
has_remove = false;
|
|
if (enable) {
|
|
EXPECT_EQ(OB_SUCCESS,
|
|
callback_list_.remove_callbacks_for_fast_commit());
|
|
EXPECT_EQ(true, has_remove);
|
|
}
|
|
|
|
return enable;
|
|
};
|
|
|
|
ObFunction<bool()> rollback_to_op =
|
|
[&]() -> bool{
|
|
bool enable = false;
|
|
if (!callback_list_.empty() &&
|
|
callback_list_.head_.next_->get_seq_no().get_seq() + 1 < seq_counter_.get_seq() - 1) {
|
|
auto from = callback_list_.head_.next_->get_seq_no();
|
|
auto range_cnt = seq_counter_.get_seq() - from.get_seq();
|
|
auto seq = from + ObRandom::rand(1, range_cnt - 1);
|
|
enable = true;
|
|
if (enable) {
|
|
bool reset_need_submit_head = false;
|
|
if (need_submit_head->get_seq_no() > seq) {
|
|
reset_need_submit_head = true;
|
|
}
|
|
auto from0 = get_seq_no() + 1;
|
|
EXPECT_EQ(OB_SUCCESS,
|
|
callback_list_.remove_callbacks_for_rollback_to(seq, from0, share::SCN::invalid_scn()));
|
|
EXPECT_EQ(false, callback_list_.empty());
|
|
if (reset_need_submit_head) {
|
|
need_submit_head = callback_list_.head_.prev_;
|
|
}
|
|
}
|
|
}
|
|
|
|
return enable;
|
|
};
|
|
|
|
|
|
for (int c = 0; c < 10000; ) {
|
|
int choice = ObRandom::rand(1, 10);
|
|
ObFunction<bool()> *op = NULL;
|
|
if (choice >= 1 && choice <= 3) {
|
|
op = &append_callback_op;
|
|
} else if (choice <= 5) {
|
|
op = &on_success_op;
|
|
} else if (choice <= 7) {
|
|
op = &remove_memtable_op;
|
|
} else if (choice <= 9) {
|
|
op = &fast_commit_op;
|
|
} else if (choice <= 10) {
|
|
op = &rollback_to_op;
|
|
}
|
|
|
|
if ((*op)()) {
|
|
TRANS_LOG(INFO, "choice on success", K(callback_list_), K(choice), K(c));
|
|
c++;
|
|
} else {
|
|
TRANS_LOG(INFO, "choice on skip", K(callback_list_), K(choice), K(c));
|
|
}
|
|
}
|
|
|
|
for (ObITransCallback* it = need_submit_head->next_;
|
|
it != &(callback_list_.head_);
|
|
it = it->next_) {
|
|
EXPECT_EQ(it->need_submit_log_, true);
|
|
// set scn because tx_commit will do check
|
|
it->scn_.convert_for_logservice(10000);
|
|
//my_calculate.add_bit(it->get_seq_no().get_seq());
|
|
}
|
|
// calc checksum of remains by fast_commit
|
|
fast_commit_reserve_cnt_ = 0;
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.remove_callbacks_for_fast_commit());
|
|
EXPECT_EQ(OB_SUCCESS, callback_list_.tx_commit());
|
|
EXPECT_EQ(true, my_calculate.equal(checksum_));
|
|
|
|
}
|
|
|
|
TEST_F(TestTxCallbackList, log_cursor) {
|
|
TRANS_LOG(INFO, "CASE: log_cursor");
|
|
int ret = 0;
|
|
auto memtable1 = create_memtable();
|
|
share::SCN scn; scn.convert_for_logservice(100);
|
|
ObTxFillRedoCtx ctx;
|
|
memtable::ObRedoLogSubmitHelper helper;
|
|
ctx.helper_ = &helper;
|
|
struct ObTxCallbackFunctorAdapter : public ObITxFillRedoFunctor {
|
|
ObTxCallbackFunctorAdapter(ObFunction<int(ObITransCallback*)> func) : func_(func) {}
|
|
int operator()(ObITransCallback*cb) { return func_(cb); }
|
|
ObFunction<int(ObITransCallback*)> func_;
|
|
};
|
|
#define APPEND_CB(branch, i) \
|
|
seq_counter_.set_branch(branch); \
|
|
auto cb_##i = create_and_append_callback(memtable1, \
|
|
true /*need_submit_log*/); \
|
|
|
|
#define LOG_SUBMITED(SCOPE_X) \
|
|
{ \
|
|
int submitted_cnt = 0; \
|
|
ObArrayHelper<ObCallbackScope> scope_a(1, &SCOPE_X, 1); \
|
|
scn = share::SCN::plus(scn, 1); \
|
|
ret = mgr_.log_submitted(scope_a, scn, submitted_cnt); \
|
|
}
|
|
|
|
#define LOG_SYNC_SUCC(SCOPE_X) \
|
|
{ \
|
|
ObArrayHelper<ObCallbackScope> scope_a(1, &SCOPE_X, 1); \
|
|
int64_t cnt = 0; \
|
|
ret = mgr_.log_sync_succ(scope_a, scn, cnt); \
|
|
}
|
|
|
|
// check log_cursor is valid and expected after every op on CallbackList
|
|
// . init callback list, insert 3 node
|
|
APPEND_CB(0, 1);
|
|
APPEND_CB(0, 2);
|
|
APPEND_CB(0, 3);
|
|
EXPECT_EQ(callback_list_.log_cursor_, cb_1);
|
|
EXPECT_TRUE(cb_3->need_submit_log_);
|
|
// . fill log for 1 and 2, log_cursor point to head
|
|
int i = 1;
|
|
ObCallbackScope scope;
|
|
ObTxCallbackFunctorAdapter f([&](ObITransCallback*cb) -> int {
|
|
if (*scope.start_ == NULL) scope.start_ = cb;
|
|
scope.end_ = cb;
|
|
scope.host_ = &callback_list_;
|
|
scope.cnt_ = i;
|
|
return i++ == 2 ? OB_EAGAIN : OB_SUCCESS;
|
|
});
|
|
ctx.callback_scope_ = &scope;
|
|
ret = callback_list_.fill_log(callback_list_.log_cursor_, ctx, f);
|
|
EXPECT_EQ(OB_EAGAIN, ret);
|
|
EXPECT_EQ(i, 3);
|
|
EXPECT_EQ(cb_1, *scope.start_);
|
|
EXPECT_EQ(cb_2, *scope.end_);
|
|
EXPECT_TRUE(cb_3->need_submit_log_);
|
|
EXPECT_EQ(callback_list_.log_cursor_, cb_1);
|
|
// . log submitted, update log_cursor, point to the next one to log
|
|
LOG_SUBMITED(scope);
|
|
EXPECT_EQ(OB_SUCCESS, ret);
|
|
EXPECT_EQ(callback_list_.log_cursor_, cb_3);
|
|
// . log_sync succ, fast_commit remove callbacks, log cursor is not affected
|
|
LOG_SYNC_SUCC(scope);
|
|
EXPECT_EQ(OB_SUCCESS, ret);
|
|
has_remove = false;
|
|
ret = callback_list_.remove_callbacks_for_fast_commit();
|
|
EXPECT_EQ(OB_SUCCESS, ret);
|
|
EXPECT_TRUE(has_remove);
|
|
EXPECT_EQ(1, callback_list_.get_length());
|
|
EXPECT_EQ(callback_list_.log_cursor_, cb_3);
|
|
// . fill redo, check continous : [1,2](logged) -> [3](will do log)
|
|
scope.reset();
|
|
i = 0;
|
|
{
|
|
ObTxCallbackFunctorAdapter f([&](ObITransCallback*cb) -> int {
|
|
if (*scope.start_ == NULL) scope.start_ = cb;
|
|
scope.end_ = cb;
|
|
scope.host_ = &callback_list_;
|
|
scope.cnt_ = ++i;
|
|
return OB_SUCCESS;
|
|
});
|
|
ret = callback_list_.fill_log(callback_list_.log_cursor_, ctx, f);
|
|
}
|
|
EXPECT_EQ(ret, OB_SUCCESS);
|
|
EXPECT_FALSE(scope.is_empty());
|
|
EXPECT_EQ(*scope.start_, *scope.end_);
|
|
EXPECT_EQ(*scope.start_, cb_3);
|
|
// . log submitted, udpate log_cursor: because no other log need to be log,
|
|
// log cursor point to head
|
|
LOG_SUBMITED(scope);
|
|
EXPECT_EQ(ret, OB_SUCCESS);
|
|
EXPECT_EQ(callback_list_.log_cursor_, &callback_list_.head_);
|
|
// . log_sync fail, because failed callbacks are removed, log cursor is unchanged
|
|
int64_t removed_cnt = 0;
|
|
ret = callback_list_.sync_log_fail(scope, scn, removed_cnt);
|
|
EXPECT_EQ(ret, OB_SUCCESS);
|
|
EXPECT_EQ(1, removed_cnt);
|
|
EXPECT_EQ(callback_list_.log_cursor_, &callback_list_.head_);
|
|
// . fill log will return empty, because of no callbacks need to log
|
|
scope.reset();
|
|
i = 0;
|
|
{
|
|
ObTxCallbackFunctorAdapter f([&](ObITransCallback*cb) -> int {
|
|
if (*scope.start_ == NULL) scope.start_ = cb;
|
|
scope.end_ = cb;
|
|
scope.host_ = &callback_list_;
|
|
scope.cnt_ = ++i;
|
|
return OB_SUCCESS;
|
|
});
|
|
ret = callback_list_.fill_log(callback_list_.log_cursor_, ctx, f);
|
|
}
|
|
EXPECT_EQ(ret, OB_SUCCESS);
|
|
EXPECT_TRUE(scope.is_empty());
|
|
EXPECT_EQ(callback_list_.log_cursor_, &callback_list_.head_);
|
|
//
|
|
// TEST rollback savepoint affect log_cursor
|
|
//
|
|
// . do append 5 callback, logging out 2, log_cursor point to 3th, rollback savepoint to 1
|
|
// the log_cusor should be adjust to the 1st.next_
|
|
//
|
|
// insert callbacks: 4, 5, then flush the log, log cursor point to head
|
|
APPEND_CB(0, 4);
|
|
EXPECT_EQ(callback_list_.log_cursor_, cb_4);
|
|
auto sp0 = get_seq_no();
|
|
APPEND_CB(0, 5);
|
|
EXPECT_EQ(callback_list_.log_cursor_, cb_4);
|
|
scope.reset();
|
|
i = 0;
|
|
{
|
|
ObTxCallbackFunctorAdapter f([&](ObITransCallback*cb) -> int {
|
|
if (*scope.start_ == NULL) scope.start_ = cb;
|
|
scope.end_ = cb;
|
|
scope.host_ = &callback_list_;
|
|
scope.cnt_ = ++i;
|
|
return OB_SUCCESS;
|
|
});
|
|
ret = callback_list_.fill_log(callback_list_.log_cursor_, ctx, f);
|
|
}
|
|
EXPECT_EQ(ret, OB_SUCCESS);
|
|
EXPECT_EQ(callback_list_.log_cursor_, cb_4);
|
|
// . all log submitted, update log_cursor, point to head
|
|
LOG_SUBMITED(scope);
|
|
EXPECT_EQ(callback_list_.log_cursor_, &callback_list_.head_);
|
|
// append new callbacks 6,7, the log cursor should point to 6
|
|
auto sp1= get_seq_no(); sp1.set_branch(1);
|
|
APPEND_CB(1, 6);
|
|
EXPECT_EQ(callback_list_.log_cursor_, cb_6);
|
|
APPEND_CB(0, 7);
|
|
APPEND_CB(0, 8);
|
|
EXPECT_EQ(callback_list_.log_cursor_, cb_6);
|
|
// must logging synced or failed before do rollback
|
|
LOG_SYNC_SUCC(scope);
|
|
EXPECT_EQ(ret, OB_SUCCESS);
|
|
auto from = get_seq_no() + 1;
|
|
from.set_branch(sp1.get_branch());
|
|
ret = callback_list_.remove_callbacks_for_rollback_to(sp1, from, share::SCN::invalid_scn());
|
|
EXPECT_EQ(ret, OB_SUCCESS);
|
|
EXPECT_EQ(callback_list_.log_cursor_, cb_7);
|
|
from.set_branch(sp0.get_branch());
|
|
ret = callback_list_.remove_callbacks_for_rollback_to(sp0, from, share::SCN::invalid_scn());
|
|
EXPECT_EQ(ret, OB_SUCCESS);
|
|
EXPECT_EQ(callback_list_.log_cursor_, cb_4->next_);
|
|
#undef APPEND_CB
|
|
}
|
|
|
|
TEST_F(TestTxCallbackList, parallel_replay_and_replay_fail_parallel_start_pos) {
|
|
using ObTxSEQ = transaction::ObTxSEQ;
|
|
using SCN = share::SCN;
|
|
TRANS_LOG(INFO, "CASE: parallel replay and replay fail");
|
|
int ret = 0;
|
|
ObMemtable *mt = create_memtable();
|
|
ObMockTxCallback *cb1 = parallel_replay_callback(mt, ObTxSEQ(10, 1), 100, false/*serial_final*/);
|
|
ObMockTxCallback *cb2 = parallel_replay_callback(mt, ObTxSEQ(20, 1), 100, false/*serial_final*/);
|
|
ObMockTxCallback *cb3 = parallel_replay_callback(mt, ObTxSEQ(30, 1), 100, false/*serial_final*/);
|
|
ASSERT_EQ(callback_list_.parallel_start_pos_, cb1);
|
|
ObMockTxCallback *cb4 = serial_replay_callback(mt, ObTxSEQ(1, 1), 10, false/*serial_final*/);
|
|
ObMockTxCallback *cb5 = serial_replay_callback(mt, ObTxSEQ(2, 1), 10, false/*serial_final*/);
|
|
ObMockTxCallback *cb6 = serial_replay_callback(mt, ObTxSEQ(3, 1), 10, false/*serial_final*/);
|
|
// check chain: serial -> parallel_start_pos -> parallel
|
|
ASSERT_EQ(cb6->next_, callback_list_.parallel_start_pos_);
|
|
ASSERT_EQ(callback_list_.parallel_start_pos_, cb1);
|
|
// replay fail, parallel start_pos
|
|
SCN scn;
|
|
scn.convert_for_tx(100);
|
|
ASSERT_EQ(OB_SUCCESS, callback_list_.replay_fail(scn, false /*serial replay*/));
|
|
ASSERT_EQ(NULL, callback_list_.parallel_start_pos_);
|
|
ObMockTxCallback *cb7 = serial_replay_callback(mt, ObTxSEQ(4, 1), 11, false/*serial_final*/);
|
|
ObMockTxCallback *cb8 = serial_replay_callback(mt, ObTxSEQ(5, 1), 11, false/*serial_final*/);
|
|
ObMockTxCallback *cb9 = serial_replay_callback(mt, ObTxSEQ(6, 1), 11, false/*serial_final*/);
|
|
// check chain order on serial part: head -> cb4 -> cb5 -> cb6 -> cb7 -> cb8 -> cb9 -> head
|
|
ASSERT_EQ(cb6->next_, cb7);
|
|
ASSERT_EQ(cb9->next_, &callback_list_.head_);
|
|
ASSERT_EQ(cb4->prev_, &callback_list_.head_);
|
|
ASSERT_EQ(callback_list_.head_.next_, cb4);
|
|
ASSERT_EQ(callback_list_.head_.prev_, cb9);
|
|
}
|
|
|
|
} // namespace unittest
|
|
|
|
namespace memtable
|
|
{
|
|
void ObMemtableCtx::free_mvcc_row_callback(ObITransCallback *cb)
|
|
{
|
|
if (OB_ISNULL(cb)) {
|
|
TRANS_LOG_RET(ERROR, OB_ERR_UNEXPECTED, "cb is null, unexpected error", KP(cb), K(*this));
|
|
} else if (cb->is_table_lock_callback()) {
|
|
free_table_lock_callback(cb);
|
|
} else {
|
|
ATOMIC_INC(&callback_free_count_);
|
|
TRANS_LOG(DEBUG, "callback release succ", KP(cb), K(*this), K(lbt()));
|
|
ctx_cb_allocator_.free(cb);
|
|
cb = NULL;
|
|
}
|
|
}
|
|
|
|
int ObTxCallbackList::remove_callbacks_for_fast_commit(const share::SCN stop_scn)
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
LockGuard guard(*this, LOCK_MODE::LOCK_ITERATE);
|
|
const int64_t fast_commit_callback_count = unittest::TestTxCallbackList::fast_commit_reserve_cnt_;
|
|
const int64_t recommand_reserve_count = (fast_commit_callback_count + 1) / 2;
|
|
const int64_t need_remove_count = length_ - recommand_reserve_count;
|
|
|
|
ObRemoveCallbacksForFastCommitFunctor functor(need_remove_count, sync_scn_);
|
|
functor.set_checksumer(checksum_scn_, &batch_checksum_);
|
|
|
|
if (OB_FAIL(callback_(functor, guard.state_))) {
|
|
TRANS_LOG(ERROR, "remove callbacks for fast commit wont report error", K(ret), K(functor));
|
|
} else {
|
|
callback_mgr_.add_fast_commit_callback_remove_cnt(functor.get_remove_cnt());
|
|
ensure_checksum_(functor.get_checksum_last_scn());
|
|
unittest::has_remove = share::SCN::min_scn() != functor.get_checksum_last_scn();
|
|
if (unittest::has_remove) {
|
|
TRANS_LOG(INFO, "remove callbacks for fast commit", K(functor), K(*this));
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
int ObTxCallbackList::remove_callbacks_for_remove_memtable(
|
|
const memtable::ObMemtableSet *memtable_set,
|
|
const share::SCN)
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
LockGuard guard(*this, LOCK_MODE::LOCK_ITERATE);
|
|
|
|
struct Functor : public ObRemoveSyncCallbacksWCondFunctor {
|
|
Functor(const bool need_remove_data = true, const bool is_reverse = false)
|
|
: ObRemoveSyncCallbacksWCondFunctor(need_remove_data, is_reverse) {}
|
|
bool cond_for_remove(ObITransCallback *callback) {
|
|
int ret = OB_SUCCESS;
|
|
int bool_ret = true;
|
|
if (OB_HASH_EXIST == (ret = memtable_set_->exist_refactored((uint64_t)callback->get_memtable()))) {
|
|
bool_ret = true;
|
|
} else if (OB_HASH_NOT_EXIST == ret) {
|
|
bool_ret = false;
|
|
} else {
|
|
// We have no idea to handle the error
|
|
ob_abort();
|
|
}
|
|
return bool_ret;
|
|
}
|
|
bool cond_for_stop(ObITransCallback *) const {
|
|
return false;
|
|
}
|
|
const memtable::ObMemtableSet *memtable_set_;
|
|
} functor(false /*need_remove_data*/);
|
|
functor.memtable_set_ = memtable_set;
|
|
functor.set_checksumer(checksum_scn_, &batch_checksum_);
|
|
|
|
if (OB_FAIL(callback_(functor, guard.state_))) {
|
|
TRANS_LOG(ERROR, "remove callbacks for remove memtable wont report error", K(ret), K(functor));
|
|
} else {
|
|
callback_mgr_.add_release_memtable_callback_remove_cnt(functor.get_remove_cnt());
|
|
ensure_checksum_(functor.get_checksum_last_scn());
|
|
if (functor.get_remove_cnt() > 0) {
|
|
TRANS_LOG(INFO, "remove callbacks for remove memtable", KP(memtable_set),
|
|
K(functor), K(*this));
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
} // namespace memtable
|
|
|
|
} // namespace oceanbase
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
system("rm -rf test_tx_callback_list.log*");
|
|
oceanbase::common::ObLogger::get_logger().set_file_name("test_tx_callback_list.log", true);
|
|
oceanbase::common::ObLogger::get_logger().set_log_level("INFO");
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|