如何在Golang中使用io进行输入输出_读取和写入数据流

Go语言io包通过io.Reader和io.Writer接口统一输入输出操作,os.Stdin、os.Stdout和*os.File均实现它们,支持逐行读取、高效写入及文件流式复制。

Go语言中,io包是处理输入输出的核心基础,它不直接提供“读字符串”或“写文件”的高层封装,而是定义统一的接口(如io.Readerio.Writer),让各种数据源和目标(文件、网络连接、内存缓冲、标准输入输出等)能以一致方式交互。掌握这些接口及其常用实现,是写出灵活、可测试、符合Go风格IO代码的关键。

理解io.Reader和io.Writer:两个核心接口

io.Reader只定义了一个方法:
Read(p []byte) (n int, err error)
它从数据源读取最多len(p)字节到切片p中,返回实际读取字节数n和错误。调用者需循环读取直到err == io.EOF(表示流结束)。

io.Writer也只有一个方法:
Write(p []byte) (n int, err error)
它将切片p中的内容写入目标,返回实际写入字节数和错误。写操作可能未完全写入,需检查n是否等于len(p),必要时重试或使用io.WriteString等辅助函数。

这两个接口极简,但威力强大——只要类型实现了它们,就能被任何接受io.Reader/io.Writer参数的函数复用。

从标准输入读取:os.Stdin 是 io.Reader

os.Stdin是预定义的*os.File,实现了io.Reader。常见用法有:

  • 逐行读取(推荐):用bufio.Scanner包装os.Stdin,自动处理换行和缓冲
  • 按字节/字节块读取:直接调用os.Stdin.Read(),需手动管理缓冲区和EOF判断

示例(逐行):

立即学习“go语言免费学习笔记(深入)”;

package main
import (
  "bufio"
  "fmt"
  "os"
)
func main() {
  scanner := bufio.NewScanner(os.Stdin)
  for scanner.Scan() {
    line := scanner.Text() // 不含换行符
    fmt.Println("你输入了:", line)
  }
  if err := scanner.Err(); err != nil {
    fmt.Fprintln(os.Stderr, "读取错误:", err)
  }
}

向标准输出写入:os.Stdout 是 io.Writer

os.Stdout实现了io.Writer,可直接传给需要io.Writer的函数(如fmt.Fprint*系列)。常见写法:

  • fmt.Println("hello") —— 默认写入os.Stdout
  • fmt.Fprintln(os.Stdout, "hello") —— 显式指定目标
  • io.WriteString(os.Stdout, "hello\n") —— 直接调用io包函数
  • bufio.Writer包装提升性能(尤其高频小写)

注意:bufio.Writer需在退出前调用Flush(),否则缓冲内容可能丢失。

文件读写:*os.File 同时实现 Reader 和 Writer

打开文件后得到*os.File,它同时满足io.Readerio.Writer。典型流程:

  • 读文件:f, _ := os.Open("input.txt") → 用io.Copy(dst, f)bufio.Scanner读取
  • 写文件:f, _ := os.Create("output.txt")os.OpenFile(..., os.O_WRONLY|os.O_CREATE) → 用fmt.Fprintio.WriteString写入
  • 务必用defer f.Close()确保关闭

高效复制示例:
io.Copy(os.Stdout, f) —— 将文件内容直接流式输出到终端,内存占用恒定。