在go语言中模拟tail -f或从日志文件读取最后一行的代码示例
在Golang中模拟带有-f选项的Linux/Unix tail命令。
这个代码例子所做的是逐行读取整个文件,直到最后一行,并计算出最后一行的大小。
从最后一行的大小,通过用文件大小减去最后一行的大小,计算出一个偏移量。
之后,我们将从偏移量读取并显示最后一行的内容,并暂停2秒的延迟。
这将模拟Golang中的tail -f logfile命令。
示例代码:
gotail.go
package main
import (
"bufio"
"flag"
"fmt"
"io"
"os"
"time"
)
var previousOffset int64 = 0
func main() {
flag.Parse()
filename := flag.Arg(0)
delay := time.Tick(2 * time.Second)
for _ = range delay {
readLastLine(filename)
}
}
func readLastLine(filename string) {
file, err := os.Open(filename)
if err != nil {
panic(err)
}
defer file.Close()
reader := bufio.NewReader(file)
// 我们需要计算最后一行的大小,以便file.ReadAt(offset)能够工作。
// 注意:这不是一个非常有效的解决方案,因为我们需要读取
// 整个文件,至少要读一次
lastLineSize := 0
for {
line, _, err := reader.ReadLine()
if err == io.EOF {
break
}
lastLineSize = len(line)
}
fileInfo, err := os.Stat(filename)
//根据lastLineSize做一个缓冲区的大小。
buffer := make([]byte, lastLineSize)
// +1以补偿该行的初始0字节
// 否则,该行的初始字符将丢失。
// 我们不需要将整个文件读入内存,而只是从某个偏移量开始读。
offset := fileInfo.Size() - int64(lastLineSize+1)
numRead, err := file.ReadAt(buffer, offset)
if previousOffset != offset {
//打印出最后一行内容
buffer = buffer[:numRead]
fmt.Printf("%s \n", buffer)
previousOffset = offset
}
}
当你第一次运行它时,打开两个终端窗口。
在一个窗口中,运行:
>gotail file.log
它将显示file.log的最后一行。
然后你可以在另一个窗口中为file.log的底部添加其他行。
附加行将出现在执行gotail file.log的第一个窗口。
注意 :
有待改进。
与其每次都读文件以获得最后一行的大小,不如尝试让它读一次。
相关文章