mirror of
https://github.com/netbirdio/netbird.git
synced 2026-04-05 09:03:54 -04:00
33 lines
535 B
Go
33 lines
535 B
Go
//go:build !ios
|
|
|
|
package dns
|
|
|
|
import (
|
|
"fmt"
|
|
)
|
|
|
|
type ShutdownState struct {
|
|
CreatedKeys []string
|
|
}
|
|
|
|
func (s *ShutdownState) Name() string {
|
|
return "dns_state"
|
|
}
|
|
|
|
func (s *ShutdownState) Cleanup() error {
|
|
manager, err := newHostManager()
|
|
if err != nil {
|
|
return fmt.Errorf("create host manager: %w", err)
|
|
}
|
|
|
|
for _, key := range s.CreatedKeys {
|
|
manager.createdKeys[key] = struct{}{}
|
|
}
|
|
|
|
if err := manager.restoreUncleanShutdownDNS(); err != nil {
|
|
return fmt.Errorf("restore unclean shutdown dns: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|