afonsofrancof/sdns-proxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(qol): try to fix the code hanging problem

Browse Source
This commit is contained in:
Afonso Franco
2025-10-12 11:14:34 +01:00
parent 315d20ad7d
commit 53fa68b525
2 changed files with 16 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
internal/qol/measurement.go
View File

@@ -60,7 +60,7 @@ func (r *MeasurementRunner) Run() error {
}
for _, upstream := range r.config.Servers {
if err := r.runMeasurement(upstream, domains, qType); err != nil {
if err := r.runPerUpstream(upstream, domains, qType); err != nil {
fmt.Fprintf(os.Stderr, "error on server %s: %v\n", upstream, err)
}
}
@@ -68,7 +68,16 @@ func (r *MeasurementRunner) Run() error {
return nil
}
func (r *MeasurementRunner) runMeasurement(upstream string, domains []string, qType uint16) error {
func (r *MeasurementRunner) setupDNSClient(upstream string) (client.DNSClient, error) {
opts := client.Options{
DNSSEC: r.config.DNSSEC,
AuthoritativeDNSSEC: r.config.AuthoritativeDNSSEC,
KeepAlive: r.config.KeepAlive,
}
return client.New(upstream, opts)
}
func (r *MeasurementRunner) runPerUpstream(upstream string, domains []string, qType uint16) error {
// Setup DNS client
dnsClient, err := r.setupDNSClient(upstream)
if err != nil {
@@ -112,21 +121,11 @@ func (r *MeasurementRunner) runMeasurement(upstream string, domains []string, qT
return r.runQueries(dnsClient, upstream, domains, qType, writer, packetCapture)
}
func (r *MeasurementRunner) setupDNSClient(upstream string) (client.DNSClient, error) {
opts := client.Options{
DNSSEC: r.config.DNSSEC,
AuthoritativeDNSSEC: r.config.AuthoritativeDNSSEC,
KeepAlive: r.config.KeepAlive,
}
return client.New(upstream, opts)
}
func (r *MeasurementRunner) runQueries(dnsClient client.DNSClient, upstream string,
domains []string, qType uint16, writer *results.MetricsWriter,
packetCapture *capture.PacketCapture) error {
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
if err := packetCapture.Start(ctx); err != nil {
return err
@@ -163,6 +162,8 @@ func (r *MeasurementRunner) runQueries(dnsClient client.DNSClient, upstream stri
}
}
cancel()
time.Sleep(100 * time.Millisecond)
if err := packetCapture.GetError(); err != nil {