golang io.Writer详解|极客笔记

golang io.Writer详解

在Go语言中，io.Writer接口定义了写入数据的通用方式。通过实现io.Writer接口，我们可以将数据写入到各种不同的目标，比如文件、网络连接、缓冲区等。

io.Writer接口

io.Writer接口定义如下：

type Writer interface {
    Write(p []byte) (n int, err error)
}

通过这个接口，我们可以将数据写入到某个目标，比如文件、网络连接等。具体的写入操作由各个实现io.Writer接口的类型来决定。

示例代码

下面是一个简单的示例代码，演示了如何使用io.Writer接口将数据写入到文件中：

package main

import (
    "fmt"
    "os"
)

func main() {
    file, err := os.Create("output.txt")
    if err != nil {
        fmt.Println("Error creating file:", err)
        return
    }
    defer file.Close()

    data := []byte("Hello, io.Writer!\n")
    n, err := file.Write(data)
    if err != nil {
        fmt.Println("Error writing to file:", err)
        return
    }

    fmt.Printf("Write %d bytes\n", n)
}

运行上面的代码，会创建一个名为output.txt的文件，并将Hello, io.Writer!写入到文件中。输出如下：

Write 16 bytes

自定义类型实现io.Writer接口

除了使用系统自带的类型实现io.Writer接口，我们也可以自定义类型来实现io.Writer接口。这样我们就可以将数据写入到自定义的目标中。

下面是一个自定义类型实现io.Writer接口的示例代码：

package main

import (
    "fmt"
)

type myWriter struct {
    data []byte
}

func (w *myWriter) Write(p []byte) (n int, err error) {
    w.data = append(w.data, p...)
    return len(p), nil
}

func main() {
    writer := &myWriter{}

    data := []byte("Hello, custom io.Writer!\n")
    n, err := writer.Write(data)
    if err != nil {
        fmt.Println("Error writing:", err)
        return
    }

    fmt.Printf("Write %d bytes\n", n)
    fmt.Printf("Data written: %s", writer.data)
}

运行上面的代码，会将Hello, custom io.Writer!写入到myWriter结构体的data字段中，并输出：

Write 24 bytes
Data written: Hello, custom io.Writer!

ioutil包

ioutil包是Go语言标准库中的一个提供I/O操作的工具包，其中包含了一些方便的函数，比如WriteFile()、ReadFile()等，用于简化文件读写操作。

下面是一个使用ioutil包写文件的示例代码：

package main

import (
    "fmt"
    "io/ioutil"
)

func main() {
    data := []byte("Hello, ioutil!\n")

    err := ioutil.WriteFile("output.txt", data, 0644)
    if err != nil {
        fmt.Println("Error writing file:", err)
        return
    }

    fmt.Println("File written successfully")
}

运行上面的代码，会将Hello, ioutil!写入到名为output.txt的文件中，并输出File written successfully。