Go语言实现在文件中插入内容

导语

此脚本写的还很稚嫩，实现插入的功能也很简单，仅供大家参考，有问题可以留言或者私信。

代码实现

目录结构：

123.txt(大家准备个此名的文件，内容随便)

123456
abcefg

main.go

package main

import (
	"fmt"
	"io"
	"os"
)

func editFile() {
	// 要实现追加的方式插入值而不是覆盖
	// 源文件
	srcfile, err := os.OpenFile("./123.txt", os.O_RDWR, 0644)
	if err != nil {
		fmt.Printf("open file failed, err:%v\n", err)
		return
	}
	// 临时文件
	tmpfile, err := os.OpenFile("./123.tmp", os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0644)
	if err != nil {
		fmt.Printf("open file failed, err:%v\n", err)
		return
	}
	// 指定在什么位置插入内容
	// 读取光标前的内容写到临时文件
	var b [3]byte
	// Read函数传入一个数组并把内容写到数组里
	n, err := srcfile.Read(b[:])
	if err != nil {
		fmt.Printf("read file failed, err:%v\n", err)
		return
	}
	tmpfile.Write(b[:n])
	// 把插入的内容也写到临时文件
	tmpfile.WriteString("hello")
	// 把源文件光标后的内容也写入到临时文件
	var x [1024]byte
	for {
		n, err := srcfile.Read(x[:])
		if err == io.EOF {
			tmpfile.Write(x[:n])
			break
		}
		if err != nil {
			fmt.Printf("read from file failed, err:%v\n", err)
			return
		}
		tmpfile.Write(x[:n])
	}
	tmpfile.Close()
	srcfile.Close()
	// 使用os.Rename不能使用defer语句去关闭文件句柄，因为defer语句是在函数指定完去执行，但是Rename操作需要时需要先关闭句柄
	ok := os.Rename("./123.tmp", "./123.txt")
	if ok == nil {
		fmt.Println("文件操作完成")
	} else {
		fmt.Printf("err:%v", ok)
	}
}

func main() {
	editFile()
}