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

Recovery: RecoveriesCollection.findRecoveryByShard should call recoveryStatus.tryIncRef before accessing fields 

The function iterates over a snapshot of on going recoveries and thus may access RecoveryStatus objects that are already finished.

Closes #8231
This commit is contained in:
Boaz Leskes 2014-10-26 19:03:33 +01:00
parent bb7ad4ab96
commit 7f14674ff5
1 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/main/java/org/elasticsearch/indices/recovery/RecoveriesCollection.java
View File

@ -103,8 +103,8 @@ public class RecoveriesCollection {
/**
* fail the recovery with the given id (if found) and remove it from the recovery collection
*
* @param id id of the recovery to fail
* @param e exception with reason for the failure
* @param id id of the recovery to fail
* @param e exception with reason for the failure
* @param sendShardFailure true a shard failed message should be sent to the master
*/
public void failRecovery(long id, RecoveryFailedException e, boolean sendShardFailure) {
@ -130,11 +130,16 @@ public class RecoveriesCollection {
@Nullable
public StatusRef findRecoveryByShard(IndexShard indexShard) {
for (RecoveryStatus recoveryStatus : onGoingRecoveries.values()) {
if (recoveryStatus.indexShard() == indexShard) {
if (recoveryStatus.tryIncRef()) {
return new StatusRef(recoveryStatus);
} else {
return null;
// check if the recovery has already finished and if not protect
// against it being closed on us while we check
if (recoveryStatus.tryIncRef()) {
try {
if (recoveryStatus.indexShard() == indexShard) {
recoveryStatus.incRef();
return new StatusRef(recoveryStatus);
}
} finally {
recoveryStatus.decRef();
}
}
}