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次便会报错。

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