test: add time limit for splitting region

Signed-off-by: ekexium <[email protected]>

Author: ekexium

config/tikv.toml

@@ -1,6 +1,10 @@
 [coprocessor]
 region-max-keys = 5
 region-split-keys = 3
+batch-split-limit = 100
 
 [raftstore]
 region-split-check-diff = "0KB"
+pd-heartbeat-tick-interval = "2s"
+pd-store-heartbeat-tick-interval = "5s"
+split-region-check-tick-interval = "1s"

tests/common/mod.rs

@@ -1,12 +1,13 @@
 mod ctl;
 
 use futures_timer::Delay;
-use log::info;
+use log::{info, warn};
 use std::{env, time::Duration};
 use tikv_client::{ColumnFamily, Key, RawClient, Result, TransactionClient};
 
 const ENV_PD_ADDRS: &str = "PD_ADDRS";
 const ENV_ENABLE_MULIT_REGION: &str = "MULTI_REGION";
+const REGION_SPLIT_TIME_LIMIT: Duration = Duration::from_secs(15);
 
 // Delete all entries in TiKV to leave a clean space for following tests.
 pub async fn clear_tikv() {
@@ -24,6 +25,11 @@ pub async fn clear_tikv() {
 // To test with multiple regions, prewrite some data. Tests that hope to test
 // with multiple regions should use keys in the corresponding ranges.
 pub async fn init() -> Result<()> {
+    // ignore SetLoggerError
+    let _ = simple_logger::SimpleLogger::new()
+        .with_level(log::LevelFilter::Warn)
+        .init();
+
     if enable_multi_region() {
         // 1000 keys: 0..1000
         let keys_1 = std::iter::successors(Some(0u32), |x| Some(x + 1))
@@ -36,14 +42,14 @@ pub async fn init() -> Result<()> {
             .take(count as usize - 1)
             .map(|x| x.to_be_bytes().to_vec());
 
-        ensure_region_splitted(keys_1.chain(keys_2), 100).await?;
+        ensure_region_split(keys_1.chain(keys_2), 100).await?;
     }
 
     clear_tikv().await;
     Ok(())
 }
 
-async fn ensure_region_splitted(
+async fn ensure_region_split(
     keys: impl IntoIterator<Item = impl Into<Key>>,
     region_count: u32,
 ) -> Result<()> {
@@ -52,7 +58,7 @@ async fn ensure_region_splitted(
     }
 
     // 1. write plenty transactional keys
-    // 2. wait until regions splitted
+    // 2. wait until regions split
 
     let client = TransactionClient::new(pd_addrs()).await?;
     let mut txn = client.begin_optimistic().await?;
@@ -65,11 +71,16 @@ async fn ensure_region_splitted(
     txn.commit().await?;
 
     info!("splitting regions...");
+    let start_time = std::time::Instant::now();
     loop {
         if ctl::get_region_count().await? as u32 >= region_count {
             break;
         }
-        Delay::new(Duration::from_secs(1)).await;
+        if start_time.elapsed() > REGION_SPLIT_TIME_LIMIT {
+            warn!("Stop splitting regions: time limit exceeded");
+            break;
+        }
+        Delay::new(Duration::from_millis(200)).await;
     }
 
     Ok(())