xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-09-19 10:06:37 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: streamline Notify function with switch-case for notification methods

Browse Source
This commit is contained in:
Laisky.Cai 2025-03-09 12:52:08 +00:00
parent 388f0ef6aa
commit 3e432c77bd
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
common/message/main.go
View File

@ -1,6 +1,8 @@
package message package message
import ( import (
"fmt"
"github.com/pkg/errors" "github.com/pkg/errors"
"github.com/songquanpeng/one-api/common/config" "github.com/songquanpeng/one-api/common/config"
) )
@ -12,8 +14,8 @@ const (
) )
func Notify(by string, title string, description string, content string) error { func Notify(by string, title string, description string, content string) error {
switch by {
if by == ByAll { case ByAll:
var errMsgs []string var errMsgs []string
if err := SendEmail(title, config.RootUserEmail, content); err != nil { if err := SendEmail(title, config.RootUserEmail, content); err != nil {
errMsgs = append(errMsgs, fmt.Sprintf("failed to send email: %v", err)) errMsgs = append(errMsgs, fmt.Sprintf("failed to send email: %v", err))
@ -26,13 +28,11 @@ func Notify(by string, title string, description string, content string) error {
return fmt.Errorf("multiple errors occurred: %v", errMsgs) return fmt.Errorf("multiple errors occurred: %v", errMsgs)
} }
return nil return nil
} case ByEmail:
if by == ByEmail {
return SendEmail(title, config.RootUserEmail, content) return SendEmail(title, config.RootUserEmail, content)
} case ByMessagePusher:
if by == ByMessagePusher {
return SendMessage(title, description, content) return SendMessage(title, description, content)
} default:
return errors.Errorf("unknown notify method: %s", by) return errors.Errorf("unknown notify method: %s", by)
} }
}