From bb0988d06a1b934c7e276461606c2761b24bce23 Mon Sep 17 00:00:00 2001 From: Ashish Agrawal Date: Thu, 4 Sep 2025 14:58:39 -0700 Subject: [PATCH] Add experimental annotations to direct-query and direct-query-core Signed-off-by: Ashish Agrawal --- .../opensearch/sql/datasource/client/DataSourceClient.java | 2 ++ .../sql/datasource/client/DataSourceClientFactory.java | 6 +++++- .../client/exceptions/DataSourceClientException.java | 6 +++++- .../org/opensearch/sql/datasource/query/QueryHandler.java | 2 ++ .../sql/datasource/query/QueryHandlerRegistry.java | 6 +++++- .../sql/directquery/DirectQueryExecutorService.java | 3 +++ .../sql/directquery/DirectQueryExecutorServiceImpl.java | 3 +++ .../opensearch/sql/directquery/model/DataSourceOptions.java | 6 +++++- .../sql/directquery/rest/model/DirectQueryResourceType.java | 6 +++++- .../directquery/rest/model/ExecuteDirectQueryRequest.java | 3 +++ .../directquery/rest/model/ExecuteDirectQueryResponse.java | 3 +++ .../rest/model/GetDirectQueryResourcesRequest.java | 3 +++ .../rest/model/GetDirectQueryResourcesResponse.java | 3 +++ .../directquery/validator/DirectQueryRequestValidator.java | 3 +++ .../opensearch/sql/prometheus/client/PrometheusClient.java | 3 +++ .../sql/prometheus/client/PrometheusClientImpl.java | 3 +++ .../sql/prometheus/exception/PrometheusClientException.java | 3 +++ .../org/opensearch/sql/prometheus/model/MetricMetadata.java | 3 +++ .../opensearch/sql/prometheus/model/PrometheusOptions.java | 3 +++ .../sql/prometheus/model/PrometheusQueryType.java | 3 +++ .../sql/prometheus/query/PrometheusQueryHandler.java | 3 +++ .../sql/prometheus/utils/PrometheusClientUtils.java | 3 +++ .../sql/datasource/client/DataSourceClientFactoryTest.java | 3 +++ .../sql/datasource/query/QueryHandlerRegistryTest.java | 3 +++ .../sql/directquery/DirectQueryExecutorServiceImplTest.java | 3 +++ .../validator/DirectQueryRequestValidatorTest.java | 3 +++ .../sql/prometheus/client/PrometheusClientImplTest.java | 3 +++ .../sql/prometheus/query/PrometheusQueryHandlerTest.java | 3 +++ .../sql/prometheus/utils/PrometheusClientUtilsTest.java | 3 +++ .../opensearch/sql/prometheus/constant/TestConstants.java | 3 +++ .../java/org/opensearch/sql/prometheus/utils/TestUtils.java | 3 +++ .../directquery/rest/RestDirectQueryManagementAction.java | 5 +++++ .../rest/RestDirectQueryResourcesManagementAction.java | 5 +++++ .../transport/TransportExecuteDirectQueryRequestAction.java | 3 +++ .../TransportGetDirectQueryResourcesRequestAction.java | 3 +++ .../sql/directquery/transport/config/DirectQueryModule.java | 3 +++ .../transport/format/DirectQueryRequestConverter.java | 3 +++ .../format/DirectQueryResourcesRequestConverter.java | 3 +++ .../transport/model/ExecuteDirectQueryActionRequest.java | 3 +++ .../transport/model/ExecuteDirectQueryActionResponse.java | 3 +++ .../model/GetDirectQueryResourcesActionRequest.java | 3 +++ .../model/GetDirectQueryResourcesActionResponse.java | 3 +++ .../transport/model/datasource/DataSourceResult.java | 3 +++ .../transport/model/datasource/PrometheusResult.java | 3 +++ .../rest/RestDirectQueryManagementActionTest.java | 3 +++ .../rest/RestDirectQueryResourcesManagementActionTest.java | 3 +++ .../TransportExecuteDirectQueryRequestActionTest.java | 3 +++ .../TransportGetDirectQueryResourcesRequestActionTest.java | 3 +++ .../directquery/transport/config/DirectQueryModuleTest.java | 3 +++ .../transport/format/DirectQueryRequestConverterTest.java | 3 +++ .../format/DirectQueryResourcesRequestConverterTest.java | 3 +++ 51 files changed, 165 insertions(+), 5 deletions(-) diff --git a/direct-query-core/src/main/java/org/opensearch/sql/datasource/client/DataSourceClient.java b/direct-query-core/src/main/java/org/opensearch/sql/datasource/client/DataSourceClient.java index 3f37b5481b6..6c319da2666 100644 --- a/direct-query-core/src/main/java/org/opensearch/sql/datasource/client/DataSourceClient.java +++ b/direct-query-core/src/main/java/org/opensearch/sql/datasource/client/DataSourceClient.java @@ -8,5 +8,7 @@ /** * Base interface for all data source clients. This interface serves as a marker interface for all * client implementations. + * + * @opensearch.experimental */ public interface DataSourceClient {} diff --git a/direct-query-core/src/main/java/org/opensearch/sql/datasource/client/DataSourceClientFactory.java b/direct-query-core/src/main/java/org/opensearch/sql/datasource/client/DataSourceClientFactory.java index 23b04572175..e0fbe7a165f 100644 --- a/direct-query-core/src/main/java/org/opensearch/sql/datasource/client/DataSourceClientFactory.java +++ b/direct-query-core/src/main/java/org/opensearch/sql/datasource/client/DataSourceClientFactory.java @@ -15,7 +15,11 @@ import org.opensearch.sql.datasource.model.DataSourceType; import org.opensearch.sql.prometheus.utils.PrometheusClientUtils; -/** Factory for creating data source clients based on the data source type. */ +/** + * Factory for creating data source clients based on the data source type. + * + * @opensearch.experimental + */ public class DataSourceClientFactory { private static final Logger LOG = LogManager.getLogger(); diff --git a/direct-query-core/src/main/java/org/opensearch/sql/datasource/client/exceptions/DataSourceClientException.java b/direct-query-core/src/main/java/org/opensearch/sql/datasource/client/exceptions/DataSourceClientException.java index 74906fc5b84..683509c76cb 100644 --- a/direct-query-core/src/main/java/org/opensearch/sql/datasource/client/exceptions/DataSourceClientException.java +++ b/direct-query-core/src/main/java/org/opensearch/sql/datasource/client/exceptions/DataSourceClientException.java @@ -5,7 +5,11 @@ package org.opensearch.sql.datasource.client.exceptions; -/** Exception thrown when there are issues with data source client operations. */ +/** + * Exception thrown when there are issues with data source client operations. + * + * @opensearch.experimental + */ public class DataSourceClientException extends RuntimeException { public DataSourceClientException(String message) { diff --git a/direct-query-core/src/main/java/org/opensearch/sql/datasource/query/QueryHandler.java b/direct-query-core/src/main/java/org/opensearch/sql/datasource/query/QueryHandler.java index 0fd4311c325..87a025c3c7a 100644 --- a/direct-query-core/src/main/java/org/opensearch/sql/datasource/query/QueryHandler.java +++ b/direct-query-core/src/main/java/org/opensearch/sql/datasource/query/QueryHandler.java @@ -16,6 +16,8 @@ * Interface for handling queries for specific data source types. * * @param The client type this handler works with, extending DataSourceClient + * + * @opensearch.experimental */ public interface QueryHandler { diff --git a/direct-query-core/src/main/java/org/opensearch/sql/datasource/query/QueryHandlerRegistry.java b/direct-query-core/src/main/java/org/opensearch/sql/datasource/query/QueryHandlerRegistry.java index ec583c96a82..c6754288c51 100644 --- a/direct-query-core/src/main/java/org/opensearch/sql/datasource/query/QueryHandlerRegistry.java +++ b/direct-query-core/src/main/java/org/opensearch/sql/datasource/query/QueryHandlerRegistry.java @@ -10,7 +10,11 @@ import org.opensearch.common.inject.Inject; import org.opensearch.sql.datasource.client.DataSourceClient; -/** Registry for all query handlers. */ +/** + * Registry for all query handlers. + * + * @opensearch.experimental + */ public class QueryHandlerRegistry { private final List> handlers; diff --git a/direct-query-core/src/main/java/org/opensearch/sql/directquery/DirectQueryExecutorService.java b/direct-query-core/src/main/java/org/opensearch/sql/directquery/DirectQueryExecutorService.java index 828f1864f14..959544b416e 100644 --- a/direct-query-core/src/main/java/org/opensearch/sql/directquery/DirectQueryExecutorService.java +++ b/direct-query-core/src/main/java/org/opensearch/sql/directquery/DirectQueryExecutorService.java @@ -10,6 +10,9 @@ import org.opensearch.sql.directquery.rest.model.GetDirectQueryResourcesRequest; import org.opensearch.sql.directquery.rest.model.GetDirectQueryResourcesResponse; +/* + * @opensearch.experimental + */ public interface DirectQueryExecutorService { /** diff --git a/direct-query-core/src/main/java/org/opensearch/sql/directquery/DirectQueryExecutorServiceImpl.java b/direct-query-core/src/main/java/org/opensearch/sql/directquery/DirectQueryExecutorServiceImpl.java index 1a2e9e58cc0..70aa8e31069 100644 --- a/direct-query-core/src/main/java/org/opensearch/sql/directquery/DirectQueryExecutorServiceImpl.java +++ b/direct-query-core/src/main/java/org/opensearch/sql/directquery/DirectQueryExecutorServiceImpl.java @@ -17,6 +17,9 @@ import org.opensearch.sql.directquery.rest.model.GetDirectQueryResourcesRequest; import org.opensearch.sql.directquery.rest.model.GetDirectQueryResourcesResponse; +/* + * @opensearch.experimental + */ public class DirectQueryExecutorServiceImpl implements DirectQueryExecutorService { private final DataSourceClientFactory dataSourceClientFactory; diff --git a/direct-query-core/src/main/java/org/opensearch/sql/directquery/model/DataSourceOptions.java b/direct-query-core/src/main/java/org/opensearch/sql/directquery/model/DataSourceOptions.java index 8faa06609e1..dd2a3afab2a 100644 --- a/direct-query-core/src/main/java/org/opensearch/sql/directquery/model/DataSourceOptions.java +++ b/direct-query-core/src/main/java/org/opensearch/sql/directquery/model/DataSourceOptions.java @@ -5,5 +5,9 @@ package org.opensearch.sql.directquery.model; -/** Interface for data source specific options. */ +/** + * Interface for data source specific options. + * + * @opensearch.experimental + */ public interface DataSourceOptions {} diff --git a/direct-query-core/src/main/java/org/opensearch/sql/directquery/rest/model/DirectQueryResourceType.java b/direct-query-core/src/main/java/org/opensearch/sql/directquery/rest/model/DirectQueryResourceType.java index 1a7070c6e83..6034d1fee7a 100644 --- a/direct-query-core/src/main/java/org/opensearch/sql/directquery/rest/model/DirectQueryResourceType.java +++ b/direct-query-core/src/main/java/org/opensearch/sql/directquery/rest/model/DirectQueryResourceType.java @@ -5,7 +5,11 @@ package org.opensearch.sql.directquery.rest.model; -/** Enum representing the types of resources that can be queried. */ +/** + * Enum representing the types of resources that can be queried. + * + * @opensearch.experimental + */ public enum DirectQueryResourceType { UNKNOWN, LABELS, diff --git a/direct-query-core/src/main/java/org/opensearch/sql/directquery/rest/model/ExecuteDirectQueryRequest.java b/direct-query-core/src/main/java/org/opensearch/sql/directquery/rest/model/ExecuteDirectQueryRequest.java index 6eae6e1ef4e..ec5d5517360 100644 --- a/direct-query-core/src/main/java/org/opensearch/sql/directquery/rest/model/ExecuteDirectQueryRequest.java +++ b/direct-query-core/src/main/java/org/opensearch/sql/directquery/rest/model/ExecuteDirectQueryRequest.java @@ -15,6 +15,9 @@ @Data @NoArgsConstructor +/* + * @opensearch.experimental + */ public class ExecuteDirectQueryRequest { // Required fields private String dataSources; // Required: From URI path parameter or request body diff --git a/direct-query-core/src/main/java/org/opensearch/sql/directquery/rest/model/ExecuteDirectQueryResponse.java b/direct-query-core/src/main/java/org/opensearch/sql/directquery/rest/model/ExecuteDirectQueryResponse.java index 8bad0693688..77750a70d54 100644 --- a/direct-query-core/src/main/java/org/opensearch/sql/directquery/rest/model/ExecuteDirectQueryResponse.java +++ b/direct-query-core/src/main/java/org/opensearch/sql/directquery/rest/model/ExecuteDirectQueryResponse.java @@ -9,6 +9,9 @@ import lombok.Data; import lombok.NoArgsConstructor; +/* + * @opensearch.experimental + */ @Data @AllArgsConstructor @NoArgsConstructor diff --git a/direct-query-core/src/main/java/org/opensearch/sql/directquery/rest/model/GetDirectQueryResourcesRequest.java b/direct-query-core/src/main/java/org/opensearch/sql/directquery/rest/model/GetDirectQueryResourcesRequest.java index 54a6b315ef7..30b54349a5e 100644 --- a/direct-query-core/src/main/java/org/opensearch/sql/directquery/rest/model/GetDirectQueryResourcesRequest.java +++ b/direct-query-core/src/main/java/org/opensearch/sql/directquery/rest/model/GetDirectQueryResourcesRequest.java @@ -9,6 +9,9 @@ import lombok.Data; import lombok.NoArgsConstructor; +/* + * @opensearch.experimental + */ @Data @NoArgsConstructor public class GetDirectQueryResourcesRequest { diff --git a/direct-query-core/src/main/java/org/opensearch/sql/directquery/rest/model/GetDirectQueryResourcesResponse.java b/direct-query-core/src/main/java/org/opensearch/sql/directquery/rest/model/GetDirectQueryResourcesResponse.java index 076423533d3..cfd5a7a79a2 100644 --- a/direct-query-core/src/main/java/org/opensearch/sql/directquery/rest/model/GetDirectQueryResourcesResponse.java +++ b/direct-query-core/src/main/java/org/opensearch/sql/directquery/rest/model/GetDirectQueryResourcesResponse.java @@ -19,6 +19,9 @@ @Data @NoArgsConstructor @JsonInclude(JsonInclude.Include.NON_NULL) +/* + * @opensearch.experimental + */ public class GetDirectQueryResourcesResponse { private T data; diff --git a/direct-query-core/src/main/java/org/opensearch/sql/directquery/validator/DirectQueryRequestValidator.java b/direct-query-core/src/main/java/org/opensearch/sql/directquery/validator/DirectQueryRequestValidator.java index 6a945574fd0..d880c1b26ba 100644 --- a/direct-query-core/src/main/java/org/opensearch/sql/directquery/validator/DirectQueryRequestValidator.java +++ b/direct-query-core/src/main/java/org/opensearch/sql/directquery/validator/DirectQueryRequestValidator.java @@ -10,6 +10,9 @@ import org.opensearch.sql.prometheus.model.PrometheusQueryType; import org.opensearch.sql.spark.rest.model.LangType; +/* + * @opensearch.experimental + */ public class DirectQueryRequestValidator { private DirectQueryRequestValidator() {} diff --git a/direct-query-core/src/main/java/org/opensearch/sql/prometheus/client/PrometheusClient.java b/direct-query-core/src/main/java/org/opensearch/sql/prometheus/client/PrometheusClient.java index 92b54f071d3..a7c0c36e5b7 100644 --- a/direct-query-core/src/main/java/org/opensearch/sql/prometheus/client/PrometheusClient.java +++ b/direct-query-core/src/main/java/org/opensearch/sql/prometheus/client/PrometheusClient.java @@ -13,6 +13,9 @@ import org.opensearch.sql.datasource.client.DataSourceClient; import org.opensearch.sql.prometheus.model.MetricMetadata; +/* + * @opensearch.experimental + */ public interface PrometheusClient extends DataSourceClient { JSONObject queryRange(String query, Long start, Long end, String step) throws IOException; diff --git a/direct-query-core/src/main/java/org/opensearch/sql/prometheus/client/PrometheusClientImpl.java b/direct-query-core/src/main/java/org/opensearch/sql/prometheus/client/PrometheusClientImpl.java index c9633e93211..abbe68397a1 100644 --- a/direct-query-core/src/main/java/org/opensearch/sql/prometheus/client/PrometheusClientImpl.java +++ b/direct-query-core/src/main/java/org/opensearch/sql/prometheus/client/PrometheusClientImpl.java @@ -27,6 +27,9 @@ import org.json.JSONObject; import org.opensearch.sql.prometheus.model.MetricMetadata; +/* + * @opensearch.experimental + */ public class PrometheusClientImpl implements PrometheusClient { private static final Logger logger = LogManager.getLogger(PrometheusClientImpl.class); diff --git a/direct-query-core/src/main/java/org/opensearch/sql/prometheus/exception/PrometheusClientException.java b/direct-query-core/src/main/java/org/opensearch/sql/prometheus/exception/PrometheusClientException.java index 6a491243d8d..ed59d566b4b 100644 --- a/direct-query-core/src/main/java/org/opensearch/sql/prometheus/exception/PrometheusClientException.java +++ b/direct-query-core/src/main/java/org/opensearch/sql/prometheus/exception/PrometheusClientException.java @@ -8,6 +8,9 @@ import org.opensearch.sql.datasource.client.exceptions.DataSourceClientException; /** PrometheusClientException. */ +/* + * @opensearch.experimental + */ public class PrometheusClientException extends DataSourceClientException { public PrometheusClientException(String message) { super(message); diff --git a/direct-query-core/src/main/java/org/opensearch/sql/prometheus/model/MetricMetadata.java b/direct-query-core/src/main/java/org/opensearch/sql/prometheus/model/MetricMetadata.java index f0f8d8cb220..530f52d5280 100644 --- a/direct-query-core/src/main/java/org/opensearch/sql/prometheus/model/MetricMetadata.java +++ b/direct-query-core/src/main/java/org/opensearch/sql/prometheus/model/MetricMetadata.java @@ -18,6 +18,9 @@ @NoArgsConstructor @EqualsAndHashCode @JsonIgnoreProperties(ignoreUnknown = true) +/* + * @opensearch.experimental + */ public class MetricMetadata { private String type; private String help; diff --git a/direct-query-core/src/main/java/org/opensearch/sql/prometheus/model/PrometheusOptions.java b/direct-query-core/src/main/java/org/opensearch/sql/prometheus/model/PrometheusOptions.java index ab4870557ae..d7a3ca2b18f 100644 --- a/direct-query-core/src/main/java/org/opensearch/sql/prometheus/model/PrometheusOptions.java +++ b/direct-query-core/src/main/java/org/opensearch/sql/prometheus/model/PrometheusOptions.java @@ -12,6 +12,9 @@ /** Prometheus-specific options for direct queries. */ @Data @NoArgsConstructor +/* + * @opensearch.experimental + */ public class PrometheusOptions implements DataSourceOptions { private PrometheusQueryType queryType; private String step; // Duration string in seconds diff --git a/direct-query-core/src/main/java/org/opensearch/sql/prometheus/model/PrometheusQueryType.java b/direct-query-core/src/main/java/org/opensearch/sql/prometheus/model/PrometheusQueryType.java index 5fb08e030fe..9d9a93fc6e7 100644 --- a/direct-query-core/src/main/java/org/opensearch/sql/prometheus/model/PrometheusQueryType.java +++ b/direct-query-core/src/main/java/org/opensearch/sql/prometheus/model/PrometheusQueryType.java @@ -9,6 +9,9 @@ /** Enum representing the types of Prometheus queries. */ @Getter +/* + * @opensearch.experimental + */ public enum PrometheusQueryType { INSTANT("instant"), RANGE("range"); diff --git a/direct-query-core/src/main/java/org/opensearch/sql/prometheus/query/PrometheusQueryHandler.java b/direct-query-core/src/main/java/org/opensearch/sql/prometheus/query/PrometheusQueryHandler.java index 102b39c727d..c9699a5d543 100644 --- a/direct-query-core/src/main/java/org/opensearch/sql/prometheus/query/PrometheusQueryHandler.java +++ b/direct-query-core/src/main/java/org/opensearch/sql/prometheus/query/PrometheusQueryHandler.java @@ -26,6 +26,9 @@ import org.opensearch.sql.prometheus.model.PrometheusOptions; import org.opensearch.sql.prometheus.model.PrometheusQueryType; +/* + * @opensearch.experimental + */ public class PrometheusQueryHandler implements QueryHandler { private static final Logger LOG = LogManager.getLogger(PrometheusQueryHandler.class); diff --git a/direct-query-core/src/main/java/org/opensearch/sql/prometheus/utils/PrometheusClientUtils.java b/direct-query-core/src/main/java/org/opensearch/sql/prometheus/utils/PrometheusClientUtils.java index fd47da9b2b9..1a95eda190c 100644 --- a/direct-query-core/src/main/java/org/opensearch/sql/prometheus/utils/PrometheusClientUtils.java +++ b/direct-query-core/src/main/java/org/opensearch/sql/prometheus/utils/PrometheusClientUtils.java @@ -25,6 +25,9 @@ import org.opensearch.sql.prometheus.client.PrometheusClient; import org.opensearch.sql.prometheus.client.PrometheusClientImpl; +/* + * @opensearch.experimental + */ public class PrometheusClientUtils { private PrometheusClientUtils() {} diff --git a/direct-query-core/src/test/java/org/opensearch/sql/datasource/client/DataSourceClientFactoryTest.java b/direct-query-core/src/test/java/org/opensearch/sql/datasource/client/DataSourceClientFactoryTest.java index 3db0e8c0fcf..749e073ae08 100644 --- a/direct-query-core/src/test/java/org/opensearch/sql/datasource/client/DataSourceClientFactoryTest.java +++ b/direct-query-core/src/test/java/org/opensearch/sql/datasource/client/DataSourceClientFactoryTest.java @@ -27,6 +27,9 @@ import org.opensearch.sql.prometheus.client.PrometheusClient; import org.opensearch.sql.prometheus.utils.PrometheusClientUtils; +/* + * @opensearch.experimental + */ @RunWith(MockitoJUnitRunner.class) public class DataSourceClientFactoryTest { diff --git a/direct-query-core/src/test/java/org/opensearch/sql/datasource/query/QueryHandlerRegistryTest.java b/direct-query-core/src/test/java/org/opensearch/sql/datasource/query/QueryHandlerRegistryTest.java index aeaa2a0e139..1c35e73a8eb 100644 --- a/direct-query-core/src/test/java/org/opensearch/sql/datasource/query/QueryHandlerRegistryTest.java +++ b/direct-query-core/src/test/java/org/opensearch/sql/datasource/query/QueryHandlerRegistryTest.java @@ -23,6 +23,9 @@ import org.mockito.junit.jupiter.MockitoExtension; import org.opensearch.sql.datasource.client.DataSourceClient; +/* + * @opensearch.experimental + */ @DisplayNameGeneration(DisplayNameGenerator.ReplaceUnderscores.class) @ExtendWith(MockitoExtension.class) public class QueryHandlerRegistryTest { diff --git a/direct-query-core/src/test/java/org/opensearch/sql/directquery/DirectQueryExecutorServiceImplTest.java b/direct-query-core/src/test/java/org/opensearch/sql/directquery/DirectQueryExecutorServiceImplTest.java index 2df9b4f9de5..856a69294ff 100644 --- a/direct-query-core/src/test/java/org/opensearch/sql/directquery/DirectQueryExecutorServiceImplTest.java +++ b/direct-query-core/src/test/java/org/opensearch/sql/directquery/DirectQueryExecutorServiceImplTest.java @@ -37,6 +37,9 @@ import org.opensearch.sql.prometheus.model.PrometheusQueryType; import org.opensearch.sql.spark.rest.model.LangType; +/* + * @opensearch.experimental + */ @ExtendWith(MockitoExtension.class) public class DirectQueryExecutorServiceImplTest { diff --git a/direct-query-core/src/test/java/org/opensearch/sql/directquery/validator/DirectQueryRequestValidatorTest.java b/direct-query-core/src/test/java/org/opensearch/sql/directquery/validator/DirectQueryRequestValidatorTest.java index 0c6fec31dcc..438586f66ea 100644 --- a/direct-query-core/src/test/java/org/opensearch/sql/directquery/validator/DirectQueryRequestValidatorTest.java +++ b/direct-query-core/src/test/java/org/opensearch/sql/directquery/validator/DirectQueryRequestValidatorTest.java @@ -15,6 +15,9 @@ import org.opensearch.sql.prometheus.model.PrometheusQueryType; import org.opensearch.sql.spark.rest.model.LangType; +/* + * @opensearch.experimental + */ public class DirectQueryRequestValidatorTest { @Test diff --git a/direct-query-core/src/test/java/org/opensearch/sql/prometheus/client/PrometheusClientImplTest.java b/direct-query-core/src/test/java/org/opensearch/sql/prometheus/client/PrometheusClientImplTest.java index 8c2444c6590..5485f7d7bba 100644 --- a/direct-query-core/src/test/java/org/opensearch/sql/prometheus/client/PrometheusClientImplTest.java +++ b/direct-query-core/src/test/java/org/opensearch/sql/prometheus/client/PrometheusClientImplTest.java @@ -35,6 +35,9 @@ import org.junit.jupiter.api.Test; import org.opensearch.sql.prometheus.exception.PrometheusClientException; +/* + * @opensearch.experimental + */ public class PrometheusClientImplTest { private MockWebServer mockWebServer; diff --git a/direct-query-core/src/test/java/org/opensearch/sql/prometheus/query/PrometheusQueryHandlerTest.java b/direct-query-core/src/test/java/org/opensearch/sql/prometheus/query/PrometheusQueryHandlerTest.java index 557e6425440..34c2bfdc58a 100644 --- a/direct-query-core/src/test/java/org/opensearch/sql/prometheus/query/PrometheusQueryHandlerTest.java +++ b/direct-query-core/src/test/java/org/opensearch/sql/prometheus/query/PrometheusQueryHandlerTest.java @@ -35,6 +35,9 @@ import org.opensearch.sql.prometheus.model.PrometheusOptions; import org.opensearch.sql.prometheus.model.PrometheusQueryType; +/* + * @opensearch.experimental + */ @RunWith(MockitoJUnitRunner.class) public class PrometheusQueryHandlerTest { diff --git a/direct-query-core/src/test/java/org/opensearch/sql/prometheus/utils/PrometheusClientUtilsTest.java b/direct-query-core/src/test/java/org/opensearch/sql/prometheus/utils/PrometheusClientUtilsTest.java index 4b7959d5428..a6f5ea5bafd 100644 --- a/direct-query-core/src/test/java/org/opensearch/sql/prometheus/utils/PrometheusClientUtilsTest.java +++ b/direct-query-core/src/test/java/org/opensearch/sql/prometheus/utils/PrometheusClientUtilsTest.java @@ -26,6 +26,9 @@ import org.opensearch.sql.datasource.model.DataSourceMetadata; import org.opensearch.sql.prometheus.client.PrometheusClient; +/* + * @opensearch.experimental + */ @RunWith(MockitoJUnitRunner.class) public class PrometheusClientUtilsTest { diff --git a/direct-query-core/src/testFixtures/java/org/opensearch/sql/prometheus/constant/TestConstants.java b/direct-query-core/src/testFixtures/java/org/opensearch/sql/prometheus/constant/TestConstants.java index 9f1342ebe45..307837f3f66 100644 --- a/direct-query-core/src/testFixtures/java/org/opensearch/sql/prometheus/constant/TestConstants.java +++ b/direct-query-core/src/testFixtures/java/org/opensearch/sql/prometheus/constant/TestConstants.java @@ -5,6 +5,9 @@ package org.opensearch.sql.prometheus.constant; +/* + * @opensearch.experimental + */ public class TestConstants { public static final String QUERY = "test_query"; public static final Long STARTTIME = 1664767694133L; diff --git a/direct-query-core/src/testFixtures/java/org/opensearch/sql/prometheus/utils/TestUtils.java b/direct-query-core/src/testFixtures/java/org/opensearch/sql/prometheus/utils/TestUtils.java index a9fcc261013..ea0aa5e0938 100644 --- a/direct-query-core/src/testFixtures/java/org/opensearch/sql/prometheus/utils/TestUtils.java +++ b/direct-query-core/src/testFixtures/java/org/opensearch/sql/prometheus/utils/TestUtils.java @@ -8,6 +8,9 @@ import java.io.IOException; import java.util.Objects; +/* + * @opensearch.experimental + */ public class TestUtils { /** diff --git a/direct-query/src/main/java/org/opensearch/sql/directquery/rest/RestDirectQueryManagementAction.java b/direct-query/src/main/java/org/opensearch/sql/directquery/rest/RestDirectQueryManagementAction.java index bc1dd30781a..f055469ac91 100644 --- a/direct-query/src/main/java/org/opensearch/sql/directquery/rest/RestDirectQueryManagementAction.java +++ b/direct-query/src/main/java/org/opensearch/sql/directquery/rest/RestDirectQueryManagementAction.java @@ -18,6 +18,7 @@ import org.apache.logging.log4j.LogManager; import org.apache.logging.log4j.Logger; import org.opensearch.OpenSearchException; +import org.opensearch.common.annotation.ExperimentalApi; import org.opensearch.core.action.ActionListener; import org.opensearch.core.rest.RestStatus; import org.opensearch.rest.BaseRestHandler; @@ -39,6 +40,10 @@ import org.opensearch.sql.protocol.response.format.JsonResponseFormatter; import org.opensearch.transport.client.node.NodeClient; +/* + * @opensearch.experimental + */ +@ExperimentalApi @RequiredArgsConstructor public class RestDirectQueryManagementAction extends BaseRestHandler { diff --git a/direct-query/src/main/java/org/opensearch/sql/directquery/rest/RestDirectQueryResourcesManagementAction.java b/direct-query/src/main/java/org/opensearch/sql/directquery/rest/RestDirectQueryResourcesManagementAction.java index 49a7609d31a..c3a388afa3b 100644 --- a/direct-query/src/main/java/org/opensearch/sql/directquery/rest/RestDirectQueryResourcesManagementAction.java +++ b/direct-query/src/main/java/org/opensearch/sql/directquery/rest/RestDirectQueryResourcesManagementAction.java @@ -16,6 +16,7 @@ import org.apache.logging.log4j.LogManager; import org.apache.logging.log4j.Logger; import org.opensearch.OpenSearchException; +import org.opensearch.common.annotation.ExperimentalApi; import org.opensearch.core.action.ActionListener; import org.opensearch.core.rest.RestStatus; import org.opensearch.rest.BaseRestHandler; @@ -35,6 +36,10 @@ import org.opensearch.sql.opensearch.util.RestRequestUtil; import org.opensearch.transport.client.node.NodeClient; +/* + * @opensearch.experimental + */ +@ExperimentalApi @RequiredArgsConstructor public class RestDirectQueryResourcesManagementAction extends BaseRestHandler { diff --git a/direct-query/src/main/java/org/opensearch/sql/directquery/transport/TransportExecuteDirectQueryRequestAction.java b/direct-query/src/main/java/org/opensearch/sql/directquery/transport/TransportExecuteDirectQueryRequestAction.java index 5626871e211..ce9c380f9d2 100644 --- a/direct-query/src/main/java/org/opensearch/sql/directquery/transport/TransportExecuteDirectQueryRequestAction.java +++ b/direct-query/src/main/java/org/opensearch/sql/directquery/transport/TransportExecuteDirectQueryRequestAction.java @@ -19,6 +19,9 @@ import org.opensearch.tasks.Task; import org.opensearch.transport.TransportService; +/* + * @opensearch.experimental + */ public class TransportExecuteDirectQueryRequestAction extends HandledTransportAction< ExecuteDirectQueryActionRequest, ExecuteDirectQueryActionResponse> { diff --git a/direct-query/src/main/java/org/opensearch/sql/directquery/transport/TransportGetDirectQueryResourcesRequestAction.java b/direct-query/src/main/java/org/opensearch/sql/directquery/transport/TransportGetDirectQueryResourcesRequestAction.java index b0b07b63257..f230c036acc 100644 --- a/direct-query/src/main/java/org/opensearch/sql/directquery/transport/TransportGetDirectQueryResourcesRequestAction.java +++ b/direct-query/src/main/java/org/opensearch/sql/directquery/transport/TransportGetDirectQueryResourcesRequestAction.java @@ -20,6 +20,9 @@ import org.opensearch.tasks.Task; import org.opensearch.transport.TransportService; +/* + * @opensearch.experimental + */ public class TransportGetDirectQueryResourcesRequestAction extends HandledTransportAction< GetDirectQueryResourcesActionRequest, GetDirectQueryResourcesActionResponse> { diff --git a/direct-query/src/main/java/org/opensearch/sql/directquery/transport/config/DirectQueryModule.java b/direct-query/src/main/java/org/opensearch/sql/directquery/transport/config/DirectQueryModule.java index d2ee17bc344..3e0194a2116 100644 --- a/direct-query/src/main/java/org/opensearch/sql/directquery/transport/config/DirectQueryModule.java +++ b/direct-query/src/main/java/org/opensearch/sql/directquery/transport/config/DirectQueryModule.java @@ -19,6 +19,9 @@ import org.opensearch.sql.directquery.DirectQueryExecutorServiceImpl; import org.opensearch.sql.prometheus.query.PrometheusQueryHandler; +/* + * @opensearch.experimental + */ public class DirectQueryModule extends AbstractModule { @Override diff --git a/direct-query/src/main/java/org/opensearch/sql/directquery/transport/format/DirectQueryRequestConverter.java b/direct-query/src/main/java/org/opensearch/sql/directquery/transport/format/DirectQueryRequestConverter.java index b75cf61be19..8d6a0d5c96e 100644 --- a/direct-query/src/main/java/org/opensearch/sql/directquery/transport/format/DirectQueryRequestConverter.java +++ b/direct-query/src/main/java/org/opensearch/sql/directquery/transport/format/DirectQueryRequestConverter.java @@ -16,6 +16,9 @@ import org.opensearch.sql.spark.rest.model.LangType; @UtilityClass +/* + * @opensearch.experimental + */ public class DirectQueryRequestConverter { public static ExecuteDirectQueryRequest fromXContentParser(XContentParser parser) diff --git a/direct-query/src/main/java/org/opensearch/sql/directquery/transport/format/DirectQueryResourcesRequestConverter.java b/direct-query/src/main/java/org/opensearch/sql/directquery/transport/format/DirectQueryResourcesRequestConverter.java index 67902bd1862..bffd275664c 100644 --- a/direct-query/src/main/java/org/opensearch/sql/directquery/transport/format/DirectQueryResourcesRequestConverter.java +++ b/direct-query/src/main/java/org/opensearch/sql/directquery/transport/format/DirectQueryResourcesRequestConverter.java @@ -13,6 +13,9 @@ import org.opensearch.sql.directquery.rest.model.GetDirectQueryResourcesRequest; @UtilityClass +/* + * @opensearch.experimental + */ public class DirectQueryResourcesRequestConverter { /** diff --git a/direct-query/src/main/java/org/opensearch/sql/directquery/transport/model/ExecuteDirectQueryActionRequest.java b/direct-query/src/main/java/org/opensearch/sql/directquery/transport/model/ExecuteDirectQueryActionRequest.java index b73dab9fcb5..24cf63f52dd 100644 --- a/direct-query/src/main/java/org/opensearch/sql/directquery/transport/model/ExecuteDirectQueryActionRequest.java +++ b/direct-query/src/main/java/org/opensearch/sql/directquery/transport/model/ExecuteDirectQueryActionRequest.java @@ -12,6 +12,9 @@ import org.opensearch.core.common.io.stream.StreamOutput; import org.opensearch.sql.directquery.rest.model.ExecuteDirectQueryRequest; +/* + * @opensearch.experimental + */ public class ExecuteDirectQueryActionRequest extends ActionRequest { private final ExecuteDirectQueryRequest directQueryRequest; diff --git a/direct-query/src/main/java/org/opensearch/sql/directquery/transport/model/ExecuteDirectQueryActionResponse.java b/direct-query/src/main/java/org/opensearch/sql/directquery/transport/model/ExecuteDirectQueryActionResponse.java index 6a080301c5c..e9689cd2394 100644 --- a/direct-query/src/main/java/org/opensearch/sql/directquery/transport/model/ExecuteDirectQueryActionResponse.java +++ b/direct-query/src/main/java/org/opensearch/sql/directquery/transport/model/ExecuteDirectQueryActionResponse.java @@ -21,6 +21,9 @@ import org.opensearch.sql.directquery.transport.model.datasource.PrometheusResult; @RequiredArgsConstructor +/* + * @opensearch.experimental + */ public class ExecuteDirectQueryActionResponse extends ActionResponse { private static final ObjectMapper OBJECT_MAPPER = diff --git a/direct-query/src/main/java/org/opensearch/sql/directquery/transport/model/GetDirectQueryResourcesActionRequest.java b/direct-query/src/main/java/org/opensearch/sql/directquery/transport/model/GetDirectQueryResourcesActionRequest.java index 388c6861bee..35a9b6c25d5 100644 --- a/direct-query/src/main/java/org/opensearch/sql/directquery/transport/model/GetDirectQueryResourcesActionRequest.java +++ b/direct-query/src/main/java/org/opensearch/sql/directquery/transport/model/GetDirectQueryResourcesActionRequest.java @@ -14,6 +14,9 @@ import org.opensearch.sql.directquery.rest.model.GetDirectQueryResourcesRequest; @Getter +/* + * @opensearch.experimental + */ public class GetDirectQueryResourcesActionRequest extends ActionRequest { private final GetDirectQueryResourcesRequest directQueryRequest; diff --git a/direct-query/src/main/java/org/opensearch/sql/directquery/transport/model/GetDirectQueryResourcesActionResponse.java b/direct-query/src/main/java/org/opensearch/sql/directquery/transport/model/GetDirectQueryResourcesActionResponse.java index 8407d4968fd..a47274fc9a4 100644 --- a/direct-query/src/main/java/org/opensearch/sql/directquery/transport/model/GetDirectQueryResourcesActionResponse.java +++ b/direct-query/src/main/java/org/opensearch/sql/directquery/transport/model/GetDirectQueryResourcesActionResponse.java @@ -14,6 +14,9 @@ @Getter @RequiredArgsConstructor +/* + * @opensearch.experimental + */ public class GetDirectQueryResourcesActionResponse extends ActionResponse { private final String result; diff --git a/direct-query/src/main/java/org/opensearch/sql/directquery/transport/model/datasource/DataSourceResult.java b/direct-query/src/main/java/org/opensearch/sql/directquery/transport/model/datasource/DataSourceResult.java index eacdce6ed88..5cc13159939 100644 --- a/direct-query/src/main/java/org/opensearch/sql/directquery/transport/model/datasource/DataSourceResult.java +++ b/direct-query/src/main/java/org/opensearch/sql/directquery/transport/model/datasource/DataSourceResult.java @@ -11,4 +11,7 @@ /** Interface for results from various data sources. */ @JsonTypeInfo(use = JsonTypeInfo.Id.NAME, include = JsonTypeInfo.As.PROPERTY, property = "type") @JsonSubTypes({@JsonSubTypes.Type(value = PrometheusResult.class, name = "prometheus")}) +/* + * @opensearch.experimental + */ public interface DataSourceResult {} diff --git a/direct-query/src/main/java/org/opensearch/sql/directquery/transport/model/datasource/PrometheusResult.java b/direct-query/src/main/java/org/opensearch/sql/directquery/transport/model/datasource/PrometheusResult.java index 33f79f3597b..0c1c4093bb5 100644 --- a/direct-query/src/main/java/org/opensearch/sql/directquery/transport/model/datasource/PrometheusResult.java +++ b/direct-query/src/main/java/org/opensearch/sql/directquery/transport/model/datasource/PrometheusResult.java @@ -18,6 +18,9 @@ @Setter @JsonTypeName("prometheus") @JsonIgnoreProperties(ignoreUnknown = true) +/* + * @opensearch.experimental + */ public class PrometheusResult implements DataSourceResult { @JsonProperty("resultType") diff --git a/direct-query/src/test/java/org/opensearch/sql/directquery/rest/RestDirectQueryManagementActionTest.java b/direct-query/src/test/java/org/opensearch/sql/directquery/rest/RestDirectQueryManagementActionTest.java index ef2f4f33bad..3193e6506dc 100644 --- a/direct-query/src/test/java/org/opensearch/sql/directquery/rest/RestDirectQueryManagementActionTest.java +++ b/direct-query/src/test/java/org/opensearch/sql/directquery/rest/RestDirectQueryManagementActionTest.java @@ -40,6 +40,9 @@ import org.opensearch.threadpool.ThreadPool; import org.opensearch.transport.client.node.NodeClient; +/* + * @opensearch.experimental + */ public class RestDirectQueryManagementActionTest { private OpenSearchSettings settings; diff --git a/direct-query/src/test/java/org/opensearch/sql/directquery/rest/RestDirectQueryResourcesManagementActionTest.java b/direct-query/src/test/java/org/opensearch/sql/directquery/rest/RestDirectQueryResourcesManagementActionTest.java index 77bf63e689d..f7f5b737f03 100644 --- a/direct-query/src/test/java/org/opensearch/sql/directquery/rest/RestDirectQueryResourcesManagementActionTest.java +++ b/direct-query/src/test/java/org/opensearch/sql/directquery/rest/RestDirectQueryResourcesManagementActionTest.java @@ -36,6 +36,9 @@ import org.opensearch.threadpool.ThreadPool; import org.opensearch.transport.client.node.NodeClient; +/* + * @opensearch.experimental + */ public class RestDirectQueryResourcesManagementActionTest { private OpenSearchSettings settings; diff --git a/direct-query/src/test/java/org/opensearch/sql/directquery/transport/TransportExecuteDirectQueryRequestActionTest.java b/direct-query/src/test/java/org/opensearch/sql/directquery/transport/TransportExecuteDirectQueryRequestActionTest.java index 0979235361b..3c514add1f4 100644 --- a/direct-query/src/test/java/org/opensearch/sql/directquery/transport/TransportExecuteDirectQueryRequestActionTest.java +++ b/direct-query/src/test/java/org/opensearch/sql/directquery/transport/TransportExecuteDirectQueryRequestActionTest.java @@ -24,6 +24,9 @@ import org.opensearch.tasks.Task; import org.opensearch.transport.TransportService; +/* + * @opensearch.experimental + */ public class TransportExecuteDirectQueryRequestActionTest { @Mock private TransportService transportService; diff --git a/direct-query/src/test/java/org/opensearch/sql/directquery/transport/TransportGetDirectQueryResourcesRequestActionTest.java b/direct-query/src/test/java/org/opensearch/sql/directquery/transport/TransportGetDirectQueryResourcesRequestActionTest.java index 114aaf66b48..4d1274d0376 100644 --- a/direct-query/src/test/java/org/opensearch/sql/directquery/transport/TransportGetDirectQueryResourcesRequestActionTest.java +++ b/direct-query/src/test/java/org/opensearch/sql/directquery/transport/TransportGetDirectQueryResourcesRequestActionTest.java @@ -24,6 +24,9 @@ import org.opensearch.tasks.Task; import org.opensearch.transport.TransportService; +/* + * @opensearch.experimental + */ public class TransportGetDirectQueryResourcesRequestActionTest { @Mock private TransportService transportService; diff --git a/direct-query/src/test/java/org/opensearch/sql/directquery/transport/config/DirectQueryModuleTest.java b/direct-query/src/test/java/org/opensearch/sql/directquery/transport/config/DirectQueryModuleTest.java index 9d96837ec11..35c52883cbb 100644 --- a/direct-query/src/test/java/org/opensearch/sql/directquery/transport/config/DirectQueryModuleTest.java +++ b/direct-query/src/test/java/org/opensearch/sql/directquery/transport/config/DirectQueryModuleTest.java @@ -19,6 +19,9 @@ import org.opensearch.sql.directquery.DirectQueryExecutorService; import org.opensearch.transport.client.node.NodeClient; +/* + * @opensearch.experimental + */ @ExtendWith(MockitoExtension.class) public class DirectQueryModuleTest { @Mock private NodeClient nodeClient; diff --git a/direct-query/src/test/java/org/opensearch/sql/directquery/transport/format/DirectQueryRequestConverterTest.java b/direct-query/src/test/java/org/opensearch/sql/directquery/transport/format/DirectQueryRequestConverterTest.java index a9ffff64688..b4bcfc9f7a2 100644 --- a/direct-query/src/test/java/org/opensearch/sql/directquery/transport/format/DirectQueryRequestConverterTest.java +++ b/direct-query/src/test/java/org/opensearch/sql/directquery/transport/format/DirectQueryRequestConverterTest.java @@ -23,6 +23,9 @@ import org.opensearch.sql.prometheus.model.PrometheusQueryType; import org.opensearch.sql.spark.rest.model.LangType; +/* + * @opensearch.experimental + */ public class DirectQueryRequestConverterTest { @Test diff --git a/direct-query/src/test/java/org/opensearch/sql/directquery/transport/format/DirectQueryResourcesRequestConverterTest.java b/direct-query/src/test/java/org/opensearch/sql/directquery/transport/format/DirectQueryResourcesRequestConverterTest.java index d8009f9a79b..aeb29d34fe3 100644 --- a/direct-query/src/test/java/org/opensearch/sql/directquery/transport/format/DirectQueryResourcesRequestConverterTest.java +++ b/direct-query/src/test/java/org/opensearch/sql/directquery/transport/format/DirectQueryResourcesRequestConverterTest.java @@ -20,6 +20,9 @@ import org.opensearch.sql.directquery.rest.model.DirectQueryResourceType; import org.opensearch.sql.directquery.rest.model.GetDirectQueryResourcesRequest; +/* + * @opensearch.experimental + */ public class DirectQueryResourcesRequestConverterTest { private RestRequest request;