optimization: only handle A and AAAA records and .lan and .internal only

main.go

@@ -39,8 +39,6 @@ type PageData struct {
 	Upstreams []Upstream
 }
 
-
-
 func initDB(dbPath string) *sql.DB {
 	db, err := sql.Open("sqlite3", dbPath)
 	if err != nil {
@@ -72,11 +70,15 @@ func (resolver *DNSResolver) handleDNSRequest(w dns.ResponseWriter, r *dns.Msg)
 	switch r.Opcode {
 	case dns.OpcodeQuery:
 		for _, q := range m.Question {
+			isLanOrInternal := strings.HasSuffix(q.Name, ".lan.") || strings.HasSuffix(q.Name, ".internal.")
+			isARecord := q.Qtype == dns.TypeA
+			isAaaaRecord := q.Qtype == dns.TypeAAAA
+
+			if (isARecord || isAaaaRecord) && isLanOrInternal {
 				var qTypeString string
-			switch q.Qtype {
-			case dns.TypeA:
+				if isARecord {
 					qTypeString = "A"
-			case dns.TypeAAAA:
+				} else {
 					qTypeString = "AAAA"
 				}
 
@@ -87,21 +89,23 @@ func (resolver *DNSResolver) handleDNSRequest(w dns.ResponseWriter, r *dns.Msg)
 					if err == nil {
 						m.Answer = append(m.Answer, rr)
 					}
-			} else {
+					continue
+				}
+			}
+
+			// Fallback to upstream for all other cases
 			resp, err := resolver.resolveUpstream(r)
 			if err == nil && resp != nil {
 				log.Printf("[UPSTREAM] Forwarded %s", q.Name)
 				m.Answer = resp.Answer
 				m.Ns = resp.Ns
 				m.Extra = resp.Extra
-					m.Rcode = resp.Rcode
 			} else {
 				log.Printf("[ERROR] Could not resolve %s", q.Name)
 				m.Rcode = dns.RcodeServerFailure
 			}
 		}
 	}
-	}
 
 	w.WriteMsg(m)
 }