实现基本的反向代理

internal/nodes/http_client_pool_test.go

@@ -0,0 +1,78 @@
+package nodes
+
+import (
+	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
+	"runtime"
+	"testing"
+	"time"
+)
+
+func TestHTTPClientPool_Client(t *testing.T) {
+	pool := NewHTTPClientPool()
+
+	{
+		origin := &serverconfigs.OriginConfig{
+			Id:      1,
+			Version: 2,
+			Addr:    &serverconfigs.NetworkAddressConfig{Host: "127.0.0.1", PortRange: "1234"},
+		}
+		err := origin.Init()
+		if err != nil {
+			t.Fatal(err)
+		}
+		{
+			client, addr, err := pool.Client(nil, origin)
+			if err != nil {
+				t.Fatal(err)
+			}
+			t.Log("addr:", addr, "client:", client)
+		}
+		for i := 0; i < 10; i++ {
+			client, addr, err := pool.Client(nil, origin)
+			if err != nil {
+				t.Fatal(err)
+			}
+			t.Log("addr:", addr, "client:", client)
+		}
+	}
+}
+
+func TestHTTPClientPool_cleanClients(t *testing.T) {
+	origin := &serverconfigs.OriginConfig{
+		Id:      1,
+		Version: 2,
+		Addr:    &serverconfigs.NetworkAddressConfig{Host: "127.0.0.1", PortRange: "1234"},
+	}
+	err := origin.Init()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	pool := NewHTTPClientPool()
+	pool.clientExpiredDuration = 2 * time.Second
+
+	for i := 0; i < 10; i++ {
+		t.Log("get", i)
+		_, _, _ = pool.Client(nil, origin)
+		time.Sleep(1 * time.Second)
+	}
+}
+
+func BenchmarkHTTPClientPool_Client(b *testing.B) {
+	runtime.GOMAXPROCS(1)
+
+	origin := &serverconfigs.OriginConfig{
+		Id:      1,
+		Version: 2,
+		Addr:    &serverconfigs.NetworkAddressConfig{Host: "127.0.0.1", PortRange: "1234"},
+	}
+	err := origin.Init()
+	if err != nil {
+		b.Fatal(err)
+	}
+
+	pool := NewHTTPClientPool()
+	for i := 0; i < b.N; i++ {
+		_, _, _ = pool.Client(nil, origin)
+	}
+}