83 lines
2.7 KiB
C#
83 lines
2.7 KiB
C#
using Microsoft.AspNetCore.Mvc;
|
|
using ScrapperAPI.Dtos;
|
|
using ScrapperAPI.Interfaces;
|
|
|
|
namespace ScrapperAPI.Controllers;
|
|
|
|
[ApiController]
|
|
[Route("extraction-runs")]
|
|
public sealed class ExtractionRunsController : ControllerBase
|
|
{
|
|
private readonly IExtractionCoordinator _coord;
|
|
private readonly IExtractionRunRepository _runs;
|
|
private readonly IExtractedDataRepository _extracted;
|
|
|
|
public ExtractionRunsController(
|
|
IExtractionCoordinator coord,
|
|
IExtractionRunRepository runs,
|
|
IExtractedDataRepository extracted)
|
|
{
|
|
_coord = coord;
|
|
_runs = runs;
|
|
_extracted = extracted;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Inicia uma extração em background (cria um run).
|
|
/// </summary>
|
|
[HttpPost]
|
|
public async Task<IActionResult> Start([FromBody] StartExtractionRequest req, CancellationToken ct)
|
|
{
|
|
var runId = await _coord.StartRunAsync(req, ct);
|
|
return Accepted(new { runId });
|
|
}
|
|
|
|
/// <summary>
|
|
/// Inicia extrações em massa (1 run por session).
|
|
/// POST /extraction-runs/bulk
|
|
/// {
|
|
/// "modelId": 10,
|
|
/// "sessionIds": [1,2,3], // opcional; se omitido/vazio, roda para todas as sessions
|
|
/// "onlyDone": true
|
|
/// }
|
|
/// </summary>
|
|
[HttpPost("bulk")]
|
|
public async Task<IActionResult> StartBulk([FromBody] BulkStartExtractionRequest req, CancellationToken ct)
|
|
{
|
|
var runIds = await _coord.StartBulkRunsAsync(req, ct);
|
|
return Accepted(new { runIds, count = runIds.Count });
|
|
}
|
|
|
|
[HttpGet("{runId:long}")]
|
|
public async Task<IActionResult> GetRun(long runId, CancellationToken ct)
|
|
{
|
|
var row = await _runs.GetByIdAsync(runId, ct);
|
|
if (row is null) return NotFound();
|
|
|
|
var runtime = _coord.GetRuntimeStatus(runId);
|
|
return Ok(new { run = row, runtime });
|
|
}
|
|
|
|
/// <summary>
|
|
/// Lista os dados extraídos de uma sessão por modelo.
|
|
/// GET /extraction-runs/session/1/model/10
|
|
/// </summary>
|
|
[HttpGet("session/{sessionId:int}/model/{modelId:long}")]
|
|
public async Task<IActionResult> ListExtracted(int sessionId, long modelId, CancellationToken ct)
|
|
{
|
|
var rows = await _extracted.ListBySessionAsync(sessionId, modelId, ct);
|
|
return Ok(rows);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Pega o JSON extraído de um item específico.
|
|
/// GET /extraction-runs/queue/123/model/10
|
|
/// </summary>
|
|
[HttpGet("queue/{queueId:int}/model/{modelId:long}")]
|
|
public async Task<IActionResult> GetByQueue(int queueId, long modelId, CancellationToken ct)
|
|
{
|
|
var row = await _extracted.GetByQueueIdAsync(queueId, modelId, ct);
|
|
return row is null ? NotFound() : Ok(row);
|
|
}
|
|
}
|