trivial: hack around IDE bug

CLion currently has difficalty indexing `#[cfg(not(test))]`, and this gets around that Closes: aptos-labs#814
n0r3k · May 5, 2022 · 28a8050 · 28a8050
1 parent 193b7cf
commit 28a8050
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 8 deletions.
diff --git a/storage/aptosdb/src/lib.rs b/storage/aptosdb/src/lib.rs
@@ -187,10 +187,7 @@ impl RocksdbPropertyReporter {
                 );
             }
             // report rocksdb properties each 10 seconds
-            #[cfg(not(test))]
-            const TIMEOUT_MS: u64 = 10000;
-            #[cfg(test)]
-            const TIMEOUT_MS: u64 = 10;
+            const TIMEOUT_MS: u64 = if cfg!(test) { 10 } else { 10000 };
 
             match recv.recv_timeout(Duration::from_millis(TIMEOUT_MS)) {
                 Ok(_) => break,

diff --git a/storage/backup/backup-cli/src/backup_types/transaction/restore.rs b/storage/backup/backup-cli/src/backup_types/transaction/restore.rs
@@ -44,10 +44,7 @@ use storage_interface::DbReaderWriter;
 use structopt::StructOpt;
 use tokio::io::BufReader;
 
-#[cfg(not(test))]
-const BATCH_SIZE: usize = 10000;
-#[cfg(test)]
-const BATCH_SIZE: usize = 2;
+const BATCH_SIZE: usize = if cfg!(test) { 2 } else { 10000 };
 
 #[derive(StructOpt)]
 pub struct TransactionRestoreOpt {