28 lines
438 B
Go
28 lines
438 B
Go
package logger
|
|
|
|
import (
|
|
"ai-gateway/internal/models"
|
|
"log"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
var saveRequestLog = false
|
|
|
|
// LogRequest 异步记录API请求日志
|
|
func LogRequest(db *gorm.DB, logEntry *models.RequestLog) {
|
|
if !saveRequestLog {
|
|
return
|
|
}
|
|
|
|
go func() {
|
|
if err := db.Create(logEntry).Error; err != nil {
|
|
log.Printf("Failed to save request log: %v", err)
|
|
}
|
|
}()
|
|
}
|
|
|
|
func SetSaveRequestLog(save bool) {
|
|
saveRequestLog = save
|
|
}
|