burble.dn42/bird-lg-go
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

frontend: specify timeout for requests (#29)

Browse Source
This commit is contained in:
Yuhui Xu 2021-08-02 18:46:43 +08:00 committed by GitHub
parent bee26f421c
commit 234aadadd9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
frontend/lgproxy.go
View File

@ -6,6 +6,7 @@ import (
"net/url" "net/url"
"strconv" "strconv"
"strings" "strings"
"time"
) )
type channelData struct { type channelData struct {
@ -46,7 +47,8 @@ func batchRequest(servers []string, endpoint string, command string) []string {
} }
url := "http://" + hostname + ":" + strconv.Itoa(setting.proxyPort) + "/" + url.PathEscape(endpoint) + "?q=" + url.QueryEscape(command) url := "http://" + hostname + ":" + strconv.Itoa(setting.proxyPort) + "/" + url.PathEscape(endpoint) + "?q=" + url.QueryEscape(command)
go func(url string, i int) { go func(url string, i int) {
response, err := http.Get(url) client := http.Client{Timeout: 5 * time.Second}
response, err := client.Get(url)
if err != nil { if err != nil {
ch <- channelData{i, "request failed: " + err.Error() + "\n"} ch <- channelData{i, "request failed: " + err.Error() + "\n"}
return return

3
frontend/whois.go
View File

@ -3,6 +3,7 @@ package main
import ( import (
"io/ioutil" "io/ioutil"
"net" "net"
"time"
) )
// Send a whois request // Send a whois request
@ -11,7 +12,7 @@ func whois(s string) string {
return "" return ""
} }
conn, err := net.Dial("tcp", setting.whoisServer+":43") conn, err := net.DialTimeout("tcp", setting.whoisServer+":43", 5*time.Second)
if err != nil { if err != nil {
return err.Error() return err.Error()
} }