diff --git a/auth/client.go b/auth/client.go index aa3edf5..8d4e71c 100644 --- a/auth/client.go +++ b/auth/client.go @@ -1,12 +1,14 @@ package auth import ( + "context" "encoding/json" "errors" "fmt" "io" "net/http" "strings" + "time" "github.com/sirupsen/logrus" ) @@ -16,14 +18,20 @@ type Client struct { // getParsedResponse gets response data from gitee func getParsedResponse(method, path string, header http.Header, body io.Reader, obj interface{}) error { + client := &http.Client{ + Timeout: 30 * time.Second, // 整个请求的最大超时时间 + } req, err := http.NewRequest(method, path, body) if err != nil { panic(err) } req.Header = header - response, err := http.DefaultClient.Do(req) + response, err := client.Do(req) if err != nil { - panic(err) + if errors.Is(err, context.DeadlineExceeded) { + return fmt.Errorf("请求超时: %w", err) + } + return fmt.Errorf("请求执行失败: %w", err) } defer response.Body.Close()