114 lines
2.1 KiB
Go
114 lines
2.1 KiB
Go
package loghandler
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"jannex/log-manager/modules/config"
|
|
"jannex/log-manager/modules/structs"
|
|
"jannex/log-manager/modules/utils"
|
|
"os"
|
|
"strconv"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
var FileMutexMap = make(map[string]*sync.Mutex)
|
|
var FileMutexMapLock sync.Mutex
|
|
|
|
func getFileMutex(filePath string) *sync.Mutex {
|
|
FileMutexMapLock.Lock()
|
|
defer FileMutexMapLock.Unlock()
|
|
|
|
mutex, ok := FileMutexMap[filePath]
|
|
|
|
if !ok {
|
|
mutex = &sync.Mutex{}
|
|
FileMutexMap[filePath] = mutex
|
|
}
|
|
|
|
return mutex
|
|
}
|
|
|
|
func AddLog(body structs.LogBody) {
|
|
year, month, day := time.Now().Date()
|
|
|
|
logFolder := config.Cfg.LogFolder
|
|
|
|
utils.CreateDirectoryIfNotExists(logFolder + body.Type)
|
|
|
|
path := logFolder + body.Type + "/" + strconv.Itoa(day) + "-" + strconv.Itoa(int(month)) + "-" + strconv.Itoa(year) + ".log"
|
|
|
|
// get the mutex for this file
|
|
mutex := getFileMutex(path)
|
|
|
|
// lock the mutex to ensure the file is only opened once at a time
|
|
mutex.Lock()
|
|
defer mutex.Unlock()
|
|
|
|
file, err := os.OpenFile(path, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
|
|
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
|
|
defer file.Close()
|
|
|
|
for _, log := range body.Logs {
|
|
if _, err := fmt.Fprintln(file, log); err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func GetAvailableLogFiles(logType string) ([]string, error) {
|
|
var availableLogs []string
|
|
|
|
path := config.Cfg.LogFolder + logType
|
|
|
|
files, err := os.ReadDir(path)
|
|
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return []string{}, err
|
|
}
|
|
|
|
for _, file := range files {
|
|
availableLogs = append(availableLogs, file.Name())
|
|
}
|
|
|
|
return availableLogs, nil
|
|
}
|
|
|
|
func GetLogByDate(logType string, date string) ([]string, error) {
|
|
var logs []string
|
|
|
|
path := config.Cfg.LogFolder + logType + "/" + date + ".log"
|
|
|
|
// get the mutex for this file
|
|
mutex := getFileMutex(path)
|
|
|
|
// lock the mutex to ensure the file is only opened once at a time
|
|
mutex.Lock()
|
|
defer mutex.Unlock()
|
|
|
|
file, err := os.Open(path)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return []string{}, err
|
|
}
|
|
defer file.Close()
|
|
|
|
scanner := bufio.NewScanner(file)
|
|
|
|
for scanner.Scan() {
|
|
logs = append(logs, scanner.Text())
|
|
}
|
|
|
|
if len(logs) == 0 {
|
|
return []string{}, nil
|
|
}
|
|
|
|
return logs, nil
|
|
}
|