74 lines
1.6 KiB
Go
74 lines
1.6 KiB
Go
package file
|
|
|
|
import (
|
|
"os"
|
|
"time"
|
|
)
|
|
|
|
// TickTime is clock resolution. By default ticks every second. Handler checks if reloadInterval has been reached on every tick.
|
|
var TickTime = 1 * time.Second
|
|
|
|
// Reload reloads a zone when it is changed on disk. If z.NoReload is true, no reloading will be done.
|
|
func (z *Zone) Reload() error {
|
|
if z.ReloadInterval == 0 {
|
|
return nil
|
|
}
|
|
tick := time.NewTicker(TickTime)
|
|
|
|
go func() {
|
|
|
|
for {
|
|
select {
|
|
|
|
case <-tick.C:
|
|
if z.LastReloaded.Add(z.ReloadInterval).After(time.Now()) {
|
|
//reload interval not reached yet
|
|
continue
|
|
}
|
|
//saving timestamp of last attempted reload
|
|
z.LastReloaded = time.Now()
|
|
|
|
zFile := z.File()
|
|
reader, err := os.Open(zFile)
|
|
if err != nil {
|
|
log.Errorf("Failed to open zone %q in %q: %v", z.origin, zFile, err)
|
|
continue
|
|
}
|
|
|
|
serial := z.SOASerialIfDefined()
|
|
zone, err := Parse(reader, z.origin, zFile, serial)
|
|
if err != nil {
|
|
if _, ok := err.(*serialErr); !ok {
|
|
log.Errorf("Parsing zone %q: %v", z.origin, err)
|
|
}
|
|
continue
|
|
}
|
|
|
|
// copy elements we need
|
|
z.reloadMu.Lock()
|
|
z.Apex = zone.Apex
|
|
z.Tree = zone.Tree
|
|
z.reloadMu.Unlock()
|
|
|
|
log.Infof("Successfully reloaded zone %q in %q with serial %d", z.origin, zFile, z.Apex.SOA.Serial)
|
|
z.Notify()
|
|
|
|
case <-z.reloadShutdown:
|
|
tick.Stop()
|
|
return
|
|
}
|
|
}
|
|
}()
|
|
return nil
|
|
}
|
|
|
|
// SOASerialIfDefined returns the SOA's serial if the zone has a SOA record in the Apex, or
|
|
// -1 otherwise.
|
|
func (z *Zone) SOASerialIfDefined() int64 {
|
|
z.reloadMu.Lock()
|
|
defer z.reloadMu.Unlock()
|
|
if z.Apex.SOA != nil {
|
|
return int64(z.Apex.SOA.Serial)
|
|
}
|
|
return -1
|
|
}
|