79 lines
2.0 KiB
Go
79 lines
2.0 KiB
Go
package handler
|
|
|
|
import (
|
|
"net/http"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"gpt-load/internal/db"
|
|
"gpt-load/internal/models"
|
|
"gpt-load/internal/response"
|
|
)
|
|
|
|
// GetLogs godoc
|
|
// @Summary Get request logs
|
|
// @Description Get request logs with pagination and filtering
|
|
// @Tags Logs
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param page query int false "Page number"
|
|
// @Param size query int false "Page size"
|
|
// @Param group_id query int false "Group ID"
|
|
// @Param start_time query string false "Start time (RFC3339)"
|
|
// @Param end_time query string false "End time (RFC3339)"
|
|
// @Param status_code query int false "Status code"
|
|
// @Success 200 {array} models.RequestLog
|
|
// @Router /api/logs [get]
|
|
func GetLogs(c *gin.Context) {
|
|
page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
|
|
size, _ := strconv.Atoi(c.DefaultQuery("size", "10"))
|
|
offset := (page - 1) * size
|
|
|
|
query := db.DB.Model(&models.RequestLog{})
|
|
|
|
if groupIDStr := c.Query("group_id"); groupIDStr != "" {
|
|
groupID, err := strconv.Atoi(groupIDStr)
|
|
if err == nil {
|
|
query = query.Where("group_id = ?", groupID)
|
|
}
|
|
}
|
|
|
|
if startTimeStr := c.Query("start_time"); startTimeStr != "" {
|
|
startTime, err := time.Parse(time.RFC3339, startTimeStr)
|
|
if err == nil {
|
|
query = query.Where("timestamp >= ?", startTime)
|
|
}
|
|
}
|
|
|
|
if endTimeStr := c.Query("end_time"); endTimeStr != "" {
|
|
endTime, err := time.Parse(time.RFC3339, endTimeStr)
|
|
if err == nil {
|
|
query = query.Where("timestamp <= ?", endTime)
|
|
}
|
|
}
|
|
|
|
if statusCodeStr := c.Query("status_code"); statusCodeStr != "" {
|
|
statusCode, err := strconv.Atoi(statusCodeStr)
|
|
if err == nil {
|
|
query = query.Where("status_code = ?", statusCode)
|
|
}
|
|
}
|
|
|
|
var logs []models.RequestLog
|
|
var total int64
|
|
|
|
query.Count(&total)
|
|
err := query.Order("timestamp desc").Offset(offset).Limit(size).Find(&logs).Error
|
|
if err != nil {
|
|
response.Error(c, http.StatusInternalServerError, "Failed to get logs")
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"total": total,
|
|
"page": page,
|
|
"size": size,
|
|
"data": logs,
|
|
})
|
|
} |