详解Golang实现http重定向https的方式

以前写代码时，都是直接将程序绑定到唯一端口提供http/https服务，在外层通过反向代理(nginx/caddy)来实现http和https的切换。随着上线后的服务越来越多，有一些服务无法直接通过反向代理来提供这种重定向，只能依靠代码自己实现。所以简要记录一下如何在代码中实现http到https的重定向。

分析

无论是反向代理还是代码自己实现，问题的本质都是判断请求是否是https请求。 如果是则直接处理，如果不是，则修改请求中的url地址，同时返回客户端一个重定向状态码(301/302/303/307)。但如果仔细分析的话，会衍生出另外的问题，返回哪个重定向码是合理的？

这个问题展开讨论，估计要写满满一大页，可能还得不出结论。 因此这里就不纠结到底返回哪个了，我使用的是307.

实现

如何我们从问题出现的场景开始分析，基本可以得出一个结论： 在需要转换的场景中，都是用户习惯性的首先发出了http请求，然后服务器才需要返回一个https的重定向。 因此实现的第一步就是创建一个监听http请求的端口:

go http.listenandserve(":8000", http.handlerfunc(redirect))

8000端口专门用来监听http请求，不能阻塞https主流程，因此单独扔给一个协程来处理。 redirect用来实现重定向:

func redirect(w http.responsewriter, req *http.request) { 
  _host := strings.split(req.host, ":")
  _host[1] = "8443"

  target := "https://" + strings.join(_host, ":") + req.url.path
  if len(req.url.rawquery) > 0 {
    target += "?" + req.url.rawquery
  }

  http.redirect(w, req, target, http.statustemporaryredirect)
}

8443是https监听的端口。 如果监听默认端口443，那么就可加可不加。 最后调用sdk中的redirect函数封装response。

处理完重定向之后，再处理https就变得很容易了:

router := mux.newrouter() 
  router.path("/").handlerfunc(handlehttps)
  c := cors.new(cors.options{
    allowedorigins:  []string{"*.devexp.cn"},
    allowedmethods:  []string{"head", "get", "post", "put", "patch", "delete", "options"},
    allowedheaders:  []string{"*"},
    allowcredentials: true,
    debug:      false,
    alloworiginfunc: func(origin string) bool {
      return true
    },
  })

  handler := c.handler(router)
  logrus.fatal(http.listenandservetls(":8443", "cert.crt", "cert.key", handler))

完整代码如下：

package main

import ( 
  "github.com/gorilla/mux"
  "github.com/rs/cors"
  "github.com/sirupsen/logrus"
  "net/http"
  "encoding/json"
  "log"
  "strings"
)

func main() { 
  go http.listenandserve(":8000", http.handlerfunc(redirect))

  router := mux.newrouter()
  router.path("/").handlerfunc(handlehttps)
  c := cors.new(cors.options{
    allowedorigins:  []string{"*.devexp.cn"},
    allowedmethods:  []string{"head", "get", "post", "put", "patch", "delete", "options"},
    allowedheaders:  []string{"*"},
    allowcredentials: true,
    debug:      false,
    alloworiginfunc: func(origin string) bool {
      return true
    },
  })

  handler := c.handler(router)
  logrus.fatal(http.listenandservetls(":8443", "cert.crt", "cert.key", handler))
}

func redirect(w http.responsewriter, req *http.request) { 
  _host := strings.split(req.host, ":")
  _host[1] = "8443"

  // remove/add not default ports from req.host
  target := "https://" + strings.join(_host, ":") + req.url.path
  if len(req.url.rawquery) > 0 {
    target += "?" + req.url.rawquery
  }
  log.printf("redirect to: %s", target)
  http.redirect(w, req, target,
    // see @andreiavrammsd comment: often 307 > 301
    http.statustemporaryredirect)
}

func handlehttps(w http.responsewriter, r *http.request) { 
  json.newencoder(w).encode(struct {
    name string
    age  int
    https bool
  }{
    "lala",
    11,
    true,
  })
}

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持。