模拟http客户端增加proxyprotocol协议(golang)

 
package main

import (
"fmt"
"io/ioutil"
"net"
"os"
)
//模拟客户端
func main() {
if len(os.Args) < 2 {
fmt.Fprintf(os.Stderr, "Usage: %s host:port", os.Args[0])
}

//获取命令行参数 socket地址
server := os.Args[1]
addr, err := net.ResolveTCPAddr("tcp4", server)
checkError(err)

//建立tcp连接
conn, err := net.DialTCP("tcp4", nil, addr)
checkError(err)

// 增加proxyprotocol 协议数据
_, err = conn.Write([]byte("PROXY TCP4 10.10.10.10 20.20.20.20 1000 2000\r\n\r\n"))
checkError(err)

//向服务端发送数据
_, err = conn.Write([]byte("HEAD /echo HTTP/1.0\r\n\r\n"))
checkError(err)
//接收响应
response, _ := ioutil.ReadAll(conn)
fmt.Println(string(response))
os.Exit(0)
}

func checkError(err error) {
if err != nil {
fmt.Println(err)
os.Exit(1)
}
}