forked from opensearch-project/OpenSearch
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge branch 'main' into add_changes_for_task_throttle_improvements
Signed-off-by: Sumit Bansal <sumitsb@amazon.com>
- Loading branch information
Showing
157 changed files
with
3,821 additions
and
1,112 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
51 changes: 51 additions & 0 deletions
51
...ement/src/main/java/org/opensearch/plugin/wlm/action/TransportUpdateQueryGroupAction.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,51 @@ | ||
/* | ||
* 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.plugin.wlm.action; | ||
|
||
import org.opensearch.action.support.ActionFilters; | ||
import org.opensearch.action.support.HandledTransportAction; | ||
import org.opensearch.common.inject.Inject; | ||
import org.opensearch.core.action.ActionListener; | ||
import org.opensearch.plugin.wlm.service.QueryGroupPersistenceService; | ||
import org.opensearch.tasks.Task; | ||
import org.opensearch.transport.TransportService; | ||
|
||
/** | ||
* Transport action to update QueryGroup | ||
* | ||
* @opensearch.experimental | ||
*/ | ||
public class TransportUpdateQueryGroupAction extends HandledTransportAction<UpdateQueryGroupRequest, UpdateQueryGroupResponse> { | ||
|
||
private final QueryGroupPersistenceService queryGroupPersistenceService; | ||
|
||
/** | ||
* Constructor for TransportUpdateQueryGroupAction | ||
* | ||
* @param actionName - action name | ||
* @param transportService - a {@link TransportService} object | ||
* @param actionFilters - a {@link ActionFilters} object | ||
* @param queryGroupPersistenceService - a {@link QueryGroupPersistenceService} object | ||
*/ | ||
@Inject | ||
public TransportUpdateQueryGroupAction( | ||
String actionName, | ||
TransportService transportService, | ||
ActionFilters actionFilters, | ||
QueryGroupPersistenceService queryGroupPersistenceService | ||
) { | ||
super(UpdateQueryGroupAction.NAME, transportService, actionFilters, UpdateQueryGroupRequest::new); | ||
this.queryGroupPersistenceService = queryGroupPersistenceService; | ||
} | ||
|
||
@Override | ||
protected void doExecute(Task task, UpdateQueryGroupRequest request, ActionListener<UpdateQueryGroupResponse> listener) { | ||
queryGroupPersistenceService.updateInClusterStateMetadata(request, listener); | ||
} | ||
} |
36 changes: 36 additions & 0 deletions
36
...oad-management/src/main/java/org/opensearch/plugin/wlm/action/UpdateQueryGroupAction.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,36 @@ | ||
/* | ||
* 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.plugin.wlm.action; | ||
|
||
import org.opensearch.action.ActionType; | ||
|
||
/** | ||
* Transport action to update QueryGroup | ||
* | ||
* @opensearch.experimental | ||
*/ | ||
public class UpdateQueryGroupAction extends ActionType<UpdateQueryGroupResponse> { | ||
|
||
/** | ||
* An instance of UpdateQueryGroupAction | ||
*/ | ||
public static final UpdateQueryGroupAction INSTANCE = new UpdateQueryGroupAction(); | ||
|
||
/** | ||
* Name for UpdateQueryGroupAction | ||
*/ | ||
public static final String NAME = "cluster:admin/opensearch/wlm/query_group/_update"; | ||
|
||
/** | ||
* Default constructor | ||
*/ | ||
private UpdateQueryGroupAction() { | ||
super(NAME, UpdateQueryGroupResponse::new); | ||
} | ||
} |
83 changes: 83 additions & 0 deletions
83
...ad-management/src/main/java/org/opensearch/plugin/wlm/action/UpdateQueryGroupRequest.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,83 @@ | ||
/* | ||
* 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.plugin.wlm.action; | ||
|
||
import org.opensearch.action.ActionRequest; | ||
import org.opensearch.action.ActionRequestValidationException; | ||
import org.opensearch.cluster.metadata.QueryGroup; | ||
import org.opensearch.core.common.io.stream.StreamInput; | ||
import org.opensearch.core.common.io.stream.StreamOutput; | ||
import org.opensearch.core.xcontent.XContentParser; | ||
import org.opensearch.wlm.MutableQueryGroupFragment; | ||
|
||
import java.io.IOException; | ||
|
||
/** | ||
* A request for update QueryGroup | ||
* | ||
* @opensearch.experimental | ||
*/ | ||
public class UpdateQueryGroupRequest extends ActionRequest { | ||
private final String name; | ||
private final MutableQueryGroupFragment mutableQueryGroupFragment; | ||
|
||
/** | ||
* Constructor for UpdateQueryGroupRequest | ||
* @param name - QueryGroup name for UpdateQueryGroupRequest | ||
* @param mutableQueryGroupFragment - MutableQueryGroupFragment for UpdateQueryGroupRequest | ||
*/ | ||
UpdateQueryGroupRequest(String name, MutableQueryGroupFragment mutableQueryGroupFragment) { | ||
this.name = name; | ||
this.mutableQueryGroupFragment = mutableQueryGroupFragment; | ||
} | ||
|
||
/** | ||
* Constructor for UpdateQueryGroupRequest | ||
* @param in - A {@link StreamInput} object | ||
*/ | ||
UpdateQueryGroupRequest(StreamInput in) throws IOException { | ||
this(in.readString(), new MutableQueryGroupFragment(in)); | ||
} | ||
|
||
/** | ||
* Generate a UpdateQueryGroupRequest from XContent | ||
* @param parser - A {@link XContentParser} object | ||
* @param name - name of the QueryGroup to be updated | ||
*/ | ||
public static UpdateQueryGroupRequest fromXContent(XContentParser parser, String name) throws IOException { | ||
QueryGroup.Builder builder = QueryGroup.Builder.fromXContent(parser); | ||
return new UpdateQueryGroupRequest(name, builder.getMutableQueryGroupFragment()); | ||
} | ||
|
||
@Override | ||
public ActionRequestValidationException validate() { | ||
QueryGroup.validateName(name); | ||
return null; | ||
} | ||
|
||
/** | ||
* name getter | ||
*/ | ||
public String getName() { | ||
return name; | ||
} | ||
|
||
/** | ||
* mutableQueryGroupFragment getter | ||
*/ | ||
public MutableQueryGroupFragment getmMutableQueryGroupFragment() { | ||
return mutableQueryGroupFragment; | ||
} | ||
|
||
@Override | ||
public void writeTo(StreamOutput out) throws IOException { | ||
out.writeString(name); | ||
mutableQueryGroupFragment.writeTo(out); | ||
} | ||
} |
Oops, something went wrong.