go代理服务器代码

最近用废弃的小笔记本搭了个centos服务器挂着，搭了ssh，tomcat,go环境，想搭个代理服务器访问路由，不想安装，想自己写一个，顺便复习一下go语言知识。

一开始，我网上搜了一下go语言写代理服务器的代理，搜索到一个:

来自:http://symphony.b3log.org/article/1357452978419

(原文地址：http://kejibo.com/golang-http-proxy-server 已失效)

package main

import (
    "http"
    "log"
    "os"
    "io/ioutil"
)

func handler(w http.ResponseWriter, r *http.Request) {
    resp, err := http.DefaultClient.Do(r)
    defer resp.Body.Close()
    if err != nil { panic(err) }
    for k, v := range resp.Header {
        for _, vv := range v {
            w.Header().Add(k, vv)
        }
    }
    for _, c := range resp.SetCookie {
        w.Header().Add("Set-Cookie", c.Raw)
    }
    w.WriteHeader(resp.StatusCode)
    result, err := ioutil.ReadAll(resp.Body)
    if err != nil && err != os.EOF { panic(err) }
    w.Write(result)
}

func main() {
    http.HandleFunc("/", handler)
    log.Println("Start serving on port 8888")
    http.ListenAndServe(":8888", nil)
    os.Exit(0)
}

写成文件上传到服务器,go build test.go 编译，发现resp.SetCookie根本没有这个方法，编译不通过，上面文章说它要最新的http源码并编译到go源码里，我都惊呆了，还要修改它的源码啊，我看了下自己的go语言版本，是1.4.2，是15年今年的了，他写的文章的时候是13年，呵呵了，我的比他的新多了，怎么可能会旧？！还有他说的编译源码，是多么复杂啊~

于是乎。机智的我，为了编译通过，活生生的把它注释了。

go build test.go

好了，编译通过，执行它

/.test

显示开启了，还是要证实一下，扫描一下端口，看了下确实开了。

好像完成了喔，于是去浏览器设置一下代理，打开baidu.com，他告诉我方法错误了,我查看了一下打印信息，发现

defer resp.Body.Close()

这行报这个错:

invalid memory address or nil pointer dereference

网上找了一下，原来是resp变量空指针错误。

我又尝试打印err信息:

http: Request.RequestURI can’t be set in client requests.

就是说，ttp.DefaultClient.Do(r)这个用法是不正确的，是不能这样设置的，于是我找了一个替代品:http.NewRequest，它的优点就是可以get也可以post,应该还有put等操作，没尝试。

于是这段handler function被我耐心的改成了这样：

func handler(w http.ResponseWriter, r *http.Request) {

    fmt.Println("url:",r.RequestURI)
    fmt.Println("method:",r.Method)
    fmt.Println("header:",r.Header)

    client0 := &http.Client{}
    req, _ := http.NewRequest(r.Method, r.RequestURI, r.Body)

    fmt.Printf("send:%+v\n", req)                                                         //看下发送的结构

    resp, err := client0.Do(req) //发送

    if err!=nil {
                fmt.Println("error:",err)
            }
    defer resp.Body.Close()

    for _, value := range resp.Request.Cookies() {
        w.Header().Add(value.Name,value.Value)
    }

    for k, v := range resp.Header {
                for _, vv := range v {
                    w.Header().Add(k, vv)
                }
            }
    w.WriteHeader(resp.StatusCode)

    result, err := ioutil.ReadAll(resp.Body)
    if (err != nil) {
        fmt.Println("error:",err)
    }
    fmt.Println("result:" ,result);

    _,err = w.Write(result)
    if (err != nil) {
        fmt.Println("error:",err)
    }

实际上我是在mac电脑上编辑的，用服务器上传在编译运行速度太慢了 = =、

看上去，代码好像写的很不错的样子。

运行起来，访问baidu死循环啦，我又改了几次，看了几次，还是这样，无奈的我只好测试其他网站，那个测试的网站是我的一个简单的jsp网页，就一个简单的列表，我一访问，竟然快速的出现了，竟然成功访问了，可是我没改到什么代码啊，我在认真的想了一想，应该是访问baidu的时候没有放请求头过去，导致它访问baidu.com的时候，源码里面的链接是post访问的，且都是baidu.com，为了证实我的猜想，我添加了请求头个cookie上去

......
    req, _ := http.NewRequest(r.Method, r.RequestURI, r.Body)
    for k, v := range req.Header {
        for _, vv := range v {
            r.Header.Add(k, vv)
        }
    }
    for _, value := range req.Cookies() {
        r.Header.Add(value.Name,value.Value)
    }
......

编译运行，浏览器重新刷新，竟然一下子就打了，我的猜想原来是正确的。

服务器基本完成，进入测试阶段，先百度一下，成功了，不过部分图片裂开了，在进入其他的，csdn可以访问，图片问题，sina正常，oschina正常，就是百度那个图片加载时裂了。读者有兴趣自行研究研究，在这，我只是先把简单的代理服务器帖上来，方便大家使用。

最后浏览了一下整改代理服务器的代码：

package main

import (
    "net/http"
    "log"
    "io/ioutil"
//  "io"
    "fmt"
)

func handler(w http.ResponseWriter, r *http.Request) {
//  http.DefaultClient.
//  w.Write([]byte("dddd"))

    fmt.Println("url:",r.RequestURI)
    fmt.Println("method:",r.Method)
    fmt.Println("header:",r.Header)

    client0 := &http.Client{}
    req, _ := http.NewRequest(r.Method, r.RequestURI, r.Body)
    for k, v := range req.Header {
        for _, vv := range v {
            r.Header.Add(k, vv)
        }
    }
    for _, value := range req.Cookies() {
        r.Header.Add(value.Name,value.Value)
    }

    fmt.Printf("send:%+v\n", req)                                                         //看下发送的结构

    resp, err := client0.Do(req) //发送

    if err!=nil {
                fmt.Println("error:",err)
            }
    defer resp.Body.Close()

    for _, value := range resp.Request.Cookies() {
        w.Header().Add(value.Name,value.Value)
    }

    for k, v := range resp.Header {
                for _, vv := range v {
                    w.Header().Add(k, vv)
                }
            }
    w.WriteHeader(resp.StatusCode)

    result, err := ioutil.ReadAll(resp.Body)
    if (err != nil) {
        fmt.Println("error:",err)
    }
    fmt.Println("result:" ,result);

    _,err = w.Write(result)
    if (err != nil) {
        fmt.Println("error:",err)
    }

}

func main() {
    http.HandleFunc("/", handler)
    log.Println("Start serving on port 8089")
     http.ListenAndServe(":8089", nil)
    log.Println("server start")
}

看看自己改的代码，和网上那个人得代码，原来相差那么多，看哭了。

版权声明：本文为博主原创文章，未经博主允许不得转载。