Add test

rp_test.go

@@ -47,7 +47,7 @@ func TestHTTPRouting(t *testing.T) {
 				us := testutil.NewUpstreamServer(t, u.hostname)
 				m[u.hostname] = us.URL + u.rootPath
 			}
-			r := testutil.NewRelayer(m)
+			r := testutil.NewSimpleRelayer(m)
 			port, err := testutil.NewPort()
 			if err != nil {
 				t.Fatal(err)

testutil/relayer.go

@@ -4,6 +4,7 @@ import (
 	"crypto/tls"
 	"fmt"
 	"net/http"
+	"net/http/httputil"
 	"net/url"
 	"os"
 	"path"
@@ -50,3 +51,32 @@ func (r *Relayer) GetCertificate(i *tls.ClientHelloInfo) (*tls.Certificate, erro
 	}
 	return &c, nil
 }
+
+func (r *Relayer) Rewrite(*httputil.ProxyRequest) error {
+	return nil
+}
+
+type SimpleRelayer struct {
+	h map[string]string
+}
+
+func NewSimpleRelayer(h map[string]string) *SimpleRelayer {
+	return &SimpleRelayer{
+		h: h,
+	}
+}
+
+func (r *SimpleRelayer) GetUpstream(req *http.Request) (*url.URL, error) {
+	host := req.Host
+	if upstream, ok := r.h[host]; ok {
+		uu, err := url.Parse(upstream)
+		if err != nil {
+			return nil, err
+		}
+		req.URL.Scheme = uu.Scheme
+		req.URL.Host = uu.Host
+		req.URL.Path = strings.ReplaceAll(path.Join(uu.Path, req.URL.Path), "//", "/")
+		return req.URL, nil
+	}
+	return nil, fmt.Errorf("not found upstream: %v", host)
+}