diff options
Diffstat (limited to 'internal')
| -rw-r--r-- | internal/client/dns.go | 12 | ||||
| -rw-r--r-- | internal/models/dns.go | 3 |
2 files changed, 13 insertions, 2 deletions
diff --git a/internal/client/dns.go b/internal/client/dns.go index 08449e0..9f9d4e8 100644 --- a/internal/client/dns.go +++ b/internal/client/dns.go @@ -2,6 +2,7 @@ package client import ( "bufio" + "context" "encoding/json" "fmt" "net/http" @@ -14,6 +15,12 @@ func (c *Client) CheckDNSTLDStream(params *models.DNSTLDParams, callback func(re term := url.PathEscape(params.Term) fullURL := c.config.BaseURL + "/dns/tld/" + term + if params.ShowOnly != "" { + queryParams := url.Values{} + queryParams.Set("showOnly", params.ShowOnly) + fullURL = fullURL + "?" + queryParams.Encode() + } + req, err := http.NewRequest(http.MethodGet, fullURL, nil) if err != nil { return fmt.Errorf("failed to create request: %w", err) @@ -21,7 +28,10 @@ func (c *Client) CheckDNSTLDStream(params *models.DNSTLDParams, callback func(re req.Header.Set("User-Agent", c.config.UserAgent) - resp, err := c.httpClient.Do(req) + streamClient := &http.Client{} + req = req.WithContext(context.Background()) + + resp, err := streamClient.Do(req) if err != nil { return fmt.Errorf("request failed: %w", err) } diff --git a/internal/models/dns.go b/internal/models/dns.go index b08d8ba..51d9e50 100644 --- a/internal/models/dns.go +++ b/internal/models/dns.go @@ -1,7 +1,8 @@ package models type DNSTLDParams struct { - Term string + Term string + ShowOnly string } type DomainResult struct { |