update error handle

netwatcherio · Mar 6, 2024 · c9615b7 · c9615b7
1 parent 65f0cab
commit c9615b7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/internal/agent/probe.go b/internal/agent/probe.go
@@ -436,6 +436,7 @@ func (c *Probe) GetAllProbesForAgent(db *mongo.Database) ([]*Probe, error) {
 							var port = strings.Split(probe.Config.Target[0].Target, ":")[1]
 							tC.Config.Target[0].Target = netResult.PublicAddress + ":" + port
 							break
+
 						}
 					}
 				} else {

diff --git a/web/agents.go b/web/agents.go
@@ -151,7 +151,10 @@ func addRouteAgents(r *Router) []*Route {
 			s := agent.Agent{ID: sId}
 
 			cAgent := new(agent.Agent)
-			ctx.ReadJSON(&cAgent)
+			err = ctx.ReadJSON(&cAgent)
+			if err != nil {
+				return err
+			}
 
 			cAgent.Site = s.ID
-Original file line number
+Diff line change
@@ Expand Up @@
     							var port = strings.Split(probe.Config.Target[0].Target, ":")[1]
     							tC.Config.Target[0].Target = netResult.PublicAddress + ":" + port
     							break
     						}
     					}
     				} else {
@@ Expand Down @@