Refactor handleAutoUpdateVersion to outside handleSync

This commit is contained in:
M Essam Hamed
2025-08-26 17:41:52 +03:00
parent d2e198bd76
commit 59ae92cf8f

View File

@@ -703,22 +703,26 @@ func (e *Engine) PopulateNetbirdConfig(netbirdConfig *mgmProto.NetbirdConfig, mg
return nil
}
func (e *Engine) handleSync(update *mgmProto.SyncResponse) error {
e.syncMsgMux.Lock()
defer e.syncMsgMux.Unlock()
if update.GetAutoUpdateVersion() != skipAutoUpdateVersion {
if e.updateManager == nil && update.GetAutoUpdateVersion() != disableAutoUpdate {
func (e *Engine) handleAutoUpdateVersion(autoUpdateVersion string) {
if autoUpdateVersion != skipAutoUpdateVersion {
if e.updateManager == nil && autoUpdateVersion != disableAutoUpdate {
e.updateManager = updatemanager.NewUpdateManager(e.statusRecorder)
e.updateManager.Start(e.ctx)
} else if e.updateManager != nil && update.GetAutoUpdateVersion() == disableAutoUpdate {
} else if e.updateManager != nil && autoUpdateVersion == disableAutoUpdate {
e.updateManager.Stop()
e.updateManager = nil
}
if e.updateManager != nil {
e.updateManager.SetVersion(update.GetAutoUpdateVersion())
e.updateManager.SetVersion(autoUpdateVersion)
}
}
}
func (e *Engine) handleSync(update *mgmProto.SyncResponse) error {
e.syncMsgMux.Lock()
defer e.syncMsgMux.Unlock()
e.handleAutoUpdateVersion(update.AutoUpdateVersion)
if update.GetNetbirdConfig() != nil {
wCfg := update.GetNetbirdConfig()
err := e.updateTURNs(wCfg.GetTurns())