github-actions[bot] commented on code in PR #45206:
URL: https://github.com/apache/doris/pull/45206#discussion_r1877434578
##########
cloud/test/meta_service_test.cpp:
##########
@@ -4530,6 +4579,223 @@ TEST(MetaServiceTest, GetDeleteBitmapUpdateLock) {
ASSERT_EQ(res.status().code(), MetaServiceCode::OK);
}
+TEST(MetaServiceTest, GetDeleteBitmapUpdateLockTabletStatsNormal) {
+ auto meta_service = get_meta_service();
+
+ std::string instance_id = "test_get_delete_bitmap_update_lock_normal";
+ [[maybe_unused]] auto* sp = SyncPoint::get_instance();
+ std::unique_ptr<int, std::function<void(int*)>> defer((int*)0x01, [](int*)
{
+ SyncPoint::get_instance()->disable_processing();
+ SyncPoint::get_instance()->clear_all_call_backs();
+ });
+ sp->set_call_back("get_instance_id", [&](auto&& args) {
+ auto* ret = try_any_cast_ret<std::string>(args);
+ ret->first = instance_id;
+ ret->second = true;
+ });
+ sp->enable_processing();
+
+ // store tablet stats
+ int64_t db_id = 1000;
+ int64_t table_id = 2001;
+ int64_t index_id = 3001;
+ // [(partition_id, tablet_id)]
+ std::vector<std::array<int64_t, 2>> tablet_idxes {{70001, 12345}, {80001,
3456}, {90001, 6789}};
+
+ add_tablet_stats(meta_service.get(), instance_id, table_id, index_id,
tablet_idxes);
+
+ GetDeleteBitmapUpdateLockResponse res;
+ get_delete_bitmap_update_lock(meta_service.get(), res, db_id, table_id,
index_id, tablet_idxes,
+ 5, 999999, -1, true);
+ ASSERT_EQ(res.status().code(), MetaServiceCode::OK);
+
+ ASSERT_EQ(res.base_compaction_cnts().size(), tablet_idxes.size());
+ for (const auto& base_compaction_cnt : res.base_compaction_cnts()) {
+ ASSERT_EQ(base_compaction_cnt, 10);
+ }
+ ASSERT_EQ(res.cumulative_compaction_cnts().size(), tablet_idxes.size());
+ for (const auto& cumu_compaction_cnt : res.cumulative_compaction_cnts()) {
+ ASSERT_EQ(cumu_compaction_cnt, 20);
+ }
+ ASSERT_EQ(res.cumulative_points().size(), tablet_idxes.size());
+ for (const auto& cumulative_point : res.cumulative_points()) {
+ ASSERT_EQ(cumulative_point, 30);
+ }
+}
+
+TEST(MetaServiceTest, GetDeleteBitmapUpdateLockTabletStatsLockExpired) {
+ auto meta_service = get_meta_service();
+
+ {
+ // 2.1 abnormal path, lock has been expired and taken by another
load/compaction during
+ // the reading of tablet stats
+ std::string instance_id =
"test_get_delete_bitmap_update_lock_abnormal1";
+ [[maybe_unused]] auto* sp = SyncPoint::get_instance();
+ std::unique_ptr<int, std::function<void(int*)>> defer((int*)0x01,
[](int*) {
+ SyncPoint::get_instance()->disable_processing();
+ SyncPoint::get_instance()->clear_all_call_backs();
+ });
+ sp->set_call_back("get_instance_id", [&](auto&& args) {
+ auto* ret = try_any_cast_ret<std::string>(args);
+ ret->first = instance_id;
+ ret->second = true;
+ });
+ sp->set_call_back("check_delete_bitmap_lock.set_lock_info", [&](auto&&
args) {
+ auto* lock_info = try_any_cast<DeleteBitmapUpdateLockPB*>(args[0]);
+ // simulate that lock_id has been modified by another load
+ lock_info->set_lock_id(345);
+ LOG(INFO) << "change lock_info.lock_id to 345, lock_info=" <<
lock_info->DebugString();
+ });
+
+ sp->enable_processing();
+
+ // store tablet stats
+ int64_t db_id = 1000;
+ int64_t table_id = 2001;
+ int64_t index_id = 3001;
+ // [(partition_id, tablet_id)]
+ std::vector<std::array<int64_t, 2>> tablet_idxes {
+ {70001, 12345}, {80001, 3456}, {90001, 6789}};
+
+ add_tablet_stats(meta_service.get(), instance_id, table_id, index_id,
tablet_idxes);
+
+ GetDeleteBitmapUpdateLockResponse res;
+ get_delete_bitmap_update_lock(meta_service.get(), res, db_id,
table_id, index_id,
+ tablet_idxes, 5, 999999, -1, true);
+ ASSERT_EQ(res.status().code(), MetaServiceCode::LOCK_EXPIRED);
+ }
+
+ {
+ // 2.2 abnormal path, lock has been taken by another load/compaction
and been released during
+ // the reading of tablet stats
+ std::string instance_id =
"test_get_delete_bitmap_update_lock_abnormal2";
+ [[maybe_unused]] auto* sp = SyncPoint::get_instance();
+ std::unique_ptr<int, std::function<void(int*)>> defer((int*)0x01,
[](int*) {
+ SyncPoint::get_instance()->disable_processing();
+ SyncPoint::get_instance()->clear_all_call_backs();
+ });
+ sp->set_call_back("get_instance_id", [&](auto&& args) {
+ auto* ret = try_any_cast_ret<std::string>(args);
+ ret->first = instance_id;
+ ret->second = true;
+ });
+ sp->set_call_back("check_delete_bitmap_lock.inject_get_lock_key_err",
[&](auto&& args) {
+ auto* err = try_any_cast<TxnErrorCode*>(args[0]);
+ // the lock has been taken by another load and been released,
+ // so the delete bitmap update lock KV will be removed
+ *err = TxnErrorCode::TXN_KEY_NOT_FOUND;
+ });
+
+ sp->enable_processing();
+
+ // store tablet stats
+ int64_t db_id = 1000;
+ int64_t table_id = 2001;
+ int64_t index_id = 3001;
+ // [(partition_id, tablet_id)]
+ std::vector<std::array<int64_t, 2>> tablet_idxes {
+ {70001, 12345}, {80001, 3456}, {90001, 6789}};
+
+ add_tablet_stats(meta_service.get(), instance_id, table_id, index_id,
tablet_idxes);
+
+ GetDeleteBitmapUpdateLockResponse res;
+ get_delete_bitmap_update_lock(meta_service.get(), res, db_id,
table_id, index_id,
+ tablet_idxes, 5, 999999, -1, true);
+ ASSERT_EQ(res.status().code(), MetaServiceCode::LOCK_EXPIRED);
+ }
+}
+
+TEST(MetaServiceTest, GetDeleteBitmapUpdateLockTabletStatsError) {
Review Comment:
warning: function 'TEST' exceeds recommended size/complexity thresholds
[readability-function-size]
```cpp
TEST(MetaServiceTest, GetDeleteBitmapUpdateLockTabletStatsError) {
^
```
<details>
<summary>Additional context</summary>
**cloud/test/meta_service_test.cpp:4707:** 89 lines including whitespace and
comments (threshold 80)
```cpp
TEST(MetaServiceTest, GetDeleteBitmapUpdateLockTabletStatsError) {
^
```
</details>
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: commits-unsubscr...@doris.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscr...@doris.apache.org
For additional commands, e-mail: commits-h...@doris.apache.org
