From 5aaab967a345124913f546b4412b0581ec570139 Mon Sep 17 00:00:00 2001
From: Devin Carr <dcarr@cloudflare.com>
Date: Wed, 5 Jul 2023 16:11:24 -0700
Subject: [PATCH] TUN-7477: Decrement UDP sessions on shutdown

When a tunnel connection is going down, any active UDP sessions
need to be cleared and the metric needs to be decremented.
---
 datagramsession/manager.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/datagramsession/manager.go b/datagramsession/manager.go
index 60b701ab..ae332bf6 100644
--- a/datagramsession/manager.go
+++ b/datagramsession/manager.go
@@ -92,7 +92,10 @@ func (m *manager) shutdownSessions(err error) {
 		byRemote: true,
 	}
 	for _, s := range m.sessions {
-		s.close(closeSessionErr)
+		m.unregisterSession(&unregisterSessionEvent{
+			sessionID: s.ID,
+			err:       closeSessionErr,
+		})
 	}
 }