Golang io.Copy()函数的使用示例

在Golang中，我们常常需要读写文件或网络连接，而io包提供了一些常用的读写接口和方法。其中，io.Copy()函数是一个非常实用的函数，它可以将一个io.Reader接口的数据拷贝到一个io.Writer接口中，以实现数据的复制。

io.Copy()函数的基本用法

io.Copy()函数的基本语法如下：

func Copy(dst Writer, src Reader) (written int64, err error)

其中，dst表示要被拷贝到的目标io.Writer接口，src表示要被拷贝的源io.Reader接口。函数返回的是拷贝的字节数和可能发生的错误。

下面是一个简单的示例，演示了如何用io.Copy()函数将一个文件的内容拷贝到另一个文件中：

package main

import (
    "io"
    "os"
)

func main() {
    // 打开源文件
    srcFile, err := os.Open("source.txt")
    if err != nil {
        panic(err)
    }
    defer srcFile.Close()

    // 创建目标文件
    dstFile, err := os.Create("destination.txt")
    if err != nil {
        panic(err)
    }
    defer dstFile.Close()

    // 拷贝文件内容
    _, err = io.Copy(dstFile, srcFile)
    if err != nil {
        panic(err)
    }
}

上述代码中，首先通过os.Open()函数打开一个名为“source.txt”的文件，如果出现错误则panic。然后通过os.Create()函数创建一个名为“destination.txt”的文件，同样如果出现错误则panic。最后调用io.Copy()函数将源文件的内容拷贝到目标文件中，如果出现错误则panic。

io.Copy()函数的高级用法

除了基本用法外，io.Copy()函数还可以应用在一些高级场景中。

拷贝网络连接

在网络编程中，我们经常需要从一个连接读取数据并将其写入另一个连接中。在这种情况下，我们可以使用io.Copy()函数来简化代码。例如：

package main

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

func handleConnection(conn net.Conn) {
    // 从连接读取数据并拷贝到标准输出
    _, err := io.Copy(os.Stdout, conn)
    if err != nil {
        log.Println("[ERROR] copying from connection:", err)
    }
}

func main() {
    // 监听端口，处理连接
    listener, err := net.Listen("tcp", ":8080")
    if err != nil {
        log.Fatal("[ERROR] listening:", err)
    }
    defer listener.Close()

    for {
        // 接受连接，并开启goroutine处理连接
        conn, err := listener.Accept()
        if err != nil {
            log.Println("[ERROR] accepting:", err)
            continue
        }
        go handleConnection(conn)
    }
}

上述代码中，我们使用了net包中的net.Listen()函数创建了一个TCP服务器，监听本地8080端口。然后在每个新连接到来时，我们开启一个新的goroutine，负责处理这个连接。在handleConnection函数中，我们调用了io.Copy()函数，将连接中的数据拷贝到标准输出中。

限定拷贝的字节数

有时候我们只想拷贝一部分数据，而不是整个源数据。在这种情况下，我们可以用io.LimitReader()函数来限定拷贝的字节数。例如：

package main

import (
    "io"
    "os"
)

func main() {
    // 打开源文件
    srcFile, err := os.Open("source.txt")
    if err != nil {
        panic(err)
    }
    defer srcFile.Close()

    // 创建目标文件
    dstFile, err := os.Create("destination.txt")
    if err != nil {
        panic(err)
    }
    defer dstFile.Close()

    // 限制拷贝的字节数const limit = 1024 * 1024 // 限制拷贝的字节数为1MB
_, err = io.CopyN(dstFile, io.LimitReader(srcFile, limit), limit)
if err != nil {
    panic(err)
}

上述代码中，我们使用了io.LimitReader()函数来限制只拷贝source.txt文件的前1MB数据。然后调用io.CopyN()函数将限制后的数据拷贝到目标文件中。