goProject/.svn/pristine/88/880f8b2789a7eb735412542d295753784c122a5c.svn-base

package impl_localfile

import (
	"fmt"
	"log"
	"os"
	"path/filepath"
	"strconv"
	"sync"
	"time"
)

const (
	con_FILE_SUFFIX = "txt"

	defaultPanicNum       = 10 // 默认支持的连续panic次数
	defaultPanicIntervals = 5  // 秒

	fileMaxSize = 1 << 27 // 128M
)

type fileLog struct {
	logPath string

	// fileDir + fileName 用于判断是否需要切换日志文件
	fileDir  string
	fileName string
	lv       levelType

	msgChan chan logObject

	f *os.File

	// 如果出现未知panic并一直发生，那么应该要真正的panic
	panicNum  int
	panicTime time.Time

	// 当前已经写入的大小
	curWriteSize int64
}

func newLog(logPath string, lv levelType) *fileLog {
	f := &fileLog{
		logPath:  logPath,
		fileDir:  "",
		fileName: "",
		msgChan:  make(chan logObject, 10240),
		lv:       lv,
		f:        nil,
	}

	go f.loop()

	return f
}

func (f *fileLog) loop() {
	defer func() {
		if r := recover(); r != nil {
			// 将错误输出，而不是记录到文件，是因为可能导致死循环
			fmt.Println("log file lv:", f.lv, " err:", r)

			// 超过间隔时间重置
			if time.Since(f.panicTime)/time.Second > defaultPanicIntervals {
				f.panicNum = 0
				f.panicTime = time.Now()
			}

			// 这里处理连续panic 也是防止循环调用形成死循环
			f.panicNum++
			if f.panicNum >= defaultPanicNum {
				panic(r)
			}
		}
		go f.loop()
	}()

	for logItem := range f.msgChan {
		f.writeLog(logItem)
	}
}

func (f *fileLog) writeLog(logObj logObject) {
	// 由于不使用直接 logUtil 调用，logUtilPlus 存在终端打印的控制，所以 logUtil 只作为纯文件日志组件
	// if PrintImportantLog && (logObj.level == warn || logObj.level == _error || logObj.level == fatal) {
	// 	fmt.Println(logObj.logInfo)
	// }

	// 检查是否需要去切换文件或者创建文件
	f.checkFileAndLoop(logObj.level, logObj.ifIncludeHour)

	f.curWriteSize += int64(len(logObj.logInfo))
	_, _ = f.f.WriteString(logObj.logInfo)
}

// 检查文件是否需要切换
func (f *fileLog) checkFileAndLoop(level levelType, ifIncludeHour bool) {
	// 获取当前时间
	now := time.Now()
	fileAbsoluteDirectory := filepath.Join(f.logPath, strconv.Itoa(now.Year()), strconv.Itoa(int(now.Month())))
	fileName := ""

	if ifIncludeHour {
		fileName = fmt.Sprintf("%s.%s.%s", time.Now().Format("2006-01-02-15"), level, con_FILE_SUFFIX)
	} else {
		fileName = fmt.Sprintf("%s.%s.%s", time.Now().Format("2006-01-02"), level, con_FILE_SUFFIX)
	}

	// 说明已经存在， 检查文件大小，并切换
	if f.f != nil && f.fileName == fileName && f.fileDir == fileAbsoluteDirectory {
		if f.curWriteSize < fileMaxSize {
			return
		}

		// 大小超额切换文件
		f.switchFile()
		return
	}

	// 创建文件夹
	if f.fileDir != fileAbsoluteDirectory {
		if err := os.MkdirAll(fileAbsoluteDirectory, os.ModePerm|os.ModeTemporary); err != nil {
			log.Println("make dir all is err :", err)
		}
		f.fileDir = fileAbsoluteDirectory
	}

	// 创建文件
	if f.fileName != fileName {
		f.fileName = fileName
	}

	// 到这里说明要切换了，关闭之前的file
	if f.f != nil {
		_ = f.f.Close()
	}

	// 正常切换文件
	f.createFile()
}

func (f *fileLog) switchFile() {
	// 关闭文件
	_ = f.f.Close()

	// 重命名
	curFileName := filepath.Join(f.fileDir, f.fileName)
	newFileName := filepath.Join(f.fileDir, fmt.Sprintf("%s.%s.%s", f.fileName, time.Now().Format("15-04-05.999"), con_FILE_SUFFIX))
	err := os.Rename(curFileName, newFileName)
	if err != nil {
		log.Println(err)
	}

	// 再次创建文件
	f.createFile()
}

func (f *fileLog) createFile() {
	// 得到最终的文件绝对路径
	fileAbsolutePath := filepath.Join(f.fileDir, f.fileName)
	// 打开文件(如果文件存在就以读写模式打开，并追加写入；如果文件不存在就创建，然后以写模式打开。)
	file, err := os.OpenFile(fileAbsolutePath, os.O_CREATE|os.O_APPEND|os.O_WRONLY, os.ModePerm|os.ModeTemporary)
	if err != nil {
		log.Println("open file is err :", err)
		return
	}
	f.f = file
	stat, _ := file.Stat()
	f.curWriteSize = stat.Size()
}

func (f *fileLog) SetLogPath(logPath string) {
	f.logPath = logPath
}

// Close 最后等待 3s 钟
func (f *fileLog) Close(wg *sync.WaitGroup, waitFinish bool) {
	ticker := time.NewTicker(3 * time.Second)
	defer func() {
		ticker.Stop()

		if f.f != nil {
			_ = f.f.Close()
		}

		wg.Done()
	}()

	for {
		if !waitFinish || len(f.msgChan) == 0 {
			break
		}

		select {
		case <-ticker.C:
			fmt.Println("wait close log file timeout:3s")
			return
		default:
			// 1ms 写入文件500条，最多等待 15ms 检查一次就行
			time.Sleep(15 * time.Millisecond)
		}
	}
}