77 lines
1.7 KiB
Go
77 lines
1.7 KiB
Go
// Package auto implements a on-the-fly loading file backend.
|
|
package auto
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/coredns/coredns/plugin/file"
|
|
"github.com/coredns/coredns/plugin/transfer"
|
|
)
|
|
|
|
// Zones maps zone names to a *Zone. This keeps track of what zones we have loaded at
|
|
// any one time.
|
|
type Zones struct {
|
|
Z map[string]*file.Zone // A map mapping zone (origin) to the Zone's data.
|
|
names []string // All the keys from the map Z as a string slice.
|
|
|
|
origins []string // Any origins from the server block.
|
|
|
|
sync.RWMutex
|
|
}
|
|
|
|
// Names returns the names from z.
|
|
func (z *Zones) Names() []string {
|
|
z.RLock()
|
|
n := z.names
|
|
z.RUnlock()
|
|
return n
|
|
}
|
|
|
|
// Origins returns the origins from z.
|
|
func (z *Zones) Origins() []string {
|
|
// doesn't need locking, because there aren't multiple Go routines accessing it.
|
|
return z.origins
|
|
}
|
|
|
|
// Zones returns a zone with origin name from z, nil when not found.
|
|
func (z *Zones) Zones(name string) *file.Zone {
|
|
z.RLock()
|
|
zo := z.Z[name]
|
|
z.RUnlock()
|
|
return zo
|
|
}
|
|
|
|
// Add adds a new zone into z. If z.ReloadInterval is not zero, the
|
|
// reload goroutine is started.
|
|
func (z *Zones) Add(zo *file.Zone, name string, t *transfer.Transfer) {
|
|
z.Lock()
|
|
|
|
if z.Z == nil {
|
|
z.Z = make(map[string]*file.Zone)
|
|
}
|
|
|
|
z.Z[name] = zo
|
|
z.names = append(z.names, name)
|
|
zo.Reload(t)
|
|
|
|
z.Unlock()
|
|
}
|
|
|
|
// Remove removes the zone named name from z. It also stops the zone's reload goroutine.
|
|
func (z *Zones) Remove(name string) {
|
|
z.Lock()
|
|
|
|
if zo, ok := z.Z[name]; ok {
|
|
zo.OnShutdown()
|
|
}
|
|
|
|
delete(z.Z, name)
|
|
|
|
// TODO(miek): just regenerate Names (might be bad if you have a lot of zones...)
|
|
z.names = []string{}
|
|
for n := range z.Z {
|
|
z.names = append(z.names, n)
|
|
}
|
|
|
|
z.Unlock()
|
|
}
|