mirror of
https://github.com/hatchet-dev/hatchet.git
synced 2026-04-30 06:30:12 -05:00
abelanger5
81 lines
1.8 KiB
Go
81 lines
1.8 KiB
Go
package workflows
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/labstack/echo/v4"
|
|
|
|
"github.com/hatchet-dev/hatchet/api/v1/server/oas/gen"
|
|
"github.com/hatchet-dev/hatchet/api/v1/server/oas/transformers"
|
|
"github.com/hatchet-dev/hatchet/pkg/repository/postgres/dbsqlc"
|
|
"github.com/hatchet-dev/hatchet/pkg/repository/postgres/sqlchelpers"
|
|
)
|
|
|
|
func (t *WorkflowService) WorkflowRunGetShape(ctx echo.Context, request gen.WorkflowRunGetShapeRequestObject) (gen.WorkflowRunGetShapeResponseObject, error) {
|
|
run := ctx.Get("workflow-run").(*dbsqlc.GetWorkflowRunByIdRow)
|
|
|
|
reqCtx, cancel := context.WithTimeout(ctx.Request().Context(), 5*time.Second)
|
|
defer cancel()
|
|
|
|
jobRuns, err := t.config.APIRepository.JobRun().ListJobRunByWorkflowRunId(
|
|
reqCtx,
|
|
sqlchelpers.UUIDToStr(run.TenantId),
|
|
sqlchelpers.UUIDToStr(run.ID),
|
|
)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
workflowVersion, _, _, _, err := t.config.APIRepository.Workflow().GetWorkflowVersionById(
|
|
sqlchelpers.UUIDToStr(run.TenantId),
|
|
sqlchelpers.UUIDToStr(run.WorkflowVersionId),
|
|
)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
jobIds := make([]string, len(jobRuns))
|
|
jobRunIds := make([]string, len(jobRuns))
|
|
|
|
for i := range jobRuns {
|
|
jobIds[i] = sqlchelpers.UUIDToStr(jobRuns[i].JobId)
|
|
jobRunIds[i] = sqlchelpers.UUIDToStr(jobRuns[i].ID)
|
|
}
|
|
|
|
// Shape of DAG
|
|
steps, err := t.config.APIRepository.WorkflowRun().GetStepsForJobs(
|
|
reqCtx,
|
|
sqlchelpers.UUIDToStr(run.TenantId),
|
|
jobIds,
|
|
)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// step runs
|
|
|
|
stepRuns, err := t.config.APIRepository.WorkflowRun().GetStepRunsForJobRuns(
|
|
reqCtx,
|
|
sqlchelpers.UUIDToStr(run.TenantId),
|
|
jobRunIds,
|
|
)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return gen.WorkflowRunGetShape200JSONResponse(
|
|
*transformers.ToWorkflowRunShape(
|
|
run,
|
|
workflowVersion,
|
|
jobRuns,
|
|
steps,
|
|
stepRuns,
|
|
),
|
|
), nil
|
|
}
|