5b7f173102
* feat(canvas): 参考素材 - 增加移除按钮并删除对应连线 * feat(canvas): 完善节点面板交互与拖拽反馈 * fix(canvas): 画布交互与小地图 - 统一浮层层级并完善节点识别 * feat(prompt-optimizer): 增加 AI 提示词优化器及宿主接入 * feat(canvas): 画布 - 增加自动整理节点按钮并补充文档 * chore(web): 前端 - 修正提示词优化器格式 * chore(backend): 后端 - 修正 Go 格式检查 * feat(workspace): 完善诊断导出与本地排障支持 * chore(web): 格式化诊断导出文件 * fix(web-test): 兼容 Bun 测试运行时 * fix(web-test): 对齐模型目录协议断言
64 lines
2.5 KiB
Go
64 lines
2.5 KiB
Go
package repository
|
|
|
|
import (
|
|
"strings"
|
|
"time"
|
|
|
|
"infinite-canvas/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
|
|
}
|