[日常]Go语言圣经--Channel习题

OStack程序员社区-中国程序员成长平台 › 门户 › 编程› Go›Go教程

原作者: [db:作者] 来自: [db:来源] 收藏邀请

练习 8.3：在netcat3例子中，conn虽然是一个interface类型的值，但是其底层真实类型是*net.TCPConn，代表一个TCP连接。一个TCP连接有读和写两个部分，可以使用CloseRead和CloseWrite方法分别关闭它们。修改netcat3的主goroutine代码，只关闭网络连接中写的部分，这样的话后台goroutine可以在标准输入被关闭后继续打印从reverb1服务器传回的数据。（要在reverb2服务器也完成同样的功能是比较困难的；参考练习 8.4。）

1.
net.Dial()
func Dial(network, address string) (Conn, error)
2.net.TCPConn
type TCPConn struct {
// contains filtered or unexported fields
}
TCPConn is an implementation of the Conn interface for TCP network connections.

package main

import (
        "io"
        "log"
        "net"
        "os"
)

func main() {
        conn, err := net.Dial("tcp", "localhost:8040")
        if err != nil {
                log.Fatal(err)
        }   
        //内置make函数创建一个channel,可以发送struct类型的数据
        done := make(chan struct{})
        //go语句调用一个函数字面量，启动goroutine的常用形式
        go func() {
                //从网络连接到标准输出，如果连接没断也会阻塞
                //如果TCP的读连接关闭会报错：use of closed network connection
                _, err := io.Copy(os.Stdout, conn)
                log.Println(err)
                log.Println("done")
                //发送channel给接收goroutine
                done <- struct{}{}
        }() 
        //从标准输入到网络连接中，这个地方会阻塞，按Control+D关闭标准输入
        mustCopy(conn, os.Stdin)
        //      conn.Close()
        //类型断言，调用*net.TCPConn的方法CloseWrite()只关闭TCP的写连接
        cw := conn.(*net.TCPConn)
        cw.CloseWrite()
        <-done // 阻塞等待后台 goroutine 完成接收channel
}
func mustCopy(dst io.Writer, src io.Reader) {
        if _, err := io.Copy(dst, src); err != nil {
                log.Fatal(err)
        }   
}