update

examples/reconnection_test/main.go

@@ -0,0 +1,71 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"log"
+	"time"
+
+	"github.com/oarkflow/mq"
+)
+
+// Simple task handler for testing
+func testHandler(ctx context.Context, task *mq.Task) mq.Result {
+	fmt.Printf("Processing task: %s\n", task.ID)
+	return mq.Result{
+		Status:  "SUCCESS",
+		Payload: []byte(fmt.Sprintf(`{"message": "Processed task %s"}`, task.ID)),
+	}
+}
+
+func main() {
+	fmt.Println("=== Consumer Reconnection Test ===")
+	fmt.Println("This test demonstrates improved reconnection behavior with jitter retry.")
+	fmt.Println("Start a broker on :8081 and then stop it to see reconnection attempts.")
+	fmt.Println()
+
+	// Create consumer with custom retry configuration
+	consumer := mq.NewConsumer(
+		"test-consumer",
+		"test-queue",
+		testHandler,
+		mq.WithBrokerURL(":8081"),
+		mq.WithMaxRetries(3),               // Limit initial connection retries
+		mq.WithInitialDelay(1*time.Second), // Start with 1 second delay
+		mq.WithMaxBackoff(30*time.Second),  // Cap at 30 seconds
+		mq.WithHTTPApi(true),
+		mq.WithWorkerPool(10, 2, 1000),
+	)
+
+	// Start consumer in a goroutine so we can observe the behavior
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	go func() {
+		fmt.Println("Starting consumer...")
+		if err := consumer.Consume(ctx); err != nil {
+			log.Printf("Consumer stopped with error: %v", err)
+		}
+	}()
+
+	// Keep the main thread alive to observe reconnection behavior
+	fmt.Println("Consumer started. You can now:")
+	fmt.Println("1. Start a broker on :8081")
+	fmt.Println("2. Stop the broker to trigger reconnection attempts")
+	fmt.Println("3. Restart the broker to see successful reconnection")
+	fmt.Println("4. Press Ctrl+C to exit")
+	fmt.Println()
+	fmt.Println("Watch the logs to see the improved reconnection behavior with:")
+	fmt.Println("- Exponential backoff with jitter")
+	fmt.Println("- Throttling of rapid reconnection attempts")
+	fmt.Println("- Long backoff periods after many failed attempts")
+	fmt.Println("- Reset of retry counters on successful connections")
+
+	// Sleep for a long time to observe behavior
+	time.Sleep(10 * time.Minute)
+
+	// Graceful shutdown
+	fmt.Println("Shutting down consumer...")
+	consumer.Close()
+	fmt.Println("Test completed.")
+}