64 lines
2.5 KiB
Go
64 lines
2.5 KiB
Go
package repository
|
|
|
|
import (
|
|
"strings"
|
|
"time"
|
|
|
|
"yingce/backend/internal/model"
|
|
)
|
|
|
|
const (
|
|
diagnosticTaskLimit = 100
|
|
diagnosticTaskLogLimit = 2000
|
|
diagnosticAPICallLimit = 1000
|
|
)
|
|
|
|
func (r *Repository) DiagnosticTasks(userID string, from time.Time, to time.Time, taskID string, projectID string) ([]model.Task, error) {
|
|
var tasks []model.Task
|
|
query := r.db.Model(&model.Task{}).Select(
|
|
"id", "user_id", "trace_id", "request_id", "project_id", "type", "status", "stage", "progress",
|
|
"operation", "provider", "model", "logical_model_id", "provider_request_id", "error", "attempts",
|
|
"started_at", "completed_at", "created_at", "updated_at",
|
|
)
|
|
if taskID = strings.TrimSpace(taskID); taskID != "" {
|
|
query = query.Where("user_id = ? AND id = ?", userID, taskID)
|
|
} else {
|
|
query = query.Where("user_id = ? AND created_at >= ? AND created_at <= ?", userID, from, to)
|
|
if projectID = strings.TrimSpace(projectID); projectID != "" {
|
|
query = query.Where("project_id = ?", projectID)
|
|
}
|
|
}
|
|
err := query.Order("created_at asc").Limit(diagnosticTaskLimit).Find(&tasks).Error
|
|
return tasks, err
|
|
}
|
|
|
|
func (r *Repository) DiagnosticTaskLogs(userID string, from time.Time, to time.Time, taskID string, projectID string) ([]model.TaskLog, error) {
|
|
var logs []model.TaskLog
|
|
query := r.db.Where("user_id = ?", userID)
|
|
if taskID = strings.TrimSpace(taskID); taskID != "" {
|
|
query = query.Where("task_id = ?", taskID)
|
|
} else {
|
|
query = query.Where("created_at >= ? AND created_at <= ?", from, to)
|
|
if projectID = strings.TrimSpace(projectID); projectID != "" {
|
|
query = query.Where("task_id IN (SELECT id FROM tasks WHERE user_id = ? AND project_id = ?)", userID, projectID)
|
|
}
|
|
}
|
|
err := query.Order("created_at asc").Limit(diagnosticTaskLogLimit).Find(&logs).Error
|
|
return logs, err
|
|
}
|
|
|
|
func (r *Repository) DiagnosticAPICallLogs(userID string, from time.Time, to time.Time, taskID string, projectID string) ([]model.ApiCallLog, error) {
|
|
var logs []model.ApiCallLog
|
|
query := r.db.Where("user_id = ?", userID)
|
|
if taskID = strings.TrimSpace(taskID); taskID != "" {
|
|
query = query.Where("task_id = ?", taskID)
|
|
} else {
|
|
query = query.Where("created_at >= ? AND created_at <= ?", from, to)
|
|
if projectID = strings.TrimSpace(projectID); projectID != "" {
|
|
query = query.Where("task_id IN (SELECT id FROM tasks WHERE user_id = ? AND project_id = ?)", userID, projectID)
|
|
}
|
|
}
|
|
err := query.Omit("RequestBody", "ResponseBody").Order("created_at asc").Limit(diagnosticAPICallLimit).Find(&logs).Error
|
|
return logs, err
|
|
}
|