goploy/core/Log.go

package core

import (
	"fmt"
	"io"
	"log"
	"os"
	"path/filepath"
	"strings"
	"time"
)

// LogLevel is log level
type LogLevel string

// log level
const (
	TRACE   LogLevel = "TRACE: "
	WARNING LogLevel = "WARNING: "
	INFO    LogLevel = "INFO: "
	ERROR   LogLevel = "ERROR: "
)

// Log information to file with logged day
func Log(lv LogLevel, content string) {
	var logFile io.Writer
	logPathEnv := os.Getenv("LOG_PATH")
	if strings.ToLower(logPathEnv) == "stdout" {
		logFile = os.Stdout
	} else {
		logPath, err := filepath.Abs(logPathEnv)
		if err != nil {
			fmt.Println(err.Error())
		}
		if _, err := os.Stat(logPath); err != nil && os.IsNotExist(err) {
			err := os.Mkdir(logPath, os.ModePerm)
			if nil != err {
				fmt.Println(err.Error())
			}
		}
		file := logPath + "/" + time.Now().Format("20060102") + ".log"
		logFile, err = os.OpenFile(file, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0766)
		if nil != err {
			fmt.Println(err.Error())
		}
	}

	logger := log.New(logFile, string(lv), log.LstdFlags|log.Lshortfile)
	logger.Output(2, content)
}