more features

2024-11-14 19:33:04 +01:00
parent d617fd9657
commit 3f6f78c839
5 changed files with 156 additions and 62 deletions
--- a/discarded-host_test.go
+++ b/discarded-host_test.go
@@ -6,9 +6,10 @@ import (
 )

 func TestDiscardedHost(t *testing.T) {
-	dh := NewDiscardedHostHandler(func(host string) {
+	dh := NewDiscardedHostHandler(8080)
+	dh.SetReconnectHandler(func(host string) {
 		t.Log(host)
-	}, 8080)
+	})
 	dh.AppendHost("localhost")
 	time.Sleep(2 * time.Second)
 	if dh.hosts[0].Tries == 0 {