714982e630
Change-Id: Ibf27b61d867eef7539fc85f92aab1b051082e964
122 lines
4.0 KiB
Java
122 lines
4.0 KiB
Java
// 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);
|
|
}
|
|
}
|