Merge pull request #490 from openinfradev/policy_notification_fix

trivial. fix messageActionProposal on policy_notifications
openinfradev · May 13, 2024 · 26c8650 · 26c8650
2 parents 2f4e0e8 + 3e9bf39
commit 26c8650
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/internal/usecase/system-notification.go b/internal/usecase/system-notification.go
@@ -136,6 +136,13 @@ func (u *SystemNotificationUsecase) Create(ctx context.Context, input domain.Cre
 
 		if systemNotification.Annotations.AlertType == "POLICY_NOTIFICATION" {
 			dto.PolicyName = systemNotification.Annotations.PolicyName
+			if strings.Contains(systemNotification.Labels.AlertName, "policy-audited") {
+				dto.MessageActionProposal = "감사"
+			} else if strings.Contains(systemNotification.Labels.AlertName, "policy-blocked") {
+				dto.MessageActionProposal = "거부"
+			} else {
+				dto.MessageActionProposal = ""
+			}
 		}
 
 		_, err = u.repo.Create(ctx, dto)
@@ -346,8 +353,6 @@ func (u *SystemNotificationUsecase) makeAdditionalInfo(systemNotification *model
 			break
 		}
 	}
-
-	log.Info(context.TODO(), systemNotification.RawData)
 }
 
 func (u *SystemNotificationUsecase) makeGrafanaUrl(ctx context.Context, primaryCluster model.Cluster, systemNotification domain.SystemNotificationRequest, clusterId domain.ClusterId) (url string) {