Go语言实现按照行数逆序读取文件 - 模拟tail命令

寒江蓑笠翁大约 4 分钟技术日志go

Go语言实现按照行数逆序读取文件 - 模拟tail命令


编写

Linux中的tail命令很多人应该都用过,我们经常用它来看日志

$ tail -n 100 /etc/nginx/access.log

如果在用go语言实现这一个功能,也是非常的简单,要点就是利用Seek这一函数

// Seek sets the offset for the next Read or Write on file to offset, interpreted
// according to whence: 0 means relative to the origin of the file, 1 means
// relative to the current offset, and 2 means relative to the end.
// It returns the new offset and an error, if any.
Seek(offset int64, whence int) (ret int64, err error)

它的第一个参数是偏移量,第二个参数总共有三个值

  • 0,相对于文件头
  • 1,相对于当前的偏移量
  • 2,相对于文件末尾

一个需要注意的点就是换行符问题,在linux上换行符是CR\n,而在windows上则是CRLF\r\n,在计算偏移量的时候这个问题不能忽视掉,在逆序读取的时候就一个字节一个字节的读,当遇到\n时就停止,然后再根据不同系统来更新偏移量。最后还需要注意的是逆序读取的偏移量不能小于文件大小的负数,否则就越过文件的起始位置了,在思路明确了以后,编码就比较轻松了,代码整体如下所示。

func TailAt(file *os.File, n int, offset int64) ([]byte, int64, error) {
	stat, err := file.Stat()
	if err != nil {
		return nil, 0, err
	}
	size := stat.Size()

	if offset == 0 {
		offset = -1
	}
	content := make([]byte, 0, n)
	char := make([]byte, 1)

	for n > 0 {
		// read byte one by one
		for offset >= -size {
			_, err := file.Seek(offset, io.SeekEnd)
			if err != nil {
				return nil, 0, err
			}

			_, err = file.Read(char)
			if err != nil {
				return nil, 0, err
			}
			if char[0] == '\n' {
				break
			}
			content = append(content, char[0])
			offset--
		}

		offset--

		if offset >= -size {
			if _, err := file.Seek(offset, io.SeekEnd); err != nil {
				return nil, 0, err
			}
			// consider CRLF
			_, err := file.Read(char)
			if err != nil {
				return nil, 0, err
			}
			if char[0] == '\r' {
				offset--
			}
		}

		if offset >= -size {
			content = append(content, '\n')
		}
		n--

		// prevent pointer overflow the head of file
		if offset < -size {
			offset = 0
			break
		}
	}

	// reverse entire contents
	slices.Reverse(content)
	return content, offset, err
}

测试

文件test.txt

hello nihao
buhao
wohao
dajiahao
aaa
bbb
ccccccdddda
z\nzcda

一个简单的测试代码

func main() {
	tails, _, err := Tail("test.txt", 2, -1)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println(string(tail))
}

输出

ccccccdddda
z\nzcda
上次编辑于:
贡献者: 246859