SubscriberMetrics: update only when replication is complete
Consider the RefReplicationDone event for updating the replication lag
metrics, so that the SHA1 included in the event is guaranteed to be
present locally as well.
Change-Id: I4fb3dde3c44e5ab66918a5bfc0e5fe82dcc69007
diff --git a/src/main/java/com/googlesource/gerrit/plugins/multisite/consumer/SubscriberMetrics.java b/src/main/java/com/googlesource/gerrit/plugins/multisite/consumer/SubscriberMetrics.java
index 1911c86..f112f07 100644
--- a/src/main/java/com/googlesource/gerrit/plugins/multisite/consumer/SubscriberMetrics.java
+++ b/src/main/java/com/googlesource/gerrit/plugins/multisite/consumer/SubscriberMetrics.java
@@ -25,6 +25,8 @@
import com.google.inject.Singleton;
import com.googlesource.gerrit.plugins.multisite.validation.ProjectVersionRefUpdate;
import com.googlesource.gerrit.plugins.replication.RefReplicatedEvent;
+import com.googlesource.gerrit.plugins.replication.RefReplicationDoneEvent;
+
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
@@ -90,9 +92,9 @@
public void updateReplicationStatusMetrics(EventMessage eventMessage) {
Event event = eventMessage.getEvent();
- if (event instanceof RefReplicatedEvent) {
- RefReplicatedEvent refReplicatedEvent = (RefReplicatedEvent) event;
- String projectName = refReplicatedEvent.getProjectNameKey().get();
+ if (event instanceof RefReplicationDoneEvent) {
+ RefReplicationDoneEvent refReplicationDoneEvent = (RefReplicationDoneEvent) event;
+ String projectName = refReplicationDoneEvent.getProjectNameKey().get();
logger.atFine().log("Updating replication lag for %s", projectName);
Optional<Long> remoteVersion = projectVersionRefUpdate.getProjectRemoteVersion(projectName);
Optional<Long> localVersion = projectVersionRefUpdate.getProjectLocalVersion(projectName);
