From 6eed458cebc081dbd28c62d9562532061600b463 Mon Sep 17 00:00:00 2001
From: Chris Eager <chris@signal.org>
Date: Thu, 18 Jul 2024 09:54:16 -0500
Subject: [PATCH] Use server timestamp for delivery duration metrics

---
 .../textsecuregcm/controllers/MessageController.java            | 2 +-
 .../textsecuregcm/websocket/WebSocketConnection.java            | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/service/src/main/java/org/whispersystems/textsecuregcm/controllers/MessageController.java b/service/src/main/java/org/whispersystems/textsecuregcm/controllers/MessageController.java
index 5aa98fbd2..c3ce840aa 100644
--- a/service/src/main/java/org/whispersystems/textsecuregcm/controllers/MessageController.java
+++ b/service/src/main/java/org/whispersystems/textsecuregcm/controllers/MessageController.java
@@ -784,7 +784,7 @@ public class MessageController {
         .thenAccept(maybeDeletedMessage -> {
           maybeDeletedMessage.ifPresent(deletedMessage -> {
 
-            WebSocketConnection.recordMessageDeliveryDuration(deletedMessage.getTimestamp(),
+            WebSocketConnection.recordMessageDeliveryDuration(deletedMessage.getServerTimestamp(),
                 auth.getAuthenticatedDevice());
 
             if (deletedMessage.hasSourceUuid() && deletedMessage.getType() != Type.SERVER_DELIVERY_RECEIPT) {
diff --git a/service/src/main/java/org/whispersystems/textsecuregcm/websocket/WebSocketConnection.java b/service/src/main/java/org/whispersystems/textsecuregcm/websocket/WebSocketConnection.java
index 10288ba73..bbc552cb8 100644
--- a/service/src/main/java/org/whispersystems/textsecuregcm/websocket/WebSocketConnection.java
+++ b/service/src/main/java/org/whispersystems/textsecuregcm/websocket/WebSocketConnection.java
@@ -237,7 +237,7 @@ public class WebSocketConnection implements MessageAvailabilityListener, Displac
                 .thenApply(ignored -> null);
 
             if (message.getType() != Envelope.Type.SERVER_DELIVERY_RECEIPT) {
-              recordMessageDeliveryDuration(message.getTimestamp(), device);
+              recordMessageDeliveryDuration(message.getServerTimestamp(), device);
               sendDeliveryReceiptFor(message);
             }
           } else {