blob: 9cdcb408904aae206256ead9322ddf1cf6fc5c9f [file] [log] [blame]
// Copyright (C) 2017 The Android Open Source Project
//
// 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.google.gerrit.acceptance.pgm;
import static com.google.common.base.Preconditions.checkState;
import com.google.auto.value.AutoValue;
import com.google.common.collect.ImmutableList;
import com.google.gerrit.extensions.registration.DynamicSet;
import com.google.gerrit.server.index.OnlineUpgradeListener;
import com.google.inject.AbstractModule;
import com.google.inject.Module;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.CountDownLatch;
class IndexUpgradeController implements OnlineUpgradeListener {
@AutoValue
abstract static class UpgradeAttempt {
static UpgradeAttempt create(String name, int oldVersion, int newVersion) {
return new AutoValue_IndexUpgradeController_UpgradeAttempt(name, oldVersion, newVersion);
}
abstract String name();
abstract int oldVersion();
abstract int newVersion();
}
private final int numExpected;
private final CountDownLatch readyToStart;
private final CountDownLatch started;
private final CountDownLatch finished;
private final List<UpgradeAttempt> startedAttempts;
private final List<UpgradeAttempt> succeededAttempts;
private final List<UpgradeAttempt> failedAttempts;
IndexUpgradeController(int numExpected) {
this.numExpected = numExpected;
readyToStart = new CountDownLatch(1);
started = new CountDownLatch(numExpected);
finished = new CountDownLatch(numExpected);
startedAttempts = new ArrayList<>();
succeededAttempts = new ArrayList<>();
failedAttempts = new ArrayList<>();
}
Module module() {
return new AbstractModule() {
@Override
public void configure() {
DynamicSet.bind(binder(), OnlineUpgradeListener.class)
.toInstance(IndexUpgradeController.this);
}
};
}
@Override
public synchronized void onStart(String name, int oldVersion, int newVersion) {
UpgradeAttempt a = UpgradeAttempt.create(name, oldVersion, newVersion);
try {
readyToStart.await();
} catch (InterruptedException e) {
throw new AssertionError("interrupted waiting to start " + a, e);
}
checkState(
started.getCount() > 0, "already started %s upgrades, can't start %s", numExpected, a);
startedAttempts.add(a);
started.countDown();
}
@Override
public synchronized void onSuccess(String name, int oldVersion, int newVersion) {
finish(UpgradeAttempt.create(name, oldVersion, newVersion), succeededAttempts);
}
@Override
public synchronized void onFailure(String name, int oldVersion, int newVersion) {
finish(UpgradeAttempt.create(name, oldVersion, newVersion), failedAttempts);
}
private synchronized void finish(UpgradeAttempt a, List<UpgradeAttempt> out) {
checkState(readyToStart.getCount() == 0, "shouldn't be finishing upgrade before starting");
checkState(
finished.getCount() > 0, "already finished %s upgrades, can't finish %s", numExpected, a);
out.add(a);
finished.countDown();
}
void runUpgrades() throws Exception {
readyToStart.countDown();
started.await();
finished.await();
}
synchronized ImmutableList<UpgradeAttempt> getStartedAttempts() {
return ImmutableList.copyOf(startedAttempts);
}
synchronized ImmutableList<UpgradeAttempt> getSucceededAttempts() {
return ImmutableList.copyOf(succeededAttempts);
}
synchronized ImmutableList<UpgradeAttempt> getFailedAttempts() {
return ImmutableList.copyOf(failedAttempts);
}
}