Joshua
e8ced74429
All checks were successful
Dev Build / build-test (pull_request) Successful in 2m23s
- Create repository/ package with pgx-backed CRUD for agents, sessions, tasks, projects - Define AgentRepo/SessionRepo/TaskRepo/ProjectRepo interfaces - Update handler to use repository interfaces instead of in-memory stores - Add SSE broker with GET /api/events endpoint (text/event-stream) - Add gateway client that polls OpenClaw for agent states - Add GATEWAY_URL and GATEWAY_POLL_INTERVAL config fields - Seed 5 demo agents (Otto, Rex, Dex, Hex, Pip) on empty DB - Update router to wire SSE broker - All 21 handler tests pass with mock repos
86 lines
2.2 KiB
Go
86 lines
2.2 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"code.cubecraftcreations.com/CubeCraft-Creations/Control-Center/go-backend/internal/models"
|
|
"github.com/jackc/pgx/v5"
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
|
)
|
|
|
|
// TaskRepository provides PostgreSQL-backed CRUD for task_logs.
|
|
type TaskRepository struct {
|
|
pool *pgxpool.Pool
|
|
}
|
|
|
|
// NewTaskRepository returns a repository wired to the given connection pool.
|
|
func NewTaskRepository(pool *pgxpool.Pool) *TaskRepository {
|
|
return &TaskRepository{pool: pool}
|
|
}
|
|
|
|
// Create inserts a new task into the task_logs table.
|
|
func (r *TaskRepository) Create(ctx context.Context, t models.Task) (models.Task, error) {
|
|
now := time.Now().UTC()
|
|
if t.CreatedAt.IsZero() {
|
|
t.CreatedAt = now
|
|
}
|
|
if t.UpdatedAt.IsZero() {
|
|
t.UpdatedAt = now
|
|
}
|
|
|
|
err := r.pool.QueryRow(ctx, `
|
|
INSERT INTO task_logs (agent_id, task, status, started_at)
|
|
VALUES ($1, $2, $3, $4)
|
|
RETURNING id, agent_id, task, status, started_at, completed_at, error_message
|
|
`, t.AgentID, t.Title, string(t.Status), t.CreatedAt).Scan(
|
|
&t.ID, &t.AgentID, &t.Title, &t.Status, &t.CreatedAt,
|
|
nil, nil,
|
|
)
|
|
if err != nil {
|
|
return t, err
|
|
}
|
|
|
|
// Rebuild the Description since task_logs only stores the title as "task".
|
|
t.Description = t.Title
|
|
return t, nil
|
|
}
|
|
|
|
// ListRecent returns the most recent tasks, newest first.
|
|
func (r *TaskRepository) ListRecent(ctx context.Context) ([]models.Task, error) {
|
|
rows, err := r.pool.Query(ctx, `
|
|
SELECT id, agent_id, task, status, started_at, completed_at, error_message
|
|
FROM task_logs
|
|
ORDER BY started_at DESC
|
|
`)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
return pgx.CollectRows(rows, func(row pgx.CollectableRow) (models.Task, error) {
|
|
var t models.Task
|
|
var completedAt *time.Time
|
|
var errMsg *string
|
|
|
|
if err := row.Scan(&t.ID, &t.AgentID, &t.Title, &t.Status,
|
|
&t.CreatedAt, &completedAt, &errMsg); err != nil {
|
|
return t, err
|
|
}
|
|
|
|
t.Description = t.Title
|
|
t.UpdatedAt = t.CreatedAt
|
|
if completedAt != nil {
|
|
t.UpdatedAt = *completedAt
|
|
}
|
|
return t, nil
|
|
})
|
|
}
|
|
|
|
// Count returns the total number of tasks.
|
|
func (r *TaskRepository) Count(ctx context.Context) (int, error) {
|
|
var n int
|
|
err := r.pool.QueryRow(ctx, `SELECT COUNT(*) FROM task_logs`).Scan(&n)
|
|
return n, err
|
|
}
|