Go语言圣经–Channel习题详解编程语言

练习 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) 
        }    
} 

  

原创文章,作者:Maggie-Hunter,如若转载,请注明出处:https://blog.ytso.com/12530.html

(0)
上一篇 2021年7月19日
下一篇 2021年7月19日

相关推荐

发表回复

登录后才能评论