org.elasticsearch.cluster.metadata.DataStreamMetadata Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of elasticsearch Show documentation
Show all versions of elasticsearch Show documentation
Elasticsearch subproject :server
/*
* 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.Collections;
import java.util.EnumSet;
import java.util.HashMap;
import java.util.Map;
import java.util.Objects;
/**
* Custom {@link Metadata} implementation for storing a map of {@link DataStream}s and their names.
*/
public class DataStreamMetadata implements Metadata.Custom {
public static final String TYPE = "data_stream";
private static final ParseField DATA_STREAM = new ParseField("data_stream");
private static final ParseField DATA_STREAM_ALIASES = new ParseField("data_stream_aliases");
@SuppressWarnings("unchecked")
private static final ConstructingObjectParser PARSER = new ConstructingObjectParser<>(TYPE, false, args -> {
Map dataStreams = (Map) args[0];
Map dataStreamAliases = (Map) args[1];
if (dataStreamAliases == null) {
dataStreamAliases = Collections.emptyMap();
}
return new DataStreamMetadata(dataStreams, dataStreamAliases);
});
static {
PARSER.declareObject(ConstructingObjectParser.constructorArg(), (p, c) -> {
Map dataStreams = new HashMap<>();
while (p.nextToken() != XContentParser.Token.END_OBJECT) {
String name = p.currentName();
dataStreams.put(name, DataStream.fromXContent(p));
}
return dataStreams;
}, DATA_STREAM);
PARSER.declareObject(ConstructingObjectParser.optionalConstructorArg(), (p, c) -> {
Map dataStreams = new HashMap<>();
while (p.nextToken() != XContentParser.Token.END_OBJECT) {
DataStreamAlias alias = DataStreamAlias.fromXContent(p);
dataStreams.put(alias.getName(), alias);
}
return dataStreams;
}, DATA_STREAM_ALIASES);
}
public static final Version DATA_STREAM_ALIAS_VERSION = Version.V_7_14_0;
private final Map dataStreams;
private final Map dataStreamAliases;
public DataStreamMetadata(Map dataStreams, Map dataStreamAliases) {
this.dataStreams = Collections.unmodifiableMap(new HashMap<>(dataStreams));
this.dataStreamAliases = Collections.unmodifiableMap(new HashMap<>(dataStreamAliases));
}
public DataStreamMetadata(StreamInput in) throws IOException {
this(
in.readMap(StreamInput::readString, DataStream::new),
in.getVersion().onOrAfter(DATA_STREAM_ALIAS_VERSION)
? in.readMap(StreamInput::readString, DataStreamAlias::new)
: Collections.emptyMap()
);
}
public Map dataStreams() {
return this.dataStreams;
}
public Map getDataStreamAliases() {
return dataStreamAliases;
}
@Override
public Diff diff(Metadata.Custom before) {
return new DataStreamMetadata.DataStreamMetadataDiff((DataStreamMetadata) before, this);
}
public static NamedDiff readDiffFrom(StreamInput in) throws IOException {
return new DataStreamMetadata.DataStreamMetadataDiff(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.dataStreams, StreamOutput::writeString, (stream, val) -> val.writeTo(stream));
if (out.getVersion().onOrAfter(DATA_STREAM_ALIAS_VERSION)) {
out.writeMap(this.dataStreamAliases, StreamOutput::writeString, (stream, val) -> val.writeTo(stream));
}
}
public static DataStreamMetadata fromXContent(XContentParser parser) throws IOException {
return PARSER.parse(parser, null);
}
@Override
public XContentBuilder toXContent(XContentBuilder builder, Params params) throws IOException {
builder.xContentValuesMap(DATA_STREAM.getPreferredName(), dataStreams);
builder.startObject(DATA_STREAM_ALIASES.getPreferredName());
for (Map.Entry dataStream : dataStreamAliases.entrySet()) {
dataStream.getValue().toXContent(builder, params);
}
builder.endObject();
return builder;
}
@Override
public int hashCode() {
return Objects.hash(this.dataStreams, dataStreamAliases);
}
@Override
public boolean equals(Object obj) {
if (obj == null) {
return false;
}
if (obj.getClass() != getClass()) {
return false;
}
DataStreamMetadata other = (DataStreamMetadata) obj;
return Objects.equals(this.dataStreams, other.dataStreams) && Objects.equals(this.dataStreamAliases, other.dataStreamAliases);
}
@Override
public String toString() {
return Strings.toString(this);
}
static class DataStreamMetadataDiff implements NamedDiff {
final Diff