From 9251b3aa1fe422bf1818e73620c5f336aa5e8dd7 Mon Sep 17 00:00:00 2001
From: cthuang <chungting@cloudflare.com>
Date: Mon, 13 Dec 2021 13:35:40 +0000
Subject: [PATCH] TUN-5530: Get current time from ticker

---
 datagramsession/session.go | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/datagramsession/session.go b/datagramsession/session.go
index f387d303..463c9c1c 100644
--- a/datagramsession/session.go
+++ b/datagramsession/session.go
@@ -77,9 +77,10 @@ func (s *Session) waitForCloseCondition(ctx context.Context, closeAfterIdle time
 			return
 		case <-s.doneChan:
 			return
-		case <-checkIdleTicker.C:
+		// TODO: TUN-5423 evaluate if using atomic is more efficient
+		case now := <-checkIdleTicker.C:
 			// The session is considered inactive if current time is after (last active time + allowed idle time)
-			if time.Now().After(activeAt.Add(closeAfterIdle)) {
+			if now.After(activeAt.Add(closeAfterIdle)) {
 				return
 			}
 		case activeAt = <-s.activeAtChan: // Update last active time