From 1c4a42009dab0d516dbee7c96a030a02cb3485ce Mon Sep 17 00:00:00 2001 From: Wenkai Yin Date: Thu, 20 Jul 2017 18:10:52 +0800 Subject: [PATCH] remove duplicate codes in notification handler (#2828) --- src/ui/service/notifications/registry/handler.go | 14 +++----------- 1 file changed, 3 insertions(+), 11 deletions(-) diff --git a/src/ui/service/notifications/registry/handler.go b/src/ui/service/notifications/registry/handler.go index 54ed82b63..12a19d365 100644 --- a/src/ui/service/notifications/registry/handler.go +++ b/src/ui/service/notifications/registry/handler.go @@ -107,7 +107,7 @@ func (n *NotificationHandler) Post() { go api.TriggerReplicationByRepository(pro.ProjectID, repository, []string{tag}, models.RepOpTransfer) - if autoScanEnabled(project) { + if autoScanEnabled(pro) { last, err := clairdao.GetLastUpdate() if err != nil { log.Errorf("Failed to get last update from Clair DB, error: %v, the auto scan will be skipped.", err) @@ -165,21 +165,13 @@ func filterEvents(notification *models.Notification) ([]*models.Event, error) { return events, nil } -func autoScanEnabled(projectName string) bool { +func autoScanEnabled(project *models.Project) bool { if !config.WithClair() { log.Debugf("Auto Scan disabled because Harbor is not deployed with Clair") return false } if config.WithAdmiral() { - p, err := config.GlobalProjectMgr.Get(projectName) - if err != nil { - log.Warningf("failed to get project, error: %v", err) - return false - } else if p == nil { - log.Warningf("project with name: %s not found.", projectName) - return false - } - return p.AutomaticallyScanImagesOnPush + return project.AutomaticallyScanImagesOnPush } return os.Getenv("ENABLE_HARBOR_SCAN_ON_PUSH") == "1" }