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 {

pkg/server/api/console/v1/schema.graphql

@@ -202,6 +202,28 @@ enum RiskOrderField
     @goEnum(
       value: "github.com/getprobo/probo/pkg/coredata.RiskOrderFieldCreatedAt"
     )
+  UPDATED_AT
+    @goEnum(
+      value: "github.com/getprobo/probo/pkg/coredata.RiskOrderFieldUpdatedAt"
+    )
+  NAME
+    @goEnum(value: "github.com/getprobo/probo/pkg/coredata.RiskOrderFieldName")
+  CATEGORY
+    @goEnum(
+      value: "github.com/getprobo/probo/pkg/coredata.RiskOrderFieldCategory"
+    )
+  TREATMENT
+    @goEnum(
+      value: "github.com/getprobo/probo/pkg/coredata.RiskOrderFieldTreatment"
+    )
+  INITIAL_RISK_SCORE
+    @goEnum(
+      value: "github.com/getprobo/probo/pkg/coredata.RiskOrderFieldInitialRiskScore"
+    )
+  RESIDUAL_RISK_SCORE
+    @goEnum(
+      value: "github.com/getprobo/probo/pkg/coredata.RiskOrderFieldResidualRiskScore"
+    )
 }
 
 enum EvidenceOrderField

pkg/server/api/console/v1/schema/schema.go

@@ -4173,6 +4173,28 @@ enum RiskOrderField
     @goEnum(
       value: "github.com/getprobo/probo/pkg/coredata.RiskOrderFieldCreatedAt"
     )
+  UPDATED_AT
+    @goEnum(
+      value: "github.com/getprobo/probo/pkg/coredata.RiskOrderFieldUpdatedAt"
+    )
+  NAME
+    @goEnum(value: "github.com/getprobo/probo/pkg/coredata.RiskOrderFieldName")
+  CATEGORY
+    @goEnum(
+      value: "github.com/getprobo/probo/pkg/coredata.RiskOrderFieldCategory"
+    )
+  TREATMENT
+    @goEnum(
+      value: "github.com/getprobo/probo/pkg/coredata.RiskOrderFieldTreatment"
+    )
+  INITIAL_RISK_SCORE
+    @goEnum(
+      value: "github.com/getprobo/probo/pkg/coredata.RiskOrderFieldInitialRiskScore"
+    )
+  RESIDUAL_RISK_SCORE
+    @goEnum(
+      value: "github.com/getprobo/probo/pkg/coredata.RiskOrderFieldResidualRiskScore"
+    )
 }
 
 enum EvidenceOrderField
@@ -42806,10 +42828,22 @@ func (ec *executionContext) marshalNRiskOrderField2githubᚗcomᚋgetproboᚋpro
 
 var (
 	unmarshalNRiskOrderField2githubᚗcomᚋgetproboᚋproboᚋpkgᚋcoredataᚐRiskOrderField = map[string]coredata.RiskOrderField{
-		"CREATED_AT": coredata.RiskOrderFieldCreatedAt,
+		"CREATED_AT":          coredata.RiskOrderFieldCreatedAt,
+		"UPDATED_AT":          coredata.RiskOrderFieldUpdatedAt,
+		"NAME":                coredata.RiskOrderFieldName,
+		"CATEGORY":            coredata.RiskOrderFieldCategory,
+		"TREATMENT":           coredata.RiskOrderFieldTreatment,
+		"INITIAL_RISK_SCORE":  coredata.RiskOrderFieldInitialRiskScore,
+		"RESIDUAL_RISK_SCORE": coredata.RiskOrderFieldResidualRiskScore,
 	}
 	marshalNRiskOrderField2githubᚗcomᚋgetproboᚋproboᚋpkgᚋcoredataᚐRiskOrderField = map[coredata.RiskOrderField]string{
-		coredata.RiskOrderFieldCreatedAt: "CREATED_AT",
+		coredata.RiskOrderFieldCreatedAt:         "CREATED_AT",
+		coredata.RiskOrderFieldUpdatedAt:         "UPDATED_AT",
+		coredata.RiskOrderFieldName:              "NAME",
+		coredata.RiskOrderFieldCategory:          "CATEGORY",
+		coredata.RiskOrderFieldTreatment:         "TREATMENT",
+		coredata.RiskOrderFieldInitialRiskScore:  "INITIAL_RISK_SCORE",
+		coredata.RiskOrderFieldResidualRiskScore: "RESIDUAL_RISK_SCORE",
 	}
 )