CalvinBackup/ctrld
1
0
Fork 0
mirror of https://github.com/Control-D-Inc/ctrld.git synced 2026-03-25 23:30:41 +01:00
Code Issues Packages Projects Releases Wiki Activity

perf(doq): implement connection pooling for improved performance

Browse Source 
Implement QUIC connection pooling for DoQ resolver to match DoH3
performance. Previously, DoQ created a new QUIC connection for every
DNS query, incurring significant handshake overhead. Now connections are
reused across queries, eliminating this overhead for subsequent requests.

The implementation follows the same pattern as DoH3, using parallel dialing
and connection pooling to achieve comparable performance characteristics.
This commit is contained in:
Cuong Manh Le
2026-01-06 14:46:00 +07:00
committed by Cuong Manh Le
parent 8d63a755ba
commit e8d1a4604e
3 changed files with 286 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
config.go
View File

@@ -276,6 +276,9 @@ type UpstreamConfig struct {
http3RoundTripper http.RoundTripper
http3RoundTripper4 http.RoundTripper
http3RoundTripper6 http.RoundTripper
doqConnPool *doqConnPool
doqConnPool4 *doqConnPool
doqConnPool6 *doqConnPool
certPool *x509.CertPool
u *url.URL
fallbackOnce sync.Once
@@ -490,7 +493,7 @@ func (uc *UpstreamConfig) SetupBootstrapIP() {
// ReBootstrap re-setup the bootstrap IP and the transport.
func (uc *UpstreamConfig) ReBootstrap() {
switch uc.Type {
case ResolverTypeDOH, ResolverTypeDOH3:
case ResolverTypeDOH, ResolverTypeDOH3, ResolverTypeDOQ:
default:
return
}
@@ -510,6 +513,27 @@ func (uc *UpstreamConfig) SetupTransport() {
uc.setupDOHTransport()
case ResolverTypeDOH3:
uc.setupDOH3Transport()
case ResolverTypeDOQ:
uc.setupDOQTransport()
}
}
func (uc *UpstreamConfig) setupDOQTransport() {
switch uc.IPStack {
case IpStackBoth, "":
uc.doqConnPool = uc.newDOQConnPool(uc.bootstrapIPs)
case IpStackV4:
uc.doqConnPool = uc.newDOQConnPool(uc.bootstrapIPs4)
case IpStackV6:
uc.doqConnPool = uc.newDOQConnPool(uc.bootstrapIPs6)
case IpStackSplit:
uc.doqConnPool4 = uc.newDOQConnPool(uc.bootstrapIPs4)
if HasIPv6() {
uc.doqConnPool6 = uc.newDOQConnPool(uc.bootstrapIPs6)
} else {
uc.doqConnPool6 = uc.doqConnPool4
}
uc.doqConnPool = uc.newDOQConnPool(uc.bootstrapIPs)
}
}
@@ -595,7 +619,7 @@ func (uc *UpstreamConfig) ErrorPing() error {
func (uc *UpstreamConfig) ping() error {
switch uc.Type {
case ResolverTypeDOH, ResolverTypeDOH3:
case ResolverTypeDOH, ResolverTypeDOH3, ResolverTypeDOQ:
default:
return nil
}
@@ -629,6 +653,10 @@ func (uc *UpstreamConfig) ping() error {
if err := ping(uc.doh3Transport(typ)); err != nil {
return err
}
case ResolverTypeDOQ:
// For DoQ, we just ensure transport is set up by calling doqTransport
// DoQ doesn't use HTTP, so we can't ping it the same way
_ = uc.doqTransport(typ)
}
}