feat: 日志持久化存储到文件+管理员后台查看/清空日志页面
This commit is contained in:
@@ -0,0 +1,74 @@
|
||||
package handlers
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
"strconv"
|
||||
|
||||
"sale/internal/utils"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
type LogHandler struct{}
|
||||
|
||||
func NewLogHandler() *LogHandler {
|
||||
return &LogHandler{}
|
||||
}
|
||||
|
||||
func (h *LogHandler) GetLogs(c *gin.Context) {
|
||||
lines := 500
|
||||
if l := c.Query("lines"); l != "" {
|
||||
if v, err := strconv.Atoi(l); err == nil && v > 0 {
|
||||
lines = v
|
||||
}
|
||||
}
|
||||
|
||||
logs, err := utils.GetLogs(lines)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to read logs"})
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, gin.H{"data": logs})
|
||||
}
|
||||
|
||||
func (h *LogHandler) ClearLogs(c *gin.Context) {
|
||||
if err := utils.ClearLogs(); err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to clear logs"})
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, gin.H{"message": "Logs cleared successfully"})
|
||||
}
|
||||
|
||||
func (h *LogHandler) GetLogInfo(c *gin.Context) {
|
||||
size, err := utils.GetLogSize()
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to get log info"})
|
||||
return
|
||||
}
|
||||
|
||||
modTime, err := utils.GetLogModTime()
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to get log info"})
|
||||
return
|
||||
}
|
||||
|
||||
var sizeStr string
|
||||
if size >= 1024*1024 {
|
||||
sizeStr = fmt.Sprintf("%.2f MB", float64(size)/(1024*1024))
|
||||
} else if size >= 1024 {
|
||||
sizeStr = fmt.Sprintf("%.2f KB", float64(size)/1024)
|
||||
} else {
|
||||
sizeStr = fmt.Sprintf("%d B", size)
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, gin.H{
|
||||
"data": gin.H{
|
||||
"size": size,
|
||||
"size_str": sizeStr,
|
||||
"mod_time": modTime,
|
||||
},
|
||||
})
|
||||
}
|
||||
@@ -25,6 +25,7 @@ func SetupRoutes(r *gin.Engine) {
|
||||
bannerHandler := handlers.NewBannerHandler()
|
||||
paymentChannelHandler := handlers.NewPaymentChannelHandler()
|
||||
paymentHandler := handlers.NewPaymentHandler()
|
||||
logHandler := handlers.NewLogHandler()
|
||||
|
||||
r.Use(middlewares.CORSMiddleware())
|
||||
|
||||
@@ -235,6 +236,10 @@ func SetupRoutes(r *gin.Engine) {
|
||||
adminPaymentChannels.DELETE("/:id", paymentChannelHandler.Delete)
|
||||
adminPaymentChannels.PUT("/:id/toggle", paymentChannelHandler.Toggle)
|
||||
}
|
||||
|
||||
admin.GET("/logs", logHandler.GetLogs)
|
||||
admin.DELETE("/logs", logHandler.ClearLogs)
|
||||
admin.GET("/logs/info", logHandler.GetLogInfo)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -0,0 +1,94 @@
|
||||
package utils
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"io"
|
||||
"log"
|
||||
"os"
|
||||
"path/filepath"
|
||||
)
|
||||
|
||||
var logFilePath = "data/app.log"
|
||||
|
||||
func InitLogger() {
|
||||
logDir := filepath.Dir(logFilePath)
|
||||
if logDir != "." && logDir != "/" {
|
||||
if err := os.MkdirAll(logDir, 0755); err != nil {
|
||||
log.Fatalf("Failed to create log directory %s: %v", logDir, err)
|
||||
}
|
||||
}
|
||||
|
||||
file, err := os.OpenFile(logFilePath, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0644)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to open log file %s: %v", logFilePath, err)
|
||||
}
|
||||
|
||||
multiWriter := io.MultiWriter(os.Stdout, file)
|
||||
log.SetOutput(multiWriter)
|
||||
|
||||
log.Printf("Logger initialized, log file: %s", logFilePath)
|
||||
}
|
||||
|
||||
func GetLogs(lines int) ([]string, error) {
|
||||
file, err := os.Open(logFilePath)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer file.Close()
|
||||
|
||||
ring := make([]string, 0, lines)
|
||||
idx := 0
|
||||
|
||||
scanner := bufio.NewScanner(file)
|
||||
buf := make([]byte, 0, 64*1024)
|
||||
scanner.Buffer(buf, 1024*1024)
|
||||
|
||||
for scanner.Scan() {
|
||||
line := scanner.Text()
|
||||
if len(ring) < cap(ring) {
|
||||
ring = append(ring, line)
|
||||
} else {
|
||||
ring[idx] = line
|
||||
}
|
||||
idx = (idx + 1) % cap(ring)
|
||||
}
|
||||
|
||||
if err := scanner.Err(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if len(ring) < cap(ring) {
|
||||
return ring, nil
|
||||
}
|
||||
|
||||
result := make([]string, len(ring))
|
||||
for i := 0; i < len(ring); i++ {
|
||||
result[i] = ring[(idx+i)%len(ring)]
|
||||
}
|
||||
return result, nil
|
||||
}
|
||||
|
||||
func ClearLogs() error {
|
||||
file, err := os.OpenFile(logFilePath, os.O_WRONLY|os.O_TRUNC, 0644)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer file.Close()
|
||||
return nil
|
||||
}
|
||||
|
||||
func GetLogSize() (int64, error) {
|
||||
info, err := os.Stat(logFilePath)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
return info.Size(), nil
|
||||
}
|
||||
|
||||
func GetLogModTime() (interface{}, error) {
|
||||
info, err := os.Stat(logFilePath)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return info.ModTime(), nil
|
||||
}
|
||||
Reference in New Issue
Block a user