Add in-flight ACK tracking and retry handling

internal/config/config.go

@@ -33,6 +33,8 @@ type Config struct {
 	HTTPRequestTimeoutMS  int
 	WorkerPollIntervalMS  int
 	MaxQueueBytesPerSOCKS int
+	AckTimeoutMS          int
+	MaxRetryCount         int
 	SessionIdleTimeoutMS  int
 	SOCKSIdleTimeoutMS    int
 	ReadBodyLimitBytes    int
@@ -52,6 +54,8 @@ func Load(path string) (Config, error) {
 		HTTPRequestTimeoutMS:  15000,
 		WorkerPollIntervalMS:  200,
 		MaxQueueBytesPerSOCKS: 1024 * 1024,
+		AckTimeoutMS:          5000,
+		MaxRetryCount:         5,
 		SessionIdleTimeoutMS:  5 * 60 * 1000,
 		SOCKSIdleTimeoutMS:    2 * 60 * 1000,
 		ReadBodyLimitBytes:    2 * 1024 * 1024,
@@ -153,6 +157,18 @@ func Load(path string) (Config, error) {
 				return Config{}, fmt.Errorf("parse MAX_QUEUE_BYTES_PER_SOCKS: %w", err)
 			}
 			cfg.MaxQueueBytesPerSOCKS = size
+		case "ACK_TIMEOUT_MS":
+			timeout, err := strconv.Atoi(value)
+			if err != nil {
+				return Config{}, fmt.Errorf("parse ACK_TIMEOUT_MS: %w", err)
+			}
+			cfg.AckTimeoutMS = timeout
+		case "MAX_RETRY_COUNT":
+			count, err := strconv.Atoi(value)
+			if err != nil {
+				return Config{}, fmt.Errorf("parse MAX_RETRY_COUNT: %w", err)
+			}
+			cfg.MaxRetryCount = count
 		case "SESSION_IDLE_TIMEOUT_MS":
 			timeout, err := strconv.Atoi(value)
 			if err != nil {
@@ -200,6 +216,12 @@ func (c Config) ValidateClient() error {
 	if c.WorkerPollIntervalMS < 1 {
 		return fmt.Errorf("invalid WORKER_POLL_INTERVAL_MS: %d", c.WorkerPollIntervalMS)
 	}
+	if c.AckTimeoutMS < 1 {
+		return fmt.Errorf("invalid ACK_TIMEOUT_MS: %d", c.AckTimeoutMS)
+	}
+	if c.MaxRetryCount < 0 {
+		return fmt.Errorf("invalid MAX_RETRY_COUNT: %d", c.MaxRetryCount)
+	}
 	if c.MaxQueueBytesPerSOCKS < c.MaxChunkSize {
 		return fmt.Errorf("MAX_QUEUE_BYTES_PER_SOCKS must be >= MAX_CHUNK_SIZE")
 	}