honeymoose
/
OpenSearch
mirror of https://github.com/honeymoose/OpenSearch.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[Rollup] Improve ID scheme for rollup documents (#32558) 

Previously, we were using a simple CRC32 for the IDs of rollup documents.
This is a very poor choice however, since 32bit IDs leads to collisions
between documents very quickly.

This commit moves Rollups over to a 128bit ID.  The ID is a concatenation
of all the keys in the document (similar to the rolling CRC before),
hashed with 128bit Murmur3, then base64 encoded.  Finally, the job
ID and a delimiter (`$`) are prepended to the ID.

This gurantees that there are 128bits per-job.  128bits should
essentially remove all chances of collisions, and the prepended
job ID means that _if_ there is a collision, it stays "within"
the job.

BWC notes:

We can only upgrade the ID scheme after we know there has been a good
checkpoint during indexing.  We don't rely on a STARTED/STOPPED
status since we can't guarantee that resulted from a real checkpoint,
or other state.  So we only upgrade the ID after we have reached
a checkpoint state during an active index run, and only after the
checkpoint has been confirmed.

Once a job has been upgraded and checkpointed, the version increments
and the new ID is used in the future.  All new jobs use the
new ID from the start
This commit is contained in:
Zachary Tong 2018-08-03 11:13:25 -04:00 committed by GitHub
parent 3d4c84f7ca
commit fc9fb64ad5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
21 changed files with 1054 additions and 170 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
x-pack/docs/en/rest-api/rollup/get-job.asciidoc
View File

@ -93,7 +93,8 @@ Which will yield the following response:
"page_size" : 1000 "page_size" : 1000
}, },
"status" : { "status" : {
"job_state" : "stopped" "job_state" : "stopped",
"upgraded_doc_id": true
}, },
"stats" : { "stats" : {
"pages_processed" : 0, "pages_processed" : 0,
@ -212,7 +213,8 @@ Which will yield the following response:
"page_size" : 1000 "page_size" : 1000
}, },
"status" : { "status" : {
"job_state" : "stopped" "job_state" : "stopped",
"upgraded_doc_id": true
}, },
"stats" : { "stats" : {
"pages_processed" : 0, "pages_processed" : 0,
@ -260,7 +262,8 @@ Which will yield the following response:
"page_size" : 1000 "page_size" : 1000
}, },
"status" : { "status" : {
"job_state" : "stopped" "job_state" : "stopped",
"upgraded_doc_id": true
}, },
"stats" : { "stats" : {
"pages_processed" : 0, "pages_processed" : 0,

37
x-pack/plugin/core/src/main/java/org/elasticsearch/xpack/core/rollup/job/RollupJobStatus.java
View File

@ -6,6 +6,7 @@
package org.elasticsearch.xpack.core.rollup.job; package org.elasticsearch.xpack.core.rollup.job;
import org.elasticsearch.Version;
import org.elasticsearch.common.Nullable; import org.elasticsearch.common.Nullable;
import org.elasticsearch.common.ParseField; import org.elasticsearch.common.ParseField;
import org.elasticsearch.common.io.stream.StreamInput; import org.elasticsearch.common.io.stream.StreamInput;
@ -39,12 +40,19 @@ public class RollupJobStatus implements Task.Status, PersistentTaskState {
@Nullable @Nullable
private final TreeMap<String, Object> currentPosition; private final TreeMap<String, Object> currentPosition;
// Flag holds the state of the ID scheme, e.g. if it has been upgraded to the
// concatenation scheme. See #32372 for more details
private boolean upgradedDocumentID;
private static final ParseField STATE = new ParseField("job_state"); private static final ParseField STATE = new ParseField("job_state");
private static final ParseField CURRENT_POSITION = new ParseField("current_position"); private static final ParseField CURRENT_POSITION = new ParseField("current_position");
private static final ParseField UPGRADED_DOC_ID = new ParseField("upgraded_doc_id");
public static final ConstructingObjectParser<RollupJobStatus, Void> PARSER = public static final ConstructingObjectParser<RollupJobStatus, Void> PARSER =
new ConstructingObjectParser<>(NAME, new ConstructingObjectParser<>(NAME,
args -> new RollupJobStatus((IndexerState) args[0], (HashMap<String, Object>) args[1])); args -> new RollupJobStatus((IndexerState) args[0],
(HashMap<String, Object>) args[1],
(Boolean)args[2]));
static { static {
PARSER.declareField(constructorArg(), p -> { PARSER.declareField(constructorArg(), p -> {
@ -62,16 +70,28 @@ public class RollupJobStatus implements Task.Status, PersistentTaskState {
} }
throw new IllegalArgumentException("Unsupported token [" + p.currentToken() + "]"); throw new IllegalArgumentException("Unsupported token [" + p.currentToken() + "]");
}, CURRENT_POSITION, ObjectParser.ValueType.VALUE_OBJECT_ARRAY); }, CURRENT_POSITION, ObjectParser.ValueType.VALUE_OBJECT_ARRAY);
// Optional to accommodate old versions of state
PARSER.declareBoolean(ConstructingObjectParser.optionalConstructorArg(), UPGRADED_DOC_ID);
} }
public RollupJobStatus(IndexerState state, @Nullable Map<String, Object> position) { public RollupJobStatus(IndexerState state, @Nullable Map<String, Object> position,
@Nullable Boolean upgradedDocumentID) {
this.state = state; this.state = state;
this.currentPosition = position == null ? null : new TreeMap<>(position); this.currentPosition = position == null ? null : new TreeMap<>(position);
this.upgradedDocumentID = upgradedDocumentID != null ? upgradedDocumentID : false; //default to false if missing
} }
public RollupJobStatus(StreamInput in) throws IOException { public RollupJobStatus(StreamInput in) throws IOException {
state = IndexerState.fromStream(in); state = IndexerState.fromStream(in);
currentPosition = in.readBoolean() ? new TreeMap<>(in.readMap()) : null; currentPosition = in.readBoolean() ? new TreeMap<>(in.readMap()) : null;
if (in.getVersion().onOrAfter(Version.V_7_0_0_alpha1)) { //TODO change this after backport
upgradedDocumentID = in.readBoolean();
} else {
// If we're getting this job from a pre-6.4.0 node,
// it is using the old ID scheme
upgradedDocumentID = false;
}
} }
public IndexerState getIndexerState() { public IndexerState getIndexerState() {
@ -82,6 +102,10 @@ public class RollupJobStatus implements Task.Status, PersistentTaskState {
return currentPosition; return currentPosition;
} }
public boolean isUpgradedDocumentID() {
return upgradedDocumentID;
}
public static RollupJobStatus fromXContent(XContentParser parser) { public static RollupJobStatus fromXContent(XContentParser parser) {
try { try {
return PARSER.parse(parser, null); return PARSER.parse(parser, null);
@ -97,6 +121,7 @@ public class RollupJobStatus implements Task.Status, PersistentTaskState {
if (currentPosition != null) { if (currentPosition != null) {
builder.field(CURRENT_POSITION.getPreferredName(), currentPosition); builder.field(CURRENT_POSITION.getPreferredName(), currentPosition);
} }
builder.field(UPGRADED_DOC_ID.getPreferredName(), upgradedDocumentID);
builder.endObject(); builder.endObject();
return builder; return builder;
} }
@ -113,6 +138,9 @@ public class RollupJobStatus implements Task.Status, PersistentTaskState {
if (currentPosition != null) { if (currentPosition != null) {
out.writeMap(currentPosition); out.writeMap(currentPosition);
} }
if (out.getVersion().onOrAfter(Version.V_7_0_0_alpha1)) { //TODO change this after backport
out.writeBoolean(upgradedDocumentID);
}
} }
@Override @Override
@ -128,11 +156,12 @@ public class RollupJobStatus implements Task.Status, PersistentTaskState {
RollupJobStatus that = (RollupJobStatus) other; RollupJobStatus that = (RollupJobStatus) other;
return Objects.equals(this.state, that.state) return Objects.equals(this.state, that.state)
&& Objects.equals(this.currentPosition, that.currentPosition); && Objects.equals(this.currentPosition, that.currentPosition)
&& Objects.equals(this.upgradedDocumentID, that.upgradedDocumentID);
} }
@Override @Override
public int hashCode() { public int hashCode() {
return Objects.hash(state, currentPosition); return Objects.hash(state, currentPosition, upgradedDocumentID);
} }
} }

2
x-pack/plugin/core/src/test/java/org/elasticsearch/xpack/core/rollup/job/JobWrapperSerializingTests.java
View File

@ -41,6 +41,6 @@ public class JobWrapperSerializingTests extends AbstractSerializingTestCase<GetR
return new GetRollupJobsAction.JobWrapper(ConfigTestHelpers.getRollupJob(randomAlphaOfLength(5)).build(), return new GetRollupJobsAction.JobWrapper(ConfigTestHelpers.getRollupJob(randomAlphaOfLength(5)).build(),
new RollupJobStats(randomNonNegativeLong(), randomNonNegativeLong(), randomNonNegativeLong(), randomNonNegativeLong()), new RollupJobStats(randomNonNegativeLong(), randomNonNegativeLong(), randomNonNegativeLong(), randomNonNegativeLong()),
new RollupJobStatus(state, Collections.emptyMap())); new RollupJobStatus(state, Collections.emptyMap(), randomBoolean()));
} }
} }

4
x-pack/plugin/core/src/test/java/org/elasticsearch/xpack/core/rollup/job/RollupJobStatusTests.java
View File

@ -8,8 +8,6 @@ package org.elasticsearch.xpack.core.rollup.job;
import org.elasticsearch.common.io.stream.Writeable; import org.elasticsearch.common.io.stream.Writeable;
import org.elasticsearch.common.xcontent.XContentParser; import org.elasticsearch.common.xcontent.XContentParser;
import org.elasticsearch.test.AbstractSerializingTestCase; import org.elasticsearch.test.AbstractSerializingTestCase;
import org.elasticsearch.xpack.core.rollup.job.IndexerState;
import org.elasticsearch.xpack.core.rollup.job.RollupJobStatus;
import java.util.HashMap; import java.util.HashMap;
import java.util.Map; import java.util.Map;
@ -35,7 +33,7 @@ public class RollupJobStatusTests extends AbstractSerializingTestCase<RollupJobS
@Override @Override
protected RollupJobStatus createTestInstance() { protected RollupJobStatus createTestInstance() {
return new RollupJobStatus(randomFrom(IndexerState.values()), randomPosition()); return new RollupJobStatus(randomFrom(IndexerState.values()), randomPosition(), randomBoolean());
} }
@Override @Override

9
x-pack/plugin/rollup/src/main/java/org/elasticsearch/xpack/rollup/Rollup.java
View File

