Go语言中实现文件实时追踪：仿tail -f功能详解

go get github.com/ActiveState/tail

import "github.com/ActiveState/tail"

package main

import (
    "context"
    "fmt"
    "log"
    "os"
    "os/signal"
    "syscall"
    "time"

    "github.com/ActiveState/tail"
)

func main() {
    // 1. 定义要追踪的文件路径
    filePath := "test_log.txt"

    // 2. 创建或清空文件，用于演示
    err := os.WriteFile(filePath, []byte("Initial log line.\n"), 0644)
    if err != nil {
        log.Fatalf("Failed to create file: %v", err)
    }
    fmt.Printf("Created file: %s\n", filePath)

    // 3. 启动一个goroutine模拟文件内容增长
    ctx, cancel := context.WithCancel(context.Background())
    defer cancel() // 确保在main函数结束时调用cancel

    go simulateFileGrowth(ctx, filePath)

    // 4. 配置tail选项
    //   - Follow: 持续追踪文件
    //   - ReOpen: 当文件被重命名或截断时重新打开
    //   - Poll: 使用轮询方式检测文件变化 (适用于不支持inotify的系统或网络文件系统)
    //   - Location: 从文件末尾开始读取 (0表示从头开始，-1表示从末尾开始)
    config := tail.Config{
        Follow:    true, // 持续追踪
        ReOpen:    true, // 文件被截断或重命名时重新打开
        MustExist: false, // 如果文件不存在，不立即报错，等待文件创建
        Poll:      true, // 使用轮询模式，更广泛兼容
        Location:  &tail.SeekInfo{Offset: 0, Whence: os.SEEK_END}, // 从文件末尾开始
        Logger:    tail.DefaultLogger, // 使用默认日志记录器
    }

    // 5. 启动文件追踪
    t, err := tail.TailFile(filePath, config)
    if err != nil {
        log.Fatalf("Failed to tail file: %v", err)
    }
    defer t.Cleanup() // 确保在程序退出时清理资源

    fmt.Println("Started tailing file. Press Ctrl+C to stop.")

    // 6. 监听系统中断信号，实现优雅退出
    sigChan := make(chan os.Signal, 1)
    signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)

    // 7. 从tail的Lines通道读取新行
    for {
        select {
        case line, ok := <-t.Lines:
            if !ok {
                // 通道关闭，通常意味着tail实例已停止
                fmt.Println("Tailer channel closed. Exiting.")
                return
            }
            fmt.Printf("[新行]: %s\n", line.Text)
        case err := <-t.Errors:
            // 处理tail内部的错误
            log.Printf("Tailer error: %v", err)
        case <-sigChan:
            // 接收到中断信号，停止tail
            fmt.Println("\nReceived interrupt signal. Stopping tailer...")
            err := t.Stop() // 停止tailer
            if err != nil {
                log.Printf("Error stopping tailer: %v", err)
            }
            // 停止文件增长模拟
            cancel()
            // 等待一段时间确保所有goroutine完成
            time.Sleep(500 * time.Millisecond)
            fmt.Println("Tailer stopped gracefully.")
            return
        }
    }
}

// simulateFileGrowth 模拟向文件追加新内容
func simulateFileGrowth(ctx context.Context, filePath string) {
    ticker := time.NewTicker(2 * time.Second) // 每2秒追加一行
    defer ticker.Stop()

    count := 0
    for {
        select {
        case <-ctx.Done():
            fmt.Println("File growth simulation stopped.")
            return
        case <-ticker.C:
            f, err := os.OpenFile(filePath, os.O_APPEND|os.O_WRONLY|os.O_CREATE, 0644)
            if err != nil {
                log.Printf("Error opening file for append: %v", err)
                continue
            }
            count++
            newLine := fmt.Sprintf("This is a new log entry #%d at %s.\n", count, time.Now().Format("15:04:05"))
            _, err = f.WriteString(newLine)
            if err != nil {
                log.Printf("Error writing to file: %v", err)
            }
            f.Close()
            fmt.Printf("Appended: %s", newLine)
        }
    }
}