rohit/probo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add risk sort keys

Browse Source 
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
Download Patch File Download Diff File Expand all files Collapse all files

10
pkg/coredata/risk.go
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))

6
pkg/coredata/risk_order_field.go
View File

@@ -20,6 +20,12 @@ type (
const (
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 {

22
pkg/server/api/console/v1/schema.graphql
View File

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

34
pkg/server/api/console/v1/schema/schema.go
View File

@@ -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
@@ -42807,9 +42829,21 @@ func (ec *executionContext) marshalNRiskOrderField2githubᚗcomᚋgetproboᚋpro
var (
unmarshalNRiskOrderField2githubᚗcomᚋgetproboᚋproboᚋpkgᚋcoredataᚐRiskOrderField = map[string]coredata.RiskOrderField{
"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.RiskOrderFieldUpdatedAt: "UPDATED_AT",
coredata.RiskOrderFieldName: "NAME",
coredata.RiskOrderFieldCategory: "CATEGORY",
coredata.RiskOrderFieldTreatment: "TREATMENT",
coredata.RiskOrderFieldInitialRiskScore: "INITIAL_RISK_SCORE",
coredata.RiskOrderFieldResidualRiskScore: "RESIDUAL_RISK_SCORE",
}
)