GoLang之网络

浏览数：48 / 时间：2015年06月10日

GoLang之网络

Go语言标准库里提供的net包，支持基于IP层、TCP/UDP层及更高层面（如HTTP、FTP、SMTP）的网络操作，其中用于IP层的称为Raw Socket。

net包的Dial()函数用于创建网络连接，函数原型如下：

func Dial(net, addr string) (Conn, error)

其中net参数是网络协议的名字，addr参数是IP地址或域名；如果连接成功，返回连接对象，否则返回error。

目前，Dial()函数支持如下几种网络协议："tcp"、"udp"、"ip"、"ip6"等，例如：

conn, err := net.Dial("tcp", "192.168.0.10:2100") 　　 // TCP连接
conn, err := net.Dial("udp", "192.168.0.12:975")  　　// UDP连接
conn, err := net.Dial("ip4:icmp", "www.baidu.com") 　// ICMP连接

在成功连接连接后，可以使用conn的Write()和Read()方法读写数据。

实际上，Dial()函数是对DialTCP()、DialUDP()、DialIP()、DialUnix()函数的封装：

func DialTCP(net string, laddr, raddr *TCPAddr) (c *TCPConn, err error)
func DialUDP(net string, laddr, raddr *UDPAddr) (c *UDPConn, err error)
func DialIP(netProto string, laddr, raddr *IPAddr) (c *IPConn, err error)
func DialUnix(net string, laddr, raddr *UnixAddr) (c *UnixConn, err error)

下面是一个TCP示例程序：

func checkError(err error) {
    if err != nil {
        fmt.Fprintf(os.Stderr, "Fatal error: %s", err.Error()) 
        os.Exit(1)
    }   
}

func readFully(conn net.Conn) ([]byte, error) {
    defer conn.Close()

    result := bytes.NewBuffer(nil)
    var buf [512]byte
    for {
        n, err := conn.Read(buf[0:]) 
        result.Write(buf[0:n])
        if err != nil {
            if err == io.EOF {
                break 
            }   

            return nil, err 
        }   
    }   

    return result.Bytes(), nil 
}

func main() {
    
    addr := "127.0.0.1:80"
    conn, err := net.Dial("tcp", addr)
    checkError(err)

    _, err = conn.Write([]byte("GET /api/v3/get HTTP/1.1\r\n\r\n"))
    checkError(err)

    result, err := readFully(conn)
    checkError(err)

    fmt.Println(string(result))

    os.Exit(0)
}

HTTP协议

Go语言标准库内建提供了net/http包，涵盖了HTTP客户端和服务端的具体实现。

HTTP客户端

net/http包的Client类型提供了如下几个方法：

func (c *Client) Get(url string) (r *Response, err error)
func (c *Client) Post(url string, bodyType string, body io.Reader) (r *Response, err error)
func (c *Client) PostForm(url string, data url.Values) (r *Response, err error)
func (c *Client) Head(url string) (r *Response, err error)
func (c *Client) Do(req *Request) (r *Response, err error)

以GET方法为例：

resp, err := http.Get("http://cre.mix.sina.com.cn/api/v3/get")

if err != nil {
　　fmt.Println("get failed", err)
　　return
}

defer resp.Body.Close()

io.Copy(os.Stdout, resp.Body)

上面这段代码请求一个网站首页，并将其内容打印到标准输出流中。

如果希望对请求做更多的控制，可以使用DO()方法：

req, err := http.NewRequest("GET", "http://cre.mix.sina.com.cn/api/v3/get", nil)

req.Header.Add("User-Agent", "Go GO")

client := &http.Client{}

resp, err := client.Do(req)

HTTP服务端

使用net/http包提供的下面两个方法

func ListenAndServe(addr string, handler Handler) error
func ListenAndServeTLS(addr string, certFile string, keyFile string, handler Handler) error

ListenAndServe()函数有2个参数，第一个参数addr即监听地址，第二个参数表示服务端处理程序，通常为空，使用默认的http.DefaultServeMux进行处理。

服务端的业务逻辑使用http.Handle()或http.HandleFunc()，会默认注入http.DefaultServeMux中，如：

http.HandleFunc("/foo", func(w http.ResponseWriter, r *http.Request) {
        fmt.Fprintf(w, "Hello, %q", html.EscapeString(r.URL.Path))
    })

http.ListenAndServe(":8001", nil)

RPC协议

在Go中，标准库提供的net/rpc包实现了PRC协议需要的相关细节，开发者可以很方便地使用该包编写RPC的服务端和客户端程序。

json处理

在Web开发领域中，JSON被广泛用于web服务端程序和客户端之间的数据通信。

Go语言内建对JSON的支持，使用encoding/json标准库，开发者可以轻松使用Go程序生成和解析JSON格式的数据。

func Marshal(v interface{}) ([]byte, error)
func Unmarshal(vdata []byte, v interface{}) error

JSON编码的一个例子：

type Book struct {
    Title string
    Authors [] string
    Publisher string
    IsPublished bool
    Price float32
}

func main() {

    gobook := Book {
        "Go programming",
        []string { "XuShiwei", "HughLv", "Johnson"},
        "isturing.com.cn",
        true,
        9.99,
    }   

    
    // encode
    b, err := json.Marshal(gobook)　　// 变量b 是一个[]byte类型
    if err == nil {
        fmt.Println(b)
    }   

    // decode
    var book Book
    json.Unmarshal(b, &book)
    fmt.Println(book)
}