From 14581a314e5e3fd463ca463aa2da65f808096dd7 Mon Sep 17 00:00:00 2001
From: Tulir Asokan <tulir@maunium.net>
Date: Thu, 30 Jun 2022 14:50:22 +0300
Subject: [PATCH] Fix checkpoint status for message taking long warnings

---
 messagetracking.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/messagetracking.go b/messagetracking.go
index 64e93ee..b74f09e 100644
--- a/messagetracking.go
+++ b/messagetracking.go
@@ -195,6 +195,9 @@ func (portal *Portal) sendMessageMetrics(evt *event.Event, err error, part strin
 		portal.log.Logfln(level, "%s %s %s from %s: %v", part, msgType, evtDescription, evt.Sender, err)
 		reason, isCertain, _, sendNotice := errorToStatusReason(err)
 		status := bridge.ReasonToCheckpointStatus(reason)
+		if errors.Is(err, errMessageTakingLong) {
+			status = bridge.MsgStatusWillRetry
+		}
 		portal.bridge.SendMessageCheckpoint(evt, bridge.MsgStepRemote, err, status, ms.getRetryNum())
 		if sendNotice {
 			ms.setNoticeID(portal.sendErrorMessage(err, isCertain, ms.getNoticeID()))