在go语言种实现模拟tail -f或从日志文件读取最后一行的示例

Go  /  管理员 发布于 1年前   310

在go中模拟带有 -f 选项的 Linux/Unix tail 命令。

此代码示例的作用是逐行读取整个文件，直到最后一行，并计算最后一行的大小。

根据最后一行的大小，用文件大小减去最后一行的大小，计算出偏移量。

之后，我们将从偏移量读取最后一行的内容，并暂停 2 秒钟显示。

这将模拟 Golang 中的 tail -f logfile 命令。

这与 

http://trac.mondorescue.org/browser/trunk/mindi-busybox/coreutils/tail.c 

中的内容并不完全相似，但本教程的目的是为了学习

示例代码：

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
         }
 }

首次运行时，打开 2 个终端窗口。

在一个窗口中，运行 ：

>gotail file.log

它将显示 file.log 的最后一行。

然后，你可以在另一个窗口的 file.log 底部添加其他行。

附加行将显示在执行 gotail file.log 的第一个窗口上。

注意：

仍有改进空间。与其每次都读取文件以获取最后一行的大小，不如尝试只读取一次。