nit

lsm1 · Dec 21, 2023 · 4219d28 · 4219d28
1 parent 31d4fc1
commit 4219d28
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/...sql-engine/src/main/scala/org/apache/kyuubi/engine/spark/operation/ExecuteStatement.scala b/...sql-engine/src/main/scala/org/apache/kyuubi/engine/spark/operation/ExecuteStatement.scala
@@ -183,18 +183,14 @@ class ExecuteStatement(
         // Rename all col name to avoid duplicate columns
         val colName = range(0, result.schema.size).map(x => "col" + x)
 
-        val compress = if (SPARK_ENGINE_RUNTIME_VERSION >= "3.2") {
-          "zstd"
-        } else {
-          "zlib"
-        }
+        val codec = if (SPARK_ENGINE_RUNTIME_VERSION >= "3.2") "zstd" else "zlib"
         // df.write will introduce an extra shuffle for the outermost limit, and hurt performance
         if (resultMaxRows > 0) {
           result.toDF(colName: _*).limit(resultMaxRows).write
-            .option("compression", compress).format("orc").save(saveFileName.get)
+            .option("compression", codec).format("orc").save(saveFileName.get)
         } else {
           result.toDF(colName: _*).write
-            .option("compression", compress).format("orc").save(saveFileName.get)
+            .option("compression", codec).format("orc").save(saveFileName.get)
         }
         info(s"Save result to $saveFileName")
         fetchOrcStatement = Some(new FetchOrcStatement(spark))