support skip dir

Author: xudong963

tests/logictest/suites/query/join.test

@@ -307,9 +307,15 @@ select t.number from numbers(10000) as t inner join numbers(1000) as t1 on t.num
 statement ok
 drop table t
 
+statement ok
+drop table if exists t3
+
 statement ok
 CREATE TABLE t3(c0 BIGINT NULL, c1 DOUBLE NULL)
 
+statement ok
+drop table if exists t4
+
 statement ok
 CREATE TABLE t4(c0 FLOAT NULL)
 

tests/sqllogictests/src/arg.rs

@@ -20,16 +20,24 @@ pub struct SqlLogicTestArgs {
     // Set specific dir to run
     #[arg(
         short = 'd',
-        long = "dir",
+        long = "run_dir",
         help = "Run sqllogictests in specific directory, the arg is optional"
     )]
     pub dir: Option<String>,
 
     // Set specific test file to run
     #[arg(
         short = 'f',
-        long = "file",
+        long = "run_file",
         help = "Run sqllogictests in specific test file, the arg is optional"
     )]
     pub file: Option<String>,
+
+    // Set specific dir to skip
+    #[arg(
+        short = 's',
+        long = "skip_dir",
+        help = "Skip sqllogictests in specific directory, the arg is optional"
+    )]
+    pub skipped_dir: Option<String>,
 }

tests/sqllogictests/src/main.rs

@@ -163,7 +163,6 @@ async fn run_suits(suits: ReadDir, databend: Databend) -> Result<()> {
                 .unwrap()
                 .to_str()
                 .unwrap();
-            dbg!(file_name);
             if let Some(ref specific_file) = args.file {
                 if file_name != specific_file {
                     continue;
@@ -185,6 +184,14 @@ fn get_files(suit: PathBuf) -> Result<Vec<walkdir::Result<DirEntry>>> {
         .max_depth(100)
         .sort_by(|a, b| a.file_name().cmp(b.file_name()))
         .into_iter()
+        .filter_entry(|e| {
+            if let Some(skipped_dir) = &args.skipped_dir {
+                if e.file_name().to_str().unwrap() == skipped_dir {
+                    return false;
+                }
+            }
+            true
+        })
         .filter(|e| {
             if args.dir.is_none() {
                 return true;