Merge pull request #14 from heejin-github/patch

Apply SDK changes for AlertTypes in sdk.CallResponse
dsrvlabs · Oct 11, 2023 · 25954ef · 25954ef
2 parents 6c8bf78 + 0ccd6a5
commit 25954ef
Show file tree

Hide file tree

Showing 4 changed files with 28 additions and 35 deletions.
diff --git a/plugins/cpu_monitor/main.go b/plugins/cpu_monitor/main.go
@@ -58,11 +58,10 @@ func pluginFeature(info, option map[string]*structpb.Value) (sdk.CallResponse, e
 	percent, err := cpu.Percent(defaultDuration*time.Second, false)
 	if err != nil {
 		ret := sdk.CallResponse{
-			FuncName:   info["execute_method"].GetStringValue(),
-			Message:    "failed to get cpu usage",
-			Severity:   pluginpb.SEVERITY_CRITICAL,
-			State:      pluginpb.STATE_FAILURE,
-			AlertTypes: []pluginpb.ALERT_TYPE{pluginpb.ALERT_TYPE_DISCORD},
+			FuncName: info["execute_method"].GetStringValue(),
+			Message:  "failed to get cpu usage",
+			Severity: pluginpb.SEVERITY_CRITICAL,
+			State:    pluginpb.STATE_FAILURE,
 		}
 
 		return ret, err
@@ -88,11 +87,10 @@ func pluginFeature(info, option map[string]*structpb.Value) (sdk.CallResponse, e
 	}
 
 	ret := sdk.CallResponse{
-		FuncName:   info["execute_method"].GetStringValue(),
-		Message:    message,
-		Severity:   severity,
-		State:      state,
-		AlertTypes: []pluginpb.ALERT_TYPE{pluginpb.ALERT_TYPE_DISCORD},
+		FuncName: info["execute_method"].GetStringValue(),
+		Message:  message,
+		Severity: severity,
+		State:    state,
 	}
 
 	return ret, nil

diff --git a/plugins/disk_monitor/main.go b/plugins/disk_monitor/main.go
@@ -87,11 +87,10 @@ func pluginFeature(info, option map[string]*structpb.Value) (sdk.CallResponse, e
 		if err != nil {
 			message = fmt.Sprint("failed to get disk usage on ", path)
 			ret := sdk.CallResponse{
-				FuncName:   info["execute_method"].GetStringValue(),
-				Message:    message,
-				Severity:   pluginpb.SEVERITY_CRITICAL,
-				State:      pluginpb.STATE_FAILURE,
-				AlertTypes: []pluginpb.ALERT_TYPE{pluginpb.ALERT_TYPE_DISCORD},
+				FuncName: info["execute_method"].GetStringValue(),
+				Message:  message,
+				Severity: pluginpb.SEVERITY_CRITICAL,
+				State:    pluginpb.STATE_FAILURE,
 			}
 
 			return ret, err
@@ -133,11 +132,10 @@ func pluginFeature(info, option map[string]*structpb.Value) (sdk.CallResponse, e
 	log.Debug().Str("module", "plugin").Msgf("severity : %s, state : %s, message : %s", severity.String(), state.String(), message)
 
 	ret := sdk.CallResponse{
-		FuncName:   info["execute_method"].GetStringValue(),
-		Message:    message,
-		Severity:   severity,
-		State:      state,
-		AlertTypes: []pluginpb.ALERT_TYPE{pluginpb.ALERT_TYPE_DISCORD},
+		FuncName: info["execute_method"].GetStringValue(),
+		Message:  message,
+		Severity: severity,
+		State:    state,
 	}
 
 	return ret, nil

diff --git a/plugins/mem_monitor/main.go b/plugins/mem_monitor/main.go
@@ -56,11 +56,10 @@ func pluginFeature(info, option map[string]*structpb.Value) (sdk.CallResponse, e
 	v, err := mem.VirtualMemory()
 	if err != nil {
 		ret := sdk.CallResponse{
-			FuncName:   info["execute_method"].GetStringValue(),
-			Message:    "failed to get memory usage",
-			Severity:   pluginpb.SEVERITY_CRITICAL,
-			State:      pluginpb.STATE_FAILURE,
-			AlertTypes: []pluginpb.ALERT_TYPE{pluginpb.ALERT_TYPE_DISCORD},
+			FuncName: info["execute_method"].GetStringValue(),
+			Message:  "failed to get memory usage",
+			Severity: pluginpb.SEVERITY_CRITICAL,
+			State:    pluginpb.STATE_FAILURE,
 		}
 
 		return ret, err
@@ -83,11 +82,10 @@ func pluginFeature(info, option map[string]*structpb.Value) (sdk.CallResponse, e
 	}
 
 	ret := sdk.CallResponse{
-		FuncName:   info["execute_method"].GetStringValue(),
-		Message:    message,
-		Severity:   severity,
-		State:      state,
-		AlertTypes: []pluginpb.ALERT_TYPE{pluginpb.ALERT_TYPE_DISCORD},
+		FuncName: info["execute_method"].GetStringValue(),
+		Message:  message,
+		Severity: severity,
+		State:    state,
 	}
 
 	return ret, nil

diff --git a/plugins/net_monitor/main.go b/plugins/net_monitor/main.go
@@ -198,11 +198,10 @@ func pluginFeature(info, option map[string]*structpb.Value) (sdk.CallResponse, e
 	}
 
 	ret := sdk.CallResponse{
-		FuncName:   info["execute_method"].GetStringValue(),
-		Message:    message,
-		Severity:   severityToSend,
-		State:      state,
-		AlertTypes: []pluginpb.ALERT_TYPE{pluginpb.ALERT_TYPE_DISCORD},
+		FuncName: info["execute_method"].GetStringValue(),
+		Message:  message,
+		Severity: severityToSend,
+		State:    state,
 	}
 
 	return ret, nil