changes for update pit

client/rest-high-level/src/main/java/org/opensearch/client/RequestConverters.java

@@ -53,12 +53,7 @@
 import org.opensearch.action.get.GetRequest;
 import org.opensearch.action.get.MultiGetRequest;
 import org.opensearch.action.index.IndexRequest;
-import org.opensearch.action.search.ClearScrollRequest;
-import org.opensearch.action.search.CreatePitRequest;
-import org.opensearch.action.search.DeletePitRequest;
-import org.opensearch.action.search.MultiSearchRequest;
-import org.opensearch.action.search.SearchRequest;
-import org.opensearch.action.search.SearchScrollRequest;
+import org.opensearch.action.search.*;
 import org.opensearch.action.support.ActiveShardCount;
 import org.opensearch.action.support.IndicesOptions;
 import org.opensearch.action.support.WriteRequest;
@@ -502,6 +497,11 @@ static Request getAllPits() {
         return new Request(HttpGet.METHOD_NAME, "/_search/point_in_time/_all");
     }
 
+    static Request updatePit(UpdatePitRequest updatePitRequest) throws IOException {
+        Request request = new Request(HttpDelete.METHOD_NAME, "/_search/point_in_time");
+        request.setEntity(createEntity(updatePitRequest, REQUEST_BODY_CONTENT_TYPE));
+        return request;
+    }
     static Request multiSearch(MultiSearchRequest multiSearchRequest) throws IOException {
         Request request = new Request(HttpPost.METHOD_NAME, "/_msearch");
 

client/rest-high-level/src/main/java/org/opensearch/client/RestHighLevelClient.java

@@ -57,18 +57,7 @@
 import org.opensearch.action.get.MultiGetResponse;
 import org.opensearch.action.index.IndexRequest;
 import org.opensearch.action.index.IndexResponse;
-import org.opensearch.action.search.ClearScrollRequest;
-import org.opensearch.action.search.ClearScrollResponse;
-import org.opensearch.action.search.CreatePitRequest;
-import org.opensearch.action.search.CreatePitResponse;
-import org.opensearch.action.search.DeletePitRequest;
-import org.opensearch.action.search.DeletePitResponse;
-import org.opensearch.action.search.GetAllPitNodesResponse;
-import org.opensearch.action.search.MultiSearchRequest;
-import org.opensearch.action.search.MultiSearchResponse;
-import org.opensearch.action.search.SearchRequest;
-import org.opensearch.action.search.SearchResponse;
-import org.opensearch.action.search.SearchScrollRequest;
+import org.opensearch.action.search.*;
 import org.opensearch.action.support.master.AcknowledgedResponse;
 import org.opensearch.action.update.UpdateRequest;
 import org.opensearch.action.update.UpdateResponse;
@@ -1403,6 +1392,16 @@ public final Cancellable getAllPitsAsync(RequestOptions options, ActionListener<
         );
     }
 
