@@ -47,7 +47,7 @@ class Conf(arguments: Seq[String]) extends ScallopConf(arguments) {
4747 addSubcommand(generateData)
4848 object generateQueries extends Subcommand (" queries" ) {
4949 val numFiles : ScallopOption [Int ] =
50- opt[Int ](required = false , descr = " Number of input files to use" )
50+ opt[Int ](required = true , descr = " Number of input files to use" )
5151 val numQueries : ScallopOption [Int ] =
5252 opt[Int ](required = true , descr = " Number of queries to generate" )
5353 val randomSeed : ScallopOption [Long ] =
@@ -58,7 +58,9 @@ class Conf(arguments: Seq[String]) extends ScallopConf(arguments) {
5858 val filename : ScallopOption [String ] =
5959 opt[String ](required = true , descr = " File to write queries to" )
6060 val numFiles : ScallopOption [Int ] =
61- opt[Int ](required = false , descr = " Number of input files to use" )
61+ opt[Int ](required = true , descr = " Number of input files to use" )
62+ val showFailedSparkQueries : ScallopOption [Boolean ] =
63+ opt[Boolean ](required = false , descr = " Whether to show failed Spark queries" )
6264 }
6365 addSubcommand(runQueries)
6466 verify()
@@ -107,7 +109,11 @@ object Main {
107109 numFiles = conf.generateQueries.numFiles(),
108110 conf.generateQueries.numQueries())
109111 case Some (conf.runQueries) =>
110- QueryRunner .runQueries(spark, conf.runQueries.numFiles(), conf.runQueries.filename())
112+ QueryRunner .runQueries(
113+ spark,
114+ conf.runQueries.numFiles(),
115+ conf.runQueries.filename(),
116+ conf.runQueries.showFailedSparkQueries())
111117 case _ =>
112118 // scalastyle:off println
113119 println(" Invalid subcommand" )
0 commit comments