add seder to transportPPLQueryResponse

Signed-off-by: zane-neo <[email protected]>

plugin/src/main/java/org/opensearch/sql/plugin/transport/TransportPPLQueryResponse.java

@@ -5,10 +5,16 @@
 
 package org.opensearch.sql.plugin.transport;
 
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
 import java.io.IOException;
+import java.io.UncheckedIOException;
+
 import lombok.Getter;
 import lombok.RequiredArgsConstructor;
 import org.opensearch.core.action.ActionResponse;
+import org.opensearch.core.common.io.stream.InputStreamStreamInput;
+import org.opensearch.core.common.io.stream.OutputStreamStreamOutput;
 import org.opensearch.core.common.io.stream.StreamInput;
 import org.opensearch.core.common.io.stream.StreamOutput;
 
@@ -25,4 +31,22 @@ public TransportPPLQueryResponse(StreamInput in) throws IOException {
   public void writeTo(StreamOutput out) throws IOException {
     out.writeString(result);
   }
+
+  public static TransportPPLQueryResponse fromActionResponse(ActionResponse actionResponse) {
+    if (actionResponse instanceof TransportPPLQueryResponse) {
+      return (TransportPPLQueryResponse) actionResponse;
+    }
+
+    try (
+        ByteArrayOutputStream baos = new ByteArrayOutputStream();
+        OutputStreamStreamOutput osso = new OutputStreamStreamOutput(baos)) {
+      actionResponse.writeTo(osso);
+      try (StreamInput input = new InputStreamStreamInput(new ByteArrayInputStream(baos.toByteArray()))) {
+        return new TransportPPLQueryResponse(input);
+      }
+    } catch (IOException e) {
+      throw new UncheckedIOException("failed to parse ActionResponse into TransportPPLQueryResponse", e);
+    }
+
+  }
 }