Update dev/kyuubi-tpcds/src/main/scala/org/apache/kyuubi/tpcds/benchm…

…ark/RunBenchmark.scala Co-authored-by: Cheng Pan <pan3793@gmail.com>
apache · Sep 27, 2022 · 11a5b5c · 11a5b5c
1 parent 780410a
commit 11a5b5c
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/dev/kyuubi-tpcds/src/main/scala/org/apache/kyuubi/tpcds/benchmark/RunBenchmark.scala b/dev/kyuubi-tpcds/src/main/scala/org/apache/kyuubi/tpcds/benchmark/RunBenchmark.scala
@@ -67,14 +67,9 @@ object RunBenchmark {
         .action((x, c) => c.copy(resultsDir = x))
         .text("dir to store benchmark results, e.g. hdfs://hdfs-nn:9870/pref")
       opt[String]('q', "queries")
-        .action((x, c) =>
-          c.copy(queries = {
-            if (x.nonEmpty || !x.isBlank) {
-              x.split(",").filter(_.nonEmpty).toSet
-            } else {
-              Set.empty[String]
-            }
-          }))
+        .action { case (x, c) =>
+          c.copy(queries = x.split(",").map(_.trim).filter(_.nonEmpty).toSet)
+        }
         .text("name of the queries to run, use , split multiple name")
       help("help")
         .text("prints this usage text")