proxy: imp code

AdguardTeam · May 17, 2023 · 6b19f15 · 6b19f15
1 parent f3f8215
commit 6b19f15
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/proxy/cache.go b/proxy/cache.go
@@ -12,6 +12,7 @@ import (
 	"github.com/AdguardTeam/dnsproxy/upstream"
 	glcache "github.com/AdguardTeam/golibs/cache"
 	"github.com/AdguardTeam/golibs/log"
+	"github.com/AdguardTeam/golibs/mathutil"
 	"github.com/miekg/dns"
 	"golang.org/x/exp/slices"
 )
@@ -533,13 +534,12 @@ func msgToKeyWithSubnet(m *dns.Msg, ecsIP net.IP, mask int) (key []byte) {
 	key = make([]byte, keyLen)
 
 	// Put DO.
-	if opt := m.IsEdns0(); opt != nil && opt.Do() {
-		key[0] = 1
-	} else {
-		key[0] = 0
-	}
+	opt := m.IsEdns0()
+	key[0] = mathutil.BoolToNumber[byte](opt != nil && opt.Do())
 
 	// Put Qtype.
+	//
+	// TODO(d.kolyshev): We should put Qtype in key[1:].
 	binary.BigEndian.PutUint16(key[:], q.Qtype)
 
 	// Put Qclass.