Add risk sort keys

Signed-off-by: Bryan Frimin <bryan@getprobo.com>

pkg/coredata/risk.go

@@ -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))

pkg/coredata/risk_order_field.go

@@ -19,7 +19,13 @@ type (
 )
 
 const (
-	RiskOrderFieldCreatedAt RiskOrderField = "CREATED_AT"
+	RiskOrderFieldCreatedAt         RiskOrderField = "CREATED_AT"
+	RiskOrderFieldUpdatedAt         RiskOrderField = "UPDATED_AT"
+	RiskOrderFieldName              RiskOrderField = "NAME"
+	RiskOrderFieldCategory          RiskOrderField = "CATEGORY"
+	RiskOrderFieldTreatment         RiskOrderField = "TREATMENT"
+	RiskOrderFieldInitialRiskScore  RiskOrderField = "INITIAL_RISK_SCORE"
+	RiskOrderFieldResidualRiskScore RiskOrderField = "RESIDUAL_RISK_SCORE"
 )
 
 func (p RiskOrderField) Column() string {