49b7051cea
This allows mux to compare and match the requests based on the HTTP method as well. If we compare the methods in the middleware chain, then mux will try to redirect the request to the first match it finds.
61 lines
1.3 KiB
Go
61 lines
1.3 KiB
Go
package middleware
|
|
|
|
import (
|
|
"log"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/markbates/goth/gothic"
|
|
)
|
|
|
|
type Middleware func(http.HandlerFunc) http.HandlerFunc
|
|
|
|
// Logging logs all requests with its path and the time it took to process
|
|
func Logging() Middleware {
|
|
|
|
// Create a new Middleware
|
|
return func(f http.HandlerFunc) http.HandlerFunc {
|
|
|
|
// Define the http.HandlerFunc
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
// Do middleware things
|
|
start := time.Now()
|
|
defer func() { log.Println(r.URL.Path, time.Since(start)) }()
|
|
|
|
// Call the next middleware/handler in chain
|
|
f(w, r)
|
|
}
|
|
}
|
|
}
|
|
|
|
func Authenticated() Middleware {
|
|
|
|
// Create a new Middleware
|
|
return func(f http.HandlerFunc) http.HandlerFunc {
|
|
|
|
// Define the http.HandlerFunc
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
sess, err := gothic.GetFromSession("google", r)
|
|
if err != nil {
|
|
// Write an error and stop the handler chain
|
|
http.Error(w, http.StatusText(http.StatusForbidden), http.StatusForbidden)
|
|
return
|
|
}
|
|
|
|
log.Printf("Got the authenticated user: %s", sess)
|
|
|
|
f(w, r)
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
// Chain applies middlewares to a http.HandlerFunc
|
|
func Chain(f http.HandlerFunc, middlewares ...Middleware) http.HandlerFunc {
|
|
for _, m := range middlewares {
|
|
f = m(f)
|
|
}
|
|
return f
|
|
}
|