如何在Gorilla Mux中将HTTP重定向到https？

Question

I am developing a golang application and I am using Gorilla Mux and i want to redirect HTTP requests to HTTPS

here is what i have so far

package main

import (
    "net/http"

    "github.com/gorilla/mux"
    "github.com/zolamk/deviant/handlers"
    "github.com/zolamk/deviant/lib"
)

func main() {

    router := mux.NewRouter()
    // this is where i am trying to redirect
    router.PathPrefix("/").Schemes("HTTP").HandlerFunc(func(res http.ResponseWriter, req *http.Request) {

        http.Redirect(res, req, fmt.Sprintf("https://%s", req.URL), http.StatusSeeOther)

    })

    router.Handle("/", handlers.ContextHandler(handlers.Index)).Methods("GET")

    router.Handle("/register/", handlers.ContextHandler(handlers.Register)).Methods("GET")

    router.Handle("/register/", handlers.ContextHandler(handlers.RegisterPost)).Methods("POST")

    router.Handle("/login/", handlers.ContextHandler(handlers.Login)).Methods("GET")

    router.Handle("/login/", handlers.ContextHandler(handlers.LoginPost)).Methods("POST")

    router.Handle("/logout/", handlers.ContextHandler(handlers.Logout)).Methods("GET")

    if lib.Settings.ServeStatic {

        router.PathPrefix("/public/").Handler(http.FileServer(http.Dir("./")))

    }

    router.NotFoundHandler = handlers.ContextHandler(handlers.NotFound)

    log.Printf("Deviant running @ http://%s
", lib.Settings.Address)

    loggedRouter := handlers.LoggedRouter(os.Stdout, router)

    log.Fatal(http.ListenAndServe(lib.Settings.Address, loggedRouter))

}

so like i said before how do i redirect HTTP traffic to HTTPS without affecting my other routes? Thank You.

Answer 1

What I ended up doing was that, I wrote a middleware that redirects HTTP requests to HTTPS

func RedirectToHTTPSRouter(next http.Handler) http.Handler {
    return http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
        proto := req.Header.Get("x-forwarded-proto")
        if proto == "http" || proto == "HTTP" {
            http.Redirect(res, req, fmt.Sprintf("https://%s%s", req.Host, req.URL), http.StatusPermanentRedirect)
            return
        }

        next.ServeHTTP(res, req)

    })
}

func main() {

    router := mux.NewRouter()

    httpsRouter := RedirectToHTTPSRouter(router)

    log.Fatal(http.ListenAndServe(lib.Settings.Address, httpsRouter))

}