loggerx/filePath.go

package loggerx

import (
	"fmt"
	"os"
	"path/filepath"
	"time"
)

// 文件操作

// 获取最新的文件名
func (l *Logger) nowFileName(event string) string {
	// ioc, _ := time.LoadLocation("Asia/Shanghai")
	// timeDir := fmt.Sprint(time.Now().In(ioc).Format("2006/01/02/15")) // 2006-01-02 15:04:05
	timeDir := fmt.Sprint(time.Now().Local().Format("2006/01/02")) // 2006-01-02 15:04:05
	if l.channel != "" {
		timeDir = l.channel + "/" + timeDir
	}
	path := l.option.dir + "/" + timeDir + "_"+event + ".log"
	// fmt.Println(filepath.Abs(path))
	return path
}

// 新建文件
func (l *Logger) getFile(event string,isRefresh bool) (*os.File, error) {
	f := l.loadFile(event)
	if f != nil && !isRefresh {
		return f, nil
	}

	l.mu.Lock()
	defer l.mu.Unlock()

	fileName := l.nowFileName(event)

	dir, _ := filepath.Split(fileName) // 识别目录与文件
	os.MkdirAll(dir, os.ModePerm)      // 创建多层目录，如果存在不会报错

	// 打开该文件，如果不存在则创建
	file, err := os.OpenFile(fileName, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0644)
	if err != nil {
		// 打开失败，尝试创建
		fmt.Println("打开日志文件失败")
		return nil, err
	}
	// 关闭原来的
	if f != nil {
		closeFile(f)
	}

	l.filePath.Store(l.channel, &filePath{
		file:     file,
		fileName: fileName,
	})

	return file, nil
}

// 加载文件
func (l *Logger) loadFile(event string) *os.File {
	val, ok := l.filePath.Load(l.channel)
	if !ok {
		return nil
	}
	f := val.(*filePath)
	if f == nil {
		return nil
	}
	if f.fileName != l.nowFileName(event) {
		// 原来的文件需关闭
		closeFile(f.file)
		return nil
	}
	return f.file
}

// 关闭文件
func closeFile(f *os.File) error {
	f.Sync()
	return f.Close()
}