/*
* Copyright 2000-2016 JetBrains s.r.o.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.intellij.util.indexing;
import com.intellij.openapi.progress.ProgressManager;
import com.intellij.util.ThrowableRunnable;
import com.intellij.util.TimeoutUtil;
import com.intellij.util.containers.ContainerUtil;
import java.util.Map;
import java.util.concurrent.atomic.AtomicReference;
/**
* @author peter
*/
enum RebuildStatus {
OK,
REQUIRES_REBUILD,
DOING_REBUILD;
private static final Map<ID<?, ?>, AtomicReference<RebuildStatus>> ourRebuildStatus = ContainerUtil.newTroveMap();
static void registerIndex(ID<?, ?> indexId) {
ourRebuildStatus.put(indexId, new AtomicReference<>(OK));
}
static boolean isOk(ID<?, ?> indexId) {
return ourRebuildStatus.get(indexId).get() == OK;
}
static boolean requestRebuild(ID<?, ?> indexId) {
return ourRebuildStatus.get(indexId).compareAndSet(OK, REQUIRES_REBUILD);
}
static void clearIndexIfNecessary(ID<?, ?> indexId, ThrowableRunnable<StorageException> clearAction) throws StorageException {
AtomicReference<RebuildStatus> rebuildStatus = ourRebuildStatus.get(indexId);
if (rebuildStatus == null) {
throw new StorageException("Problem updating " + indexId);
}
if (rebuildStatus.compareAndSet(REQUIRES_REBUILD, DOING_REBUILD)) {
doClear(clearAction, rebuildStatus);
} else {
waitUntilIndexReady(rebuildStatus);
}
}
private static void doClear(ThrowableRunnable<StorageException> clearAction, AtomicReference<RebuildStatus> status) throws StorageException {
try {
clearAction.run();
}
catch (StorageException e) {
status.compareAndSet(DOING_REBUILD, REQUIRES_REBUILD);
throw e;
}
if (!status.compareAndSet(DOING_REBUILD, OK)) {
FileBasedIndexImpl.LOG.error("Unexpected status " + status.get());
}
}
private static void waitUntilIndexReady(AtomicReference<RebuildStatus> rebuildStatus) {
while (rebuildStatus.get() != OK) {
ProgressManager.checkCanceled();
TimeoutUtil.sleep(50);
}
}
}