A simple BasicAuth example

examples/basic-auth/main.go

@@ -0,0 +1,74 @@
+package main
+
+import (
+	"github.com/telanflow/mps"
+	"github.com/telanflow/mps/middleware"
+	"io/ioutil"
+	"log"
+	"net/http"
+	"net/url"
+	"time"
+)
+
+// A simple BasicAuth example
+func main() {
+	// endPoint server
+	go http.ListenAndServe("localhost:8080", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		// Basic Authentication
+		usr, pwd, ok := r.BasicAuth()
+		if !ok {
+			w.WriteHeader(401)
+			w.Write([]byte("401 Authentication Required"))
+			return
+		}
+		if usr != "test" || pwd != "test" {
+			w.WriteHeader(401)
+			w.Write([]byte("401 Authentication Required"))
+			return
+		}
+		w.Write([]byte("successful endPoint"))
+	}))
+
+	// proxy server
+	proxy := mps.NewHttpProxy()
+	// proxy BasicAuth
+	proxy.Use(middleware.BasicAuth("mps_realm", func(username, password string) bool {
+		return username == "mps" && password == "mps"
+	}))
+	proxy.UseFunc(func(req *http.Request, ctx *mps.Context) (*http.Response, error) {
+		// set endPoint BasicAuth
+		// Or you can set the endPoint BasicAuth on the client
+		req.SetBasicAuth("test", "test")
+		return ctx.Next(req)
+	})
+	go http.ListenAndServe("localhost:8081", proxy)
+
+	// wait proxy started
+	time.Sleep(2 * time.Second)
+
+	// send request
+	// request ==> proxy ==> http://localhost:8080
+	// response <== proxy <== http://localhost:8080
+	request, _ := http.NewRequest(http.MethodGet, "http://localhost:8080", nil)
+	http.DefaultClient.Transport = &http.Transport{
+		Proxy: func(req *http.Request) (*url.URL, error) {
+			// set proxy server BasicAuth
+			middleware.SetBasicAuth(req, "mps", "mps")
+
+			// set endPoint BasicAuth
+			// Or you can set the endPoint to BasicAuth on the proxy server
+			//req.SetBasicAuth("test", "test")
+
+			return url.Parse("http://localhost:8081")
+		},
+	}
+	resp, err := http.DefaultClient.Do(request)
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer resp.Body.Close()
+
+	body, _ := ioutil.ReadAll(resp.Body)
+	log.Println(string(body))
+}
+

examples/cascade-proxy/main.go

@@ -38,7 +38,7 @@ func main() {
 	}
 	go http.ListenAndServe("localhost:9992", proxy2)
 
-	// wait proxy server run
+	// wait proxy server started
 	time.Sleep(2 * time.Second)
 
 	// send request

examples/simple-http-proxy/main.go

@@ -18,7 +18,7 @@ func main() {
 	// create a http proxy server
 	proxy := mps.NewHttpProxy()
 	proxy.UseFunc(func(req *http.Request, ctx *mps.Context) (*http.Response, error) {
-		log.Printf("[INFO] middleware -- %s", req.URL)
+		log.Printf("[INFO] middleware -- %s\n", req.URL)
 		return ctx.Next(req)
 	})