redo

internal/database/schema.go

@@ -119,14 +119,43 @@ func (db *DB) migrate() error {
 		FOREIGN KEY (created_by) REFERENCES users(id) ON DELETE SET NULL
 	);
 
+	CREATE TABLE IF NOT EXISTS task_logs (
+		id INTEGER PRIMARY KEY AUTOINCREMENT,
+		task_id INTEGER NOT NULL,
+		runner_id INTEGER,
+		log_level TEXT NOT NULL,
+		message TEXT NOT NULL,
+		step_name TEXT,
+		created_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
+		FOREIGN KEY (task_id) REFERENCES tasks(id) ON DELETE CASCADE,
+		FOREIGN KEY (runner_id) REFERENCES runners(id) ON DELETE SET NULL
+	);
+
+	CREATE TABLE IF NOT EXISTS task_steps (
+		id INTEGER PRIMARY KEY AUTOINCREMENT,
+		task_id INTEGER NOT NULL,
+		step_name TEXT NOT NULL,
+		status TEXT NOT NULL DEFAULT 'pending',
+		started_at DATETIME,
+		completed_at DATETIME,
+		duration_ms INTEGER,
+		error_message TEXT,
+		FOREIGN KEY (task_id) REFERENCES tasks(id) ON DELETE CASCADE
+	);
+
 	CREATE INDEX IF NOT EXISTS idx_jobs_user_id ON jobs(user_id);
 	CREATE INDEX IF NOT EXISTS idx_jobs_status ON jobs(status);
 	CREATE INDEX IF NOT EXISTS idx_tasks_job_id ON tasks(job_id);
 	CREATE INDEX IF NOT EXISTS idx_tasks_runner_id ON tasks(runner_id);
 	CREATE INDEX IF NOT EXISTS idx_tasks_status ON tasks(status);
+	CREATE INDEX IF NOT EXISTS idx_tasks_started_at ON tasks(started_at);
 	CREATE INDEX IF NOT EXISTS idx_job_files_job_id ON job_files(job_id);
 	CREATE INDEX IF NOT EXISTS idx_registration_tokens_token ON registration_tokens(token);
 	CREATE INDEX IF NOT EXISTS idx_registration_tokens_expires_at ON registration_tokens(expires_at);
+	CREATE INDEX IF NOT EXISTS idx_task_logs_task_id_created_at ON task_logs(task_id, created_at);
+	CREATE INDEX IF NOT EXISTS idx_task_logs_runner_id ON task_logs(runner_id);
+	CREATE INDEX IF NOT EXISTS idx_task_steps_task_id ON task_steps(task_id);
+	CREATE INDEX IF NOT EXISTS idx_runners_last_heartbeat ON runners(last_heartbeat);
 	`
 
 	if _, err := db.Exec(schema); err != nil {
@@ -142,6 +171,15 @@ func (db *DB) migrate() error {
 		`ALTER TABLE runners ADD COLUMN runner_secret TEXT`,
 		`ALTER TABLE runners ADD COLUMN manager_secret TEXT`,
 		`ALTER TABLE runners ADD COLUMN verified BOOLEAN NOT NULL DEFAULT 0`,
+		// Add allow_parallel_runners to jobs if it doesn't exist
+		`ALTER TABLE jobs ADD COLUMN allow_parallel_runners BOOLEAN NOT NULL DEFAULT 1`,
+		// Add timeout_seconds to jobs if it doesn't exist
+		`ALTER TABLE jobs ADD COLUMN timeout_seconds INTEGER DEFAULT 86400`,
+		// Add new columns to tasks if they don't exist
+		`ALTER TABLE tasks ADD COLUMN current_step TEXT`,
+		`ALTER TABLE tasks ADD COLUMN retry_count INTEGER DEFAULT 0`,
+		`ALTER TABLE tasks ADD COLUMN max_retries INTEGER DEFAULT 3`,
+		`ALTER TABLE tasks ADD COLUMN timeout_seconds INTEGER`,
 	}
 
 	for _, migration := range migrations {