func newTestServer(t *testing.T, c bool) *server {
Port += 10
StrPort = strconv.Itoa(Port)
s := new(server)
client := etcd.NewClient([]string{"http://127.0.0.1:4001"})
// TODO(miek): why don't I use NewServer??
s.group = new(sync.WaitGroup)
s.scache = cache.New(100, 0)
s.rcache = cache.New(100, 0)
if c {
s.rcache = cache.New(100, 60) // 100 items, 60s ttl
}
s.config = new(Config)
s.config.Domain = "skydns.test."
s.config.DnsAddr = "127.0.0.1:" + StrPort
s.config.Nameservers = []string{"8.8.4.4:53"}
SetDefaults(s.config)
s.config.Local = "104.server1.development.region1.skydns.test."
s.config.Priority = 10
s.config.RCacheTtl = RCacheTtl
s.config.Ttl = 3600
s.config.Ndots = 2
s.dnsUDPclient = &dns.Client{Net: "udp", ReadTimeout: 2 * s.config.ReadTimeout, WriteTimeout: 2 * s.config.ReadTimeout, SingleInflight: true}
s.dnsTCPclient = &dns.Client{Net: "tcp", ReadTimeout: 2 * s.config.ReadTimeout, WriteTimeout: 2 * s.config.ReadTimeout, SingleInflight: true}
s.backend = backendetcd.NewBackend(client, &backendetcd.Config{
Ttl: s.config.Ttl,
Priority: s.config.Priority,
})
go s.Run()
// Yeah, yeah, should do a proper fix.
time.Sleep(500 * time.Millisecond)
return s
}
func main() {
flag.Parse()
machines := strings.Split(machine, ",")
client := newClient(machines, tlspem, tlskey, cacert)
if nameserver != "" {
for _, hostPort := range strings.Split(nameserver, ",") {
if err := validateHostPort(hostPort); err != nil {
log.Fatalf("skydns: nameserver is invalid: %s", err)
}
config.Nameservers = append(config.Nameservers, hostPort)
}
}
if err := validateHostPort(config.DnsAddr); err != nil {
log.Fatalf("skydns: addr is invalid: %s", err)
}
if err := loadConfig(client, config); err != nil {
log.Fatalf("skydns: %s", err)
}
server.SetDefaults(config)
if config.Local != "" {
config.Local = dns.Fqdn(config.Local)
}
backend := backendetcd.NewBackend(client, &backendetcd.Config{
Ttl: config.Ttl,
Priority: config.Priority,
})
s := server.New(backend, config)
if discover {
go func() {
recv := make(chan *etcd.Response)
go client.Watch("/_etcd/machines/", 0, true, recv, nil)
duration := 1 * time.Second
for {
select {
case n := <-recv:
if n != nil {
if client := updateClient(n, tlskey, tlspem, cacert); client != nil {
backend.UpdateClient(client)
}
duration = 1 * time.Second // reset
} else {
// we can see an n == nil, probably when we can't connect to etcd.
log.Printf("skydns: etcd machine cluster update failed, sleeping %s + ~3s", duration)
time.Sleep(duration + (time.Duration(rand.Float32() * 3e9))) // Add some random.
duration *= 2
if duration > 32*time.Second {
duration = 32 * time.Second
}
}
}
}
}()
}
if stub {
s.UpdateStubZones()
go func() {
recv := make(chan *etcd.Response)
go client.Watch(msg.Path(config.Domain)+"/dns/stub/", 0, true, recv, nil)
duration := 1 * time.Second
for {
select {
case n := <-recv:
if n != nil {
s.UpdateStubZones()
log.Printf("skydns: stubzone update")
duration = 1 * time.Second // reset
} else {
// we can see an n == nil, probably when we can't connect to etcd.
log.Printf("skydns: stubzone update failed, sleeping %s + ~3s", duration)
time.Sleep(duration + (time.Duration(rand.Float32() * 3e9))) // Add some random.
duration *= 2
if duration > 32*time.Second {
duration = 32 * time.Second
}
}
}
}
}()
}
stats.Collect() // Graphite
server.Metrics() // Prometheus
if err := s.Run(); err != nil {
log.Fatalf("skydns: %s", err)
}
}
