Add risk sort keys

Signed-off-by: Bryan Frimin <bryan@getprobo.com>
This commit is contained in:
Bryan Frimin
2025-05-22 16:16:06 -07:00
parent 62a0fb96b7
commit c646808828
4 changed files with 75 additions and 3 deletions

View File

@@ -53,6 +53,16 @@ func (r *Risk) CursorKey(orderBy RiskOrderField) page.CursorKey {
switch orderBy {
case RiskOrderFieldCreatedAt:
return page.CursorKey{ID: r.ID, Value: r.CreatedAt}
case RiskOrderFieldName:
return page.CursorKey{ID: r.ID, Value: r.Name}
case RiskOrderFieldCategory:
return page.CursorKey{ID: r.ID, Value: r.Category}
case RiskOrderFieldTreatment:
return page.CursorKey{ID: r.ID, Value: r.Treatment}
case RiskOrderFieldInitialRiskScore:
return page.CursorKey{ID: r.ID, Value: r.InherentRiskScore}
case RiskOrderFieldResidualRiskScore:
return page.CursorKey{ID: r.ID, Value: r.ResidualRiskScore}
}
panic(fmt.Sprintf("unsupported order by: %s", orderBy))