transfer_middleware/until/sysLog/log.go

package sysLog

import (
	"context"
	"fmt"
	"github.com/zeromicro/go-zero/core/logx"
	"os"
	"path/filepath"
	"time"
)

func ErrLog(ctx context.Context, errContent ...any) {
	path, _ := errLogFile("err")
	// 创建一个文件对象
	file, err := os.OpenFile(path, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
	if err != nil {
		logx.Error(err)
		return
	}
	defer file.Close()
	// 创建一个日志写入器，将日志写入文件
	writer := logx.NewWriter(file)
	// 使用该写入器记录日志
	logx.SetWriter(writer)
	logx.WithContext(ctx).WithCallerSkip(2).Errorf("errlog:", errContent)
}

func LogSendMq(ctx context.Context, content ...any) {
	path, _ := errLogFile("produce")
	// 创建一个文件对象
	file, err := os.OpenFile(path, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
	if err != nil {
		logx.Error(err)
		return
	}
	defer file.Close()
	// 创建一个日志写入器，将日志写入文件
	writer := logx.NewWriter(file)
	// 使用该写入器记录日志
	logx.SetWriter(writer)
	logx.WithContext(ctx).WithCallerSkip(2).Info("sendMq:", content)
}

func ErrQueueLog(ctx context.Context, content ...any) {
	path, _ := logQueuePath()
	// 创建一个文件对象
	file, err := os.OpenFile(path, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
	if err != nil {
		logx.Error(err)
		return
	}
	defer file.Close()
	// 创建一个日志写入器，将日志写入文件
	writer := logx.NewWriter(file)
	// 使用该写入器记录日志
	logx.SetWriter(writer)
	logx.WithContext(ctx).WithCallerSkip(2).Errorf("queue:", content)
}

func logQueuePath() (string, error) {
	logPath, err := runtimePathForQueue()
	if err != nil {
		return "", err
	}
	path := fmt.Sprintf("%s/%s", logPath, "queue")
	err = CheckDir(path)
	if err != nil {
		return "", err
	}
	return fmt.Sprintf("%s/%s", path, time.Now().Format(time.DateOnly)), nil
}

func errLogFile(pathName string) (string, error) {
	logPath, err := runtimePath()
	if err != nil {
		return "", err
	}
	path := fmt.Sprintf("%s/%s", logPath, pathName)
	err = CheckDir(path)
	if err != nil {
		return "", err
	}
	return fmt.Sprintf("%s/%s", path, time.Now().Format(time.DateOnly)), nil
}

func runtimePath() (string, error) {
	path, err := os.Getwd()
	return fmt.Sprintf("%s/%s/", filepath.Dir(filepath.Dir(path)), "runtime"), err
}

func runtimePathForQueue() (string, error) {
	path, err := os.Getwd()
	return fmt.Sprintf("%s/%s/", filepath.Dir(filepath.Dir(filepath.Dir(path))), "runtime"), err
}

func CheckDir(path string) error {
	// 判断目录是否存在
	if _, err := os.Stat(path); os.IsNotExist(err) {
		// 如果目录不存在，则创建它
		err = os.MkdirAll(path, os.ModePerm)
		if err != nil {
			return err
		}
	} else if err != nil {
		// 如果Stat返回了其他错误（比如权限问题）
		return err
	}
	return nil
}