feat: support send alert with specific providers

casbin · Sep 27, 2024 · 0170799 · 0170799
1 parent 25d2a77
commit 0170799
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 14 deletions.
diff --git a/object/site_cache.go b/object/site_cache.go
@@ -107,8 +107,8 @@ func refreshSiteMap() error {
 		}
 
 		newSiteMap[strings.ToLower(site.Domain)] = site
-		if site.EnableAlert {
-			newHealthCheckNeededDomains = append(newHealthCheckNeededDomains, site.Domain)
+		if !shouldStopHealthCheck(site) {
+			newHealthCheckNeededDomains = append(newHealthCheckNeededDomains, strings.ToLower(site.Domain))
 		}
 		for _, domain := range site.OtherDomains {
 			if domain != "" {

diff --git a/object/site_timer_health.go b/object/site_timer_health.go
@@ -16,6 +16,7 @@ package object
 
 import (
 	"fmt"
+	"strings"
 	"time"
 
 	"github.com/casdoor/casdoor-go-sdk/casdoorsdk"
@@ -58,17 +59,19 @@ func healthCheck(site *Site, domain string) error {
 	if user == nil {
 		return fmt.Errorf("user not found")
 	}
-	if user.Email != "" {
-		err = casdoorsdk.SendEmail("CasWAF HealthCheck Check Alert", pingResponse, "CasWAF", user.Email)
-	}
-	if err != nil {
-		fmt.Println(err)
-	}
-	if user.Phone != "" {
-		err = casdoorsdk.SendSms(pingResponse, user.Phone)
-	}
-	if err != nil {
-		fmt.Println(err)
+	for _, provider := range site.AlertProviders {
+		if strings.HasPrefix(provider, "Email/") {
+			err := casdoorsdk.SendEmailByProvider("CasWAF HealthCheck Check Alert", pingResponse, "CasWAF", provider[6:], user.Email)
+			if err != nil {
+				fmt.Println(err)
+			}
+		}
+		if strings.HasPrefix(provider, "SMS/") {
+			err := casdoorsdk.SendSmsByProvider(pingResponse, provider[4:], user.Phone)
+			if err != nil {
+				fmt.Println(err)
+			}
+		}
 	}
 	return nil
 }
@@ -83,7 +86,8 @@ func startHealthCheckLoop() {
 		go func() {
 			for {
 				site := GetSiteByDomain(domain)
-				if site == nil || !site.EnableAlert || site.Domain == "" || site.Status == "Inactive" {
+				if shouldStopHealthCheck(site) {
+					delete(healthCheckTryTimesMap, domain)
 					return
 				}
 
@@ -96,3 +100,7 @@ func startHealthCheckLoop() {
 		}()
 	}
 }
+
+func shouldStopHealthCheck(site *Site) bool {
+	return site == nil || !site.EnableAlert || site.Domain == "" || site.Status == "Inactive"
+}