zettio/weave#362 - moved Check*() to common

common/errors.go

@@ -0,0 +1,13 @@
+package common
+
+func CheckFatal(e error) {
+	if e != nil {
+		Error.Fatal(e)
+	}
+}
+
+func CheckWarn(e error) {
+	if e != nil {
+		Warning.Println(e)
+	}
+}

nameserver/mdns_client_test.go

@@ -2,6 +2,7 @@ package nameserver
 
 import (
 	"github.com/miekg/dns"
+	. "github.com/zettio/weave/common"
 	wt "github.com/zettio/weave/testing"
 	"log"
 	"net"
@@ -20,18 +21,18 @@ func minimalServer(w dns.ResponseWriter, req *dns.Msg) {
 
 	sendAnswer := func(m *dns.Msg) {
 		buf, err := m.Pack()
-		checkFatal(err)
+		CheckFatal(err)
 		if buf == nil {
 			log.Fatal("Nil buffer")
 		}
 		//log.Println("minimalServer sending:", buf)
 		// This is a bit of a kludge - per the RFC we should send responses from 5353, but that doesn't seem to work
 		sendconn, err := net.DialUDP("udp", nil, ipv4Addr)
-		checkFatal(err)
+		CheckFatal(err)
 
 		_, err = sendconn.Write(buf)
 		sendconn.Close()
-		checkFatal(err)
+		CheckFatal(err)
 	}
 
 	//log.Println("minimalServer received:", req)

nameserver/server.go

@@ -42,7 +42,7 @@ func NewDNSServer(config *dns.ClientConfig, zone Zone, iface *net.Interface, por
 // Start the DNS server
 func (s *DNSServer) Start() error {
 	mdnsClient, err := NewMDNSClient()
-	checkFatal(err)
+	CheckFatal(err)
 
 	ifaceName := "default interface"
 	if s.iface != nil {
@@ -50,7 +50,7 @@ func (s *DNSServer) Start() error {
 	}
 	Info.Printf("Using mDNS on %s", ifaceName)
 	err = mdnsClient.Start(s.iface)
-	checkFatal(err)
+	CheckFatal(err)
 
 	// create two DNS request multiplexerers, depending on the protocol used by clients
 	// (we use the same protocol for asking upstream servers)
@@ -63,10 +63,10 @@ func (s *DNSServer) Start() error {
 	}
 
 	mdnsServer, err := NewMDNSServer(s.zone)
-	checkFatal(err)
+	CheckFatal(err)
 
 	err = mdnsServer.Start(s.iface)
-	checkFatal(err)
+	CheckFatal(err)
 
 	address := fmt.Sprintf(":%d", s.port)
 	s.udpSrv = &dns.Server{Addr: address, Net: "udp", Handler: mux(&dns.Client{Net: "udp", UDPSize: UDPBufSize})}
@@ -75,13 +75,13 @@ func (s *DNSServer) Start() error {
 	go func() {
 		Info.Printf("Listening for DNS on %s (UDP)", address)
 		err = s.udpSrv.ListenAndServe()
-		checkFatal(err)
+		CheckFatal(err)
 	}()
 
 	go func() {
 		Info.Printf("Listening for DNS on %s (TCP)", address)
 		err = s.tcpSrv.ListenAndServe()
-		checkFatal(err)
+		CheckFatal(err)
 	}()
 
 	return nil

nameserver/util.go

@@ -1,17 +0,0 @@
-package nameserver
-
-import (
-	. "github.com/zettio/weave/common"
-)
-
-func checkFatal(e error) {
-	if e != nil {
-		Error.Fatal(e)
-	}
-}
-
-func checkWarn(e error) {
-	if e != nil {
-		Warning.Println(e)
-	}
-}