Go 如何在文件开头处插入一行话

发布时间:2023-03-18浏览次数:0

支持注册ChatGPT Plus的OneKey虚拟卡
绑定Apple Pay、Google Pay、支付宝和微信支付进行日常消费

注册和了解更多 ->

silver
package main

import (
	"fmt"
	"io/ioutil"
	"os"
)

func main() {
	fileName := "test.txt" // 源文件名
	insertContent := "This is a new line\n" // 要插入的内容

	fileBytes, err := ioutil.ReadFile(fileName) // 读取源文件
	if err != nil {
		fmt.Println(err)
		return
	}

	tmpFile, err := ioutil.TempFile("", "tmp") // 创建一个临时文件
	if err != nil {
		fmt.Println(err)
		return
	}
	defer os.Remove(tmpFile.Name()) // 程序结束后删除临时文件

	tmpFile.Write([]byte(insertContent)) // 将要插入的内容写入到临时文件中
	tmpFile.Write(fileBytes)             // 将源文件的内容追加到临时文件中

	err = os.Rename(tmpFile.Name(), fileName) // 将临时文件重命名为源文件,覆盖原来的内容
	if err != nil {
		fmt.Println(err)
		return
	}

	fmt.Println("Insert success!")
}
字节笔记本扫描二维码查看更多内容