Fix linter warnings and go fmt

This commit is contained in:
George 2020-09-23 14:49:57 -04:00
parent db6b46d933
commit eaffe705b1
5 changed files with 25 additions and 25 deletions

View File

@ -53,8 +53,7 @@ func init() {
func main() { func main() {
flag.Parse() flag.Parse()
var logger *log.Logger logger := log.New(os.Stdout, "", log.Flags())
logger = log.New(os.Stdout, "", log.Flags())
logger.EnableLevel("error") logger.EnableLevel("error")
logger.EnableLevel("warn") logger.EnableLevel("warn")
@ -86,7 +85,9 @@ func main() {
} }
} }
s.Start() if err := s.Start(); err != nil {
logger.Fatal(err)
}
logger.Infoln("Listening on:", listenAddr) logger.Infoln("Listening on:", listenAddr)
c := make(chan os.Signal, 1) c := make(chan os.Signal, 1)
@ -96,9 +97,9 @@ func main() {
defer s.Stop() defer s.Stop()
for { for {
select { select {
case _ = <-c: case <-c:
return return
case _ = <-r: case <-r:
if useconffile != "" { if useconffile != "" {
if err := loadConfig(s, useconffile); err != nil { if err := loadConfig(s, useconffile); err != nil {
logger.Errorln(err) logger.Errorln(err)

View File

@ -10,7 +10,6 @@ import (
"github.com/miekg/dns" "github.com/miekg/dns"
) )
func GenConf(target, zone string) (string, error) { func GenConf(target, zone string) (string, error) {
ip := net.ParseIP(target) ip := net.ParseIP(target)
if ip == nil { if ip == nil {
@ -50,4 +49,3 @@ func ParseZoneConfigMap(zoneConfigMap map[string][]string) (map[string][]dns.RR,
} }
return zoneConfig, nil return zoneConfig, nil
} }

View File

@ -2,9 +2,9 @@ package meshname
import ( import (
"bytes" "bytes"
"fmt"
"net" "net"
"testing" "testing"
"fmt"
"github.com/zhoreeq/meshname/pkg/meshname" "github.com/zhoreeq/meshname/pkg/meshname"
) )

View File

@ -38,16 +38,15 @@ func New(log *log.Logger, listenAddr string) *MeshnameServer {
} }
} }
func (s *MeshnameServer) Stop() error { func (s *MeshnameServer) Stop() {
s.startedLock.Lock() s.startedLock.Lock()
defer s.startedLock.Unlock() defer s.startedLock.Unlock()
if s.started == true { if s.started {
s.dnsServer.Shutdown() if err := s.dnsServer.Shutdown(); err != nil {
s.log.Debugln(err)
}
s.started = false s.started = false
return nil
} else {
return errors.New("MeshnameServer is not running")
} }
} }
@ -55,7 +54,7 @@ func (s *MeshnameServer) Start() error {
s.startedLock.Lock() s.startedLock.Lock()
defer s.startedLock.Unlock() defer s.startedLock.Unlock()
if s.started == false { if !s.started {
s.dnsServer = &dns.Server{Addr: s.listenAddr, Net: "udp"} s.dnsServer = &dns.Server{Addr: s.listenAddr, Net: "udp"}
for tld, subnet := range s.networks { for tld, subnet := range s.networks {
dns.HandleFunc(tld, s.handleRequest) dns.HandleFunc(tld, s.handleRequest)
@ -129,7 +128,10 @@ func (s *MeshnameServer) handleRequest(w dns.ResponseWriter, r *dns.Msg) {
} }
m.Answer = append(m.Answer, resp.Answer...) m.Answer = append(m.Answer, resp.Answer...)
} }
w.WriteMsg(m)
if err := w.WriteMsg(m); err != nil {
s.log.Debugln("Error writing response:", err)
}
} }
func (s *MeshnameServer) isRemoteLookupAllowed(addr net.Addr) bool { func (s *MeshnameServer) isRemoteLookupAllowed(addr net.Addr) bool {

View File

@ -31,7 +31,6 @@ func TestServerLocalDomain(t *testing.T) {
t.Fatalf("meshname: Failed to parse Meshname config: %s", err) t.Fatalf("meshname: Failed to parse Meshname config: %s", err)
} }
ts.Start() ts.Start()
tc := new(dns.Client) tc := new(dns.Client)