Golang 小技巧：使用 io.LimitReader 限制读取大小

在 Go 中读取文件或 HTTP 请求体时，如果直接使用：

data, err := io.ReadAll(r)

会一直读取到数据结束。

如果数据来源不可信，内容特别大，就可能占用过多内存。

这时候可以使用 io.LimitReader 限制最多读取多少字节：

reader := io.LimitReader(r, 1024*1024)

data, err := io.ReadAll(reader)
if err != nil {
	return err
}

这里表示最多读取：

1 MB

例如读取 HTTP 请求体：

func handler(w http.ResponseWriter, r *http.Request) {
	reader := io.LimitReader(r.Body, 1024*1024)

	body, err := io.ReadAll(reader)
	if err != nil {
		http.Error(w, "读取失败", http.StatusBadRequest)
		return
	}

	fmt.Println(string(body))
}

小结

当读取的数据大小不可控时，可以使用：

io.LimitReader(reader, maxSize)

给读取操作加一个上限。

这个小技巧很适合 HTTP 请求体、上传文件、外部数据流等场景，可以避免一次读取过多数据导致内存占用异常。