blob: c4c0b0b4e597035e62158cdc88ba540e8a358b9f [file] [log] [blame]
// Copyright (C) 2018 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.ericsson.gerrit.plugins.highavailability.autoreindex;
import com.ericsson.gerrit.plugins.highavailability.forwarder.rest.AbstractIndexRestApiServlet;
import com.ericsson.gerrit.plugins.highavailability.forwarder.rest.AbstractIndexRestApiServlet.IndexName;
import com.google.common.flogger.FluentLogger;
import com.google.gerrit.extensions.annotations.PluginData;
import com.google.gerrit.extensions.events.AccountIndexedListener;
import com.google.gerrit.extensions.events.ChangeIndexedListener;
import com.google.gerrit.extensions.events.GroupIndexedListener;
import com.google.gerrit.extensions.events.ProjectIndexedListener;
import com.google.gerrit.reviewdb.server.ReviewDb;
import com.google.gerrit.server.change.ChangeFinder;
import com.google.gerrit.server.git.WorkQueue;
import com.google.gerrit.server.notedb.ChangeNotes;
import com.google.gwtorm.server.SchemaFactory;
import com.google.inject.Inject;
import com.google.inject.Singleton;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Path;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
import java.util.Optional;
import java.util.concurrent.ScheduledExecutorService;
@Singleton
public class IndexTs
implements ChangeIndexedListener,
AccountIndexedListener,
GroupIndexedListener,
ProjectIndexedListener {
private static final FluentLogger log = FluentLogger.forEnclosingClass();
private static final DateTimeFormatter formatter = DateTimeFormatter.ISO_DATE_TIME;
private final Path dataDir;
private final ScheduledExecutorService exec;
private final FlusherRunner flusher;
private final SchemaFactory<ReviewDb> schemaFactory;
private final ChangeFinder changeFinder;
private volatile LocalDateTime changeTs;
private volatile LocalDateTime accountTs;
private volatile LocalDateTime groupTs;
private volatile LocalDateTime projectTs;
class FlusherRunner implements Runnable {
@Override
public void run() {
store(AbstractIndexRestApiServlet.IndexName.CHANGE, changeTs);
store(AbstractIndexRestApiServlet.IndexName.ACCOUNT, accountTs);
store(AbstractIndexRestApiServlet.IndexName.GROUP, groupTs);
store(AbstractIndexRestApiServlet.IndexName.PROJECT, projectTs);
}
private void store(AbstractIndexRestApiServlet.IndexName index, LocalDateTime latestTs) {
Optional<LocalDateTime> currTs = getUpdateTs(index);
if (!currTs.isPresent() || latestTs.isAfter(currTs.get())) {
Path indexTsFile = dataDir.resolve(index.name().toLowerCase());
try {
Files.write(indexTsFile, latestTs.format(formatter).getBytes(StandardCharsets.UTF_8));
} catch (IOException e) {
log.atSevere().withCause(e).log("Unable to update last timestamp for index %s", index);
}
}
}
}
@Inject
public IndexTs(
@PluginData Path dataDir,
WorkQueue queue,
SchemaFactory<ReviewDb> schemaFactory,
ChangeFinder changeFinder) {
this.dataDir = dataDir;
this.exec = queue.getDefaultQueue();
this.flusher = new FlusherRunner();
this.schemaFactory = schemaFactory;
this.changeFinder = changeFinder;
}
@Override
public void onProjectIndexed(String project) {
update(IndexName.PROJECT, LocalDateTime.now());
}
@Override
public void onGroupIndexed(String uuid) {
update(IndexName.GROUP, LocalDateTime.now());
}
@Override
public void onAccountIndexed(int id) {
update(IndexName.ACCOUNT, LocalDateTime.now());
}
@Override
public void onChangeIndexed(String projectName, int id) {
try (ReviewDb db = schemaFactory.open()) {
ChangeNotes changeNotes = changeFinder.findOne(projectName + "~" + id);
update(
IndexName.CHANGE,
changeNotes == null
? LocalDateTime.now()
: changeNotes.getChange().getLastUpdatedOn().toLocalDateTime());
} catch (Exception e) {
log.atWarning().withCause(e).log("Unable to update the latest TS for change %d", id);
}
}
@Override
public void onChangeDeleted(int id) {
update(IndexName.CHANGE, LocalDateTime.now());
}
public Optional<LocalDateTime> getUpdateTs(AbstractIndexRestApiServlet.IndexName index) {
try {
Path indexTsFile = dataDir.resolve(index.name().toLowerCase());
if (indexTsFile.toFile().exists()) {
String tsString = Files.readAllLines(indexTsFile).get(0);
return Optional.of(LocalDateTime.parse(tsString, formatter));
}
} catch (Exception e) {
log.atWarning().withCause(e).log("Unable to read last timestamp for index %s", index);
}
return Optional.empty();
}
void update(AbstractIndexRestApiServlet.IndexName index, LocalDateTime dateTime) {
switch (index) {
case CHANGE:
changeTs = dateTime;
break;
case ACCOUNT:
accountTs = dateTime;
break;
case GROUP:
groupTs = dateTime;
break;
case PROJECT:
projectTs = dateTime;
break;
default:
throw new IllegalArgumentException("Unsupported index " + index);
}
exec.execute(flusher);
}
}