Pārlūkot izejas kodu

Minor code cleanup

Ask Bjørn Hansen 10 gadi atpakaļ
vecāks
revīzija
e3a9bf0a3f
4 mainītis faili ar 17 papildinājumiem un 28 dzēšanām
  1. 13 13
      picker.go
  2. 2 4
      targeting_test.go
  3. 1 4
      zone.go
  4. 1 7
      zones.go

+ 13 - 13
picker.go

@@ -9,35 +9,35 @@ import (
 func (label *Label) Picker(qtype uint16, max int) Records {
 func (label *Label) Picker(qtype uint16, max int) Records {
 
 
 	if qtype == dns.TypeANY {
 	if qtype == dns.TypeANY {
-		result := make([]Record, 0)
+		var result []Record
 		for rtype := range label.Records {
 		for rtype := range label.Records {
 
 
-			rtype_records := label.Picker(rtype, max)
+			rtypeRecords := label.Picker(rtype, max)
 
 
-			tmp_result := make(Records, len(result)+len(rtype_records))
+			tmpResult := make(Records, len(result)+len(rtypeRecords))
 
 
-			copy(tmp_result, result)
-			copy(tmp_result[len(result):], rtype_records)
-			result = tmp_result
+			copy(tmpResult, result)
+			copy(tmpResult[len(result):], rtypeRecords)
+			result = tmpResult
 		}
 		}
 
 
 		return result
 		return result
 	}
 	}
 
 
-	if label_rr := label.Records[qtype]; label_rr != nil {
+	if labelRR := label.Records[qtype]; labelRR != nil {
 
 
 		// not "balanced", just return all
 		// not "balanced", just return all
 		if label.Weight[qtype] == 0 {
 		if label.Weight[qtype] == 0 {
-			return label_rr
+			return labelRR
 		}
 		}
 
 
-		rr_count := len(label_rr)
-		if max > rr_count {
-			max = rr_count
+		rrCount := len(labelRR)
+		if max > rrCount {
+			max = rrCount
 		}
 		}
 
 
-		servers := make([]Record, len(label_rr))
-		copy(servers, label_rr)
+		servers := make([]Record, len(labelRR))
+		copy(servers, labelRR)
 		result := make([]Record, max)
 		result := make([]Record, max)
 		sum := label.Weight[qtype]
 		sum := label.Weight[qtype]
 
 

+ 2 - 4
targeting_test.go

@@ -1,8 +1,8 @@
 package main
 package main
 
 
 import (
 import (
-	"net"
 	. "gopkg.in/check.v1"
 	. "gopkg.in/check.v1"
+	"net"
 )
 )
 
 
 type TargetingSuite struct {
 type TargetingSuite struct {
@@ -15,15 +15,13 @@ func (s *TargetingSuite) SetUpSuite(c *C) {
 }
 }
 
 
 func (s *TargetingSuite) TestTargetString(c *C) {
 func (s *TargetingSuite) TestTargetString(c *C) {
-	var tgt TargetOptions
-	tgt = TargetGlobal + TargetCountry + TargetContinent
+	tgt := TargetOptions(TargetGlobal + TargetCountry + TargetContinent)
 
 
 	str := tgt.String()
 	str := tgt.String()
 	c.Check(str, Equals, "@ continent country")
 	c.Check(str, Equals, "@ continent country")
 }
 }
 
 
 func (s *TargetingSuite) TestTargetParse(c *C) {
 func (s *TargetingSuite) TestTargetParse(c *C) {
-
 	tgt, err := parseTargets("@ foo country")
 	tgt, err := parseTargets("@ foo country")
 	str := tgt.String()
 	str := tgt.String()
 	c.Check(str, Equals, "@ country")
 	c.Check(str, Equals, "@ country")

+ 1 - 4
zone.go

@@ -126,9 +126,7 @@ func (z *Zone) SoaRR() dns.RR {
 // first available qType at each targeting level. Return a Label
 // first available qType at each targeting level. Return a Label
 // and the qtype that was "found"
 // and the qtype that was "found"
 func (z *Zone) findLabels(s string, targets []string, qts qTypes) (*Label, uint16) {
 func (z *Zone) findLabels(s string, targets []string, qts qTypes) (*Label, uint16) {
-
 	for _, target := range targets {
 	for _, target := range targets {
-
 		var name string
 		var name string
 
 
 		switch target {
 		switch target {
@@ -143,9 +141,8 @@ func (z *Zone) findLabels(s string, targets []string, qts qTypes) (*Label, uint1
 		}
 		}
 
 
 		if label, ok := z.Labels[name]; ok {
 		if label, ok := z.Labels[name]; ok {
-
+			var name string
 			for _, qtype := range qts {
 			for _, qtype := range qts {
-
 				switch qtype {
 				switch qtype {
 				case dns.TypeANY:
 				case dns.TypeANY:
 					// short-circuit mostly to avoid subtle bugs later
 					// short-circuit mostly to avoid subtle bugs later

+ 1 - 7
zones.go

@@ -159,7 +159,6 @@ func readZoneFile(zoneName, fileName string) (zone *Zone, zerr error) {
 		//log.Printf("k: %s v: %#v, T: %T\n", k, v, v)
 		//log.Printf("k: %s v: %#v, T: %T\n", k, v, v)
 
 
 		switch k {
 		switch k {
-
 		case "ttl":
 		case "ttl":
 			zone.Options.Ttl = valueToInt(v)
 			zone.Options.Ttl = valueToInt(v)
 		case "serial":
 		case "serial":
@@ -219,7 +218,6 @@ func readZoneFile(zoneName, fileName string) (zone *Zone, zerr error) {
 }
 }
 
 
 func setupZoneData(data map[string]interface{}, Zone *Zone) {
 func setupZoneData(data map[string]interface{}, Zone *Zone) {
-
 	recordTypes := map[string]uint16{
 	recordTypes := map[string]uint16{
 		"a":     dns.TypeA,
 		"a":     dns.TypeA,
 		"aaaa":  dns.TypeAAAA,
 		"aaaa":  dns.TypeAAAA,
@@ -233,7 +231,6 @@ func setupZoneData(data map[string]interface{}, Zone *Zone) {
 	}
 	}
 
 
 	for dk, dv_inter := range data {
 	for dk, dv_inter := range data {
-
 		dv := dv_inter.(map[string]interface{})
 		dv := dv_inter.(map[string]interface{})
 
 
 		//log.Printf("K %s V %s TYPE-V %T\n", dk, dv, dv)
 		//log.Printf("K %s V %s TYPE-V %T\n", dk, dv, dv)
@@ -241,7 +238,6 @@ func setupZoneData(data map[string]interface{}, Zone *Zone) {
 		label := Zone.AddLabel(dk)
 		label := Zone.AddLabel(dk)
 
 
 		for rType, rdata := range dv {
 		for rType, rdata := range dv {
-
 			switch rType {
 			switch rType {
 			case "max_hosts":
 			case "max_hosts":
 				label.MaxHosts = valueToInt(rdata)
 				label.MaxHosts = valueToInt(rdata)
@@ -291,7 +287,6 @@ func setupZoneData(data map[string]interface{}, Zone *Zone) {
 			label.Records[dnsType] = make(Records, len(records[rType]))
 			label.Records[dnsType] = make(Records, len(records[rType]))
 
 
 			for i := 0; i < len(records[rType]); i++ {
 			for i := 0; i < len(records[rType]); i++ {
-
 				//log.Printf("RT %T %#v\n", records[rType][i], records[rType][i])
 				//log.Printf("RT %T %#v\n", records[rType][i], records[rType][i])
 
 
 				record := new(Record)
 				record := new(Record)
@@ -524,14 +519,13 @@ func setupZoneData(data map[string]interface{}, Zone *Zone) {
 }
 }
 
 
 func getStringWeight(rec []interface{}) (string, int) {
 func getStringWeight(rec []interface{}) (string, int) {
-
 	str := rec[0].(string)
 	str := rec[0].(string)
 	var weight int
 	var weight int
-	var err error
 
 
 	if len(rec) > 1 {
 	if len(rec) > 1 {
 		switch rec[1].(type) {
 		switch rec[1].(type) {
 		case string:
 		case string:
+			var err error
 			weight, err = strconv.Atoi(rec[1].(string))
 			weight, err = strconv.Atoi(rec[1].(string))
 			if err != nil {
 			if err != nil {
 				panic("Error converting weight to integer")
 				panic("Error converting weight to integer")