refactor
This commit is contained in:
parent
f7a59668ed
commit
00bd9a5486
@ -3,7 +3,6 @@ package main
|
|||||||
import (
|
import (
|
||||||
"flag"
|
"flag"
|
||||||
"fmt"
|
"fmt"
|
||||||
"net"
|
|
||||||
"os"
|
"os"
|
||||||
"strings"
|
"strings"
|
||||||
"os/signal"
|
"os/signal"
|
||||||
@ -43,17 +42,11 @@ func main() {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
networks := make(map[string]*net.IPNet)
|
networks := make(map[string]string)
|
||||||
for _, item := range strings.Split(*networksconf, ",") {
|
for _, item := range strings.Split(*networksconf, ",") {
|
||||||
tokens := strings.SplitN(item, "=", 2)
|
if tokens := strings.SplitN(item, "=", 2); len(tokens) == 2 {
|
||||||
domain, subnet := tokens[0], tokens[1]
|
networks[tokens[0]] = tokens[1]
|
||||||
|
|
||||||
_, validSubnet, err := net.ParseCIDR(subnet)
|
|
||||||
if err != nil {
|
|
||||||
logger.Errorln(err)
|
|
||||||
os.Exit(1)
|
|
||||||
}
|
}
|
||||||
networks[domain] = validSubnet
|
|
||||||
}
|
}
|
||||||
|
|
||||||
s := new(meshname.MeshnameServer)
|
s := new(meshname.MeshnameServer)
|
||||||
|
@ -57,10 +57,18 @@ type MeshnameServer struct {
|
|||||||
networks map[string]*net.IPNet
|
networks map[string]*net.IPNet
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *MeshnameServer) Init(log *log.Logger, listenAddr string, zoneConfigPath string, networks map[string]*net.IPNet) {
|
func (s *MeshnameServer) Init(log *log.Logger, listenAddr string, zoneConfigPath string, networks map[string]string) {
|
||||||
s.log = log
|
s.log = log
|
||||||
s.listenAddr = listenAddr
|
s.listenAddr = listenAddr
|
||||||
s.networks = networks
|
s.networks = make(map[string]*net.IPNet)
|
||||||
|
for domain, subnet := range networks {
|
||||||
|
_, validSubnet, err := net.ParseCIDR(subnet)
|
||||||
|
if err != nil {
|
||||||
|
s.log.Errorln(err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
s.networks[domain] = validSubnet
|
||||||
|
}
|
||||||
s.zoneConfigPath = zoneConfigPath
|
s.zoneConfigPath = zoneConfigPath
|
||||||
s.zoneConfig = make(map[string][]dns.RR)
|
s.zoneConfig = make(map[string][]dns.RR)
|
||||||
if s.dnsClient == nil {
|
if s.dnsClient == nil {
|
||||||
|
Loading…
Reference in New Issue
Block a user