forked from elastic/elasticsearch
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add xpack info and usage endpoints for runtime fields
Relates to elastic#59332
- Loading branch information
Showing
7 changed files
with
701 additions
and
8 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
274 changes: 274 additions & 0 deletions
274
...rc/main/java/org/elasticsearch/xpack/core/runtimefields/RuntimeFieldsFeatureSetUsage.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,274 @@ | ||
| /* | ||
| * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one | ||
| * or more contributor license agreements. Licensed under the Elastic License; | ||
| * you may not use this file except in compliance with the Elastic License. | ||
| */ | ||
|
|
||
| package org.elasticsearch.xpack.core.runtimefields; | ||
|
|
||
| import org.elasticsearch.Version; | ||
| import org.elasticsearch.cluster.metadata.IndexMetadata; | ||
| import org.elasticsearch.cluster.metadata.MappingMetadata; | ||
| import org.elasticsearch.common.io.stream.StreamInput; | ||
| import org.elasticsearch.common.io.stream.StreamOutput; | ||
| import org.elasticsearch.common.io.stream.Writeable; | ||
| import org.elasticsearch.common.xcontent.ToXContentObject; | ||
| import org.elasticsearch.common.xcontent.XContentBuilder; | ||
| import org.elasticsearch.xpack.core.XPackFeatureSet; | ||
| import org.elasticsearch.xpack.core.XPackField; | ||
|
|
||
| import java.io.IOException; | ||
| import java.util.ArrayList; | ||
| import java.util.Collections; | ||
| import java.util.Comparator; | ||
| import java.util.HashMap; | ||
| import java.util.HashSet; | ||
| import java.util.List; | ||
| import java.util.Map; | ||
| import java.util.Objects; | ||
| import java.util.Set; | ||
| import java.util.regex.Matcher; | ||
| import java.util.regex.Pattern; | ||
|
|
||
| public class RuntimeFieldsFeatureSetUsage extends XPackFeatureSet.Usage { | ||
|
|
||
| public static RuntimeFieldsFeatureSetUsage fromMetadata(Iterable<IndexMetadata> metadata) { | ||
| Map<String, RuntimeFieldStats> fieldTypes = new HashMap<>(); | ||
| for (IndexMetadata indexMetadata : metadata) { | ||
| if (indexMetadata.isSystem()) { | ||
| // Don't include system indices in statistics about mappings, we care about the user's indices. | ||
| continue; | ||
| } | ||
| Set<String> indexFieldTypes = new HashSet<>(); | ||
| MappingMetadata mappingMetadata = indexMetadata.mapping(); | ||
| if (mappingMetadata != null) { | ||
| Object runtimeObject = mappingMetadata.getSourceAsMap().get("runtime"); | ||
| if (runtimeObject instanceof Map == false) { | ||
| continue; | ||
| } | ||
| Map<?, ?> runtimeMappings = (Map<?, ?>) runtimeObject; | ||
| for (Object runtimeFieldMappingObject : runtimeMappings.values()) { | ||
| if (runtimeFieldMappingObject instanceof Map == false) { | ||
| continue; | ||
| } | ||
| Map<?, ?> runtimeFieldMapping = (Map<?, ?>) runtimeFieldMappingObject; | ||
| Object typeObject = runtimeFieldMapping.get("type"); | ||
| if (typeObject == null) { | ||
| continue; | ||
| } | ||
| String type = typeObject.toString(); | ||
| RuntimeFieldStats stats = fieldTypes.computeIfAbsent(type, RuntimeFieldStats::new); | ||
| stats.count++; | ||
| if (indexFieldTypes.add(type)) { | ||
| stats.indexCount++; | ||
| } | ||
| Object scriptObject = runtimeFieldMapping.get("script"); | ||
| if (scriptObject == null) { | ||
| stats.scriptLessCount++; | ||
| } else if (scriptObject instanceof Map) { | ||
| Map<?, ?> script = (Map<?, ?>) scriptObject; | ||
| Object sourceObject = script.get("source"); | ||
| if (sourceObject != null) { | ||
| String scriptSource = sourceObject.toString(); | ||
| int chars = scriptSource.length(); | ||
| long lines = scriptSource.split("\\n").length; | ||
| int docUsages = countOccurrences(scriptSource, "doc[\\[\\.]"); | ||
| int sourceUsages = countOccurrences(scriptSource, "params\\._source"); | ||
| stats.update(chars, lines, sourceUsages, docUsages); | ||
| } | ||
| Object langObject = script.get("lang"); | ||
| if (langObject != null) { | ||
| stats.scriptLangs.add(langObject.toString()); | ||
| } | ||
| } | ||
| } | ||
| } | ||
| } | ||
| List<RuntimeFieldStats> runtimeFieldStats = new ArrayList<>(fieldTypes.values()); | ||
| runtimeFieldStats.sort(Comparator.comparing(RuntimeFieldStats::type)); | ||
| return new RuntimeFieldsFeatureSetUsage(Collections.unmodifiableList(runtimeFieldStats)); | ||
| } | ||
|
|
||
| private final List<RuntimeFieldStats> stats; | ||
|
|
||
| RuntimeFieldsFeatureSetUsage(List<RuntimeFieldStats> stats) { | ||
| super(XPackField.RUNTIME_FIELDS, true, true); | ||
| this.stats = stats; | ||
| } | ||
|
|
||
| public RuntimeFieldsFeatureSetUsage(StreamInput in) throws IOException { | ||
| super(in); | ||
| this.stats = in.readList(RuntimeFieldStats::new); | ||
| } | ||
|
|
||
| @Override | ||
| public void writeTo(StreamOutput out) throws IOException { | ||
| super.writeTo(out); | ||
| out.writeList(stats); | ||
| } | ||
|
|
||
| List<RuntimeFieldStats> getRuntimeFieldStats() { | ||
| return stats; | ||
| } | ||
|
|
||
| @Override | ||
| protected void innerXContent(XContentBuilder builder, Params params) throws IOException { | ||
| super.innerXContent(builder, params); | ||
| builder.startArray("field_types"); | ||
| for (RuntimeFieldStats stats : stats) { | ||
| stats.toXContent(builder, params); | ||
| } | ||
| builder.endArray(); | ||
| } | ||
|
|
||
| @Override | ||
| public Version getMinimalSupportedVersion() { | ||
| return Version.V_7_11_0; | ||
| } | ||
|
|
||
| @Override | ||
| public boolean equals(Object o) { | ||
| if (this == o) { | ||
| return true; | ||
| } | ||
| if (o == null || getClass() != o.getClass()) { | ||
| return false; | ||
| } | ||
| RuntimeFieldsFeatureSetUsage that = (RuntimeFieldsFeatureSetUsage) o; | ||
| return stats.equals(that.stats); | ||
| } | ||
|
|
||
| @Override | ||
| public int hashCode() { | ||
| return Objects.hash(stats); | ||
| } | ||
|
|
||
| private static int countOccurrences(String script, String keyword) { | ||
| int occurrences = 0; | ||
| Pattern pattern = Pattern.compile(keyword); | ||
| Matcher matcher = pattern.matcher(script); | ||
| while (matcher.find()) { | ||
| occurrences++; | ||
| } | ||
| return occurrences; | ||
| } | ||
|
|
||
| static final class RuntimeFieldStats implements Writeable, ToXContentObject { | ||
| private final String type; | ||
| private int count = 0; | ||
| private int indexCount = 0; | ||
| private final Set<String> scriptLangs; | ||
| private long scriptLessCount = 0; | ||
| private long maxLines = 0; | ||
| private long totalLines = 0; | ||
| private long maxChars = 0; | ||
| private long totalChars = 0; | ||
| private long maxSourceUsages = 0; | ||
| private long totalSourceUsages = 0; | ||
| private long maxDocUsages = 0; | ||
| private long totalDocUsages = 0; | ||
|
|
||
| RuntimeFieldStats(String type) { | ||
| this.type = Objects.requireNonNull(type); | ||
| this.scriptLangs = new HashSet<>(); | ||
| } | ||
|
|
||
| RuntimeFieldStats(StreamInput in) throws IOException { | ||
| this.type = in.readString(); | ||
| this.count = in.readInt(); | ||
| this.indexCount = in.readInt(); | ||
| this.scriptLangs = in.readSet(StreamInput::readString); | ||
| this.scriptLessCount = in.readLong(); | ||
| this.maxLines = in.readLong(); | ||
| this.totalLines = in.readLong(); | ||
| this.maxChars = in.readLong(); | ||
| this.totalChars = in.readLong(); | ||
| this.maxSourceUsages = in.readLong(); | ||
| this.totalSourceUsages = in.readLong(); | ||
| this.maxDocUsages = in.readLong(); | ||
| this.totalDocUsages = in.readLong(); | ||
| } | ||
|
|
||
| String type() { | ||
| return type; | ||
| } | ||
|
|
||
| @Override | ||
| public void writeTo(StreamOutput out) throws IOException { | ||
| out.writeString(type); | ||
| out.writeInt(count); | ||
| out.writeInt(indexCount); | ||
| out.writeCollection(scriptLangs, StreamOutput::writeString); | ||
| out.writeLong(scriptLessCount); | ||
| out.writeLong(maxLines); | ||
| out.writeLong(totalLines); | ||
| out.writeLong(maxChars); | ||
| out.writeLong(totalChars); | ||
| out.writeLong(maxSourceUsages); | ||
| out.writeLong(totalSourceUsages); | ||
| out.writeLong(maxDocUsages); | ||
| out.writeLong(totalDocUsages); | ||
| } | ||
|
|
||
| @Override | ||
| public XContentBuilder toXContent(XContentBuilder builder, Params params) throws IOException { | ||
| builder.startObject(); | ||
| builder.field("name", type); | ||
| builder.field("count", count); | ||
| builder.field("index_count", indexCount); | ||
| builder.field("scriptless_count", scriptLessCount); | ||
| builder.array("lang", scriptLangs.toArray(new String[0])); | ||
| builder.field("lines_max", maxLines); | ||
| builder.field("lines_total", totalLines); | ||
| builder.field("chars_max", maxChars); | ||
| builder.field("chars_total", totalChars); | ||
| builder.field("source_max", maxSourceUsages); | ||
| builder.field("source_total", totalSourceUsages); | ||
| builder.field("doc_max", maxDocUsages); | ||
| builder.field("doc_total", totalDocUsages); | ||
| builder.endObject(); | ||
| return builder; | ||
| } | ||
|
|
||
| void update(int chars, long lines, int sourceUsages, int docUsages) { | ||
| this.maxChars = Math.max(this.maxChars, chars); | ||
| this.totalChars += chars; | ||
| this.maxLines = Math.max(this.maxLines, lines); | ||
| this.totalLines += lines; | ||
| this.totalSourceUsages += sourceUsages; | ||
| this.maxSourceUsages = Math.max(this.maxSourceUsages, sourceUsages); | ||
| this.totalDocUsages += docUsages; | ||
| this.maxDocUsages = Math.max(this.maxDocUsages, docUsages); | ||
| } | ||
|
|
||
| @Override | ||
| public boolean equals(Object o) { | ||
| if (this == o) { | ||
| return true; | ||
| } | ||
| if (o == null || getClass() != o.getClass()) { | ||
| return false; | ||
| } | ||
| RuntimeFieldStats that = (RuntimeFieldStats) o; | ||
| return count == that.count && | ||
| indexCount == that.indexCount && | ||
| scriptLessCount == that.scriptLessCount && | ||
| maxLines == that.maxLines && | ||
| totalLines == that.totalLines && | ||
| maxChars == that.maxChars && | ||
| totalChars == that.totalChars && | ||
| maxSourceUsages == that.maxSourceUsages && | ||
| totalSourceUsages == that.totalSourceUsages && | ||
| maxDocUsages == that.maxDocUsages && | ||
| totalDocUsages == that.totalDocUsages && | ||
| type.equals(that.type) && | ||
| scriptLangs.equals(that.scriptLangs); | ||
| } | ||
|
|
||
| @Override | ||
| public int hashCode() { | ||
| return Objects.hash(type, count, indexCount, scriptLangs, scriptLessCount, maxLines, totalLines, maxChars, totalChars, | ||
| maxSourceUsages, totalSourceUsages, maxDocUsages, totalDocUsages); | ||
| } | ||
| } | ||
| } |
Oops, something went wrong.