hcq/oceanbase
hcq/oceanbase
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
oceanbase/deps/oblib/unittest/lib/stat/test_diagnose_info.cpp
AntiTopQuark e0b1dda713 [FEAT MERGE] Phase one of 3A project 
Co-authored-by: rolandqi <qikai456@126.com>
Co-authored-by: yangzhifeng <yangzhifeng83@gmail.com>
2023-08-25 08:44:14 +00:00

215 lines
11 KiB
C++
Raw Blame History

/**
* 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
#include "lib/stat/ob_diagnose_info.h"
#include "lib/stat/ob_session_stat.h"
#include "lib/lock/ob_spin_rwlock.h"
#include "lib/coro/testing.h"
namespace oceanbase
{
namespace common
{
#define TENANT_EVENT_GET(stat_no) \
({ \
int64_t ret = 0; \
oceanbase::common::ObDiagnoseTenantInfo *tenant_info \
= oceanbase::common::ObDiagnoseTenantInfo::get_local_diagnose_info(); \
if (NULL != tenant_info) { \
if (oceanbase::common::stat_no < oceanbase::common::ObStatEventIds::STAT_EVENT_ADD_END) { \
oceanbase::common::ObStatEventAddStat *stat \
= tenant_info->get_add_stat_stats().get( \
::oceanbase::common::stat_no); \
if (NULL != stat) { \
ret = stat->get_stat_value(); \
} \
} \
} \
ret; \
})
TEST(ObDiagnoseSessionInfo, guard)
{
EVENT_INC(SYS_TIME_MODEL_ENCRYPT_CPU);
EXPECT_EQ(1, GET_TSI(ObSessionDIBuffer)->get_tenant_id());
EXPECT_EQ(1, TENANT_EVENT_GET(ObStatEventIds::SYS_TIME_MODEL_ENCRYPT_CPU));
{
ObTenantStatEstGuard tenant_guard(2);
EVENT_INC(SYS_TIME_MODEL_ENCRYPT_CPU);
EXPECT_EQ(2, GET_TSI(ObSessionDIBuffer)->get_tenant_id());
EXPECT_EQ(1, TENANT_EVENT_GET(ObStatEventIds::SYS_TIME_MODEL_ENCRYPT_CPU));
{
ObTenantStatEstGuard tenant_guard(3);
EVENT_INC(SYS_TIME_MODEL_ENCRYPT_CPU);
EXPECT_EQ(3, GET_TSI(ObSessionDIBuffer)->get_tenant_id());
EXPECT_EQ(1, TENANT_EVENT_GET(ObStatEventIds::SYS_TIME_MODEL_ENCRYPT_CPU));
}
EXPECT_EQ(2, GET_TSI(ObSessionDIBuffer)->get_tenant_id());
}
EXPECT_EQ(1, GET_TSI(ObSessionDIBuffer)->get_tenant_id());
}
TEST(ObDISessionCache, multithread)
{
ObDiagnoseSessionInfo info;
ObDiagnoseTenantInfo tenant_info;
ObWaitEventHistory &history = info.get_event_history();
for (uint64_t i = 0; i <2; i++) {
info.notify_wait_begin(ObWaitEventIds::DEFAULT_COND_WAIT, 0, 0, 0, 0);
EXPECT_EQ(1, history.curr_pos_);
EXPECT_EQ(1, history.item_cnt_);
EXPECT_EQ(1, history.nest_cnt_);
EXPECT_EQ(ObWaitEventIds::DEFAULT_COND_WAIT, history.items_[(history.current_wait_ + SESSION_WAIT_HISTORY_CNT) % SESSION_WAIT_HISTORY_CNT].event_no_);
EXPECT_EQ(0, history.items_[(history.current_wait_ + SESSION_WAIT_HISTORY_CNT) % SESSION_WAIT_HISTORY_CNT].level_);
::usleep(1);
info.notify_wait_end(&tenant_info);
EXPECT_EQ(1, history.curr_pos_);
EXPECT_EQ(1, history.item_cnt_);
EXPECT_EQ(0, history.nest_cnt_);
EXPECT_EQ(ObWaitEventIds::DEFAULT_COND_WAIT, history.items_[(history.current_wait_ + SESSION_WAIT_HISTORY_CNT) % SESSION_WAIT_HISTORY_CNT].event_no_);
EXPECT_EQ(0, history.items_[(history.current_wait_ + SESSION_WAIT_HISTORY_CNT) % SESSION_WAIT_HISTORY_CNT].level_);
info.notify_wait_begin(ObWaitEventIds::MT_READ_LOCK_WAIT, 0, 0, 0, 0);
EXPECT_EQ(2, history.curr_pos_);
EXPECT_EQ(2, history.item_cnt_);
EXPECT_EQ(1, history.nest_cnt_);
EXPECT_EQ(ObWaitEventIds::MT_READ_LOCK_WAIT, history.items_[(history.current_wait_ + SESSION_WAIT_HISTORY_CNT) % SESSION_WAIT_HISTORY_CNT].event_no_);
EXPECT_EQ(0, history.items_[(history.current_wait_ + SESSION_WAIT_HISTORY_CNT) % SESSION_WAIT_HISTORY_CNT].level_);
info.notify_wait_begin(ObWaitEventIds::MT_READ_LOCK_WAIT, 0, 0, 0, 0);
EXPECT_EQ(3, history.curr_pos_);
EXPECT_EQ(3, history.item_cnt_);
EXPECT_EQ(2, history.nest_cnt_);
EXPECT_EQ(ObWaitEventIds::MT_READ_LOCK_WAIT, history.items_[(history.current_wait_ + SESSION_WAIT_HISTORY_CNT) % SESSION_WAIT_HISTORY_CNT].event_no_);
EXPECT_EQ(1, history.items_[(history.current_wait_ + SESSION_WAIT_HISTORY_CNT) % SESSION_WAIT_HISTORY_CNT].level_);
info.notify_wait_end(&tenant_info);
EXPECT_EQ(3, history.curr_pos_);
EXPECT_EQ(3, history.item_cnt_);
EXPECT_EQ(2, history.nest_cnt_);
EXPECT_EQ(ObWaitEventIds::MT_READ_LOCK_WAIT, history.items_[(history.current_wait_ + SESSION_WAIT_HISTORY_CNT) % SESSION_WAIT_HISTORY_CNT].event_no_);
EXPECT_EQ(0, history.items_[(history.current_wait_ + SESSION_WAIT_HISTORY_CNT) % SESSION_WAIT_HISTORY_CNT].level_);
info.notify_wait_begin(ObWaitEventIds::MT_READ_LOCK_WAIT, 0, 0, 0, 0);
EXPECT_EQ(4, history.curr_pos_);
EXPECT_EQ(4, history.item_cnt_);
EXPECT_EQ(3, history.nest_cnt_);
EXPECT_EQ(ObWaitEventIds::MT_READ_LOCK_WAIT, history.items_[(history.current_wait_ + SESSION_WAIT_HISTORY_CNT) % SESSION_WAIT_HISTORY_CNT].event_no_);
EXPECT_EQ(1, history.items_[(history.current_wait_ + SESSION_WAIT_HISTORY_CNT) % SESSION_WAIT_HISTORY_CNT].level_);
info.notify_wait_end(&tenant_info);
info.notify_wait_begin(ObWaitEventIds::MT_READ_LOCK_WAIT, 0, 0, 0, 0);
info.notify_wait_end(&tenant_info);
info.notify_wait_begin(ObWaitEventIds::MT_READ_LOCK_WAIT, 0, 0, 0, 0);
info.notify_wait_begin(ObWaitEventIds::MT_READ_LOCK_WAIT, 0, 0, 0, 0);
EXPECT_EQ(7, history.curr_pos_);
EXPECT_EQ(7, history.item_cnt_);
EXPECT_EQ(6, history.nest_cnt_);
EXPECT_EQ(ObWaitEventIds::MT_READ_LOCK_WAIT, history.items_[(history.current_wait_ + SESSION_WAIT_HISTORY_CNT) % SESSION_WAIT_HISTORY_CNT].event_no_);
EXPECT_EQ(2, history.items_[(history.current_wait_ + SESSION_WAIT_HISTORY_CNT) % SESSION_WAIT_HISTORY_CNT].level_);
info.notify_wait_begin(ObWaitEventIds::MT_READ_LOCK_WAIT, 0, 0, 0, 0);
EXPECT_EQ(8, history.curr_pos_);
EXPECT_EQ(8, history.item_cnt_);
EXPECT_EQ(7, history.nest_cnt_);
EXPECT_EQ(ObWaitEventIds::MT_READ_LOCK_WAIT, history.items_[(history.current_wait_ + SESSION_WAIT_HISTORY_CNT) % SESSION_WAIT_HISTORY_CNT].event_no_);
EXPECT_EQ(3, history.items_[(history.current_wait_ + SESSION_WAIT_HISTORY_CNT) % SESSION_WAIT_HISTORY_CNT].level_);
info.notify_wait_end(&tenant_info);
info.notify_wait_end(&tenant_info);
EXPECT_EQ(8, history.curr_pos_);
EXPECT_EQ(8, history.item_cnt_);
EXPECT_EQ(7, history.nest_cnt_);
EXPECT_EQ(ObWaitEventIds::MT_READ_LOCK_WAIT, history.items_[(history.current_wait_ + SESSION_WAIT_HISTORY_CNT) % SESSION_WAIT_HISTORY_CNT].event_no_);
EXPECT_EQ(1, history.items_[(history.current_wait_ + SESSION_WAIT_HISTORY_CNT) % SESSION_WAIT_HISTORY_CNT].level_);
info.notify_wait_end(&tenant_info);
if (1 == i) {
info.notify_wait_begin(ObWaitEventIds::DEFAULT_COND_WAIT, 0, 0, 0, 0);
::usleep(1);
info.notify_wait_end(&tenant_info);
info.notify_wait_begin(ObWaitEventIds::DEFAULT_COND_WAIT, 0, 0, 0, 0);
::usleep(1);
info.notify_wait_end(&tenant_info);
info.notify_wait_end(&tenant_info);
EXPECT_EQ(2, history.curr_pos_);
EXPECT_EQ(2, history.item_cnt_);
EXPECT_EQ(0, history.nest_cnt_);
EXPECT_EQ(ObWaitEventIds::MT_READ_LOCK_WAIT, history.items_[(history.curr_pos_ - 1 + SESSION_WAIT_HISTORY_CNT) % SESSION_WAIT_HISTORY_CNT].event_no_);
EXPECT_EQ(0, history.items_[(history.curr_pos_ - 1 + SESSION_WAIT_HISTORY_CNT) % SESSION_WAIT_HISTORY_CNT].level_);
EXPECT_EQ(history.items_[8].wait_begin_time_, history.items_[(history.curr_pos_ - 1 + SESSION_WAIT_HISTORY_CNT) % SESSION_WAIT_HISTORY_CNT].wait_begin_time_);
EXPECT_EQ(history.items_[9].wait_end_time_, history.items_[(history.curr_pos_ - 1 + SESSION_WAIT_HISTORY_CNT) % SESSION_WAIT_HISTORY_CNT].wait_end_time_);
} else {
info.notify_wait_end(&tenant_info);
EXPECT_EQ(1, history.curr_pos_);
EXPECT_EQ(1, history.item_cnt_);
EXPECT_EQ(0, history.nest_cnt_);
EXPECT_EQ(ObWaitEventIds::DEFAULT_COND_WAIT, history.items_[(history.curr_pos_ - 1 + SESSION_WAIT_HISTORY_CNT) % SESSION_WAIT_HISTORY_CNT].event_no_);
EXPECT_EQ(0, history.items_[(history.curr_pos_ - 1 + SESSION_WAIT_HISTORY_CNT) % SESSION_WAIT_HISTORY_CNT].level_);
}
info.reset();
}
}
TEST(ObDiagnoseSessionInfo, normal)
{
ObWaitEventDesc max_wait;
ObWaitEventStat total_wait;
ObWaitEventDesc max_wait1;
ObWaitEventStat total_wait1;
{
ObSessionStatEstGuard session_guard(1, 1);
{
ObMaxWaitGuard max_guard(&max_wait);
ObTotalWaitGuard total_guard(&total_wait);
for (int i = 0; i < 30; i++) {
{
ObWaitEventGuard wait_guard(i);
::usleep(i);
}
ObMaxWaitGuard max_guard1(&max_wait1);
ObTotalWaitGuard total_guard1(&total_wait1);
{
ObWaitEventGuard wait_guard1(29-i);
::usleep(i*200);
}
}
ObDiagnoseSessionInfo *info = ObDiagnoseSessionInfo::get_local_diagnose_info();
EXPECT_EQ(1, info->get_tenant_id());
ObWaitEventDesc *last_wait = NULL;
info->get_event_history().get_last_wait(last_wait);
EXPECT_EQ(0, last_wait->event_no_);
EXPECT_EQ(0, info->get_curr_wait().event_no_);
//EXPECT_EQ(0, max_wait.event_no_);
EXPECT_EQ(54, total_wait.total_waits_);
uint64_t begin_time = ::oceanbase::common::ObTimeUtility::current_time();
for (int i = 0; i < 1000000; i++) {
ObWaitEventGuard wait_guard(ObWaitEventIds::MT_READ_LOCK_WAIT);
ObWaitEventGuard wait_guard1(ObWaitEventIds::MT_READ_LOCK_WAIT);
}
uint64_t end_time = ::oceanbase::common::ObTimeUtility::current_time();
printf("scan sysstat time: %ld us\n", (end_time - begin_time));
begin_time = ::oceanbase::common::ObTimeUtility::current_time();
for (int i = 0; i < 1000000; i++) {
ObWaitEventGuard wait_guard(ObWaitEventIds::MT_READ_LOCK_WAIT);
ObWaitEventGuard wait_guard1(ObWaitEventIds::DEFAULT_COND_WAIT);
}
end_time = ::oceanbase::common::ObTimeUtility::current_time();
printf("scan sysstat time: %ld us\n", (end_time - begin_time));
}
}
}
}
}
int main(int argc, char **argv)
{
oceanbase::common::ObLogger::get_logger().set_log_level("INFO");
testing::InitGoogleTest(&argc, argv);
return RUN_ALL_TESTS();
}
Reference in New Issue View Git Blame Copy Permalink