org.elasticsearch.cluster.metadata.ComponentTemplateMetadata Maven / Gradle / Ivy
/*
* Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
* or more contributor license agreements. Licensed under the Elastic License
* 2.0 and the Server Side Public License, v 1; you may not use this file except
* in compliance with, at your election, the Elastic License 2.0 or the Server
* Side Public License, v 1.
*/
package org.elasticsearch.cluster.metadata;
import org.elasticsearch.Version;
import org.elasticsearch.cluster.Diff;
import org.elasticsearch.cluster.DiffableUtils;
import org.elasticsearch.cluster.NamedDiff;
import org.elasticsearch.common.Strings;
import org.elasticsearch.common.io.stream.StreamInput;
import org.elasticsearch.common.io.stream.StreamOutput;
import org.elasticsearch.xcontent.ConstructingObjectParser;
import org.elasticsearch.xcontent.ParseField;
import org.elasticsearch.xcontent.XContentBuilder;
import org.elasticsearch.xcontent.XContentParser;
import java.io.IOException;
import java.util.EnumSet;
import java.util.HashMap;
import java.util.Map;
import java.util.Objects;
/**
* {@link ComponentTemplateMetadata} is a custom {@link Metadata} implementation for storing a map
* of component templates and their names.
*/
public class ComponentTemplateMetadata implements Metadata.Custom {
public static final String TYPE = "component_template";
private static final ParseField COMPONENT_TEMPLATE = new ParseField("component_template");
@SuppressWarnings("unchecked")
private static final ConstructingObjectParser PARSER = new ConstructingObjectParser<>(
TYPE,
false,
a -> new ComponentTemplateMetadata((Map) a[0])
);
static {
PARSER.declareObject(ConstructingObjectParser.constructorArg(), (p, c) -> {
Map templates = new HashMap<>();
while (p.nextToken() != XContentParser.Token.END_OBJECT) {
String name = p.currentName();
templates.put(name, ComponentTemplate.parse(p));
}
return templates;
}, COMPONENT_TEMPLATE);
}
private final Map componentTemplates;
public ComponentTemplateMetadata(Map componentTemplates) {
this.componentTemplates = componentTemplates;
}
public ComponentTemplateMetadata(StreamInput in) throws IOException {
this.componentTemplates = in.readMap(StreamInput::readString, ComponentTemplate::new);
}
public Map componentTemplates() {
return this.componentTemplates;
}
@Override
public Diff diff(Metadata.Custom before) {
return new ComponentTemplateMetadataDiff((ComponentTemplateMetadata) before, this);
}
public static NamedDiff readDiffFrom(StreamInput in) throws IOException {
return new ComponentTemplateMetadataDiff(in);
}
@Override
public EnumSet context() {
return Metadata.ALL_CONTEXTS;
}
@Override
public String getWriteableName() {
return TYPE;
}
@Override
public Version getMinimalSupportedVersion() {
return Version.V_7_7_0;
}
@Override
public void writeTo(StreamOutput out) throws IOException {
out.writeMap(this.componentTemplates, StreamOutput::writeString, (stream, val) -> val.writeTo(stream));
}
public static ComponentTemplateMetadata fromXContent(XContentParser parser) throws IOException {
return PARSER.parse(parser, null);
}
@Override
public XContentBuilder toXContent(XContentBuilder builder, Params params) throws IOException {
builder.startObject(COMPONENT_TEMPLATE.getPreferredName());
for (Map.Entry template : componentTemplates.entrySet()) {
builder.field(template.getKey(), template.getValue(), params);
}
builder.endObject();
return builder;
}
@Override
public int hashCode() {
return Objects.hash(this.componentTemplates);
}
@Override
public boolean equals(Object obj) {
if (obj == null) {
return false;
}
if (obj.getClass() != getClass()) {
return false;
}
ComponentTemplateMetadata other = (ComponentTemplateMetadata) obj;
return Objects.equals(this.componentTemplates, other.componentTemplates);
}
@Override
public String toString() {
return Strings.toString(this);
}
static class ComponentTemplateMetadataDiff implements NamedDiff {
final Diff