golang io.copy詳解

golang io.copy是一個用於將數據從一個讀取器（Reader）複製到一個寫入器（Writer）的函數。該函數的調用非常簡單，只需要提供源讀取器和目標寫入器即可：

io.Copy(destination, source)

在這個函數中，數據從源讀取器傳輸到目標寫入器，直到源讀取器到達結尾。

一、更快的文件複製

通過使用io.copy，可以實現更快的文件複製，因為它使用了內置的bufio緩存。以下是一個使用io.copy來複制文件的示例：

func copyFile(dst, src string) (written int64, err error) {
    srcFile, err := os.Open(src)
    if err != nil {
        return
    }
    defer srcFile.Close()

    dstFile, err := os.Create(dst)
    if err != nil {
        return
    }
    defer dstFile.Close()

    return io.Copy(dstFile, srcFile)
}

在這個示例中，我們創建了一個名為copyFile的函數，該函數使用了io.copy來複制一個文件。該函數首先打開了源文件和目標文件，然後調用了io.Copy函數將數據從源文件複製到目標文件。

二、處理HTTP請求

io.copy還可以用於處理HTTP請求，這通常是在從一個URL下載文件時使用的。以下是一個使用io.copy從HTTP響應中複製數據的示例：

resp, err := http.Get("http://example.com/")
if err != nil {
    // handle error
}
defer resp.Body.Close()

_, err = io.Copy(os.Stdout, resp.Body)
if err != nil {
    // handle error
}

在這個示例中，我們首先使用http.Get函數向一個URL發送GET請求，並將響應存儲在resp變數中。然後，我們使用resp.Body來獲取HTTP響應的主體。接下來我們將os.Stdout作為目標寫入器傳遞給io.Copy函數，該函數將HTTP響應的主體傳輸到標準輸出。

三、處理Websocket數據

io.copy同樣適用於處理Websocket數據，以下是一個使用io.copy處理Websocket數據的示例：

func handleWebsocket(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println(err)
        return
    }
    defer conn.Close()

    for {
        messageType, message, err := conn.ReadMessage()
        if err != nil {
            log.Println(err)
            return
        }

        err = conn.WriteMessage(messageType, message)
        if err != nil {
            log.Println(err)
            return
        }

        _, err = io.Copy(conn, conn)
        if err != nil {
            log.Println(err)
            return
        }
    }
}

在這個示例中，我們首先使用upgrader.Upgrade函數升級HTTP連接為Websocket連接，並將連接存儲在變數conn中。我們然後在一個無限循環中讀取來自客戶端的Websocket消息，並將其回顯回客戶端。最後，我們使用io.Copy將Websocket連接複製到它自己上，從而實現了一個簡單的Websocket回聲伺服器。

四、處理TCP連接數據

io.copy還可以用於處理TCP連接數據，以下是一個使用io.copy處理TCP連接數據的示例：

func handleConnection(conn net.Conn) {
    defer conn.Close()

    _, err := io.Copy(conn, conn)
    if err != nil {
        log.Println(err)
    }
}

在這個示例中，我們定義了一個名為handleConnection的函數，該函數使用io.copy將TCP連接複製到它自己上。如果發生任何錯誤，則該函數會記錄該錯誤並退出。

五、結語

io.copy是一個非常有用的函數，它是在golang io包中定義的。通過使用io.copy，我們可以輕鬆地從一個讀取器複製數據到另一個寫入器並處理HTTP請求、Websocket數據和TCP連接數據等各種場景。