+    public final UpdatePitResponse updatePit(UpdatePitRequest updatePitRequest, RequestOptions options) throws IOException {
+        return performRequestAndParseEntity(
+            updatePitRequest,
+            RequestConverters::updatePit,
+            options,
+            UpdatePitResponse::fromXContent,
+            emptySet()
+        );
+    }
+
     /**
      * Clears one or more scroll ids using the Clear Scroll API.
      *

client/rest-high-level/src/test/java/org/opensearch/client/PitIT.java

@@ -32,7 +32,7 @@
  * Tests point in time API with rest high level client
  */
 public class PitIT extends OpenSearchRestHighLevelClientTestCase {
-
+//--this is is PITIT
     @Before
     public void indexDocuments() throws IOException {
         Request doc1 = new Request(HttpPut.METHOD_NAME, "/index/_doc/1");

server/src/main/java/org/opensearch/action/search/UpdatePitAction.java

@@ -0,0 +1,20 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * The OpenSearch Contributors require contributions made to
+ * this file be licensed under the Apache-2.0 license or a
+ * compatible open source license.
+ */
+
+package org.opensearch.action.search;
+
+import org.opensearch.action.ActionType;
+
+public class UpdatePitAction extends ActionType<UpdatePitResponse> {
+    public static final UpdatePitAction INSTANCE = new UpdatePitAction();
+    public static final String NAME = "indices:data/read/point_in_time/update";
+
+    private UpdatePitAction(){
+        super(NAME, UpdatePitResponse::new);
+    }
+}

server/src/main/java/org/opensearch/action/search/UpdatePitInfo.java

@@ -0,0 +1,75 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * The OpenSearch Contributors require contributions made to
+ * this file be licensed under the Apache-2.0 license or a
+ * compatible open source license.
+ */
+
+package org.opensearch.action.search;
+
+import org.opensearch.common.ParseField;
+import org.opensearch.common.io.stream.StreamInput;
+import org.opensearch.common.io.stream.StreamOutput;
+import org.opensearch.common.io.stream.Writeable;
+import org.opensearch.common.xcontent.ConstructingObjectParser;
+import org.opensearch.common.xcontent.ToXContent;
+import org.opensearch.common.xcontent.XContentBuilder;
+import org.opensearch.transport.TransportResponse;
+
+import java.io.IOException;
+
+import static org.opensearch.common.xcontent.ConstructingObjectParser.constructorArg;
+
+public class UpdatePitInfo extends TransportResponse implements Writeable, ToXContent {
+    private final boolean successful;
+    private final String pitId;
+
+    public UpdatePitInfo(boolean successful, String pitId){
+        this.successful = successful;
+        this.pitId = pitId;
+    }
+
+    public UpdatePitInfo(StreamInput in) throws IOException{
+        successful = in.readBoolean();
+        pitId = in.readString();
+    }
+
+    public boolean isSuccessful(){
+        return successful;
+    }
+
+    public String getPitId(){
+        return pitId;
+    }
+
+    @Override
+    public void writeTo(StreamOutput out) throws IOException {
+        out.writeBoolean(successful);
+        out.writeString(pitId);
+    }
+
+    static final ConstructingObjectParser<UpdatePitInfo, Void> PARSER = new ConstructingObjectParser<>(
+        "update_pit_info",
+        true,
+        args -> new UpdatePitInfo((boolean) args[0], (String) args[1])
+    );
+
+    static {
+        PARSER.declareBoolean(constructorArg(), new ParseField("successful"));
+        PARSER.declareString(constructorArg(), new ParseField("pit_id"));
+    }
+
+    private static final ParseField SUCCESSFUL = new ParseField("successful");
+    private static final ParseField PIT_ID = new ParseField("pit_id");
+
+    @Override
+    public XContentBuilder toXContent(XContentBuilder builder, Params params) throws IOException {
+        builder.startObject();
+        builder.field(SUCCESSFUL.getPreferredName(), successful);
+        builder.field(PIT_ID.getPreferredName(), pitId);
+        builder.endObject();
+        return builder;
+    }
+}
+

server/src/main/java/org/opensearch/action/search/UpdatePitRequest.java

@@ -0,0 +1,77 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * The OpenSearch Contributors require contributions made to
+ * this file be licensed under the Apache-2.0 license or a
+ * compatible open source license.
+ */
+
+package org.opensearch.action.search;
+
+import org.opensearch.action.ActionRequest;
+import org.opensearch.action.ActionRequestValidationException;
+import org.opensearch.common.io.stream.StreamInput;
+import org.opensearch.common.unit.TimeValue;
+import org.opensearch.common.xcontent.ToXContent;
+import org.opensearch.common.xcontent.ToXContentObject;
+import org.opensearch.common.xcontent.XContentBuilder;
+import org.opensearch.common.xcontent.XContentParser;
+
+import java.io.IOException;
+
+import static org.opensearch.action.ValidateActions.addValidationError;
+
+public class UpdatePitRequest extends ActionRequest implements ToXContentObject {
+
+    private String pit_id;
+    private TimeValue keepAlive;
+
+    public UpdatePitRequest(String pit_id){
+        this.keepAlive = null;
+        this.pit_id = pit_id;
+    }
+
+    public UpdatePitRequest(StreamInput in) throws IOException {
+        super(in);
+        pit_id = in.readString();
+        keepAlive = in.readTimeValue();
+    }
+
+    public UpdatePitRequest() {}
+
+    @Override
+    public ActionRequestValidationException validate() {
+        ActionRequestValidationException validationException = null;
+        if (keepAlive == null) {
+            validationException = addValidationError("keep alive not specified", validationException);
+        }
+        return validationException;
+    }
+
+    @Override
+    public XContentBuilder toXContent(XContentBuilder builder, ToXContent.Params params) throws IOException {
+        builder.field("keep_alive", keepAlive);
+        builder.field("pit_id", pit_id);
+        return builder;
+    }
+
+    public void fromXContent(XContentParser parser) throws IOException {
+        if(parser.nextToken() != XContentParser.Token.START_OBJECT){
+            throw new IllegalArgumentException("Malformed content, must start with an object");
+        } else {
+            XContentParser.Token token;
+            String currentFieldName = null;
+            while((token = parser.nextToken()) != XContentParser.Token.END_OBJECT){
+                if (token == XContentParser.Token.FIELD_NAME){
+                    currentFieldName = parser.currentName();
+                } else if("keep_alive".equals(currentFieldName)){
+                    if(token.isValue() == false){
+                        throw new IllegalArgumentException("keep_alive should only contain a time value");
+                    }
+                    keepAlive = TimeValue.parseTimeValue(parser.text(),"keep_alive");
+                }
+
+            }
+        }
+    }
+}