@ -80,7 +80,14 @@ import static java.util.Collections.emptyList;
public class Rollup extends Plugin implements ActionPlugin, PersistentTaskPlugin { public class Rollup extends Plugin implements ActionPlugin, PersistentTaskPlugin {
public static final String BASE_PATH = "/_xpack/rollup/"; public static final String BASE_PATH = "/_xpack/rollup/";
public static final int ROLLUP_VERSION = 1;
// Introduced in ES version 6.3
public static final int ROLLUP_VERSION_V1 = 1;
// Introduced in ES Version 6.4
// Bumped due to ID collision, see #32372
public static final int ROLLUP_VERSION_V2 = 2;
public static final int CURRENT_ROLLUP_VERSION = ROLLUP_VERSION_V2;
public static final String TASK_THREAD_POOL_NAME = RollupField.NAME + "_indexing"; public static final String TASK_THREAD_POOL_NAME = RollupField.NAME + "_indexing";
public static final String SCHEDULE_THREAD_POOL_NAME = RollupField.NAME + "_scheduler"; public static final String SCHEDULE_THREAD_POOL_NAME = RollupField.NAME + "_scheduler";

4
x-pack/plugin/rollup/src/main/java/org/elasticsearch/xpack/rollup/action/TransportRollupSearchAction.java
View File

@ -191,7 +191,9 @@ public class TransportRollupSearchAction extends TransportAction<SearchRequest,
copiedSource.query(new BoolQueryBuilder() copiedSource.query(new BoolQueryBuilder()
.must(rewritten) .must(rewritten)
.filter(new TermQueryBuilder(RollupField.formatMetaField(RollupField.ID.getPreferredName()), id)) .filter(new TermQueryBuilder(RollupField.formatMetaField(RollupField.ID.getPreferredName()), id))
.filter(new TermQueryBuilder(RollupField.formatMetaField(RollupField.VERSION_FIELD), Rollup.ROLLUP_VERSION))); // Both versions are acceptable right now since they are compatible at search time
.filter(new TermsQueryBuilder(RollupField.formatMetaField(RollupField.VERSION_FIELD),
new long[]{Rollup.ROLLUP_VERSION_V1, Rollup.ROLLUP_VERSION_V2})));
// And add a new msearch per JobID // And add a new msearch per JobID
msearch.add(new SearchRequest(context.getRollupIndices(), copiedSource).types(request.types())); msearch.add(new SearchRequest(context.getRollupIndices(), copiedSource).types(request.types()));

54
x-pack/plugin/rollup/src/main/java/org/elasticsearch/xpack/rollup/job/IndexerUtils.java
View File

@ -8,7 +8,6 @@ package org.elasticsearch.xpack.rollup.job;
import org.apache.log4j.Logger; import org.apache.log4j.Logger;
import org.elasticsearch.ElasticsearchException; import org.elasticsearch.ElasticsearchException;
import org.elasticsearch.action.index.IndexRequest; import org.elasticsearch.action.index.IndexRequest;
import org.elasticsearch.common.Numbers;
import org.elasticsearch.search.aggregations.Aggregation; import org.elasticsearch.search.aggregations.Aggregation;
import org.elasticsearch.search.aggregations.bucket.composite.CompositeAggregation; import org.elasticsearch.search.aggregations.bucket.composite.CompositeAggregation;
import org.elasticsearch.search.aggregations.bucket.histogram.DateHistogramAggregationBuilder; import org.elasticsearch.search.aggregations.bucket.histogram.DateHistogramAggregationBuilder;
@ -21,14 +20,12 @@ import org.elasticsearch.xpack.core.rollup.job.GroupConfig;
import org.elasticsearch.xpack.core.rollup.job.RollupJobStats; import org.elasticsearch.xpack.core.rollup.job.RollupJobStats;
import org.elasticsearch.xpack.rollup.Rollup; import org.elasticsearch.xpack.rollup.Rollup;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.HashMap; import java.util.HashMap;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
import java.util.TreeMap; import java.util.TreeMap;
import java.util.stream.Collectors; import java.util.stream.Collectors;
import java.util.zip.CRC32;
/** /**
* These utilities are used to convert agg responses into a set of rollup documents. * These utilities are used to convert agg responses into a set of rollup documents.
@ -43,10 +40,14 @@ class IndexerUtils {
* *
* @param agg The aggregation response that you want to rollup * @param agg The aggregation response that you want to rollup
* @param rollupIndex The index that holds rollups for this job * @param rollupIndex The index that holds rollups for this job
* @param stats The stats accumulator for this job's task
* @param groupConfig The grouping configuration for the job
* @param jobId The ID for the job
* @param isUpgradedDocID `true` if this job is using the new ID scheme
* @return A list of rolled documents derived from the response * @return A list of rolled documents derived from the response
*/ */
static List<IndexRequest> processBuckets(CompositeAggregation agg, String rollupIndex, RollupJobStats stats, static List<IndexRequest> processBuckets(CompositeAggregation agg, String rollupIndex, RollupJobStats stats,
GroupConfig groupConfig, String jobId) { GroupConfig groupConfig, String jobId, boolean isUpgradedDocID) {
logger.debug("Buckets: [" + agg.getBuckets().size() + "][" + jobId + "]"); logger.debug("Buckets: [" + agg.getBuckets().size() + "][" + jobId + "]");
return agg.getBuckets().stream().map(b ->{ return agg.getBuckets().stream().map(b ->{
@ -57,24 +58,30 @@ class IndexerUtils {
TreeMap<String, Object> keys = new TreeMap<>(b.getKey()); TreeMap<String, Object> keys = new TreeMap<>(b.getKey());
List<Aggregation> metrics = b.getAggregations().asList(); List<Aggregation> metrics = b.getAggregations().asList();
RollupIDGenerator idGenerator;
if (isUpgradedDocID) {
idGenerator = new RollupIDGenerator.Murmur3(jobId);
} else {
idGenerator = new RollupIDGenerator.CRC();
}
Map<String, Object> doc = new HashMap<>(keys.size() + metrics.size()); Map<String, Object> doc = new HashMap<>(keys.size() + metrics.size());
CRC32 docId = processKeys(keys, doc, b.getDocCount(), groupConfig);
byte[] vs = jobId.getBytes(StandardCharsets.UTF_8); processKeys(keys, doc, b.getDocCount(), groupConfig, idGenerator);
docId.update(vs, 0, vs.length); idGenerator.add(jobId);
processMetrics(metrics, doc); processMetrics(metrics, doc);
doc.put(RollupField.ROLLUP_META + "." + RollupField.VERSION_FIELD, Rollup.ROLLUP_VERSION); doc.put(RollupField.ROLLUP_META + "." + RollupField.VERSION_FIELD,
isUpgradedDocID ? Rollup.CURRENT_ROLLUP_VERSION : Rollup.ROLLUP_VERSION_V1);
doc.put(RollupField.ROLLUP_META + "." + RollupField.ID.getPreferredName(), jobId); doc.put(RollupField.ROLLUP_META + "." + RollupField.ID.getPreferredName(), jobId);
IndexRequest request = new IndexRequest(rollupIndex, RollupField.TYPE_NAME, String.valueOf(docId.getValue())); IndexRequest request = new IndexRequest(rollupIndex, RollupField.TYPE_NAME, idGenerator.getID());
request.source(doc); request.source(doc);
return request; return request;
}).collect(Collectors.toList()); }).collect(Collectors.toList());
} }
private static CRC32 processKeys(Map<String, Object> keys, Map<String, Object> doc, long count, GroupConfig groupConfig) { private static void processKeys(Map<String, Object> keys, Map<String, Object> doc,
CRC32 docID = new CRC32(); long count, GroupConfig groupConfig, RollupIDGenerator idGenerator) {
keys.forEach((k, v) -> { keys.forEach((k, v) -> {
// Also add a doc count for each key. This will duplicate data, but makes search easier later // Also add a doc count for each key. This will duplicate data, but makes search easier later
doc.put(k + "." + RollupField.COUNT_FIELD, count); doc.put(k + "." + RollupField.COUNT_FIELD, count);
@ -83,37 +90,34 @@ class IndexerUtils {
assert v != null; assert v != null;
doc.put(k + "." + RollupField.TIMESTAMP, v); doc.put(k + "." + RollupField.TIMESTAMP, v);
doc.put(k + "." + RollupField.INTERVAL, groupConfig.getDateHisto().getInterval()); doc.put(k + "." + RollupField.INTERVAL, groupConfig.getDateHisto().getInterval());
doc.put(k + "." + DateHistogramGroupConfig.TIME_ZONE, groupConfig.getDateHisto().getTimeZone()); doc.put(k + "." + DateHistogramGroupConfig.TIME_ZONE, groupConfig.getDateHisto().getTimeZone().toString());
docID.update(Numbers.longToBytes((Long)v), 0, 8); idGenerator.add((Long)v);
} else if (k.endsWith("." + HistogramAggregationBuilder.NAME)) { } else if (k.endsWith("." + HistogramAggregationBuilder.NAME)) {
doc.put(k + "." + RollupField.VALUE, v); doc.put(k + "." + RollupField.VALUE, v);
doc.put(k + "." + RollupField.INTERVAL, groupConfig.getHisto().getInterval()); doc.put(k + "." + RollupField.INTERVAL, groupConfig.getHisto().getInterval());
if (v == null) { if (v == null) {
// Arbitrary value to update the doc ID with for nulls idGenerator.addNull();
docID.update(19);
} else { } else {
docID.update(Numbers.doubleToBytes((Double) v), 0, 8); idGenerator.add((Double) v);
} }
} else if (k.endsWith("." + TermsAggregationBuilder.NAME)) { } else if (k.endsWith("." + TermsAggregationBuilder.NAME)) {
doc.put(k + "." + RollupField.VALUE, v); doc.put(k + "." + RollupField.VALUE, v);
if (v == null) { if (v == null) {
// Arbitrary value to update the doc ID with for nulls idGenerator.addNull();
docID.update(19);
} else if (v instanceof String) { } else if (v instanceof String) {
byte[] vs = ((String) v).getBytes(StandardCharsets.UTF_8); idGenerator.add((String)v);
docID.update(vs, 0, vs.length);
} else if (v instanceof Long) { } else if (v instanceof Long) {
docID.update(Numbers.longToBytes((Long)v), 0, 8); idGenerator.add((Long)v);
} else if (v instanceof Double) { } else if (v instanceof Double) {
docID.update(Numbers.doubleToBytes((Double)v), 0, 8); idGenerator.add((Double)v);
} else { } else {
throw new RuntimeException("Encountered value of type [" + v.getClass() + "], which was unable to be processed."); throw new RuntimeException("Encountered value of type ["
+ v.getClass() + "], which was unable to be processed.");
} }
} else { } else {
throw new ElasticsearchException("Could not identify key in agg [" + k + "]"); throw new ElasticsearchException("Could not identify key in agg [" + k + "]");
} }
}); });
return docID;
} }
private static void processMetrics(List<Aggregation> metrics, Map<String, Object> doc) { private static void processMetrics(List<Aggregation> metrics, Map<String, Object> doc) {

178
x-pack/plugin/rollup/src/main/java/org/elasticsearch/xpack/rollup/job/RollupIDGenerator.java Normal file
View File

@ -0,0 +1,178 @@
/*
* 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.rollup.job;
import org.apache.lucene.util.BytesRef;
import org.apache.lucene.util.BytesRefBuilder;
import org.elasticsearch.common.Numbers;
import org.elasticsearch.common.hash.MurmurHash3;
import java.nio.charset.StandardCharsets;
import java.util.Base64;
import java.util.zip.CRC32;
/**
* The ID Generator creates a deterministic document ID to be used for rollup docs.
* It does this by accepting values (e.g. composite keys for the rollup bucket) and
* hashes those together in a deterministic manner.
*
* Once the ID has generated, the generator instance becomes locked and will not
* accept any more values. This is a safety mechanism to prevent accidentally
* changing the ID at runtime.
*
* NOTE: this class is not thread safe; there is no synchronization on the "generated"
* flag and it is unsafe to use from multiple threads.
*/
public abstract class RollupIDGenerator {
public abstract void add(Integer v);
public abstract void add(Long v);
public abstract void add(Double v);
public abstract void add(String v);
public abstract void addNull();
public abstract String getID();
private boolean generated = false;
final boolean check(Object v) {
if (generated) {
throw new RuntimeException("Cannot update ID as it has already been generated.");
}
if (v == null) {
addNull();
return false;
}
return true;
}
final void setFlag() {
if (generated) {
throw new RuntimeException("Cannot generate ID as it has already been generated.");
}
generated = true;
}
/**
* The "old" style ID used in Rollup V1. A rolling 32 bit CRC.
*
* Null values are hashed as (int)19.
*/
@Deprecated
public static class CRC extends RollupIDGenerator {
private final CRC32 crc = new CRC32();
@Override
public void add(Integer v) {
if (check(v)) {
crc.update(v);
}
}
@Override
public void add(Long v) {
if (check(v)) {
crc.update(Numbers.longToBytes(v), 0, 8);
}
}
@Override
public void add(Double v) {
if (check(v)) {
crc.update(Numbers.doubleToBytes(v), 0, 8);
}
}
@Override
public void add(String v) {
if (check(v)) {
byte[] vs = (v).getBytes(StandardCharsets.UTF_8);
crc.update(vs, 0, vs.length);
}
}
@Override
public void addNull() {
// Old ID scheme used (int)19 as the null placeholder.
// Not great but we're stuck with it :(
crc.update(19);
}
@Override
public String getID() {
setFlag();
return String.valueOf(crc.getValue());
}
}
/**
* The "new" style ID, used in Rollup V2. A 128 bit Murmur3 hash of
* all the keys concatenated together, base64-encoded, then prepended
* with the job ID and a `$` delimiter
*
* Null values are hashed as a (hopefully) unique string `__NULL_PLACEHOLDER__830f1de2__`
*/
public static class Murmur3 extends RollupIDGenerator {
private static final long SEED = 19;
private static final BytesRef DELIM = new BytesRef("$");
private static final BytesRef NULL_PLACEHOLDER = new BytesRef("__NULL_PLACEHOLDER__830f1de2__");
private final BytesRefBuilder id = new BytesRefBuilder();
private final String jobId;
Murmur3(String jobId) {
this.jobId = jobId;
}
@Override
public void add(Integer v) {
if (check(v)) {
update(Numbers.intToBytes(v));
}
}
@Override
public void add(Long v) {
if (check(v)) {
update(Numbers.longToBytes(v));
}
}
@Override
public void add(Double v) {
if (check(v)) {
update(Numbers.doubleToBytes(v));
}
}
@Override
public void add(String v) {
if (check(v)) {
update((v).getBytes(StandardCharsets.UTF_8));
}
}
@Override
public void addNull() {
// New ID scheme uses a (hopefully) unique placeholder for null
update(NULL_PLACEHOLDER.bytes);
}
private void update(byte[] v) {
id.append(v, 0, v.length);
id.append(DELIM);
}
@Override
public String getID() {
setFlag();
MurmurHash3.Hash128 hasher
= MurmurHash3.hash128(id.bytes(), 0, id.length(), SEED, new MurmurHash3.Hash128());
byte[] hashedBytes = new byte[16];
System.arraycopy(Numbers.longToBytes(hasher.h1), 0, hashedBytes, 0, 8);
System.arraycopy(Numbers.longToBytes(hasher.h2), 0, hashedBytes, 8, 8);
return jobId + "$" + Base64.getUrlEncoder().withoutPadding().encodeToString(hashedBytes);
}
}
}

17
x-pack/plugin/rollup/src/main/java/org/elasticsearch/xpack/rollup/job/RollupIndexer.java
View File

@ -34,6 +34,7 @@ import java.util.HashMap;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
import java.util.concurrent.Executor; import java.util.concurrent.Executor;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.atomic.AtomicReference; import java.util.concurrent.atomic.AtomicReference;
/** /**
@ -54,6 +55,7 @@ public abstract class RollupIndexer {
private final AtomicReference<IndexerState> state; private final AtomicReference<IndexerState> state;
private final AtomicReference<Map<String, Object>> position; private final AtomicReference<Map<String, Object>> position;
private final Executor executor; private final Executor executor;
protected final AtomicBoolean upgradedDocumentID;
private final CompositeAggregationBuilder compositeBuilder; private final CompositeAggregationBuilder compositeBuilder;
private long maxBoundary; private long maxBoundary;
@ -65,13 +67,15 @@ public abstract class RollupIndexer {
* @param initialState Initial state for the indexer * @param initialState Initial state for the indexer
* @param initialPosition The last indexed bucket of the task * @param initialPosition The last indexed bucket of the task
*/ */
RollupIndexer(Executor executor, RollupJob job, AtomicReference<IndexerState> initialState, Map<String, Object> initialPosition) { RollupIndexer(Executor executor, RollupJob job, AtomicReference<IndexerState> initialState,
Map<String, Object> initialPosition, AtomicBoolean upgradedDocumentID) {
this.executor = executor; this.executor = executor;
this.job = job; this.job = job;
this.stats = new RollupJobStats(); this.stats = new RollupJobStats();
this.state = initialState; this.state = initialState;
this.position = new AtomicReference<>(initialPosition); this.position = new AtomicReference<>(initialPosition);
this.compositeBuilder = createCompositeBuilder(job.getConfig()); this.compositeBuilder = createCompositeBuilder(job.getConfig());
this.upgradedDocumentID = upgradedDocumentID;
} }
/** /**
@ -140,6 +144,13 @@ public abstract class RollupIndexer {
return stats; return stats;
} }
/**
* Returns if this job has upgraded it's ID scheme yet or not
*/
public boolean isUpgradedDocumentID() {
return upgradedDocumentID.get();
}
/** /**
* Sets the internal state to {@link IndexerState#STARTED} if the previous state was {@link IndexerState#STOPPED}. Setting the state to * Sets the internal state to {@link IndexerState#STARTED} if the previous state was {@link IndexerState#STOPPED}. Setting the state to
* STARTED allows a job to run in the background when {@link #maybeTriggerAsyncJob(long)} is called. * STARTED allows a job to run in the background when {@link #maybeTriggerAsyncJob(long)} is called.
@ -312,8 +323,10 @@ public abstract class RollupIndexer {
} }
final BulkRequest bulkRequest = new BulkRequest(); final BulkRequest bulkRequest = new BulkRequest();
// Indexer is single-threaded, and only place that the ID scheme can get upgraded is doSaveState(), so
// we can pass down the boolean value rather than the atomic here
final List<IndexRequest> docs = IndexerUtils.processBuckets(response, job.getConfig().getRollupIndex(), final List<IndexRequest> docs = IndexerUtils.processBuckets(response, job.getConfig().getRollupIndex(),
stats, job.getConfig().getGroupConfig(), job.getConfig().getId()); stats, job.getConfig().getGroupConfig(), job.getConfig().getId(), upgradedDocumentID.get());
docs.forEach(bulkRequest::add); docs.forEach(bulkRequest::add);
assert bulkRequest.requests().size() > 0; assert bulkRequest.requests().size() > 0;
doNextBulk(bulkRequest, doNextBulk(bulkRequest,

56
x-pack/plugin/rollup/src/main/java/org/elasticsearch/xpack/rollup/job/RollupJobTask.java
View File

@ -37,6 +37,7 @@ import org.elasticsearch.xpack.core.scheduler.SchedulerEngine;
import org.elasticsearch.xpack.rollup.Rollup; import org.elasticsearch.xpack.rollup.Rollup;
import java.util.Map; import java.util.Map;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.atomic.AtomicReference; import java.util.concurrent.atomic.AtomicReference;
/** /**
@ -97,8 +98,10 @@ public class RollupJobTask extends AllocatedPersistentTask implements SchedulerE
private final Client client; private final Client client;
private final RollupJob job; private final RollupJob job;
ClientRollupPageManager(RollupJob job, IndexerState initialState, Map<String, Object> initialPosition, Client client) { ClientRollupPageManager(RollupJob job, IndexerState initialState, Map<String, Object> initialPosition,
super(threadPool.executor(ThreadPool.Names.GENERIC), job, new AtomicReference<>(initialState), initialPosition); Client client, AtomicBoolean upgradedDocumentID) {
super(threadPool.executor(ThreadPool.Names.GENERIC), job, new AtomicReference<>(initialState),
initialPosition, upgradedDocumentID);
this.client = client; this.client = client;
this.job = job; this.job = job;
} }
@ -122,9 +125,16 @@ public class RollupJobTask extends AllocatedPersistentTask implements SchedulerE
next.run(); next.run();
} else { } else {
// Otherwise, attempt to persist our state // Otherwise, attempt to persist our state
final RollupJobStatus state = new RollupJobStatus(indexerState, getPosition()); // Upgrade to the new ID scheme while we are at it
boolean oldState = upgradedDocumentID.getAndSet(true);
final RollupJobStatus state = new RollupJobStatus(indexerState, getPosition(), upgradedDocumentID.get());
logger.debug("Updating persistent state of job [" + job.getConfig().getId() + "] to [" + indexerState.toString() + "]"); logger.debug("Updating persistent state of job [" + job.getConfig().getId() + "] to [" + indexerState.toString() + "]");
updatePersistentTaskState(state, ActionListener.wrap(task -> next.run(), exc -> next.run())); updatePersistentTaskState(state, ActionListener.wrap(task -> next.run(), exc -> {
// We failed to update the persistent task for some reason,
// set our flag back to what it was before
upgradedDocumentID.set(oldState);
next.run();
}));
} }
} }
@ -148,6 +158,7 @@ public class RollupJobTask extends AllocatedPersistentTask implements SchedulerE
private final SchedulerEngine schedulerEngine; private final SchedulerEngine schedulerEngine;
private final ThreadPool threadPool; private final ThreadPool threadPool;
private final RollupIndexer indexer; private final RollupIndexer indexer;
private AtomicBoolean upgradedDocumentID;
RollupJobTask(long id, String type, String action, TaskId parentTask, RollupJob job, RollupJobStatus state, RollupJobTask(long id, String type, String action, TaskId parentTask, RollupJob job, RollupJobStatus state,
Client client, SchedulerEngine schedulerEngine, ThreadPool threadPool, Map<String, String> headers) { Client client, SchedulerEngine schedulerEngine, ThreadPool threadPool, Map<String, String> headers) {
@ -156,6 +167,9 @@ public class RollupJobTask extends AllocatedPersistentTask implements SchedulerE
this.schedulerEngine = schedulerEngine; this.schedulerEngine = schedulerEngine;
this.threadPool = threadPool; this.threadPool = threadPool;
// We can assume the new ID scheme only for new jobs
this.upgradedDocumentID = new AtomicBoolean(true);
// If status is not null, we are resuming rather than starting fresh. // If status is not null, we are resuming rather than starting fresh.
Map<String, Object> initialPosition = null; Map<String, Object> initialPosition = null;
IndexerState initialState = IndexerState.STOPPED; IndexerState initialState = IndexerState.STOPPED;
@ -169,25 +183,35 @@ public class RollupJobTask extends AllocatedPersistentTask implements SchedulerE
* it is indexing but without the actual indexing thread running. * it is indexing but without the actual indexing thread running.
*/ */
initialState = IndexerState.STARTED; initialState = IndexerState.STARTED;
} else if (existingState.equals(IndexerState.ABORTING) || existingState.equals(IndexerState.STOPPING)) { } else if (existingState.equals(IndexerState.ABORTING) || existingState.equals(IndexerState.STOPPING)) {
// It shouldn't be possible to persist ABORTING, but if for some reason it does, // It shouldn't be possible to persist ABORTING, but if for some reason it does,
// play it safe and restore the job as STOPPED. An admin will have to clean it up, // play it safe and restore the job as STOPPED. An admin will have to clean it up,
// but it won't be running, and won't delete itself either. Safest option. // but it won't be running, and won't delete itself either. Safest option.
// If we were STOPPING, that means it persisted but was killed before finally stopped... so ok // If we were STOPPING, that means it persisted but was killed before finally stopped... so ok
// to restore as STOPEPD // to restore as STOPPED
initialState = IndexerState.STOPPED; initialState = IndexerState.STOPPED;
} else { } else {
initialState = existingState; initialState = existingState;
} }
initialPosition = state.getPosition(); initialPosition = state.getPosition();
// Since we have state, we are resuming a job/checkpoint. Although we are resuming
// from something that was checkpointed, we can't guarantee it was the _final_ checkpoint
// before the job ended (e.g. it could have been STOPPING, still indexing and killed, leaving
// us with an interval of time partially indexed).
//
// To be safe, if we are resuming any job, use it's ID upgrade status. It will only
// be true if it actually finished a full checkpoint.
this.upgradedDocumentID.set(state.isUpgradedDocumentID());
} }
this.indexer = new ClientRollupPageManager(job, initialState, initialPosition, this.indexer = new ClientRollupPageManager(job, initialState, initialPosition,
new ParentTaskAssigningClient(client, new TaskId(getPersistentTaskId()))); new ParentTaskAssigningClient(client, new TaskId(getPersistentTaskId())), upgradedDocumentID);
} }
@Override @Override
public Status getStatus() { public Status getStatus() {
return new RollupJobStatus(indexer.getState(), indexer.getPosition()); return new RollupJobStatus(indexer.getState(), indexer.getPosition(), upgradedDocumentID.get());
} }
/** /**
@ -223,13 +247,16 @@ public class RollupJobTask extends AllocatedPersistentTask implements SchedulerE
+ " state was [" + prevState + "]")); + " state was [" + prevState + "]"));
return; return;
} }
final IndexerState newState = indexer.start(); final IndexerState newState = indexer.start();
if (newState != IndexerState.STARTED) { if (newState != IndexerState.STARTED) {
listener.onFailure(new ElasticsearchException("Cannot start task for Rollup Job [" + job.getConfig().getId() + "] because" listener.onFailure(new ElasticsearchException("Cannot start task for Rollup Job [" + job.getConfig().getId() + "] because"
+ " state was [" + newState + "]")); + " state was [" + newState + "]"));
return; return;
} }
final RollupJobStatus state = new RollupJobStatus(IndexerState.STARTED, indexer.getPosition());
final RollupJobStatus state = new RollupJobStatus(IndexerState.STARTED, indexer.getPosition(), upgradedDocumentID.get());
logger.debug("Updating state for rollup job [" + job.getConfig().getId() + "] to [" + state.getIndexerState() + "][" + logger.debug("Updating state for rollup job [" + job.getConfig().getId() + "] to [" + state.getIndexerState() + "][" +
state.getPosition() + "]"); state.getPosition() + "]");
updatePersistentTaskState(state, updatePersistentTaskState(state,
@ -240,6 +267,8 @@ public class RollupJobTask extends AllocatedPersistentTask implements SchedulerE
listener.onResponse(new StartRollupJobAction.Response(true)); listener.onResponse(new StartRollupJobAction.Response(true));
}, },
(exc) -> { (exc) -> {
// We were unable to update the persistent status, so we need to shutdown the indexer too.
indexer.stop();
listener.onFailure( listener.onFailure(
new ElasticsearchException("Error while updating state for rollup job [" + job.getConfig().getId() new ElasticsearchException("Error while updating state for rollup job [" + job.getConfig().getId()
+ "] to [" + state.getIndexerState() + "].", exc) + "] to [" + state.getIndexerState() + "].", exc)
@ -261,6 +290,7 @@ public class RollupJobTask extends AllocatedPersistentTask implements SchedulerE
* @param listener The listener that is requesting the stop, so that we can signal completion/failure * @param listener The listener that is requesting the stop, so that we can signal completion/failure
*/ */
public synchronized void stop(ActionListener<StopRollupJobAction.Response> listener) { public synchronized void stop(ActionListener<StopRollupJobAction.Response> listener) {
final IndexerState newState = indexer.stop(); final IndexerState newState = indexer.stop();
switch (newState) { switch (newState) {
case STOPPED: case STOPPED:
@ -268,9 +298,13 @@ public class RollupJobTask extends AllocatedPersistentTask implements SchedulerE
break; break;
case STOPPING: case STOPPING:
// update the persistent state only if there is no background job running, // update the persistent state to STOPPED. There are two scenarios and both are safe:
// otherwise the state is updated by the indexer when the background job detects the STOPPING state. // 1. we persist STOPPED now, indexer continues a bit then sees the flag and checkpoints another
RollupJobStatus state = new RollupJobStatus(IndexerState.STOPPED, indexer.getPosition()); // STOPPED with the more recent position. That will also upgrade the ID scheme
// 2. we persist STOPPED now, indexer continues a bit but then dies. When/if we resume we'll pick up
// at last checkpoint, overwrite some docs and eventually checkpoint. At that time we'll also
// upgrade the ID scheme
RollupJobStatus state = new RollupJobStatus(IndexerState.STOPPED, indexer.getPosition(), upgradedDocumentID.get());
updatePersistentTaskState(state, updatePersistentTaskState(state,
ActionListener.wrap( ActionListener.wrap(
(task) -> { (task) -> {

3
x-pack/plugin/rollup/src/test/java/org/elasticsearch/xpack/rollup/action/SearchActionTests.java
View File

@ -541,7 +541,8 @@ public class SearchActionTests extends ESTestCase {
BoolQueryBuilder bool1 = new BoolQueryBuilder() BoolQueryBuilder bool1 = new BoolQueryBuilder()
.must(TransportRollupSearchAction.rewriteQuery(request.source().query(), caps)) .must(TransportRollupSearchAction.rewriteQuery(request.source().query(), caps))
.filter(new TermQueryBuilder(RollupField.formatMetaField(RollupField.ID.getPreferredName()), "foo")) .filter(new TermQueryBuilder(RollupField.formatMetaField(RollupField.ID.getPreferredName()), "foo"))
.filter(new TermQueryBuilder(RollupField.formatMetaField(RollupField.VERSION_FIELD), Rollup.ROLLUP_VERSION)); .filter(new TermsQueryBuilder(RollupField.formatMetaField(RollupField.VERSION_FIELD),
new long[]{Rollup.ROLLUP_VERSION_V1, Rollup.ROLLUP_VERSION_V2}));
assertThat(msearch.requests().get(1).source().query(), equalTo(bool1)); assertThat(msearch.requests().get(1).source().query(), equalTo(bool1));
} }

119
x-pack/plugin/rollup/src/test/java/org/elasticsearch/xpack/rollup/job/IndexerUtilsTests.java
View File

@ -10,6 +10,7 @@ import org.apache.lucene.document.LongPoint;
import org.apache.lucene.document.SortedNumericDocValuesField; import org.apache.lucene.document.SortedNumericDocValuesField;
import org.apache.lucene.index.DirectoryReader; import org.apache.lucene.index.DirectoryReader;
import org.apache.lucene.index.IndexReader; import org.apache.lucene.index.IndexReader;
import org.apache.lucene.index.IndexWriter;
import org.apache.lucene.index.RandomIndexWriter; import org.apache.lucene.index.RandomIndexWriter;
import org.apache.lucene.search.IndexSearcher; import org.apache.lucene.search.IndexSearcher;
import org.apache.lucene.search.MatchAllDocsQuery; import org.apache.lucene.search.MatchAllDocsQuery;
@ -112,7 +113,7 @@ public class IndexerUtilsTests extends AggregatorTestCase {
directory.close(); directory.close();
List<IndexRequest> docs = IndexerUtils.processBuckets(composite, indexName, stats, List<IndexRequest> docs = IndexerUtils.processBuckets(composite, indexName, stats,
ConfigTestHelpers.getGroupConfig().build(), "foo"); ConfigTestHelpers.getGroupConfig().build(), "foo", randomBoolean());
assertThat(docs.size(), equalTo(numDocs)); assertThat(docs.size(), equalTo(numDocs));
for (IndexRequest doc : docs) { for (IndexRequest doc : docs) {
@ -179,7 +180,7 @@ public class IndexerUtilsTests extends AggregatorTestCase {
directory.close(); directory.close();
List<IndexRequest> docs = IndexerUtils.processBuckets(composite, indexName, stats, List<IndexRequest> docs = IndexerUtils.processBuckets(composite, indexName, stats,
ConfigTestHelpers.getGroupConfig().build(), "foo"); ConfigTestHelpers.getGroupConfig().build(), "foo", randomBoolean());
assertThat(docs.size(), equalTo(numDocs)); assertThat(docs.size(), equalTo(numDocs));
for (IndexRequest doc : docs) { for (IndexRequest doc : docs) {
@ -235,7 +236,7 @@ public class IndexerUtilsTests extends AggregatorTestCase {
directory.close(); directory.close();
List<IndexRequest> docs = IndexerUtils.processBuckets(composite, indexName, stats, List<IndexRequest> docs = IndexerUtils.processBuckets(composite, indexName, stats,
ConfigTestHelpers.getGroupConfig().build(), "foo"); ConfigTestHelpers.getGroupConfig().build(), "foo", randomBoolean());
assertThat(docs.size(), equalTo(numDocs)); assertThat(docs.size(), equalTo(numDocs));
for (IndexRequest doc : docs) { for (IndexRequest doc : docs) {
@ -301,7 +302,7 @@ public class IndexerUtilsTests extends AggregatorTestCase {
directory.close(); directory.close();
List<IndexRequest> docs = IndexerUtils.processBuckets(composite, indexName, stats, List<IndexRequest> docs = IndexerUtils.processBuckets(composite, indexName, stats,
ConfigTestHelpers.getGroupConfig().build(), "foo"); ConfigTestHelpers.getGroupConfig().build(), "foo", randomBoolean());
assertThat(docs.size(), equalTo(numDocs)); assertThat(docs.size(), equalTo(numDocs));
for (IndexRequest doc : docs) { for (IndexRequest doc : docs) {
@ -312,7 +313,7 @@ public class IndexerUtilsTests extends AggregatorTestCase {
} }
} }
public void testKeyOrdering() { public void testKeyOrderingOldID() {
CompositeAggregation composite = mock(CompositeAggregation.class); CompositeAggregation composite = mock(CompositeAggregation.class);
when(composite.getBuckets()).thenAnswer((Answer<List<CompositeAggregation.Bucket>>) invocationOnMock -> { when(composite.getBuckets()).thenAnswer((Answer<List<CompositeAggregation.Bucket>>) invocationOnMock -> {
@ -355,11 +356,112 @@ public class IndexerUtilsTests extends AggregatorTestCase {
GroupConfig.Builder groupConfig = ConfigTestHelpers.getGroupConfig(); GroupConfig.Builder groupConfig = ConfigTestHelpers.getGroupConfig();
groupConfig.setHisto(new HistogramGroupConfig(123L, "abc")); groupConfig.setHisto(new HistogramGroupConfig(123L, "abc"));
List<IndexRequest> docs = IndexerUtils.processBuckets(composite, "foo", new RollupJobStats(), groupConfig.build(), "foo"); List<IndexRequest> docs = IndexerUtils.processBuckets(composite, "foo", new RollupJobStats(),
groupConfig.build(), "foo", false);
assertThat(docs.size(), equalTo(1)); assertThat(docs.size(), equalTo(1));
assertThat(docs.get(0).id(), equalTo("1237859798")); assertThat(docs.get(0).id(), equalTo("1237859798"));
} }
public void testKeyOrderingNewID() {
CompositeAggregation composite = mock(CompositeAggregation.class);
when(composite.getBuckets()).thenAnswer((Answer<List<CompositeAggregation.Bucket>>) invocationOnMock -> {
List<CompositeAggregation.Bucket> foos = new ArrayList<>();
CompositeAggregation.Bucket bucket = mock(CompositeAggregation.Bucket.class);
LinkedHashMap<String, Object> keys = new LinkedHashMap<>(3);
keys.put("foo.date_histogram", 123L);
keys.put("bar.terms", "baz");
keys.put("abc.histogram", 1.9);
keys = shuffleMap(keys, Collections.emptySet());
when(bucket.getKey()).thenReturn(keys);
List<Aggregation> list = new ArrayList<>(3);
InternalNumericMetricsAggregation.SingleValue mockAgg = mock(InternalNumericMetricsAggregation.SingleValue.class);
when(mockAgg.getName()).thenReturn("123");
list.add(mockAgg);
InternalNumericMetricsAggregation.SingleValue mockAgg2 = mock(InternalNumericMetricsAggregation.SingleValue.class);
when(mockAgg2.getName()).thenReturn("abc");
list.add(mockAgg2);
InternalNumericMetricsAggregation.SingleValue mockAgg3 = mock(InternalNumericMetricsAggregation.SingleValue.class);
when(mockAgg3.getName()).thenReturn("yay");
list.add(mockAgg3);
Collections.shuffle(list, random());
Aggregations aggs = new Aggregations(list);
when(bucket.getAggregations()).thenReturn(aggs);
when(bucket.getDocCount()).thenReturn(1L);
foos.add(bucket);
return foos;
});
GroupConfig.Builder groupConfig = ConfigTestHelpers.getGroupConfig();
groupConfig.setHisto(new HistogramGroupConfig(1, "abc"));
List<IndexRequest> docs = IndexerUtils.processBuckets(composite, "foo", new RollupJobStats(),
groupConfig.build(), "foo", true);
assertThat(docs.size(), equalTo(1));
assertThat(docs.get(0).id(), equalTo("foo$c9LcrFqeFW92uN_Z7sv1hA"));
}
/*
A test to make sure very long keys don't break the hash
*/
public void testKeyOrderingNewIDLong() {
CompositeAggregation composite = mock(CompositeAggregation.class);
when(composite.getBuckets()).thenAnswer((Answer<List<CompositeAggregation.Bucket>>) invocationOnMock -> {
List<CompositeAggregation.Bucket> foos = new ArrayList<>();
CompositeAggregation.Bucket bucket = mock(CompositeAggregation.Bucket.class);
LinkedHashMap<String, Object> keys = new LinkedHashMap<>(3);
keys.put("foo.date_histogram", 123L);
char[] charArray = new char[IndexWriter.MAX_TERM_LENGTH];
Arrays.fill(charArray, 'a');
keys.put("bar.terms", new String(charArray));
keys.put("abc.histogram", 1.9);
keys = shuffleMap(keys, Collections.emptySet());
when(bucket.getKey()).thenReturn(keys);
List<Aggregation> list = new ArrayList<>(3);
InternalNumericMetricsAggregation.SingleValue mockAgg = mock(InternalNumericMetricsAggregation.SingleValue.class);
when(mockAgg.getName()).thenReturn("123");
list.add(mockAgg);
InternalNumericMetricsAggregation.SingleValue mockAgg2 = mock(InternalNumericMetricsAggregation.SingleValue.class);
when(mockAgg2.getName()).thenReturn("abc");
list.add(mockAgg2);
InternalNumericMetricsAggregation.SingleValue mockAgg3 = mock(InternalNumericMetricsAggregation.SingleValue.class);
when(mockAgg3.getName()).thenReturn("yay");
list.add(mockAgg3);
Collections.shuffle(list, random());
Aggregations aggs = new Aggregations(list);
when(bucket.getAggregations()).thenReturn(aggs);
when(bucket.getDocCount()).thenReturn(1L);
foos.add(bucket);
return foos;
});
GroupConfig.Builder groupConfig = ConfigTestHelpers.getGroupConfig();
groupConfig.setHisto(new HistogramGroupConfig(1, "abc"));
List<IndexRequest> docs = IndexerUtils.processBuckets(composite, "foo", new RollupJobStats(),
groupConfig.build(), "foo", true);
assertThat(docs.size(), equalTo(1));
assertThat(docs.get(0).id(), equalTo("foo$VAFKZpyaEqYRPLyic57_qw"));
}
public void testNullKeys() { public void testNullKeys() {
CompositeAggregation composite = mock(CompositeAggregation.class); CompositeAggregation composite = mock(CompositeAggregation.class);
@ -384,7 +486,8 @@ public class IndexerUtilsTests extends AggregatorTestCase {
GroupConfig.Builder groupConfig = ConfigTestHelpers.getGroupConfig(); GroupConfig.Builder groupConfig = ConfigTestHelpers.getGroupConfig();
groupConfig.setHisto(randomHistogramGroupConfig(random())); groupConfig.setHisto(randomHistogramGroupConfig(random()));
List<IndexRequest> docs = IndexerUtils.processBuckets(composite, "foo", new RollupJobStats(), groupConfig.build(), "foo"); List<IndexRequest> docs = IndexerUtils.processBuckets(composite, "foo", new RollupJobStats(),
groupConfig.build(), "foo", randomBoolean());
assertThat(docs.size(), equalTo(1)); assertThat(docs.size(), equalTo(1));
assertFalse(Strings.isNullOrEmpty(docs.get(0).id())); assertFalse(Strings.isNullOrEmpty(docs.get(0).id()));
} }
@ -446,7 +549,7 @@ public class IndexerUtilsTests extends AggregatorTestCase {
directory.close(); directory.close();
List<IndexRequest> docs = IndexerUtils.processBuckets(composite, indexName, stats, List<IndexRequest> docs = IndexerUtils.processBuckets(composite, indexName, stats,
ConfigTestHelpers.getGroupConfig().build(), "foo"); ConfigTestHelpers.getGroupConfig().build(), "foo", randomBoolean());
assertThat(docs.size(), equalTo(6)); assertThat(docs.size(), equalTo(6));
for (IndexRequest doc : docs) { for (IndexRequest doc : docs) {

32
x-pack/plugin/rollup/src/test/java/org/elasticsearch/xpack/rollup/job/RollupIndexerIndexingTests.java
View File

@ -71,6 +71,7 @@ import java.util.concurrent.Executor;
import java.util.concurrent.ExecutorService; import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors; import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit; import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.atomic.AtomicReference; import java.util.concurrent.atomic.AtomicReference;
import java.util.function.Consumer; import java.util.function.Consumer;
@ -81,6 +82,7 @@ import static org.hamcrest.number.OrderingComparison.greaterThan;
public class RollupIndexerIndexingTests extends AggregatorTestCase { public class RollupIndexerIndexingTests extends AggregatorTestCase {
private QueryShardContext queryShardContext; private QueryShardContext queryShardContext;
private IndexSettings settings; private IndexSettings settings;
private final boolean newIDScheme = randomBoolean();
@Before @Before
private void setup() { private void setup() {
@ -111,7 +113,7 @@ public class RollupIndexerIndexingTests extends AggregatorTestCase {
assertThat(request.type(), equalTo("_doc")); assertThat(request.type(), equalTo("_doc"));
assertThat(request.sourceAsMap(), equalTo( assertThat(request.sourceAsMap(), equalTo(
asMap( asMap(
"_rollup.version", 1, "_rollup.version", newIDScheme ? 2 : 1,
"the_histo.date_histogram.timestamp", 3, "the_histo.date_histogram.timestamp", 3,
"the_histo.date_histogram.interval", "1ms", "the_histo.date_histogram.interval", "1ms",
"the_histo.date_histogram._count", 2, "the_histo.date_histogram._count", 2,
@ -124,7 +126,7 @@ public class RollupIndexerIndexingTests extends AggregatorTestCase {
assertThat(request.type(), equalTo("_doc")); assertThat(request.type(), equalTo("_doc"));
assertThat(request.sourceAsMap(), equalTo( assertThat(request.sourceAsMap(), equalTo(
asMap( asMap(
"_rollup.version", 1, "_rollup.version", newIDScheme ? 2 : 1,
"the_histo.date_histogram.timestamp", 7, "the_histo.date_histogram.timestamp", 7,
"the_histo.date_histogram.interval", "1ms", "the_histo.date_histogram.interval", "1ms",
"the_histo.date_histogram._count", 1, "the_histo.date_histogram._count", 1,
@ -170,7 +172,7 @@ public class RollupIndexerIndexingTests extends AggregatorTestCase {
assertThat(request.type(), equalTo("_doc")); assertThat(request.type(), equalTo("_doc"));
assertThat(request.sourceAsMap(), equalTo( assertThat(request.sourceAsMap(), equalTo(
asMap( asMap(
"_rollup.version", 1, "_rollup.version", newIDScheme ? 2 : 1,
"the_histo.date_histogram.timestamp", asLong("2015-03-31T03:00:00"), "the_histo.date_histogram.timestamp", asLong("2015-03-31T03:00:00"),
"the_histo.date_histogram.interval", "1h", "the_histo.date_histogram.interval", "1h",
"the_histo.date_histogram._count", 3, "the_histo.date_histogram._count", 3,
@ -188,7 +190,7 @@ public class RollupIndexerIndexingTests extends AggregatorTestCase {
assertThat(request.type(), equalTo("_doc")); assertThat(request.type(), equalTo("_doc"));
assertThat(request.sourceAsMap(), equalTo( assertThat(request.sourceAsMap(), equalTo(
asMap( asMap(
"_rollup.version", 1, "_rollup.version", newIDScheme ? 2 : 1,
"the_histo.date_histogram.timestamp", asLong("2015-03-31T04:00:00"), "the_histo.date_histogram.timestamp", asLong("2015-03-31T04:00:00"),
"the_histo.date_histogram.interval", "1h", "the_histo.date_histogram.interval", "1h",
"the_histo.date_histogram._count", 3, "the_histo.date_histogram._count", 3,
@ -206,7 +208,7 @@ public class RollupIndexerIndexingTests extends AggregatorTestCase {
assertThat(request.type(), equalTo("_doc")); assertThat(request.type(), equalTo("_doc"));
assertThat(request.sourceAsMap(), equalTo( assertThat(request.sourceAsMap(), equalTo(
asMap( asMap(
"_rollup.version", 1, "_rollup.version", newIDScheme ? 2 : 1,
"the_histo.date_histogram.timestamp", asLong("2015-03-31T05:00:00"), "the_histo.date_histogram.timestamp", asLong("2015-03-31T05:00:00"),
"the_histo.date_histogram.interval", "1h", "the_histo.date_histogram.interval", "1h",
"the_histo.date_histogram._count", 4, "the_histo.date_histogram._count", 4,
@ -224,7 +226,7 @@ public class RollupIndexerIndexingTests extends AggregatorTestCase {
assertThat(request.type(), equalTo("_doc")); assertThat(request.type(), equalTo("_doc"));
assertThat(request.sourceAsMap(), equalTo( assertThat(request.sourceAsMap(), equalTo(
asMap( asMap(
"_rollup.version", 1, "_rollup.version", newIDScheme ? 2 : 1,
"the_histo.date_histogram.timestamp", asLong("2015-03-31T06:00:00"), "the_histo.date_histogram.timestamp", asLong("2015-03-31T06:00:00"),
"the_histo.date_histogram.interval", "1h", "the_histo.date_histogram.interval", "1h",
"the_histo.date_histogram._count", 3, "the_histo.date_histogram._count", 3,
@ -242,7 +244,7 @@ public class RollupIndexerIndexingTests extends AggregatorTestCase {
assertThat(request.type(), equalTo("_doc")); assertThat(request.type(), equalTo("_doc"));
assertThat(request.sourceAsMap(), equalTo( assertThat(request.sourceAsMap(), equalTo(
asMap( asMap(
"_rollup.version", 1, "_rollup.version", newIDScheme ? 2 : 1,
"the_histo.date_histogram.timestamp", asLong("2015-03-31T07:00:00"), "the_histo.date_histogram.timestamp", asLong("2015-03-31T07:00:00"),
"the_histo.date_histogram.interval", "1h", "the_histo.date_histogram.interval", "1h",
"the_histo.date_histogram._count", 3, "the_histo.date_histogram._count", 3,
@ -289,7 +291,7 @@ public class RollupIndexerIndexingTests extends AggregatorTestCase {
assertThat(request.type(), equalTo("_doc")); assertThat(request.type(), equalTo("_doc"));
assertThat(request.sourceAsMap(), equalTo( assertThat(request.sourceAsMap(), equalTo(
asMap( asMap(
"_rollup.version", 1, "_rollup.version", newIDScheme ? 2 : 1,
"the_histo.date_histogram.timestamp", rounding.round(now - TimeValue.timeValueHours(5).getMillis()), "the_histo.date_histogram.timestamp", rounding.round(now - TimeValue.timeValueHours(5).getMillis()),
"the_histo.date_histogram.interval", "1m", "the_histo.date_histogram.interval", "1m",
"the_histo.date_histogram._count", 2, "the_histo.date_histogram._count", 2,
@ -302,7 +304,7 @@ public class RollupIndexerIndexingTests extends AggregatorTestCase {
assertThat(request.type(), equalTo("_doc")); assertThat(request.type(), equalTo("_doc"));
assertThat(request.sourceAsMap(), equalTo( assertThat(request.sourceAsMap(), equalTo(
asMap( asMap(
"_rollup.version", 1, "_rollup.version", newIDScheme ? 2 : 1,
"the_histo.date_histogram.timestamp", rounding.round(now - TimeValue.timeValueMinutes(75).getMillis()), "the_histo.date_histogram.timestamp", rounding.round(now - TimeValue.timeValueMinutes(75).getMillis()),
"the_histo.date_histogram.interval", "1m", "the_histo.date_histogram.interval", "1m",
"the_histo.date_histogram._count", 2, "the_histo.date_histogram._count", 2,
@ -315,7 +317,7 @@ public class RollupIndexerIndexingTests extends AggregatorTestCase {
assertThat(request.type(), equalTo("_doc")); assertThat(request.type(), equalTo("_doc"));
assertThat(request.sourceAsMap(), equalTo( assertThat(request.sourceAsMap(), equalTo(
asMap( asMap(
"_rollup.version", 1, "_rollup.version", newIDScheme ? 2 : 1,
"the_histo.date_histogram.timestamp", rounding.round(now - TimeValue.timeValueMinutes(61).getMillis()), "the_histo.date_histogram.timestamp", rounding.round(now - TimeValue.timeValueMinutes(61).getMillis()),
"the_histo.date_histogram.interval", "1m", "the_histo.date_histogram.interval", "1m",
"the_histo.date_histogram._count", 1, "the_histo.date_histogram._count", 1,
@ -355,7 +357,7 @@ public class RollupIndexerIndexingTests extends AggregatorTestCase {
assertThat(request.type(), equalTo("_doc")); assertThat(request.type(), equalTo("_doc"));
assertThat(request.sourceAsMap(), equalTo( assertThat(request.sourceAsMap(), equalTo(
asMap( asMap(
"_rollup.version", 1, "_rollup.version", newIDScheme ? 2 : 1,
"the_histo.date_histogram.timestamp", asLong("2015-03-31T03:00:00"), "the_histo.date_histogram.timestamp", asLong("2015-03-31T03:00:00"),
"the_histo.date_histogram.interval", "1d", "the_histo.date_histogram.interval", "1d",
"the_histo.date_histogram._count", 2, "the_histo.date_histogram._count", 2,
@ -374,7 +376,7 @@ public class RollupIndexerIndexingTests extends AggregatorTestCase {
assertThat(request.type(), equalTo("_doc")); assertThat(request.type(), equalTo("_doc"));
assertThat(request.sourceAsMap(), equalTo( assertThat(request.sourceAsMap(), equalTo(
asMap( asMap(
"_rollup.version", 1, "_rollup.version", newIDScheme ? 2 : 1,
"the_histo.date_histogram.timestamp", asLong("2015-03-31T03:00:00"), "the_histo.date_histogram.timestamp", asLong("2015-03-31T03:00:00"),
"the_histo.date_histogram.interval", "1d", "the_histo.date_histogram.interval", "1d",
"the_histo.date_histogram._count", 2, "the_histo.date_histogram._count", 2,
@ -387,7 +389,7 @@ public class RollupIndexerIndexingTests extends AggregatorTestCase {
assertThat(request.type(), equalTo("_doc")); assertThat(request.type(), equalTo("_doc"));
assertThat(request.sourceAsMap(), equalTo( assertThat(request.sourceAsMap(), equalTo(
asMap( asMap(
"_rollup.version", 1, "_rollup.version", newIDScheme ? 2 : 1,
"the_histo.date_histogram.timestamp", asLong("2015-04-01T03:00:00"), "the_histo.date_histogram.timestamp", asLong("2015-04-01T03:00:00"),
"the_histo.date_histogram.interval", "1d", "the_histo.date_histogram.interval", "1d",
"the_histo.date_histogram._count", 5, "the_histo.date_histogram._count", 5,
@ -425,7 +427,7 @@ public class RollupIndexerIndexingTests extends AggregatorTestCase {
Map<String, Object> source = ((IndexRequest) request).sourceAsMap(); Map<String, Object> source = ((IndexRequest) request).sourceAsMap();
assertThat(source.get("_rollup.version"), equalTo(1)); assertThat(source.get("_rollup.version"), equalTo(newIDScheme ? 2 : 1));
assertThat(source.get("ts.date_histogram.interval"), equalTo(timeInterval.toString())); assertThat(source.get("ts.date_histogram.interval"), equalTo(timeInterval.toString()));
assertNotNull(source.get("the_avg.avg._count")); assertNotNull(source.get("the_avg.avg._count"));
assertNotNull(source.get("the_avg.avg.value")); assertNotNull(source.get("the_avg.avg.value"));
@ -580,7 +582,7 @@ public class RollupIndexerIndexingTests extends AggregatorTestCase {
SyncRollupIndexer(Executor executor, RollupJob job, IndexSearcher searcher, SyncRollupIndexer(Executor executor, RollupJob job, IndexSearcher searcher,
MappedFieldType[] fieldTypes, MappedFieldType timestampField) { MappedFieldType[] fieldTypes, MappedFieldType timestampField) {
super(executor, job, new AtomicReference<>(IndexerState.STARTED), null); super(executor, job, new AtomicReference<>(IndexerState.STARTED), null, new AtomicBoolean(newIDScheme));
this.searcher = searcher; this.searcher = searcher;
this.fieldTypes = fieldTypes; this.fieldTypes = fieldTypes;
this.timestampField = timestampField; this.timestampField = timestampField;

37
x-pack/plugin/rollup/src/test/java/org/elasticsearch/xpack/rollup/job/RollupIndexerStateTests.java
View File

@ -20,11 +20,11 @@ import org.elasticsearch.search.aggregations.Aggregations;
import org.elasticsearch.search.aggregations.InternalAggregations; import org.elasticsearch.search.aggregations.InternalAggregations;
import org.elasticsearch.search.aggregations.bucket.composite.CompositeAggregation; import org.elasticsearch.search.aggregations.bucket.composite.CompositeAggregation;
import org.elasticsearch.test.ESTestCase; import org.elasticsearch.test.ESTestCase;
import org.elasticsearch.xpack.core.rollup.ConfigTestHelpers;
import org.elasticsearch.xpack.core.rollup.RollupField; import org.elasticsearch.xpack.core.rollup.RollupField;
import org.elasticsearch.xpack.core.rollup.job.GroupConfig; import org.elasticsearch.xpack.core.rollup.job.GroupConfig;
import org.elasticsearch.xpack.core.rollup.job.IndexerState; import org.elasticsearch.xpack.core.rollup.job.IndexerState;
import org.elasticsearch.xpack.core.rollup.job.RollupJob; import org.elasticsearch.xpack.core.rollup.job.RollupJob;
import org.elasticsearch.xpack.core.rollup.ConfigTestHelpers;
import org.elasticsearch.xpack.core.rollup.job.RollupJobConfig; import org.elasticsearch.xpack.core.rollup.job.RollupJobConfig;
import org.mockito.stubbing.Answer; import org.mockito.stubbing.Answer;
@ -49,14 +49,16 @@ import static org.mockito.Mockito.when;
public class RollupIndexerStateTests extends ESTestCase { public class RollupIndexerStateTests extends ESTestCase {
private static class EmptyRollupIndexer extends RollupIndexer { private static class EmptyRollupIndexer extends RollupIndexer {
EmptyRollupIndexer(Executor executor, RollupJob job, AtomicReference<IndexerState> initialState, EmptyRollupIndexer(Executor executor, RollupJob job, AtomicReference<IndexerState> initialState,
Map<String, Object> initialPosition) { Map<String, Object> initialPosition, boolean upgraded) {
super(executor, job, initialState, initialPosition); super(executor, job, initialState, initialPosition, new AtomicBoolean(upgraded));
} }
EmptyRollupIndexer(Executor executor, RollupJob job, IndexerState initialState, Map<String, Object> initialPosition) { EmptyRollupIndexer(Executor executor, RollupJob job, AtomicReference<IndexerState> initialState,
this(executor, job, new AtomicReference<>(initialState), initialPosition); Map<String, Object> initialPosition) {
this(executor, job, initialState, initialPosition, randomBoolean());
} }
@Override @Override
protected void doNextSearch(SearchRequest request, ActionListener<SearchResponse> nextPhase) { protected void doNextSearch(SearchRequest request, ActionListener<SearchResponse> nextPhase) {
// TODO Should use InternalComposite constructor but it is package protected in core. // TODO Should use InternalComposite constructor but it is package protected in core.
@ -127,13 +129,13 @@ public class RollupIndexerStateTests extends ESTestCase {
protected CountDownLatch latch; protected CountDownLatch latch;
DelayedEmptyRollupIndexer(Executor executor, RollupJob job, AtomicReference<IndexerState> initialState, DelayedEmptyRollupIndexer(Executor executor, RollupJob job, AtomicReference<IndexerState> initialState,
Map<String, Object> initialPosition) { Map<String, Object> initialPosition, boolean upgraded) {
super(executor, job, initialState, initialPosition); super(executor, job, initialState, initialPosition, upgraded);
} }
DelayedEmptyRollupIndexer(Executor executor, RollupJob job, IndexerState initialState, DelayedEmptyRollupIndexer(Executor executor, RollupJob job, AtomicReference<IndexerState> initialState,
Map<String, Object> initialPosition) { Map<String, Object> initialPosition) {
this(executor, job, new AtomicReference<>(initialState), initialPosition); super(executor, job, initialState, initialPosition, randomBoolean());
} }
private CountDownLatch newLatch() { private CountDownLatch newLatch() {
@ -161,7 +163,7 @@ public class RollupIndexerStateTests extends ESTestCase {
NonEmptyRollupIndexer(Executor executor, RollupJob job, AtomicReference<IndexerState> initialState, NonEmptyRollupIndexer(Executor executor, RollupJob job, AtomicReference<IndexerState> initialState,
Map<String, Object> initialPosition, Function<SearchRequest, SearchResponse> searchFunction, Map<String, Object> initialPosition, Function<SearchRequest, SearchResponse> searchFunction,
Function<BulkRequest, BulkResponse> bulkFunction, Consumer<Exception> failureConsumer) { Function<BulkRequest, BulkResponse> bulkFunction, Consumer<Exception> failureConsumer) {
super(executor, job, initialState, initialPosition); super(executor, job, initialState, initialPosition, new AtomicBoolean(randomBoolean()));
this.searchFunction = searchFunction; this.searchFunction = searchFunction;
this.bulkFunction = bulkFunction; this.bulkFunction = bulkFunction;
this.failureConsumer = failureConsumer; this.failureConsumer = failureConsumer;
@ -216,10 +218,11 @@ public class RollupIndexerStateTests extends ESTestCase {
public void testStarted() throws Exception { public void testStarted() throws Exception {
RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(ESTestCase.randomAlphaOfLengthBetween(1, 10)).build(), RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(ESTestCase.randomAlphaOfLengthBetween(1, 10)).build(),
Collections.emptyMap()); Collections.emptyMap());
IndexerState state = IndexerState.STOPPED; AtomicReference<IndexerState> state = new AtomicReference<>(IndexerState.STOPPED);
final ExecutorService executor = Executors.newFixedThreadPool(1); final ExecutorService executor = Executors.newFixedThreadPool(1);
try { try {
RollupIndexer indexer = new EmptyRollupIndexer(executor, job, state, null); RollupIndexer indexer = new EmptyRollupIndexer(executor, job, state, null, true);
assertTrue(indexer.isUpgradedDocumentID());
indexer.start(); indexer.start();
assertThat(indexer.getState(), equalTo(IndexerState.STARTED)); assertThat(indexer.getState(), equalTo(IndexerState.STARTED));
assertTrue(indexer.maybeTriggerAsyncJob(System.currentTimeMillis())); assertTrue(indexer.maybeTriggerAsyncJob(System.currentTimeMillis()));
@ -235,7 +238,7 @@ public class RollupIndexerStateTests extends ESTestCase {
public void testIndexing() throws Exception { public void testIndexing() throws Exception {
RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(ESTestCase.randomAlphaOfLengthBetween(1, 10)).build(), RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(ESTestCase.randomAlphaOfLengthBetween(1, 10)).build(),
Collections.emptyMap()); Collections.emptyMap());
IndexerState state = IndexerState.STOPPED; AtomicReference<IndexerState> state = new AtomicReference<>(IndexerState.STOPPED);
final ExecutorService executor = Executors.newFixedThreadPool(1); final ExecutorService executor = Executors.newFixedThreadPool(1);
try { try {
AtomicBoolean isFinished = new AtomicBoolean(false); AtomicBoolean isFinished = new AtomicBoolean(false);
@ -434,7 +437,7 @@ public class RollupIndexerStateTests extends ESTestCase {
public void testStopIndexing() throws Exception { public void testStopIndexing() throws Exception {
RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(ESTestCase.randomAlphaOfLengthBetween(1, 10)).build(), RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(ESTestCase.randomAlphaOfLengthBetween(1, 10)).build(),
Collections.emptyMap()); Collections.emptyMap());
IndexerState state = IndexerState.STOPPED; AtomicReference<IndexerState> state = new AtomicReference<>(IndexerState.STOPPED);
final ExecutorService executor = Executors.newFixedThreadPool(1); final ExecutorService executor = Executors.newFixedThreadPool(1);
try { try {
DelayedEmptyRollupIndexer indexer = new DelayedEmptyRollupIndexer(executor, job, state, null); DelayedEmptyRollupIndexer indexer = new DelayedEmptyRollupIndexer(executor, job, state, null);
@ -457,7 +460,7 @@ public class RollupIndexerStateTests extends ESTestCase {
public void testAbortIndexing() throws Exception { public void testAbortIndexing() throws Exception {
RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(ESTestCase.randomAlphaOfLengthBetween(1, 10)).build(), RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(ESTestCase.randomAlphaOfLengthBetween(1, 10)).build(),
Collections.emptyMap()); Collections.emptyMap());
IndexerState state = IndexerState.STOPPED; AtomicReference<IndexerState> state = new AtomicReference<>(IndexerState.STOPPED);
final ExecutorService executor = Executors.newFixedThreadPool(1); final ExecutorService executor = Executors.newFixedThreadPool(1);
try { try {
final AtomicBoolean isAborted = new AtomicBoolean(false); final AtomicBoolean isAborted = new AtomicBoolean(false);
@ -485,7 +488,7 @@ public class RollupIndexerStateTests extends ESTestCase {
public void testAbortStarted() throws Exception { public void testAbortStarted() throws Exception {
RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(ESTestCase.randomAlphaOfLengthBetween(1, 10)).build(), RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(ESTestCase.randomAlphaOfLengthBetween(1, 10)).build(),
Collections.emptyMap()); Collections.emptyMap());
IndexerState state = IndexerState.STOPPED; AtomicReference<IndexerState> state = new AtomicReference<>(IndexerState.STOPPED);
final ExecutorService executor = Executors.newFixedThreadPool(1); final ExecutorService executor = Executors.newFixedThreadPool(1);
try { try {
final AtomicBoolean isAborted = new AtomicBoolean(false); final AtomicBoolean isAborted = new AtomicBoolean(false);
@ -512,7 +515,7 @@ public class RollupIndexerStateTests extends ESTestCase {
public void testMultipleJobTriggering() throws Exception { public void testMultipleJobTriggering() throws Exception {
RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(ESTestCase.randomAlphaOfLengthBetween(1, 10)).build(), RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(ESTestCase.randomAlphaOfLengthBetween(1, 10)).build(),
Collections.emptyMap()); Collections.emptyMap());
IndexerState state = IndexerState.STOPPED; AtomicReference<IndexerState> state = new AtomicReference<>(IndexerState.STOPPED);
final ExecutorService executor = Executors.newFixedThreadPool(1); final ExecutorService executor = Executors.newFixedThreadPool(1);
try { try {
final AtomicBoolean isAborted = new AtomicBoolean(false); final AtomicBoolean isAborted = new AtomicBoolean(false);

135
x-pack/plugin/rollup/src/test/java/org/elasticsearch/xpack/rollup/job/RollupJobTaskTests.java
View File

@ -59,7 +59,7 @@ public class RollupJobTaskTests extends ESTestCase {
public void testInitialStatusStopped() { public void testInitialStatusStopped() {
RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(randomAlphaOfLength(5)).build(), Collections.emptyMap()); RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(randomAlphaOfLength(5)).build(), Collections.emptyMap());
RollupJobStatus status = new RollupJobStatus(IndexerState.STOPPED, Collections.singletonMap("foo", "bar")); RollupJobStatus status = new RollupJobStatus(IndexerState.STOPPED, Collections.singletonMap("foo", "bar"), randomBoolean());
Client client = mock(Client.class); Client client = mock(Client.class);
when(client.settings()).thenReturn(Settings.EMPTY); when(client.settings()).thenReturn(Settings.EMPTY);
SchedulerEngine schedulerEngine = new SchedulerEngine(Clock.systemUTC()); SchedulerEngine schedulerEngine = new SchedulerEngine(Clock.systemUTC());
@ -72,7 +72,7 @@ public class RollupJobTaskTests extends ESTestCase {
public void testInitialStatusAborting() { public void testInitialStatusAborting() {
RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(randomAlphaOfLength(5)).build(), Collections.emptyMap()); RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(randomAlphaOfLength(5)).build(), Collections.emptyMap());
RollupJobStatus status = new RollupJobStatus(IndexerState.ABORTING, Collections.singletonMap("foo", "bar")); RollupJobStatus status = new RollupJobStatus(IndexerState.ABORTING, Collections.singletonMap("foo", "bar"), randomBoolean());
Client client = mock(Client.class); Client client = mock(Client.class);
when(client.settings()).thenReturn(Settings.EMPTY); when(client.settings()).thenReturn(Settings.EMPTY);
SchedulerEngine schedulerEngine = new SchedulerEngine(Clock.systemUTC()); SchedulerEngine schedulerEngine = new SchedulerEngine(Clock.systemUTC());
@ -85,7 +85,7 @@ public class RollupJobTaskTests extends ESTestCase {
public void testInitialStatusStopping() { public void testInitialStatusStopping() {
RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(randomAlphaOfLength(5)).build(), Collections.emptyMap()); RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(randomAlphaOfLength(5)).build(), Collections.emptyMap());
RollupJobStatus status = new RollupJobStatus(IndexerState.STOPPING, Collections.singletonMap("foo", "bar")); RollupJobStatus status = new RollupJobStatus(IndexerState.STOPPING, Collections.singletonMap("foo", "bar"), randomBoolean());
Client client = mock(Client.class); Client client = mock(Client.class);
when(client.settings()).thenReturn(Settings.EMPTY); when(client.settings()).thenReturn(Settings.EMPTY);
SchedulerEngine schedulerEngine = new SchedulerEngine(Clock.systemUTC()); SchedulerEngine schedulerEngine = new SchedulerEngine(Clock.systemUTC());
@ -98,7 +98,7 @@ public class RollupJobTaskTests extends ESTestCase {
public void testInitialStatusStarted() { public void testInitialStatusStarted() {
RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(randomAlphaOfLength(5)).build(), Collections.emptyMap()); RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(randomAlphaOfLength(5)).build(), Collections.emptyMap());
RollupJobStatus status = new RollupJobStatus(IndexerState.STARTED, Collections.singletonMap("foo", "bar")); RollupJobStatus status = new RollupJobStatus(IndexerState.STARTED, Collections.singletonMap("foo", "bar"), randomBoolean());
Client client = mock(Client.class); Client client = mock(Client.class);
when(client.settings()).thenReturn(Settings.EMPTY); when(client.settings()).thenReturn(Settings.EMPTY);
SchedulerEngine schedulerEngine = new SchedulerEngine(Clock.systemUTC()); SchedulerEngine schedulerEngine = new SchedulerEngine(Clock.systemUTC());
@ -109,9 +109,9 @@ public class RollupJobTaskTests extends ESTestCase {
assertTrue(((RollupJobStatus)task.getStatus()).getPosition().containsKey("foo")); assertTrue(((RollupJobStatus)task.getStatus()).getPosition().containsKey("foo"));
} }
public void testInitialStatusIndexing() { public void testInitialStatusIndexingOldID() {
RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(randomAlphaOfLength(5)).build(), Collections.emptyMap()); RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(randomAlphaOfLength(5)).build(), Collections.emptyMap());
RollupJobStatus status = new RollupJobStatus(IndexerState.INDEXING, Collections.singletonMap("foo", "bar")); RollupJobStatus status = new RollupJobStatus(IndexerState.INDEXING, Collections.singletonMap("foo", "bar"), false);
Client client = mock(Client.class); Client client = mock(Client.class);
when(client.settings()).thenReturn(Settings.EMPTY); when(client.settings()).thenReturn(Settings.EMPTY);
SchedulerEngine schedulerEngine = new SchedulerEngine(Clock.systemUTC()); SchedulerEngine schedulerEngine = new SchedulerEngine(Clock.systemUTC());
@ -120,6 +120,21 @@ public class RollupJobTaskTests extends ESTestCase {
assertThat(((RollupJobStatus)task.getStatus()).getIndexerState(), equalTo(IndexerState.STARTED)); assertThat(((RollupJobStatus)task.getStatus()).getIndexerState(), equalTo(IndexerState.STARTED));
assertThat(((RollupJobStatus)task.getStatus()).getPosition().size(), equalTo(1)); assertThat(((RollupJobStatus)task.getStatus()).getPosition().size(), equalTo(1));
assertTrue(((RollupJobStatus)task.getStatus()).getPosition().containsKey("foo")); assertTrue(((RollupJobStatus)task.getStatus()).getPosition().containsKey("foo"));
assertFalse(((RollupJobStatus) task.getStatus()).isUpgradedDocumentID());
}
public void testInitialStatusIndexingNewID() {
RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(randomAlphaOfLength(5)).build(), Collections.emptyMap());
RollupJobStatus status = new RollupJobStatus(IndexerState.INDEXING, Collections.singletonMap("foo", "bar"), true);
Client client = mock(Client.class);
when(client.settings()).thenReturn(Settings.EMPTY);
SchedulerEngine schedulerEngine = new SchedulerEngine(Clock.systemUTC());
RollupJobTask task = new RollupJobTask(1, "type", "action", new TaskId("node", 123), job,
status, client, schedulerEngine, pool, Collections.emptyMap());
assertThat(((RollupJobStatus)task.getStatus()).getIndexerState(), equalTo(IndexerState.STARTED));
assertThat(((RollupJobStatus)task.getStatus()).getPosition().size(), equalTo(1));
assertTrue(((RollupJobStatus)task.getStatus()).getPosition().containsKey("foo"));
assertTrue(((RollupJobStatus) task.getStatus()).isUpgradedDocumentID());
} }
public void testNoInitialStatus() { public void testNoInitialStatus() {
@ -131,11 +146,12 @@ public class RollupJobTaskTests extends ESTestCase {
null, client, schedulerEngine, pool, Collections.emptyMap()); null, client, schedulerEngine, pool, Collections.emptyMap());
assertThat(((RollupJobStatus)task.getStatus()).getIndexerState(), equalTo(IndexerState.STOPPED)); assertThat(((RollupJobStatus)task.getStatus()).getIndexerState(), equalTo(IndexerState.STOPPED));
assertNull(((RollupJobStatus)task.getStatus()).getPosition()); assertNull(((RollupJobStatus)task.getStatus()).getPosition());
assertTrue(((RollupJobStatus) task.getStatus()).isUpgradedDocumentID());
} }
public void testStartWhenStarted() throws InterruptedException { public void testStartWhenStarted() throws InterruptedException {
RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(randomAlphaOfLength(5)).build(), Collections.emptyMap()); RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(randomAlphaOfLength(5)).build(), Collections.emptyMap());
RollupJobStatus status = new RollupJobStatus(IndexerState.STARTED, Collections.singletonMap("foo", "bar")); RollupJobStatus status = new RollupJobStatus(IndexerState.STARTED, Collections.singletonMap("foo", "bar"), randomBoolean());
Client client = mock(Client.class); Client client = mock(Client.class);
when(client.settings()).thenReturn(Settings.EMPTY); when(client.settings()).thenReturn(Settings.EMPTY);
SchedulerEngine schedulerEngine = new SchedulerEngine(Clock.systemUTC()); SchedulerEngine schedulerEngine = new SchedulerEngine(Clock.systemUTC());
@ -243,7 +259,7 @@ public class RollupJobTaskTests extends ESTestCase {
public void testStartWhenStopped() throws InterruptedException { public void testStartWhenStopped() throws InterruptedException {
RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(randomAlphaOfLength(5)).build(), Collections.emptyMap()); RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(randomAlphaOfLength(5)).build(), Collections.emptyMap());
RollupJobStatus status = new RollupJobStatus(IndexerState.STOPPED, Collections.singletonMap("foo", "bar")); RollupJobStatus status = new RollupJobStatus(IndexerState.STOPPED, Collections.singletonMap("foo", "bar"), randomBoolean());
Client client = mock(Client.class); Client client = mock(Client.class);
when(client.settings()).thenReturn(Settings.EMPTY); when(client.settings()).thenReturn(Settings.EMPTY);
SchedulerEngine schedulerEngine = mock(SchedulerEngine.class); SchedulerEngine schedulerEngine = mock(SchedulerEngine.class);
@ -281,7 +297,7 @@ public class RollupJobTaskTests extends ESTestCase {
public void testTriggerUnrelated() throws InterruptedException { public void testTriggerUnrelated() throws InterruptedException {
RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(randomAlphaOfLength(5)).build(), Collections.emptyMap()); RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(randomAlphaOfLength(5)).build(), Collections.emptyMap());
RollupJobStatus status = new RollupJobStatus(IndexerState.STOPPED, Collections.singletonMap("foo", "bar")); RollupJobStatus status = new RollupJobStatus(IndexerState.STOPPED, Collections.singletonMap("foo", "bar"), randomBoolean());
Client client = mock(Client.class); Client client = mock(Client.class);
when(client.settings()).thenReturn(Settings.EMPTY); when(client.settings()).thenReturn(Settings.EMPTY);
SchedulerEngine schedulerEngine = mock(SchedulerEngine.class); SchedulerEngine schedulerEngine = mock(SchedulerEngine.class);
@ -406,6 +422,10 @@ public class RollupJobTaskTests extends ESTestCase {
listener.onResponse(new PersistentTasksCustomMetaData.PersistentTask<>("foo", RollupField.TASK_NAME, job, 1, listener.onResponse(new PersistentTasksCustomMetaData.PersistentTask<>("foo", RollupField.TASK_NAME, job, 1,
new PersistentTasksCustomMetaData.Assignment("foo", "foo"))); new PersistentTasksCustomMetaData.Assignment("foo", "foo")));
} else if (counterValue == 1) { } else if (counterValue == 1) {
// When we get here, doSaveState() was just invoked so we will have
// have upgraded IDs
RollupJobStatus s = (RollupJobStatus)this.getStatus();
assertTrue(s.isUpgradedDocumentID());
finished.set(true); finished.set(true);
} }
@ -490,6 +510,99 @@ public class RollupJobTaskTests extends ESTestCase {
listener.onResponse(new PersistentTasksCustomMetaData.PersistentTask<>("foo", RollupField.TASK_NAME, job, 1, listener.onResponse(new PersistentTasksCustomMetaData.PersistentTask<>("foo", RollupField.TASK_NAME, job, 1,
new PersistentTasksCustomMetaData.Assignment("foo", "foo"))); new PersistentTasksCustomMetaData.Assignment("foo", "foo")));
} else if (counterValue == 1) { } else if (counterValue == 1) {
// When we get here, doSaveState() was just invoked so we will have
// have upgraded IDs
RollupJobStatus s = (RollupJobStatus)this.getStatus();
assertTrue(s.isUpgradedDocumentID());
finished.set(true);
}
}
};
assertThat(((RollupJobStatus)task.getStatus()).getIndexerState(), equalTo(IndexerState.STOPPED));
assertNull(((RollupJobStatus)task.getStatus()).getPosition());
task.start(new ActionListener<StartRollupJobAction.Response>() {
@Override
public void onResponse(StartRollupJobAction.Response response) {
assertTrue(response.isStarted());
assertThat(((RollupJobStatus)task.getStatus()).getIndexerState(), equalTo(IndexerState.STARTED));
started.set(true);
}
@Override
public void onFailure(Exception e) {
fail("Should not have entered onFailure");
}
});
ESTestCase.awaitBusy(started::get);
task.triggered(new SchedulerEngine.Event(RollupJobTask.SCHEDULE_NAME + "_" + job.getConfig().getId(), 123, 123));
assertThat(((RollupJobStatus)task.getStatus()).getIndexerState(), equalTo(IndexerState.INDEXING));
assertThat(task.getStats().getNumInvocations(), equalTo(1L));
// Allow search response to return now
latch.countDown();
// Wait for the final persistent status to finish
ESTestCase.awaitBusy(finished::get);
}
@SuppressWarnings("unchecked")
public void testSaveStateChangesIDScheme() throws InterruptedException {
final ThreadContext threadContext = new ThreadContext(Settings.EMPTY);
Map<String, String> headers = new HashMap<>(1);
headers.put("es-security-runas-user", "foo");
headers.put("_xpack_security_authentication", "bar");
RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(randomAlphaOfLength(5)).build(), headers);
Client client = mock(Client.class);
when(client.settings()).thenReturn(Settings.EMPTY);
AtomicBoolean started = new AtomicBoolean(false);
AtomicBoolean finished = new AtomicBoolean(false);
AtomicInteger counter = new AtomicInteger(0);
CountDownLatch latch = new CountDownLatch(1);
final ThreadPool threadPool = mock(ThreadPool.class);
when(client.threadPool()).thenReturn(threadPool);
when(threadPool.getThreadContext()).thenReturn(threadContext);
doAnswer(invocationOnMock -> {
assertFalse(threadContext.getHeaders().isEmpty());
assertThat(threadContext.getHeaders().get("es-security-runas-user"), equalTo("foo"));
assertThat(threadContext.getHeaders().get("_xpack_security_authentication"), equalTo("bar"));
SearchResponse r = mock(SearchResponse.class);
when(r.getShardFailures()).thenReturn(ShardSearchFailure.EMPTY_ARRAY);
CompositeAggregation compositeAgg = mock(CompositeAggregation.class);
when(compositeAgg.getBuckets()).thenReturn(Collections.emptyList());
when(compositeAgg.getName()).thenReturn(RollupField.NAME);
Aggregations aggs = new Aggregations(Collections.singletonList(compositeAgg));
when(r.getAggregations()).thenReturn(aggs);
// Wait before progressing
latch.await();
((ActionListener)invocationOnMock.getArguments()[2]).onResponse(r);
return null;
}).when(client).execute(anyObject(), anyObject(), anyObject());
SchedulerEngine schedulerEngine = mock(SchedulerEngine.class);
RollupJobStatus status = new RollupJobStatus(IndexerState.STOPPED, null, false);
RollupJobTask task = new RollupJobTask(1, "type", "action", new TaskId("node", 123), job,
status, client, schedulerEngine, pool, Collections.emptyMap()) {
@Override
public void updatePersistentTaskState(PersistentTaskState taskState,
ActionListener<PersistentTasksCustomMetaData.PersistentTask<?>> listener) {
Integer counterValue = counter.getAndIncrement();
if (counterValue == 0) {
assertThat(taskState, instanceOf(RollupJobStatus.class));
assertThat(((RollupJobStatus) taskState).getIndexerState(), equalTo(IndexerState.STARTED));
listener.onResponse(new PersistentTasksCustomMetaData.PersistentTask<>("foo", RollupField.TASK_NAME, job, 1,
new PersistentTasksCustomMetaData.Assignment("foo", "foo")));
} else if (counterValue == 1) {
// When we get here, doSaveState() was just invoked so we will have
// have upgraded IDs
RollupJobStatus s = (RollupJobStatus)this.getStatus();
assertTrue(s.isUpgradedDocumentID());
finished.set(true); finished.set(true);
} }
@ -525,7 +638,7 @@ public class RollupJobTaskTests extends ESTestCase {
public void testStopWhenStopped() throws InterruptedException { public void testStopWhenStopped() throws InterruptedException {
RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(randomAlphaOfLength(5)).build(), Collections.emptyMap()); RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(randomAlphaOfLength(5)).build(), Collections.emptyMap());
RollupJobStatus status = new RollupJobStatus(IndexerState.STOPPED, null); RollupJobStatus status = new RollupJobStatus(IndexerState.STOPPED, null, randomBoolean());
Client client = mock(Client.class); Client client = mock(Client.class);
when(client.settings()).thenReturn(Settings.EMPTY); when(client.settings()).thenReturn(Settings.EMPTY);
SchedulerEngine schedulerEngine = new SchedulerEngine(Clock.systemUTC()); SchedulerEngine schedulerEngine = new SchedulerEngine(Clock.systemUTC());
@ -632,7 +745,7 @@ public class RollupJobTaskTests extends ESTestCase {
public void testStopWhenAborting() throws InterruptedException { public void testStopWhenAborting() throws InterruptedException {
RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(randomAlphaOfLength(5)).build(), Collections.emptyMap()); RollupJob job = new RollupJob(ConfigTestHelpers.getRollupJob(randomAlphaOfLength(5)).build(), Collections.emptyMap());
RollupJobStatus status = new RollupJobStatus(IndexerState.STOPPED, null); RollupJobStatus status = new RollupJobStatus(IndexerState.STOPPED, null, randomBoolean());
Client client = mock(Client.class); Client client = mock(Client.class);
when(client.settings()).thenReturn(Settings.EMPTY); when(client.settings()).thenReturn(Settings.EMPTY);
SchedulerEngine schedulerEngine = new SchedulerEngine(Clock.systemUTC()); SchedulerEngine schedulerEngine = new SchedulerEngine(Clock.systemUTC());

3
x-pack/plugin/src/test/resources/rest-api-spec/test/rollup/delete_job.yml
View File

@ -70,6 +70,7 @@ setup:
trigger_count: 0 trigger_count: 0
status: status:
job_state: "stopped" job_state: "stopped"
upgraded_doc_id: true
- do: - do:
xpack.rollup.delete_job: xpack.rollup.delete_job:
@ -116,6 +117,7 @@ setup:
trigger_count: 0 trigger_count: 0
status: status:
job_state: "stopped" job_state: "stopped"
upgraded_doc_id: true
- do: - do:
xpack.rollup.delete_job: xpack.rollup.delete_job:
@ -162,6 +164,7 @@ setup:
trigger_count: 0 trigger_count: 0
status: status:
job_state: "stopped" job_state: "stopped"
upgraded_doc_id: true
- do: - do:
xpack.rollup.start_job: xpack.rollup.start_job:

3
x-pack/plugin/src/test/resources/rest-api-spec/test/rollup/get_jobs.yml
View File

@ -71,6 +71,7 @@ setup:
trigger_count: 0 trigger_count: 0
status: status:
job_state: "stopped" job_state: "stopped"
upgraded_doc_id: true
--- ---
"Test get with no jobs": "Test get with no jobs":
@ -181,6 +182,7 @@ setup:
trigger_count: 0 trigger_count: 0
status: status:
job_state: "stopped" job_state: "stopped"
upgraded_doc_id: true
- config: - config:
id: "bar" id: "bar"
index_pattern: "bar" index_pattern: "bar"
@ -206,4 +208,5 @@ setup:
trigger_count: 0 trigger_count: 0
status: status:
job_state: "stopped" job_state: "stopped"
upgraded_doc_id: true

1
x-pack/plugin/src/test/resources/rest-api-spec/test/rollup/put_job.yml
View File

@ -71,6 +71,7 @@ setup:
trigger_count: 0 trigger_count: 0
status: status:
job_state: "stopped" job_state: "stopped"
upgraded_doc_id: true
--- ---
"Test put_job with existing name": "Test put_job with existing name":

94
x-pack/qa/full-cluster-restart/src/test/java/org/elasticsearch/xpack/restart/FullClusterRestartIT.java
View File

@ -32,6 +32,7 @@ import org.junit.Before;
import java.io.IOException; import java.io.IOException;
import java.nio.charset.StandardCharsets; import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.Base64; import java.util.Base64;
import java.util.List; import java.util.List;
import java.util.Locale; import java.util.Locale;
@ -41,6 +42,7 @@ import java.util.stream.Collectors;
import static org.elasticsearch.common.unit.TimeValue.timeValueSeconds; import static org.elasticsearch.common.unit.TimeValue.timeValueSeconds;
import static org.hamcrest.Matchers.anyOf; import static org.hamcrest.Matchers.anyOf;
import static org.hamcrest.Matchers.containsInAnyOrder;
import static org.hamcrest.Matchers.containsString; import static org.hamcrest.Matchers.containsString;
import static org.hamcrest.Matchers.equalTo; import static org.hamcrest.Matchers.equalTo;
import static org.hamcrest.Matchers.everyItem; import static org.hamcrest.Matchers.everyItem;
@ -323,6 +325,98 @@ public class FullClusterRestartIT extends ESRestTestCase {
} }
} }
public void testRollupIDSchemeAfterRestart() throws Exception {
assumeTrue("Rollup can be tested with 6.3.0 and onwards", oldClusterVersion.onOrAfter(Version.V_6_3_0));
if (runningAgainstOldCluster) {
final Request indexRequest = new Request("POST", "/id-test-rollup/_doc/1");
indexRequest.setJsonEntity("{\"timestamp\":\"2018-01-01T00:00:01\",\"value\":123}");
client().performRequest(indexRequest);
// create the rollup job
final Request createRollupJobRequest = new Request("PUT", "/_xpack/rollup/job/rollup-id-test");
createRollupJobRequest.setJsonEntity("{"
+ "\"index_pattern\":\"id-test-rollup\","
+ "\"rollup_index\":\"id-test-results-rollup\","
+ "\"cron\":\"*/1 * * * * ?\","
+ "\"page_size\":100,"
+ "\"groups\":{"
+ " \"date_histogram\":{"
+ " \"field\":\"timestamp\","
+ " \"interval\":\"5m\""
+ " },"
+ "\"histogram\":{"
+ " \"fields\": [\"value\"],"
+ " \"interval\":1"
+ " },"
+ "\"terms\":{"
+ " \"fields\": [\"value\"]"
+ " }"
+ "},"
+ "\"metrics\":["
+ " {\"field\":\"value\",\"metrics\":[\"min\",\"max\",\"sum\"]}"
+ "]"
+ "}");
Map<String, Object> createRollupJobResponse = entityAsMap(client().performRequest(createRollupJobRequest));
assertThat(createRollupJobResponse.get("acknowledged"), equalTo(Boolean.TRUE));
// start the rollup job
final Request startRollupJobRequest = new Request("POST", "_xpack/rollup/job/rollup-id-test/_start");
Map<String, Object> startRollupJobResponse = entityAsMap(client().performRequest(startRollupJobRequest));
assertThat(startRollupJobResponse.get("started"), equalTo(Boolean.TRUE));
assertRollUpJob("rollup-id-test");
assertBusy(() -> {
client().performRequest(new Request("POST", "id-test-results-rollup/_refresh"));
final Request searchRequest = new Request("GET", "id-test-results-rollup/_search");
try {
Map<String, Object> searchResponse = entityAsMap(client().performRequest(searchRequest));
assertNotNull(ObjectPath.eval("hits.total", searchResponse));
assertThat(ObjectPath.eval("hits.total", searchResponse), equalTo(1));
assertThat(ObjectPath.eval("hits.hits.0._id", searchResponse), equalTo("3310683722"));
} catch (IOException e) {
fail();
}
});
} else {
final Request indexRequest = new Request("POST", "/id-test-rollup/_doc/2");
indexRequest.setJsonEntity("{\"timestamp\":\"2018-01-02T00:00:01\",\"value\":345}");
client().performRequest(indexRequest);
assertRollUpJob("rollup-id-test");
assertBusy(() -> {
client().performRequest(new Request("POST", "id-test-results-rollup/_refresh"));
final Request searchRequest = new Request("GET", "id-test-results-rollup/_search");
try {
Map<String, Object> searchResponse = entityAsMap(client().performRequest(searchRequest));
assertNotNull(ObjectPath.eval("hits.total", searchResponse));
assertThat(ObjectPath.eval("hits.total", searchResponse), equalTo(2));
List<String> ids = new ArrayList<>(2);
ids.add(ObjectPath.eval("hits.hits.0._id", searchResponse));
ids.add(ObjectPath.eval("hits.hits.1._id", searchResponse));
// should have both old and new ID formats
assertThat(ids, containsInAnyOrder("3310683722", "rollup-id-test$ehY4NAyVSy8xxUDZrNXXIA"));
List<Double> values = new ArrayList<>(2);
Map<String, Object> doc = ObjectPath.eval("hits.hits.0._source", searchResponse);
values.add((Double)doc.get("value.min.value"));
doc = ObjectPath.eval("hits.hits.1._source", searchResponse);
values.add((Double)doc.get("value.min.value"));
assertThat(values, containsInAnyOrder(123.0, 345.0));
} catch (IOException e) {
fail();
}
});
}
}
public void testSqlFailsOnIndexWithTwoTypes() throws IOException { public void testSqlFailsOnIndexWithTwoTypes() throws IOException {
// TODO this isn't going to trigger until we backport to 6.1 // TODO this isn't going to trigger until we backport to 6.1
assumeTrue("It is only possible to build an index that sql doesn't like before 6.0.0", assumeTrue("It is only possible to build an index that sql doesn't like before 6.0.0",

293
x-pack/qa/rolling-upgrade/src/test/java/org/elasticsearch/upgrades/RollupIDUpgradeIT.java Normal file
View File

@ -0,0 +1,293 @@
/*
* 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.upgrades;
import org.elasticsearch.client.Request;
import org.elasticsearch.client.Response;
import org.elasticsearch.common.Booleans;
import org.elasticsearch.common.xcontent.support.XContentMapValues;
import org.elasticsearch.rest.RestStatus;
import org.elasticsearch.test.rest.ESRestTestCase;
import org.elasticsearch.xpack.core.watcher.support.xcontent.ObjectPath;
import org.hamcrest.Matcher;
import java.io.IOException;
import java.time.Instant;
import java.time.OffsetDateTime;
import java.time.ZoneOffset;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.concurrent.TimeUnit;
import static org.hamcrest.Matchers.anyOf;
import static org.hamcrest.Matchers.containsInAnyOrder;
import static org.hamcrest.Matchers.equalTo;
public class RollupIDUpgradeIT extends AbstractUpgradeTestCase {
/**
* This test verifies that as a cluster is upgraded incrementally, new documents eventually switch
* over to the "new" form of ID (128 bit Murmur3 ids).
*
* Rollup IDs are essentially the hashed concatenation of keys returned by the composite aggregation,
* so the field values that are being indexed (timestamp, value, etc) directly affect the
* ID that is generated.
*
* We don't know which node will get the Rollup task to start, so we don't know when it will migrate.
* The first doc is guaranteed to be the "old" style since all nodes are un-upgraded. The second
* and third phase will have a mixed cluster, and the rollup task may or may not migrate. In those
* phases we have two options (old and new) for the document added in the phase.
*
* The last phase is guaranteed to be new as it's a fully upgraded cluster.
*/
public void testIDsUpgradeCorrectly() throws Exception {
switch (CLUSTER_TYPE) {
case OLD:
break;
case MIXED:
Request waitForYellow = new Request("GET", "/_cluster/health");
waitForYellow.addParameter("wait_for_nodes", "3");
waitForYellow.addParameter("wait_for_status", "yellow");
client().performRequest(waitForYellow);
break;
case UPGRADED:
Request waitForGreen = new Request("GET", "/_cluster/health/target,rollup");
waitForGreen.addParameter("wait_for_nodes", "3");
waitForGreen.addParameter("wait_for_status", "green");
// wait for long enough that we give delayed unassigned shards to stop being delayed
waitForGreen.addParameter("timeout", "70s");
waitForGreen.addParameter("level", "shards");
client().performRequest(waitForGreen);
break;
default:
throw new UnsupportedOperationException("Unknown cluster type [" + CLUSTER_TYPE + "]");
}
OffsetDateTime timestamp = Instant.parse("2018-01-01T00:00:01.000Z").atOffset(ZoneOffset.UTC);
if (CLUSTER_TYPE == ClusterType.OLD) {
String recoverQuickly = "{\"settings\": {\"index.unassigned.node_left.delayed_timeout\": \"100ms\"}}";
Request createTargetIndex = new Request("PUT", "/target");
createTargetIndex.setJsonEntity(recoverQuickly);
client().performRequest(createTargetIndex);
final Request indexRequest = new Request("POST", "/target/_doc/1");
indexRequest.setJsonEntity("{\"timestamp\":\"" + timestamp.toString() + "\",\"value\":123}");
client().performRequest(indexRequest);
// create the rollup job
final Request createRollupJobRequest = new Request("PUT", "/_xpack/rollup/job/rollup-id-test");
createRollupJobRequest.setJsonEntity("{"
+ "\"index_pattern\":\"target\","
+ "\"rollup_index\":\"rollup\","
+ "\"cron\":\"*/1 * * * * ?\","
+ "\"page_size\":100,"
+ "\"groups\":{"
+ " \"date_histogram\":{"
+ " \"field\":\"timestamp\","
+ " \"interval\":\"5m\""
+ " },"
+ "\"histogram\":{"
+ " \"fields\": [\"value\"],"
+ " \"interval\":1"
+ " },"
+ "\"terms\":{"
+ " \"fields\": [\"value\"]"
+ " }"
+ "},"
+ "\"metrics\":["
+ " {\"field\":\"value\",\"metrics\":[\"min\",\"max\",\"sum\"]}"
+ "]"
+ "}");
Map<String, Object> createRollupJobResponse = entityAsMap(client().performRequest(createRollupJobRequest));
assertThat(createRollupJobResponse.get("acknowledged"), equalTo(Boolean.TRUE));
Request updateSettings = new Request("PUT", "/rollup/_settings");
updateSettings.setJsonEntity(recoverQuickly);
client().performRequest(updateSettings);
// start the rollup job
final Request startRollupJobRequest = new Request("POST", "_xpack/rollup/job/rollup-id-test/_start");
Map<String, Object> startRollupJobResponse = entityAsMap(client().performRequest(startRollupJobRequest));
assertThat(startRollupJobResponse.get("started"), equalTo(Boolean.TRUE));
assertRollUpJob("rollup-id-test");
List<String> ids = getSearchResults(1);
assertThat(ids, containsInAnyOrder("3310683722"));
}
if (CLUSTER_TYPE == ClusterType.MIXED && Booleans.parseBoolean(System.getProperty("tests.first_round"))) {
final Request indexRequest = new Request("POST", "/target/_doc/2");
indexRequest.setJsonEntity("{\"timestamp\":\"" + timestamp.plusDays(1).toString() + "\",\"value\":345}");
client().performRequest(indexRequest);
assertRollUpJob("rollup-id-test");
client().performRequest(new Request("POST", "rollup/_refresh"));
List<String> ids = getSearchResults(2);
// first doc, guaranteed old style
ids.remove("3310683722");
// next doc may be either style
ids.removeAll(Arrays.asList("621059582", "rollup-id-test$ehY4NAyVSy8xxUDZrNXXIA"));
assertThat(ids.toString(),ids.size(), equalTo(0));
}
if (CLUSTER_TYPE == ClusterType.MIXED && Booleans.parseBoolean(System.getProperty("tests.first_round")) == false) {
final Request indexRequest = new Request("POST", "/target/_doc/3");
indexRequest.setJsonEntity("{\"timestamp\":\"" + timestamp.plusDays(2).toString() + "\",\"value\":456}");
client().performRequest(indexRequest);
assertRollUpJob("rollup-id-test");
client().performRequest(new Request("POST", "rollup/_refresh"));
List<String> ids = getSearchResults(3);
// first doc, guaranteed old style
ids.remove("3310683722");
// next two docs may be either style
ids.removeAll(Arrays.asList("621059582", "4288019978",
"rollup-id-test$ehY4NAyVSy8xxUDZrNXXIA", "rollup-id-test$60RGDSb92YI5LH4_Fnq_1g"));
assertThat(ids.toString(), ids.size(), equalTo(0));
}
if (CLUSTER_TYPE == ClusterType.UPGRADED) {
final Request indexRequest = new Request("POST", "/target/_doc/4");
indexRequest.setJsonEntity("{\"timestamp\":\"" + timestamp.plusDays(3).toString() + "\",\"value\":567}");
client().performRequest(indexRequest);
assertRollUpJob("rollup-id-test");
client().performRequest(new Request("POST", "rollup/_refresh"));
List<String> ids = getSearchResults(4);
// first doc, guaranteed old style
ids.remove("3310683722");
// next two docs may be either style
ids.removeAll(Arrays.asList("621059582", "4288019978",
"rollup-id-test$ehY4NAyVSy8xxUDZrNXXIA", "rollup-id-test$60RGDSb92YI5LH4_Fnq_1g"));
// Last is guaranteed to be new
ids.remove("rollup-id-test$LAKZftDeQwsUtdPixrkkzQ");
assertThat(ids.toString(), ids.size(), equalTo(0));
}
}
private List<String> getSearchResults(int expectedCount) throws Exception {
final List<String> collectedIDs = new ArrayList<>();
assertBusy(() -> {
collectedIDs.clear();
client().performRequest(new Request("POST", "rollup/_refresh"));
final Request searchRequest = new Request("GET", "rollup/_search");
try {
Map<String, Object> searchResponse = entityAsMap(client().performRequest(searchRequest));
assertNotNull(ObjectPath.eval("hits.total", searchResponse));
assertThat(ObjectPath.eval("hits.total", searchResponse), equalTo(expectedCount));
for (int i = 0; i < expectedCount; i++) {
String id = ObjectPath.eval("hits.hits." + i + "._id", searchResponse);
collectedIDs.add(id);
Map<String, Object> doc = ObjectPath.eval("hits.hits." + i + "._source", searchResponse);
assertNotNull(doc);
if (id.startsWith("rollup-id-test")) {
assertThat(doc.get("_rollup.version"), equalTo(2));
} else {
assertThat(doc.get("_rollup.version"), equalTo(1));
}
}
} catch (IOException e) {
fail();
}
});
return collectedIDs;
}
@SuppressWarnings("unchecked")
private void assertRollUpJob(final String rollupJob) throws Exception {
final Matcher<?> expectedStates = anyOf(equalTo("indexing"), equalTo("started"));
waitForRollUpJob(rollupJob, expectedStates);
// check that the rollup job is started using the RollUp API
final Request getRollupJobRequest = new Request("GET", "_xpack/rollup/job/" + rollupJob);
Map<String, Object> getRollupJobResponse = entityAsMap(client().performRequest(getRollupJobRequest));
Map<String, Object> job = getJob(getRollupJobResponse, rollupJob);
if (job != null) {
assertThat(ObjectPath.eval("status.job_state", job), expectedStates);
}
// check that the rollup job is started using the Tasks API
final Request taskRequest = new Request("GET", "_tasks");
taskRequest.addParameter("detailed", "true");
taskRequest.addParameter("actions", "xpack/rollup/*");
Map<String, Object> taskResponse = entityAsMap(client().performRequest(taskRequest));
Map<String, Object> taskResponseNodes = (Map<String, Object>) taskResponse.get("nodes");
Map<String, Object> taskResponseNode = (Map<String, Object>) taskResponseNodes.values().iterator().next();
Map<String, Object> taskResponseTasks = (Map<String, Object>) taskResponseNode.get("tasks");
Map<String, Object> taskResponseStatus = (Map<String, Object>) taskResponseTasks.values().iterator().next();
assertThat(ObjectPath.eval("status.job_state", taskResponseStatus), expectedStates);
// check that the rollup job is started using the Cluster State API
final Request clusterStateRequest = new Request("GET", "_cluster/state/metadata");
Map<String, Object> clusterStateResponse = entityAsMap(client().performRequest(clusterStateRequest));
List<Map<String, Object>> rollupJobTasks = ObjectPath.eval("metadata.persistent_tasks.tasks", clusterStateResponse);
boolean hasRollupTask = false;
for (Map<String, Object> task : rollupJobTasks) {
if (ObjectPath.eval("id", task).equals(rollupJob)) {
hasRollupTask = true;
break;
}
}
if (hasRollupTask == false) {
fail("Expected persistent task for [" + rollupJob + "] but none found.");
}
}
private void waitForRollUpJob(final String rollupJob, final Matcher<?> expectedStates) throws Exception {
assertBusy(() -> {
final Request getRollupJobRequest = new Request("GET", "_xpack/rollup/job/" + rollupJob);
Response getRollupJobResponse = client().performRequest(getRollupJobRequest);
assertThat(getRollupJobResponse.getStatusLine().getStatusCode(), equalTo(RestStatus.OK.getStatus()));
Map<String, Object> job = getJob(getRollupJobResponse, rollupJob);
if (job != null) {
assertThat(ObjectPath.eval("status.job_state", job), expectedStates);
}
}, 30L, TimeUnit.SECONDS);
}
private static Map<String, Object> getJob(Response response, String targetJobId) throws IOException {
return getJob(ESRestTestCase.entityAsMap(response), targetJobId);
}
@SuppressWarnings("unchecked")
private static Map<String, Object> getJob(Map<String, Object> jobsMap, String targetJobId) throws IOException {
List<Map<String, Object>> jobs =
(List<Map<String, Object>>) XContentMapValues.extractValue("jobs", jobsMap);
if (jobs == null) {
return null;
}
for (Map<String, Object> job : jobs) {
String jobId = (String) ((Map<String, Object>) job.get("config")).get("id");
if (jobId.equals(targetJobId)) {
return job;
}
}
return null;
}
}