add notify info for pagerduty

Author: ThreeAndTwo

notify.go

@@ -23,6 +23,8 @@ type Config struct {
 	Platform Platform
 	Token    string
 	Channel  string
+	Source   string
+	Severity string
 }
 
 func NewNotify(config *Config) *Notify {
@@ -66,6 +68,8 @@ func (n *Notify) sendSlackNotify(msg string) error {
 func (n *Notify) sendPagerdutyNotify(msg string) error {
 	app := pagerduty.New(pagerduty.Options{
 		Token: n.config.Token,
+		Source: n.config.Source,
+		Severity: n.config.Severity,
 	})
 	err := app.Send(msg)
 	return err

notify_test.go

@@ -36,13 +36,25 @@ func TestNotify_Send(t *testing.T) {
 			args{msg: "test case"},
 		},
 		{
-			"test pagerduty notify",
+			"test pagerduty severity is null",
 			fields{config: &Config{
 				Platform: Platform("pagerduty"),
 				Token: os.Getenv("PAGERDUTY_TOKEN"),
+				Source: "api-test",
+				Severity: "",
 			}},
 			args{msg: "test pagerduty"},
 		},
+		{
+			"test pagerduty severity is error",
+			fields{config: &Config{
+				Platform: Platform("pagerduty"),
+				Token: os.Getenv("PAGERDUTY_TOKEN"),
+				Source: "api-test",
+				Severity: "error",
+			}},
+			args{msg: "test pagerduty is error"},
+		},
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {

pagerduty/notify.go

@@ -15,8 +15,10 @@ const (
 )
 
 type Options struct {
-	Token string `json:"token"`
-	Text  string `json:"text"`
+	Token    string `json:"token"`
+	Source   string `json:"source"`
+	Severity string `json:"severity"`
+	Text     string `json:"text"`
 }
 
 type pagerduty struct {
@@ -59,8 +61,8 @@ func (c *client) Send(message string) error {
 	pdOpt := &pagerduty{
 		Payload: payload{
 			Summary:  message,
-			Source:   "monitoringtool:cloudvendor:central-region-dc-01:852559987:cluster/api-stats-prod-003",
-			Severity: "info",
+			Source:   c.opt.Source,
+			Severity: c.opt.Severity,
 		},
 		RoutingKey:  c.opt.Token,
 		EventAction: "trigger",
@@ -93,8 +95,12 @@ func (c *client) Send(message string) error {
 }
 
 func (c *client) check(msg string) error {
-	if c.opt.Token == "" {
-		return errors.New("missing auth token")
+	if c.opt.Token == "" || c.opt.Source == "" {
+		return errors.New("missing config")
+	}
+
+	if  c.opt.Severity == "" {
+		c.opt.Severity = "critical"
 	}
 
 	if msg == "" {