package main

import (
	"fmt"
	"log"
	"os"
	"runtime/debug"
	"time"

	"gitee.com/xuesongtao/gotool/base"
	"gitee.com/xuesongtao/gotool/xfile"
	pslog "gitee.com/xuesongtao/ps-log"
	"gitee.com/xuesongtao/ps-log/line"
)

func main() {
	ps, err := pslog.NewPsLog(pslog.WithAsync2Tos())
	if err != nil {
		panic(err)
	}
	defer ps.Close()

	// 实时监听
	if err := ps.TailLogs(); err != nil {
		panic(err)
	}

	tmp := "log/test.log"

	// 日志行 merge 规则
	mergeLine := line.NewMulti()
	// 以时间格式开头, 如: 2023-02-10 16:13:53.441
	if err := mergeLine.StartPattern(`\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}.\d{3}`); err != nil {
		panic(err)
	}

	handler := &pslog.Handler{
		CleanOffset: true,           // 重新加载时, 清理已保存的 文件偏移量
		Change:      -1,             // 每次都保存文件偏移量
		Tail:        true,           // 实时监听
		ExpireAt:    pslog.NoExpire, // 不过期
		MergeRule:   mergeLine,
		Targets: []*pslog.Target{
			{
				Content:  " ",        // 目标内容
				Excludes: []string{}, // 排查内容
				To:       []pslog.PsLogWriter{&pslog.Stdout{}},
			},
		},
	}

	// 注册
	if err := ps.Register(handler); err != nil {
		panic(err)
	}
	closeCh := make(chan int)
	go func() {
		fh := xfile.NewFileHandle(tmp)
		if err := fh.Initf(os.O_RDWR | os.O_APPEND | os.O_TRUNC); err != nil {
			log.Println(err)
			return
		}
		defer fh.Close()

		appendContent := func(content string) {
			_, err := fh.AppendContent(content)
			if err != nil {
				log.Println("write err:", err)
			}
			time.Sleep(time.Second)
		}

		for i := 0; i <= 10; i++ {
			// 前缀
			timeStr := time.Now().Format(base.DatetimeFmt+".000") + " "
			content := timeStr + fmt.Sprint(i) + "\n"
			if i%5 == 0 {
				// 模拟 stack
				content = timeStr + string(debug.Stack())
			}
			appendContent(content)
		}
		close(closeCh)
	}()

	// 添加待监听的 path
	if err := ps.AddPaths(tmp); err != nil {
		panic(err)
	}

	// dump
	log.Println(ps.List())
	for range closeCh {
	}
}