middlewareManage

middleware.go

// With trilopin fixes

package main

import "fmt"
import "net/http"
import "log"

type Middleware func(http.Handler) http.Handler
type MiddlewareManager struct {
	ms []Middleware
}

func (mm *MiddlewareManager) Compose(m ...Middleware) {
	mm.ms = append(mm.ms, m...)
}

func (mm *MiddlewareManager) DoMiddle(h http.Handler) http.Handler {
	for _, m := range mm.ms {
		h = m(h)
	}
	return h
}

func logger(h http.Handler) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		log.Printf("%s requested %s", r.RemoteAddr, r.URL)
		h.ServeHTTP(w, r)
	})
}

func test(h http.Handler) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		log.Printf("test function %s", r.URL)
		h.ServeHTTP(w, r)
	})
}

func main() {
	mm := MiddlewareManager{}
	mm.Compose(logger, test)

	mux := http.NewServeMux()
	mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprintln(w, "Are you here?")
	})

	mmux := mm.DoMiddle(mux)
	err := http.ListenAndServe(":9999", mmux)
	log.Fatal(err)
}