Support creating/updating prometheus rules (#5228)

* Support creating/updating prometheus rules

Signed-off-by: Ashish Agrawal <ashisagr@amazon.com>

* Merge delete resources to write resources

Signed-off-by: Ashish Agrawal <ashisagr@amazon.com>

* Add error handling improvements

Signed-off-by: Ashish Agrawal <ashisagr@amazon.com>

* Fix bugs in the code and support a different ruler uri

Signed-off-by: Ashish Agrawal <ashisagr@amazon.com>

* Add support to delete silences and fetch alert manager status

Signed-off-by: Ashish Agrawal <ashisagr@amazon.com>

* Fix instant query and address resource leak issue

Signed-off-by: Ashish Agrawal <ashisagr@amazon.com>

* Resolve test issue

Signed-off-by: Ashish Agrawal <ashisagr@amazon.com>

* Additional fixes

Signed-off-by: Ashish Agrawal <ashisagr@amazon.com>

---------

Signed-off-by: Ashish Agrawal <ashisagr@amazon.com>

Author: lezzago

direct-query-core/src/main/java/org/opensearch/sql/datasource/query/QueryHandler.java

@@ -18,7 +18,6 @@
  * Interface for handling queries for specific data source types.
  *
  * @param <T> The client type this handler works with, extending DataSourceClient
- *
  * @opensearch.experimental
  */
 public interface QueryHandler<T extends DataSourceClient> {
@@ -52,15 +51,15 @@ GetDirectQueryResourcesResponse<?> getResources(T client, GetDirectQueryResource
       throws IOException;
 
   /**
-   * Writes resources to the data source.
+   * Writes resources to the data source. Supports create, update, and delete operations.
    *
    * @param client The client instance to use
    * @param request The resources request
-   * @return Response containing the requested resources
-   * @throws IOException If resource retrieval fails
+   * @return Response containing the result
+   * @throws IOException If the operation fails
    */
-  WriteDirectQueryResourcesResponse<?> writeResources(T client, WriteDirectQueryResourcesRequest request)
-          throws IOException;
+  WriteDirectQueryResourcesResponse<?> writeResources(
+      T client, WriteDirectQueryResourcesRequest request) throws IOException;
 
   /**
    * Checks if this handler can handle the given client type.

direct-query-core/src/main/java/org/opensearch/sql/directquery/DirectQueryExecutorService.java

@@ -35,11 +35,11 @@ GetDirectQueryResourcesResponse<?> getDirectQueryResources(
       GetDirectQueryResourcesRequest request);
 
   /**
-   * Write resources to a data source.
+   * Write resources to a data source. Supports create, update, and delete operations.
    *
    * @param request The resources request
-   * @return A response containing the resources to create
+   * @return A response containing the result
    */
   WriteDirectQueryResourcesResponse<?> writeDirectQueryResources(
-          WriteDirectQueryResourcesRequest request);
+      WriteDirectQueryResourcesRequest request);
 }

direct-query-core/src/main/java/org/opensearch/sql/directquery/DirectQueryExecutorServiceImpl.java

@@ -118,4 +118,5 @@ public WriteDirectQueryResourcesResponse<?> writeDirectQueryResources(
                         "Unsupported data source type: " + request.getDataSource())
             );
     }
+
 }

direct-query-core/src/main/java/org/opensearch/sql/directquery/rest/model/DirectQueryResourceType.java

@@ -21,7 +21,8 @@ public enum DirectQueryResourceType {
   ALERTMANAGER_ALERTS,
   ALERTMANAGER_ALERT_GROUPS,
   ALERTMANAGER_RECEIVERS,
-  ALERTMANAGER_SILENCES;
+  ALERTMANAGER_SILENCES,
+  ALERTMANAGER_STATUS;
 
   /**
    * Convert a string to the corresponding enum value, case-insensitive.

direct-query-core/src/main/java/org/opensearch/sql/directquery/rest/model/WriteDirectQueryResourcesRequest.java

@@ -21,6 +21,8 @@ public class WriteDirectQueryResourcesRequest {
   private String resourceName;
   private String request;
   private Map<String, String> RequestOptions;
+  private String groupName;
+  private boolean delete;
 
   /**
    * Sets the resource type from a string value.

direct-query-core/src/main/java/org/opensearch/sql/prometheus/client/PrometheusClient.java

@@ -58,10 +58,12 @@ Map<String, List<MetricMetadata>> getAllMetrics(Map<String, String> queryParams)
   JSONObject getAlerts() throws IOException;
 
   /**
-   * Get all recording and alerting rules.
+   * Get all recording and alerting rules, normalized to a consistent JSON format. Handles
+   * Prometheus JSON, Cortex/Thanos YAML, and AMP JSON responses, returning them all as a
+   * {"groups":[...]} structure.
    *
    * @param queryParams Map of query parameters to include in the request
-   * @return JSONObject containing all rules
+   * @return JSONObject with {"groups":[...]} structure
    * @throws IOException If there is an issue with the request
    */
   JSONObject getRules(Map<String, String> queryParams) throws IOException;
@@ -107,4 +109,62 @@ Map<String, List<MetricMetadata>> getAllMetrics(Map<String, String> queryParams)
    * @throws IOException If there is an issue with the request
    */
   String createAlertmanagerSilences(String silenceJson) throws IOException;
+
+  /**
+   * Expire (delete) a silence in Alertmanager.
+   *
+   * @param silenceId The ID of the silence to expire
+   * @return String containing the response
+   * @throws IOException If there is an issue with the request
+   */
+  String deleteAlertmanagerSilence(String silenceId) throws IOException;
+
+  /**
+   * Get Alertmanager status including configuration, version, and cluster info.
+   *
+   * @return JSONObject containing the Alertmanager status
+   * @throws IOException If there is an issue with the request
+   */
+  JSONObject getAlertmanagerStatus() throws IOException;
+
+  /**
+   * Get rules for a specific namespace, normalized to a consistent JSON format. Handles
+   * Cortex/Thanos YAML and AMP JSON responses, returning them all as a {"groups":[...]} structure.
+   *
+   * @param namespace The rules namespace
+   * @param queryParams Map of query parameters to include in the request
+   * @return JSONObject with {"groups":[...]} structure
+   * @throws IOException If there is an issue with the request
+   */
+  JSONObject getRulesByNamespace(String namespace, Map<String, String> queryParams)
+      throws IOException;
+
+  /**
+   * Create or update a rule group in a namespace via the Cortex/Thanos Ruler API.
+   *
+   * @param namespace The rules namespace
+   * @param yamlBody The rule group definition in YAML format
+   * @return String containing the response
+   * @throws IOException If there is an issue with the request
+   */
+  String createOrUpdateRuleGroup(String namespace, String yamlBody) throws IOException;
+
+  /**
+   * Delete all rules in a namespace via the Cortex/Thanos Ruler API.
+   *
+   * @param namespace The rules namespace to delete
+   * @return String containing the response
+   * @throws IOException If there is an issue with the request
+   */
+  String deleteRuleNamespace(String namespace) throws IOException;
+
+  /**
+   * Delete a specific rule group in a namespace via the Cortex/Thanos Ruler API.
+   *
+   * @param namespace The rules namespace
+   * @param groupName The specific rule group name to delete
+   * @return String containing the response
+   * @throws IOException If there is an issue with the request
+   */
+  String deleteRuleGroup(String namespace, String groupName) throws IOException;
 }