 e0b1dda713
			
		
	
	e0b1dda713
	
	
	
		
			
			Co-authored-by: rolandqi <qikai456@126.com> Co-authored-by: yangzhifeng <yangzhifeng83@gmail.com>
		
			
				
	
	
		
			215 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			215 lines
		
	
	
		
			11 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
 | |
| #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();
 | |
| }
 |