浏览代码

Fix time service wrapper

世界 2 月之前
父节点
当前提交
a11384b286
共有 1 个文件被更改,包括 6 次插入3 次删除
  1. 6 3
      common/tls/time_wrapper.go

+ 6 - 3
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 {