[SPARK-41752][SQL][FOLLOW-UP] Fix Protobuf serializer for SQLExecutionUIData

### What changes were proposed in this pull request?

The new field `rootExecutionId` of `SQLExecutionUIData` is not correctly serialized/deserialized in #39268. This PR is to fix it.

### Why are the changes needed?

Bug fix

### Does this PR introduce _any_ user-facing change?

No

### How was this patch tested?

UT

Closes #39489 from gengliangwang/SPARK-41752.

Authored-by: Gengliang Wang <[email protected]>
Signed-off-by: Gengliang Wang <[email protected]>

Author: gengliangwang

core/src/main/protobuf/org/apache/spark/status/protobuf/store_types.proto

@@ -405,19 +405,19 @@ message SQLPlanMetric {
 
 message SQLExecutionUIData {
   int64 execution_id = 1;
-  string description = 2;
-  string details = 3;
-  string physical_plan_description = 4;
-  map<string, string> modified_configs = 5;
-  repeated SQLPlanMetric metrics = 6;
-  int64 submission_time = 7;
-  optional int64 completion_time = 8;
-  optional string error_message = 9;
-  map<int64, JobExecutionStatus> jobs = 10;
-  repeated int64 stages = 11;
-  bool metric_values_is_null = 12;
-  map<int64, string> metric_values = 13;
-  optional int64 root_execution_id = 14;
+  int64 root_execution_id = 2;
+  string description = 3;
+  string details = 4;
+  string physical_plan_description = 5;
+  map<string, string> modified_configs = 6;
+  repeated SQLPlanMetric metrics = 7;
+  int64 submission_time = 8;
+  optional int64 completion_time = 9;
+  optional string error_message = 10;
+  map<int64, JobExecutionStatus> jobs = 11;
+  repeated int64 stages = 12;
+  bool metric_values_is_null = 13;
+  map<int64, string> metric_values = 14;
 }
 
 message SparkPlanGraphNode {

sql/core/src/main/scala/org/apache/spark/status/protobuf/sql/SQLExecutionUIDataSerializer.scala

@@ -33,6 +33,7 @@ class SQLExecutionUIDataSerializer extends ProtobufSerDe {
     val ui = input.asInstanceOf[SQLExecutionUIData]
     val builder = StoreTypes.SQLExecutionUIData.newBuilder()
     builder.setExecutionId(ui.executionId)
+    builder.setRootExecutionId(ui.rootExecutionId)
     builder.setDescription(ui.description)
     builder.setDetails(ui.details)
     builder.setPhysicalPlanDescription(ui.physicalPlanDescription)

sql/core/src/test/scala/org/apache/spark/status/protobuf/sql/KVStoreProtobufSerializerSuite.scala

@@ -34,6 +34,7 @@ class KVStoreProtobufSerializerSuite extends SparkFunSuite {
     val bytes = serializer.serialize(input)
     val result = serializer.deserialize(bytes, classOf[SQLExecutionUIData])
     assert(result.executionId == input.executionId)
+    assert(result.rootExecutionId == input.rootExecutionId)
     assert(result.description == input.description)
     assert(result.details == input.details)
     assert(result.physicalPlanDescription == input.physicalPlanDescription)