-
Notifications
You must be signed in to change notification settings - Fork 7
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add data models for health stats API (#120)
* Add data models for health stats api Signed-off-by: Chenyang Ji <[email protected]> * update PR based on comments Signed-off-by: Chenyang Ji <[email protected]> --------- Signed-off-by: Chenyang Ji <[email protected]> (cherry picked from commit d74adb3) Signed-off-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
- Loading branch information
1 parent
6843023
commit e69c7cd
Showing
15 changed files
with
712 additions
and
23 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
92 changes: 92 additions & 0 deletions
92
.../java/org/opensearch/plugin/insights/rules/model/healthStats/QueryGrouperHealthStats.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,92 @@ | ||
| /* | ||
| * 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.insights.rules.model.healthStats; | ||
|
|
||
| import java.io.IOException; | ||
| import org.opensearch.core.common.io.stream.StreamInput; | ||
| import org.opensearch.core.common.io.stream.StreamOutput; | ||
| import org.opensearch.core.common.io.stream.Writeable; | ||
| import org.opensearch.core.xcontent.ToXContentFragment; | ||
| import org.opensearch.core.xcontent.XContentBuilder; | ||
|
|
||
| /** | ||
| * Represents the health statistics of the query grouper. | ||
| */ | ||
| public class QueryGrouperHealthStats implements ToXContentFragment, Writeable { | ||
| private final int queryGroupCount; | ||
| private final int queryGroupHeapSize; | ||
| private static final String QUERY_GROUP_COUNT_TOTAL = "QueryGroupCount_Total"; | ||
| private static final String QUERY_GROUP_COUNT_MAX_HEAP = "QueryGroupCount_MaxHeap"; | ||
|
|
||
| /** | ||
| * Constructor to read QueryGrouperHealthStats from a StreamInput. | ||
| * | ||
| * @param in the StreamInput to read the QueryGrouperHealthStats from | ||
| * @throws IOException IOException | ||
| */ | ||
| public QueryGrouperHealthStats(final StreamInput in) throws IOException { | ||
| this.queryGroupCount = in.readInt(); | ||
| this.queryGroupHeapSize = in.readInt(); | ||
| } | ||
|
|
||
| /** | ||
| * Constructor of QueryGrouperHealthStats | ||
| * | ||
| * @param queryGroupCount Number of groups in the grouper | ||
| * @param queryGroupHeapSize Heap size of the grouper | ||
| */ | ||
| public QueryGrouperHealthStats(final int queryGroupCount, final int queryGroupHeapSize) { | ||
| this.queryGroupCount = queryGroupCount; | ||
| this.queryGroupHeapSize = queryGroupHeapSize; | ||
| } | ||
|
|
||
| /** | ||
| * Write QueryGrouperHealthStats Object to output stream | ||
| * @param out streamOutput | ||
| * @throws IOException IOException | ||
| */ | ||
| @Override | ||
| public void writeTo(StreamOutput out) throws IOException { | ||
| out.writeInt(queryGroupCount); | ||
| out.writeInt(queryGroupHeapSize); | ||
| } | ||
|
|
||
| /** | ||
| * Write QueryGrouperHealthStats object to XContent | ||
| * | ||
| * @param builder XContentBuilder | ||
| * @param params Parameters | ||
| * @return XContentBuilder | ||
| * @throws IOException IOException | ||
| */ | ||
| @Override | ||
| public XContentBuilder toXContent(XContentBuilder builder, Params params) throws IOException { | ||
| builder.field(QUERY_GROUP_COUNT_TOTAL, queryGroupCount); | ||
| builder.field(QUERY_GROUP_COUNT_MAX_HEAP, queryGroupHeapSize); | ||
| return builder; | ||
| } | ||
|
|
||
| /** | ||
| * Gets the number of query groups. | ||
| * | ||
| * @return the query group count | ||
| */ | ||
| public int getQueryGroupCount() { | ||
| return queryGroupCount; | ||
| } | ||
|
|
||
| /** | ||
| * Gets the query group heap size. | ||
| * | ||
| * @return the query group heap size | ||
| */ | ||
| public int getQueryGroupHeapSize() { | ||
| return queryGroupHeapSize; | ||
| } | ||
| } |
Oops, something went wrong.