golang NewSingleHostReverseProxy + gorilla mux

NewSingleHostReverseProxy+gorillamux

package main

import (
	"net/http"
	"net/http/httputil"
	"net/url"

	"github.com/gorilla/mux"
)

// package main

// import (
// 	"net/http"

// 	"github.com/gorilla/mux"
// )

// func main() {
// 	r := mux.NewRouter()
// 	r.HandleFunc("/actions", func(w http.ResponseWriter, r *http.Request) {
// 		w.Write([]byte("actions"))
// 		w.WriteHeader(200)
// 	})
// 	http.Handle("/", r)
// 	http.ListenAndServe(":3333", r)
// }

// Run server above
// call curl http://localhost:3222/target/actions
// Url through proxy:  http://localhost:3333/actions

func main() {
	target := "http://localhost:3333"
	remote, err := url.Parse(target)
	if err != nil {
		panic(err)
	}

	proxy := httputil.NewSingleHostReverseProxy(remote)
	r := mux.NewRouter()

	r.HandleFunc("/target/{forward:.*}", proxyHandler(remote, proxy))
	http.Handle("/", r)

	http.ListenAndServe(":3222", r)
}

func proxyHandler(url *url.URL, p *httputil.ReverseProxy) func(http.ResponseWriter, *http.Request) {
	return func(w http.ResponseWriter, r *http.Request) {
		r.URL.Host = url.Host
		r.URL.Scheme = url.Scheme
		r.Header.Set("X-Forwarded-Host", r.Header.Get("Host"))
		r.Host = url.Host

		r.URL.Path = mux.Vars(r)["forward"]
		p.ServeHTTP(w, r)
	}
}