change in update pit response

Arpit-Bandejiya · Nov 26, 2022 · 555d8ba · 555d8ba
1 parent 2a2c038
commit 555d8ba
Showing 1 changed file with 20 additions and 1 deletion.
diff --git a/server/src/main/java/org/opensearch/action/search/UpdatePitResponse.java b/server/src/main/java/org/opensearch/action/search/UpdatePitResponse.java
@@ -20,14 +20,14 @@
 
 import java.io.IOException;
 
+import static org.opensearch.common.xcontent.XContentParserUtils.ensureExpectedToken;
 import static org.opensearch.rest.RestStatus.OK;
 
 public class UpdatePitResponse extends ActionResponse implements StatusToXContentObject {
 
     private static final ParseField ID = new ParseField("pit_id");
     private static final ParseField CREATION_TIME = new ParseField("creation_time");
     private static final ParseField EXPIRY_TIME = new ParseField("expiry_time");
-    private static final ParseField SUCCESS_FULL = new ParseField("successfull");
 
     private final String id;
     private final int totalShards;
@@ -78,11 +78,28 @@ public UpdatePitResponse(
         this.shardFailures = shardFailures;
     }
 
+    public static CreatePitResponse fromXContent(XContentParser parser) throws IOException{
+        ensureExpectedToken(XContentParser.Token.START_OBJECT, parser.nextToken(), parser);
+        parser.nextToken();
+        return innerFromXContent(parser);
+    }
+
+    public static CreatePitResponse innerFromXContent(XContentParser parser) throws IOException {
+
+    }
     @Override
     public void writeTo(StreamOutput out) throws IOException{
         out.writeString(id);
+        out.writeVInt(totalShards);
+        out.writeVInt(successfulShards);
+        out.writeVInt(failedShards);
+        out.writeVInt(skippedShards);
         out.writeLong(creationTime);
         out.writeLong(expiryTime);
+        out.writeVInt(shardFailures.length);
+        for (ShardSearchFailure shardSearchFailure : shardFailures) {
+            shardSearchFailure.writeTo(out);
+        }
     }
 
     @Override
@@ -98,6 +115,8 @@ public XContentBuilder toXContent(XContentBuilder builder, Params params) throws
             getFailedShards(),
             getShardFailures()
         );
+        builder.field(CREATION_TIME.getPreferredName(), creationTime);
+        builder.field(EXPIRY_TIME.getPreferredName(), expiryTime);
         builder.endObject();
         return builder;
     }