From fd126b941496c7a0b1de4ca8cfa48dacd0e3c1bd Mon Sep 17 00:00:00 2001 From: gearnode Date: Mon, 31 Mar 2025 10:31:39 +0200 Subject: [PATCH] Support importance in mitigation import Signed-off-by: gearnode --- pkg/probo/mitigation_service.go | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/pkg/probo/mitigation_service.go b/pkg/probo/mitigation_service.go index 50922fdb6..5df221d20 100644 --- a/pkg/probo/mitigation_service.go +++ b/pkg/probo/mitigation_service.go @@ -49,9 +49,10 @@ type ( ImportMitigationRequest struct { Mitigations []struct { - Name string `json:"name"` - Description string `json:"description"` - Category string `json:"category"` + Name string `json:"name"` + Description string `json:"description"` + Category string `json:"category"` + Importance coredata.MitigationImportance `json:"importance"` } `json:"mitigations"` } ) @@ -99,7 +100,7 @@ func (s MitigationService) Import( Category: mitigation.Category, State: coredata.MitigationStateNotStarted, Standards: []string{}, - Importance: coredata.MitigationImportancePreferred, + Importance: mitigation.Importance, CreatedAt: now, UpdatedAt: now, })