mirror of
https://github.com/coredns/coredns.git
synced 2025-10-31 18:23:13 -04:00
219bfd0493
* middleware/metrics: add more metrics middleware/cache: Add metrics for number of elements in the cache. Also export the total size. Update README to detail the new metrics. middleware/metrics Move metrics into subpackage called "vars". This breaks the import cycle and is cleaner. This allows vars.Report to be used in the the dnsserver to log refused queries. middleware/metrics: tests Add tests to the metrics framework. The metrics/test subpackage allows scraping of the local server. Do a few test scrape of the metrics that are defined in the metrics middleware. This also allows metrics integration tests to check if the caching and dnssec middleware export their metrics correctly. * update README * typos * fix tests
107 lines
2.0 KiB
Go
107 lines
2.0 KiB
Go
package auto
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
"path"
|
|
"path/filepath"
|
|
"regexp"
|
|
|
|
"github.com/miekg/coredns/middleware/file"
|
|
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
// Walk will recursively walk of the file under l.directory and adds the one that match l.re.
|
|
func (a Auto) Walk() error {
|
|
|
|
// TODO(miek): should add something so that we don't stomp on each other.
|
|
|
|
toDelete := make(map[string]bool)
|
|
for _, n := range a.Zones.Names() {
|
|
toDelete[n] = true
|
|
}
|
|
|
|
filepath.Walk(a.loader.directory, func(path string, info os.FileInfo, err error) error {
|
|
if info.IsDir() {
|
|
return nil
|
|
}
|
|
|
|
match, origin := matches(a.loader.re, info.Name(), a.loader.template)
|
|
if !match {
|
|
return nil
|
|
}
|
|
|
|
if _, ok := a.Zones.Z[origin]; ok {
|
|
// we already have this zone
|
|
toDelete[origin] = false
|
|
return nil
|
|
}
|
|
|
|
reader, err := os.Open(path)
|
|
if err != nil {
|
|
log.Printf("[WARNING] Opening %s failed: %s", path, err)
|
|
return nil
|
|
}
|
|
|
|
zo, err := file.Parse(reader, origin, path)
|
|
if err != nil {
|
|
// Parse barfs warning by itself...
|
|
return nil
|
|
}
|
|
|
|
zo.NoReload = a.loader.noReload
|
|
zo.TransferTo = a.loader.transferTo
|
|
|
|
a.Zones.Add(zo, origin)
|
|
|
|
if a.metrics != nil {
|
|
a.metrics.AddZone(origin)
|
|
}
|
|
|
|
zo.Notify()
|
|
|
|
log.Printf("[INFO] Inserting zone `%s' from: %s", origin, path)
|
|
|
|
toDelete[origin] = false
|
|
|
|
return nil
|
|
})
|
|
|
|
for origin, ok := range toDelete {
|
|
if !ok {
|
|
continue
|
|
}
|
|
|
|
if a.metrics != nil {
|
|
a.metrics.RemoveZone(origin)
|
|
}
|
|
|
|
a.Zones.Remove(origin)
|
|
|
|
log.Printf("[INFO] Deleting zone `%s'", origin)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// matches matches re to filename, if is is a match, the subexpression will be used to expand
|
|
// template to an origin. When match is true that origin is returned. Origin is fully qualified.
|
|
func matches(re *regexp.Regexp, filename, template string) (match bool, origin string) {
|
|
base := path.Base(filename)
|
|
|
|
matches := re.FindStringSubmatchIndex(base)
|
|
if matches == nil {
|
|
return false, ""
|
|
}
|
|
|
|
by := re.ExpandString(nil, template, base, matches)
|
|
if by == nil {
|
|
return false, ""
|
|
}
|
|
|
|
origin = dns.Fqdn(string(by))
|
|
|
|
return true, origin
|
|
}
|