add get rule api logic

Signed-off-by: Ruirui Zhang <mariazrr@amazon.com>

Author: ruai0511

CHANGELOG.md

@@ -20,6 +20,7 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 - Add vertical scaling and SoftReference for snapshot repository data cache ([#16489](https://github.com/opensearch-project/OpenSearch/pull/16489))
 - [Workload Management] Add Workload Management IT ([#16359](https://github.com/opensearch-project/OpenSearch/pull/16359))
 - Support prefix list for remote repository attributes([#16271](https://github.com/opensearch-project/OpenSearch/pull/16271))
+- Add create rule api logic for rule-based auto tagging ([#17336](https://github.com/opensearch-project/OpenSearch/pull/17336))
 - Add new configuration setting `synonym_analyzer`, to the `synonym` and `synonym_graph` filters, enabling the specification of a custom analyzer for reading the synonym file ([#16488](https://github.com/opensearch-project/OpenSearch/pull/16488)).
 - Add stats for remote publication failure and move download failure stats to remote methods([#16682](https://github.com/opensearch-project/OpenSearch/pull/16682/))
 - Update script supports java.lang.String.sha1() and java.lang.String.sha256() methods ([#16923](https://github.com/opensearch-project/OpenSearch/pull/16923))

plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/WorkloadManagementPlugin.java

@@ -18,32 +18,39 @@
 import org.opensearch.common.settings.Settings;
 import org.opensearch.common.settings.SettingsFilter;
 import org.opensearch.core.action.ActionResponse;
-import org.opensearch.plugin.wlm.action.CreateQueryGroupAction;
-import org.opensearch.plugin.wlm.action.DeleteQueryGroupAction;
-import org.opensearch.plugin.wlm.action.GetQueryGroupAction;
-import org.opensearch.plugin.wlm.action.TransportCreateQueryGroupAction;
-import org.opensearch.plugin.wlm.action.TransportDeleteQueryGroupAction;
-import org.opensearch.plugin.wlm.action.TransportGetQueryGroupAction;
-import org.opensearch.plugin.wlm.action.TransportUpdateQueryGroupAction;
-import org.opensearch.plugin.wlm.action.UpdateQueryGroupAction;
-import org.opensearch.plugin.wlm.rest.RestCreateQueryGroupAction;
-import org.opensearch.plugin.wlm.rest.RestDeleteQueryGroupAction;
-import org.opensearch.plugin.wlm.rest.RestGetQueryGroupAction;
-import org.opensearch.plugin.wlm.rest.RestUpdateQueryGroupAction;
-import org.opensearch.plugin.wlm.service.QueryGroupPersistenceService;
+import org.opensearch.indices.SystemIndexDescriptor;
+import org.opensearch.plugin.wlm.querygroup.action.CreateQueryGroupAction;
+import org.opensearch.plugin.wlm.querygroup.action.DeleteQueryGroupAction;
+import org.opensearch.plugin.wlm.querygroup.action.GetQueryGroupAction;
+import org.opensearch.plugin.wlm.querygroup.action.TransportCreateQueryGroupAction;
+import org.opensearch.plugin.wlm.querygroup.action.TransportDeleteQueryGroupAction;
+import org.opensearch.plugin.wlm.querygroup.action.TransportGetQueryGroupAction;
+import org.opensearch.plugin.wlm.querygroup.action.TransportUpdateQueryGroupAction;
+import org.opensearch.plugin.wlm.querygroup.action.UpdateQueryGroupAction;
+import org.opensearch.plugin.wlm.querygroup.rest.RestCreateQueryGroupAction;
+import org.opensearch.plugin.wlm.querygroup.rest.RestDeleteQueryGroupAction;
+import org.opensearch.plugin.wlm.querygroup.rest.RestGetQueryGroupAction;
+import org.opensearch.plugin.wlm.querygroup.rest.RestUpdateQueryGroupAction;
+import org.opensearch.plugin.wlm.querygroup.service.QueryGroupPersistenceService;
+import org.opensearch.plugin.wlm.rule.action.GetRuleAction;
+import org.opensearch.plugin.wlm.rule.action.TransportGetRuleAction;
+import org.opensearch.plugin.wlm.rule.rest.RestGetRuleAction;
 import org.opensearch.plugins.ActionPlugin;
 import org.opensearch.plugins.Plugin;
+import org.opensearch.plugins.SystemIndexPlugin;
 import org.opensearch.rest.RestController;
 import org.opensearch.rest.RestHandler;
 
 import java.util.Collection;
 import java.util.List;
 import java.util.function.Supplier;
 
+import static org.opensearch.plugin.wlm.rule.service.RulePersistenceService.RULE_INDEX;
+
 /**
  * Plugin class for WorkloadManagement
  */
-public class WorkloadManagementPlugin extends Plugin implements ActionPlugin {
+public class WorkloadManagementPlugin extends Plugin implements ActionPlugin, SystemIndexPlugin {
 
     /**
      * Default constructor
@@ -56,10 +63,16 @@ public WorkloadManagementPlugin() {}
             new ActionPlugin.ActionHandler<>(CreateQueryGroupAction.INSTANCE, TransportCreateQueryGroupAction.class),
             new ActionPlugin.ActionHandler<>(GetQueryGroupAction.INSTANCE, TransportGetQueryGroupAction.class),
             new ActionPlugin.ActionHandler<>(DeleteQueryGroupAction.INSTANCE, TransportDeleteQueryGroupAction.class),
-            new ActionPlugin.ActionHandler<>(UpdateQueryGroupAction.INSTANCE, TransportUpdateQueryGroupAction.class)
+            new ActionPlugin.ActionHandler<>(UpdateQueryGroupAction.INSTANCE, TransportUpdateQueryGroupAction.class),
+            new ActionPlugin.ActionHandler<>(GetRuleAction.INSTANCE, TransportGetRuleAction.class)
         );
     }
 
+    @Override
+    public Collection<SystemIndexDescriptor> getSystemIndexDescriptors(Settings settings) {
+        return List.of(new SystemIndexDescriptor(RULE_INDEX, "System index used for storing rules"));
+    }
+
     @Override
     public List<RestHandler> getRestHandlers(
         Settings settings,
@@ -74,7 +87,8 @@ public List<RestHandler> getRestHandlers(
             new RestCreateQueryGroupAction(),
             new RestGetQueryGroupAction(),
             new RestDeleteQueryGroupAction(),
-            new RestUpdateQueryGroupAction()
+            new RestUpdateQueryGroupAction(),
+            new RestGetRuleAction()
         );
     }
 

plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/WorkloadManagementPluginModule.java

@@ -10,7 +10,7 @@
 
 import org.opensearch.common.inject.AbstractModule;
 import org.opensearch.common.inject.Singleton;
-import org.opensearch.plugin.wlm.service.QueryGroupPersistenceService;
+import org.opensearch.plugin.wlm.querygroup.service.QueryGroupPersistenceService;
 
 /**
  * Guice Module to manage WorkloadManagement related objects

plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/action/CreateQueryGroupAction.java renamed to plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/querygroup/action/CreateQueryGroupAction.java

@@ -6,7 +6,7 @@
  * compatible open source license.
  */
 
-package org.opensearch.plugin.wlm.action;
+package org.opensearch.plugin.wlm.querygroup.action;
 
 import org.opensearch.action.ActionType;
 

plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/action/CreateQueryGroupRequest.java renamed to plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/querygroup/action/CreateQueryGroupRequest.java

@@ -6,7 +6,7 @@
  * compatible open source license.
  */
 
-package org.opensearch.plugin.wlm.action;
+package org.opensearch.plugin.wlm.querygroup.action;
 
 import org.opensearch.action.ActionRequestValidationException;
 import org.opensearch.action.support.clustermanager.ClusterManagerNodeRequest;

plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/action/CreateQueryGroupResponse.java renamed to plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/querygroup/action/CreateQueryGroupResponse.java

@@ -6,7 +6,7 @@
  * compatible open source license.
  */
 
-package org.opensearch.plugin.wlm.action;
+package org.opensearch.plugin.wlm.querygroup.action;
 
 import org.opensearch.cluster.metadata.QueryGroup;
 import org.opensearch.core.action.ActionResponse;

plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/action/DeleteQueryGroupAction.java renamed to plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/querygroup/action/DeleteQueryGroupAction.java

@@ -6,7 +6,7 @@
  * compatible open source license.
  */
 
-package org.opensearch.plugin.wlm.action;
+package org.opensearch.plugin.wlm.querygroup.action;
 
 import org.opensearch.action.ActionType;
 import org.opensearch.action.support.master.AcknowledgedResponse;

plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/action/DeleteQueryGroupRequest.java renamed to plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/querygroup/action/DeleteQueryGroupRequest.java

@@ -6,7 +6,7 @@
  * compatible open source license.
  */
 
-package org.opensearch.plugin.wlm.action;
+package org.opensearch.plugin.wlm.querygroup.action;
 
 import org.opensearch.action.ActionRequestValidationException;
 import org.opensearch.action.support.master.AcknowledgedRequest;

plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/action/GetQueryGroupAction.java renamed to plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/querygroup/action/GetQueryGroupAction.java

@@ -6,7 +6,7 @@
  * compatible open source license.
  */
 
-package org.opensearch.plugin.wlm.action;
+package org.opensearch.plugin.wlm.querygroup.action;
 
 import org.opensearch.action.ActionType;
 

plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/action/GetQueryGroupRequest.java renamed to plugins/workload-management/src/main/java/org/opensearch/plugin/wlm/querygroup/action/GetQueryGroupRequest.java

@@ -6,7 +6,7 @@
  * compatible open source license.
  */
 
-package org.opensearch.plugin.wlm.action;
+package org.opensearch.plugin.wlm.querygroup.action;
 
 import org.opensearch.action.ActionRequestValidationException;
 import org.opensearch.action.support.clustermanager.ClusterManagerNodeReadRequest;