Implement httpClientTimeout

Author: ycrash

internal/agent/api/action_api.go

@@ -175,7 +175,9 @@ func (s *Server) handleYcrashForward(writer http.ResponseWriter, request *http.R
 	fr.RequestURI = ""
 	fr.Header.Del("ycrash-forward")
 	fr.Close = true
-	client := http.Client{}
+	client := http.Client{
+		Timeout: config.GlobalConfig.HttpClientTimeout,
+	}
 	r, err := client.Do(fr)
 	if err != nil {
 		resp.Code = -2

internal/agent/ondemand/ondemand.go

@@ -995,6 +995,7 @@ func RequestFin(endpoint string) (resp []byte, err error) {
 	}
 	httpClient := &http.Client{
 		Transport: transport,
+		Timeout:   config.GlobalConfig.HttpClientTimeout,
 	}
 	req, err := http.NewRequest("POST", endpoint, nil)
 	if err != nil {

internal/capture/healthcheck.go

@@ -150,6 +150,7 @@ func newHTTPClient() *http.Client {
 			DisableKeepAlives:  true,
 			ForceAttemptHTTP2:  true,
 		},
+		Timeout: config.GlobalConfig.HttpClientTimeout,
 	}
 }
 

internal/capture/post.go

@@ -114,6 +114,7 @@ func PostCustomDataWithPositionFunc(endpoint, params string, file *os.File, posi
 	}
 	httpClient := &http.Client{
 		Transport: transport,
+		Timeout:   config.GlobalConfig.HttpClientTimeout,
 	}
 	err = position(file)
 	if err != nil {
@@ -169,6 +170,7 @@ func GetData(endpoint string) (msg string, ok bool) {
 	}
 	httpClient := &http.Client{
 		Transport: transport,
+		Timeout:   config.GlobalConfig.HttpClientTimeout,
 	}
 	req, err := http.NewRequest("GET", endpoint, nil)
 	if err != nil {

internal/config/config.go

@@ -93,6 +93,8 @@ type Options struct {
 	// Extended Data
 	EdScript     string `yaml:"edScript" usage:"Extended Data: Path to a custom script that collects extended diagnostics data"`
 	EdDataFolder string `yaml:"edDataFolder" usage:"Extended Data: Directory path where artifacts generated by (-edScript) will be stored and collected. WARNING: All existing files in this folder will be cleared before script execution"`
+
+	HttpClientTimeout time.Duration `yaml:"httpClientTimeout" usage:"HTTP client timeout for API requests (e.g., 1m, 30s). Default is 60 seconds."`
 }
 
 type HealthChecks map[string]HealthCheck
@@ -174,6 +176,7 @@ func defaultConfig() Config {
 			DeferDelete:       true,
 			AppLogLineCount:   10000,
 			TDCaptureDuration: 0 * time.Second, // Setting here 0 seconds as default since handling it in jstack.go
+			HttpClientTimeout: 60 * time.Second,
 		},
 	}
 }