From 83238d6a81fbf892b7526121e174469ac09b10d5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=B8=96=E7=95=8C?= <i@sekai.icu>
Date: Wed, 23 Jul 2025 18:22:52 +0800
Subject: [PATCH] Fix time service wrapper

---
 common/tls/time_wrapper.go | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/common/tls/time_wrapper.go b/common/tls/time_wrapper.go
index 491fca98..5cbecedc 100644
--- a/common/tls/time_wrapper.go
+++ b/common/tls/time_wrapper.go
@@ -11,10 +11,13 @@ type TimeServiceWrapper struct {
 }
 
 func (w *TimeServiceWrapper) TimeFunc() func() time.Time {
-	if w.TimeService == nil {
-		return nil
+	return func() time.Time {
+		if w.TimeService != nil {
+			return w.TimeService.TimeFunc()()
+		} else {
+			return time.Now()
+		}
 	}
-	return w.TimeService.TimeFunc()
 }
 
 func (w *TimeServiceWrapper) Upstream() any {