39 lines
1.0 KiB
Go
39 lines
1.0 KiB
Go
package dblogger
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"time"
|
|
|
|
"mini-chat/internal/proposal"
|
|
"mini-chat/internal/repository/mysql"
|
|
"mini-chat/internal/repository/mysql/dao"
|
|
"mini-chat/internal/repository/mysql/model"
|
|
|
|
"github.com/spf13/cast"
|
|
)
|
|
|
|
func LoggerHandler(db mysql.Repo) func(msg *proposal.RequestLoggerMessage) {
|
|
return func(msg *proposal.RequestLoggerMessage) {
|
|
logData := new(model.LogRequest)
|
|
logData.Tid = msg.Tid
|
|
logData.Username = msg.Username
|
|
logData.Host = msg.HOST
|
|
logData.Path = msg.Path
|
|
logData.Method = msg.Method
|
|
logData.HTTPCode = int32(msg.HTTPCode)
|
|
logData.BusinessCode = int32(msg.BusinessCode)
|
|
logData.CostMilliseconds = float64(cast.ToFloat32(fmt.Sprintf("%.2f", msg.CostSeconds*1000)))
|
|
logData.IsSuccess = -1
|
|
if msg.IsSuccess == true {
|
|
logData.IsSuccess = 1
|
|
}
|
|
logData.Content = msg.Content
|
|
logData.CreatedAt = time.Now()
|
|
|
|
if err := dao.Use(db.GetDbW()).LogRequest.Create(logData); err != nil {
|
|
log.Println(fmt.Sprintf("日志写入失败: %s", err.Error()))
|
|
}
|
|
}
|
|
}
|