feat: 日志持久化存储到文件+管理员后台查看/清空日志页面

This commit is contained in:
2026-05-28 00:56:19 +08:00
parent 8319b4b7ed
commit 65965c409f
12 changed files with 473 additions and 224 deletions
+1
View File
@@ -14,6 +14,7 @@ func main() {
config.Load()
utils.InitDB()
utils.InitLogger()
utils.AutoMigrate()
if config.AppConfig.Server.Mode == "release" {
+74
View File
@@ -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,
},
})
}
+5
View File
@@ -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)
}
}
}
+94
View File
@@ -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
}