1. 程式人生 > >[日常] Go語言聖經--Channel習題

[日常] Go語言聖經--Channel習題

Go語言 inter fun ron goroutin close 創建 n) std

練習 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)
        }   
}

  

[日常] Go語言聖經--Channel習題