chore: cleanup ntp code

This commit is contained in:
wwqgtxx 2025-09-13 01:01:41 +08:00
parent 1b99759eaf
commit fe0e364807
2 changed files with 48 additions and 32 deletions

View File

@ -231,6 +231,8 @@ func updateNTP(c *config.NTP) {
c.DialerProxy,
c.WriteToSystem,
)
} else {
ntp.ReCreateNTPService("", 0, "", false)
}
}

View File

@ -3,6 +3,7 @@ package ntp
import (
"context"
"sync"
"sync/atomic"
"time"
"github.com/metacubex/mihomo/component/dialer"
@ -13,8 +14,8 @@ import (
"github.com/metacubex/sing/common/ntp"
)
var offset time.Duration
var service *Service
var globalSrv atomic.Pointer[Service]
var globalMu sync.Mutex
type Service struct {
server M.Socksaddr
@ -22,15 +23,22 @@ type Service struct {
ticker *time.Ticker
ctx context.Context
cancel context.CancelFunc
mu sync.Mutex
mu sync.RWMutex
offset time.Duration
syncSystemTime bool
running bool
}
func ReCreateNTPService(server string, interval time.Duration, dialerProxy string, syncSystemTime bool) {
globalMu.Lock()
defer globalMu.Unlock()
service := globalSrv.Swap(nil)
if service != nil {
service.Stop()
}
if server == "" {
return
}
ctx, cancel := context.WithCancel(context.Background())
service = &Service{
server: M.ParseSocksaddr(server),
@ -41,6 +49,7 @@ func ReCreateNTPService(server string, interval time.Duration, dialerProxy strin
syncSystemTime: syncSystemTime,
}
service.Start()
globalSrv.Store(service)
}
func (srv *Service) Start() {
@ -52,46 +61,49 @@ func (srv *Service) Start() {
log.Errorln("Initialize NTP time failed: %s", err)
return
}
service.running = true
srv.running = true
go srv.loopUpdate()
}
func (srv *Service) Stop() {
srv.mu.Lock()
defer srv.mu.Unlock()
if service.running {
if srv.running {
srv.ticker.Stop()
srv.cancel()
service.running = false
srv.running = false
}
}
func (srv *Service) Running() bool {
func (srv *Service) Offset() time.Duration {
if srv == nil {
return false
return 0
}
srv.mu.Lock()
defer srv.mu.Unlock()
return srv.running
srv.mu.RLock()
defer srv.mu.RUnlock()
if srv.running {
return srv.offset
}
return 0
}
func (srv *Service) update() error {
var response *ntp.Response
var err error
for i := 0; i < 3; i++ {
if response, err = ntp.Exchange(context.Background(), srv.dialer, srv.server); err == nil {
break
response, err = ntp.Exchange(srv.ctx, srv.dialer, srv.server)
if err != nil {
continue
}
if i == 2 {
return err
}
}
offset = response.ClockOffset
offset := response.ClockOffset
if offset > time.Duration(0) {
log.Infoln("System clock is ahead of NTP time by %s", offset)
} else if offset < time.Duration(0) {
log.Infoln("System clock is behind NTP time by %s", -offset)
}
srv.mu.Lock()
srv.offset = offset
srv.mu.Unlock()
if srv.syncSystemTime {
timeNow := response.Time
syncErr := setSystemTime(timeNow)
@ -103,6 +115,8 @@ func (srv *Service) update() error {
}
}
return nil
}
return err
}
func (srv *Service) loopUpdate() {
@ -121,7 +135,7 @@ func (srv *Service) loopUpdate() {
func Now() time.Time {
now := time.Now()
if service.Running() && offset.Abs() > 0 {
if offset := globalSrv.Load().Offset(); offset.Abs() > 0 {
now = now.Add(offset)
}
return now