Register ProfilingUsage as named writeable

danielmitterdorfer · Oct 6, 2023 · 1518e35 · 1518e35
1 parent 8ef3b35
commit 1518e35
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/x-pack/plugin/core/src/main/java/org/elasticsearch/xpack/core/XPackClientPlugin.java b/x-pack/plugin/core/src/main/java/org/elasticsearch/xpack/core/XPackClientPlugin.java
@@ -23,6 +23,7 @@
 import org.elasticsearch.xpack.core.aggregatemetric.AggregateMetricFeatureSetUsage;
 import org.elasticsearch.xpack.core.analytics.AnalyticsFeatureSetUsage;
 import org.elasticsearch.xpack.core.application.EnterpriseSearchFeatureSetUsage;
+import org.elasticsearch.xpack.core.application.ProfilingUsage;
 import org.elasticsearch.xpack.core.archive.ArchiveFeatureSetUsage;
 import org.elasticsearch.xpack.core.ccr.AutoFollowMetadata;
 import org.elasticsearch.xpack.core.datastreams.DataStreamFeatureSetUsage;
@@ -277,7 +278,8 @@ public List<NamedWriteableRegistry.Entry> getNamedWriteables() {
                 XPackFeatureSet.Usage.class,
                 XPackField.ENTERPRISE_SEARCH,
                 EnterpriseSearchFeatureSetUsage::new
-            )
+            ),
+            new NamedWriteableRegistry.Entry(XPackFeatureSet.Usage.class, XPackField.UNIVERSAL_PROFILING, ProfilingUsage::new)
         ).filter(Objects::nonNull).toList();
     }
 

diff --git a/...earch/xpack/profiling/ProfilingUsage.java → ...pack/core/application/ProfilingUsage.java b/...earch/xpack/profiling/ProfilingUsage.java → ...pack/core/application/ProfilingUsage.java
@@ -5,7 +5,7 @@
  * 2.0.
  */
 
-package org.elasticsearch.xpack.profiling;
+package org.elasticsearch.xpack.core.application;
 
 import org.elasticsearch.TransportVersion;
 import org.elasticsearch.TransportVersions;

diff --git a/...filing/src/main/java/org/elasticsearch/xpack/profiling/ProfilingUsageTransportAction.java b/...filing/src/main/java/org/elasticsearch/xpack/profiling/ProfilingUsageTransportAction.java
@@ -22,6 +22,7 @@
 import org.elasticsearch.xpack.core.action.XPackUsageFeatureAction;
 import org.elasticsearch.xpack.core.action.XPackUsageFeatureResponse;
 import org.elasticsearch.xpack.core.action.XPackUsageFeatureTransportAction;
+import org.elasticsearch.xpack.core.application.ProfilingUsage;
 
 public class ProfilingUsageTransportAction extends XPackUsageFeatureTransportAction {
     private final ProfilingLicenseChecker licenseChecker;