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

Go  /  管理员 发布于 2年前   890

在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的第一个窗口。

注意 : 

有待改进。

与其每次都读文件以获得最后一行的大小，不如尝试让它读一次。