From fcb1a7813ad0013411247f789ca59e0ab95399ce Mon Sep 17 00:00:00 2001
From: Dreamacro <305009791@qq.com>
Date: Thu, 6 Dec 2018 13:29:43 +0800
Subject: [PATCH] Fix: dns msg to ip

---
 dns/client.go | 16 ++++------------
 1 file changed, 4 insertions(+), 12 deletions(-)

diff --git a/dns/client.go b/dns/client.go
index 7559ea37..dad0ff2f 100644
--- a/dns/client.go
+++ b/dns/client.go
@@ -174,20 +174,12 @@ func (r *Resolver) msgToIP(msg *D.Msg) ([]net.IP, error) {
 	var ips []net.IP
 
 	for _, answer := range msg.Answer {
-		if r.ipv6 {
-			ans, ok := answer.(*D.AAAA)
-			if !ok {
-				continue
-			}
+		switch ans := answer.(type) {
+		case *D.AAAA:
 			ips = append(ips, ans.AAAA)
-			continue
+		case *D.A:
+			ips = append(ips, ans.A)
 		}
-
-		ans, ok := answer.(*D.A)
-		if !ok {
-			continue
-		}
-		ips = append(ips, ans.A)
 	}
 
 	if len(ips) == 0 {