Replace code by fragment

common/utils/src/main/java/org/apache/spark/QueryContext.java

@@ -49,9 +49,6 @@ public interface QueryContext {
     // The corresponding fragment of the query which throws the exception.
     String fragment();
 
-    // The Spark code (API) that caused throwing the exception.
-    String code();
-
     // The user code (call site of the API) that caused throwing the exception.
     String callSite();
 

common/utils/src/main/scala/org/apache/spark/SparkThrowableHelper.scala

@@ -124,7 +124,7 @@ private[spark] object SparkThrowableHelper {
                   if (stopIndex > 0) g.writeNumberField("stopIndex", stopIndex)
                   g.writeStringField("fragment", c.fragment())
                 case QueryContextType.Dataset =>
-                  g.writeStringField("code", c.code())
+                  g.writeStringField("fragment", c.fragment())
                   g.writeStringField("callSite", c.callSite())
               }
               g.writeEndObject()

connector/connect/common/src/main/scala/org/apache/spark/sql/connect/client/GrpcExceptionConverter.scala

@@ -334,7 +334,6 @@ private[client] object GrpcExceptionConverter {
         override def stopIndex(): Int = queryCtx.getStopIndex
         override def fragment(): String = queryCtx.getFragment
         override def callSite(): String = queryCtx.getCallSite
-        override def code(): String = queryCtx.getCode
         override def summary(): String = queryCtx.getSummary
       }
     }.toArray

core/src/test/scala/org/apache/spark/SparkFunSuite.scala

@@ -378,8 +378,8 @@ abstract class SparkFunSuite
         assert(actual.fragment() === expected.fragment,
           "Invalid fragment of a query context. Actual:" + actual.toString)
       } else if (actual.contextType() == QueryContextType.Dataset) {
-        assert(actual.code() === expected.code,
-          "Invalid code of a query context. Actual:" + actual.toString)
+        assert(actual.fragment() === expected.fragment,
+          "Invalid code fragment of a query context. Actual:" + actual.toString)
         assert(actual.callSite().matches(expected.callSitePattern),
           "Invalid callSite of a query context. Actual:" + actual.toString)
       }

core/src/test/scala/org/apache/spark/SparkThrowableSuite.scala

@@ -509,7 +509,6 @@ class SparkThrowableSuite extends SparkFunSuite {
       override val startIndex = 2
       override val stopIndex = -1
       override val fragment = "1 / 0"
-      override def code: String = throw new UnsupportedOperationException
       override def callSite: String = throw new UnsupportedOperationException
       override val summary = ""
     }
@@ -588,8 +587,7 @@ class SparkThrowableSuite extends SparkFunSuite {
       override def objectType: String = throw new UnsupportedOperationException
       override def startIndex: Int = throw new UnsupportedOperationException
       override def stopIndex: Int = throw new UnsupportedOperationException
-      override def fragment: String = throw new UnsupportedOperationException
-      override val code: String = "div"
+      override val fragment: String = "div"
       override val callSite: String = "SimpleApp$.main(SimpleApp.scala:9)"
       override val summary = ""
     }

sql/api/src/main/scala/org/apache/spark/sql/catalyst/trees/QueryContexts.scala

@@ -131,27 +131,25 @@ case class SQLQueryContext(
       originStartIndex.get <= originStopIndex.get
   }
 
-  override def code: String = throw new UnsupportedOperationException
   override def callSite: String = throw new UnsupportedOperationException
 }
 
 case class DatasetQueryContext(
-    override val code: String,
+    override val fragment: String,
     override val callSite: String) extends QueryContext {
   override val contextType = QueryContextType.Dataset
 
   override def objectType: String = throw new UnsupportedOperationException
   override def objectName: String = throw new UnsupportedOperationException
   override def startIndex: Int = throw new UnsupportedOperationException
   override def stopIndex: Int = throw new UnsupportedOperationException
-  override def fragment: String = throw new UnsupportedOperationException
 
   override lazy val summary: String = {
     val builder = new StringBuilder
     builder ++= "== Dataset ==\n"
     builder ++= "\""
 
-    builder ++= code
+    builder ++= fragment
     builder ++= "\""
     builder ++= " was called from "
     builder ++= callSite