golang 实现Location跳转方式

golang作为互联网时代的C语言，对网络的支持是非常友好的，最近想做个短网址转发使用，自然想到用Golang开发。

闲话少说，直接上源码：

package main 
import (
 "fmt"
 "log"
 "net/http"
)
 
func login(w http.ResponseWriter, r *http.Request) {
 fmt.Print(fmt.Sprintf("%v+", r))
 w.Header().Set("Cache-Control", "must-revalidate, no-store")
 w.Header().Set("Content-Type", " text/html;charset=UTF-8")
 w.Header().Set("Location", "http://wap.baidu.com/")//跳转地址设置
 w.WriteHeader(307)//关键在这里！
}
func main() {
 http.HandleFunc("/", login)              //设置访问的路由
 err := http.ListenAndServe(":9090", nil) //设置监听的端口
 if err != nil {
  log.Fatal("ListenAndServe: ", err)
 }
}

补充：Go获取Location重定向url

go本身不提供重定向的url，如果想要获取，只能从源码下手。

http.Get()接收到一个url后，会进一步确认，是否包含重定向url，如果包含，自动转到新的url，我们需要做的就是在这之前截取重定向url。

进入到http包中找到函数：

//自用，保留url地址信息，头部大写进行暴露
var RedirectMy string //此行源码中没有，为自己添加
 
func defaultCheckRedirect(req *Request, via []*Request) error {
 //自用，将url根据需求进行组合
 RedirectMy = "//"+req.URL.Host+req.URL.Path//此行源码中没有，为自己添加
 if len(via) >= 10 {
  return errors.New("stopped after 10 redirects")
 }
 return nil
}

从源码中可以看出，go本身支持10次重定向，超过10次便会报错。

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。如有错误或未考虑完全的地方，望不吝赐教。