Skip to content

Commit

Permalink
Core: fix index throttling to notice a change to max_merge_count on t…
Browse files Browse the repository at this point in the history
…he next merge start/finish

Previously such changes were never noticed, and index throttling kept
kicking in at the original setting from startup.

Closes elastic#8136

Closes elastic#8132
  • Loading branch information
mikemccand committed Oct 18, 2014
1 parent 991841b commit df6aeba
Showing 1 changed file with 7 additions and 5 deletions.
Expand Up @@ -272,7 +272,7 @@ public void start() throws EngineException {
try {
this.indexWriter = createWriter();
mergeScheduler.removeListener(this.throttle);
this.throttle = new IndexThrottle(mergeScheduler.getMaxMerges(), logger);
this.throttle = new IndexThrottle(mergeScheduler, logger);
mergeScheduler.addListener(throttle);
} catch (IOException e) {
maybeFailEngine(e, "start");
Expand Down Expand Up @@ -844,7 +844,7 @@ public void flush(Flush flush) throws EngineException {
currentIndexWriter().close(false);
indexWriter = createWriter();
mergeScheduler.removeListener(this.throttle);
this.throttle = new IndexThrottle(mergeScheduler.getMaxMerges(), this.logger);
this.throttle = new IndexThrottle(mergeScheduler, this.logger);
mergeScheduler.addListener(throttle);
// commit on a just opened writer will commit even if there are no changes done to it
// we rely on that for the commit data translog id key
Expand Down Expand Up @@ -1722,13 +1722,13 @@ private static final class IndexThrottle implements MergeSchedulerProvider.Liste
private final InternalLock lockReference = new InternalLock(new ReentrantLock());
private final AtomicInteger numMergesInFlight = new AtomicInteger(0);
private final AtomicBoolean isThrottling = new AtomicBoolean();
private final int maxNumMerges;
private final MergeSchedulerProvider mergeScheduler;
private final ESLogger logger;

private volatile InternalLock lock = NOOP_LOCK;

public IndexThrottle(int maxNumMerges, ESLogger logger) {
this.maxNumMerges = maxNumMerges;
public IndexThrottle(MergeSchedulerProvider mergeScheduler, ESLogger logger) {
this.mergeScheduler = mergeScheduler;
this.logger = logger;
}

Expand All @@ -1738,6 +1738,7 @@ public Releasable acquireThrottle() {

@Override
public synchronized void beforeMerge(OnGoingMerge merge) {
int maxNumMerges = mergeScheduler.getMaxMerges();
if (numMergesInFlight.incrementAndGet() > maxNumMerges) {
if (isThrottling.getAndSet(true) == false) {
logger.info("now throttling indexing: numMergesInFlight={}, maxNumMerges={}", numMergesInFlight, maxNumMerges);
Expand All @@ -1748,6 +1749,7 @@ public synchronized void beforeMerge(OnGoingMerge merge) {

@Override
public synchronized void afterMerge(OnGoingMerge merge) {
int maxNumMerges = mergeScheduler.getMaxMerges();
if (numMergesInFlight.decrementAndGet() < maxNumMerges) {
if (isThrottling.getAndSet(false)) {
logger.info("stop throttling indexing: numMergesInFlight={}, maxNumMerges={}", numMergesInFlight, maxNumMerges);
Expand Down

0 comments on commit df6aeba

Please sign in to comment.