Add evidence node query support

Signed-off-by: gearnode <bryan@frimin.fr>
This commit is contained in:
gearnode
2025-01-31 09:04:10 -08:00
parent cd6a388672
commit 3dd6310c72
3 changed files with 85 additions and 0 deletions

View File

@@ -145,6 +145,13 @@ func (r *queryResolver) Node(ctx context.Context, id gid.GID) (types.Node, error
}
return types.NewTask(task), nil
case coredata.EvidenceEntityType:
evidence, err := r.svc.GetEvidence(ctx, id)
if err != nil {
return nil, err
}
return types.NewEvidence(evidence), nil
default:
}

View File

@@ -59,6 +59,64 @@ func (e *Evidence) scan(r pgx.Row) error {
)
}
func (e *Evidence) LoadByID(
ctx context.Context,
conn pg.Conn,
scope *Scope,
evidenceID gid.GID,
) error {
q := `
WITH
evidence_states AS (
SELECT
evidence_id,
to_state AS state,
reason,
RANK() OVER w
FROM
evidence_state_transitions
WHERE
evidence_id = @evidence_id
WINDOW
w AS (PARTITION BY evidence_id ORDER BY created_at DESC)
)
SELECT
id,
task_id,
es.state,
object_key,
mime_type,
size,
created_at,
updated_at
FROM
evidences
INNER JOIN
evidence_states es ON es.evidence_id = evidences.id
WHERE
%s
AND id = @evidence_id
AND es.rank = 1
LIMIT 1;
`
q = fmt.Sprintf(q, scope.SQLFragment())
args := pgx.NamedArgs{"evidence_id": evidenceID}
maps.Copy(args, scope.SQLArguments())
r := conn.QueryRow(ctx, q, args)
e2 := Evidence{}
if err := e2.scan(r); err != nil {
return err
}
*e = e2
return nil
}
func (e *Evidences) LoadByTaskID(
ctx context.Context,
conn pg.Conn,

View File

@@ -169,6 +169,26 @@ func (s Service) GetTask(
return task, nil
}
func (s Service) GetEvidence(
ctx context.Context,
evidenceID gid.GID,
) (*coredata.Evidence, error) {
evidence := &coredata.Evidence{}
err := s.pg.WithConn(
ctx,
func(conn pg.Conn) error {
return evidence.LoadByID(ctx, conn, s.scope, evidenceID)
},
)
if err != nil {
return nil, err
}
return evidence, nil
}
func (s Service) ListOrganizationFrameworks(
ctx context.Context,
organizationID gid.GID,