router

package main

import (
	"fmt"
	"io/ioutil"
	"net/http"
	"time"

	"github.com/trencat/netutils/http/router"
)

func handler(w http.ResponseWriter, r *http.Request) {
	fmt.Fprint(w, "Test handler\n")
}

func middleware(fn http.HandlerFunc) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprint(w, "Middleware acting before handler")

		fn(w, r)

		fmt.Fprint(w, "Middleware acting after handler")
	}
}

func funnyMiddleware(fn http.HandlerFunc) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprint(w, "FunnyMiddleware acting before handler")

		fn(w, r)

		fmt.Fprint(w, "FunnyMiddleware acting after handler")
	}
}

func main() {
	r := make(router.Router)

	// Request with no middleware
	r.HandlerFunc("/nomiddleware", handler)

	// Request with middleware
	r.AddRoute("/middleware", middleware)
	r.AddRoute("/middleware", funnyMiddleware)
	r.HandlerFunc("/middleware", handler)

	r.Build()

	// Run server, omitting error handling for clarity
	go http.ListenAndServe(":8080", nil)

	// Wait a second until server is up
	time.Sleep(time.Duration(1) * time.Second)

	// Do request. Omitting error handling for clarity
	response, _ := http.Get("http://127.0.0.1:8080/nomiddleware")
	body, _ := ioutil.ReadAll(response.Body)
	fmt.Printf("%s", body)

	// Do request. Omitting error handling for clarity
	response, _ = http.Get("http://127.0.0.1:8080/middleware")
	body, _ = ioutil.ReadAll(response.Body)
	fmt.Printf("%s", body)
}

Output: