From c9615b77efc42d0b8038249e43bc7fa8244f67ce Mon Sep 17 00:00:00 2001
From: Shaun Agostinho <shaunagostinho@gmail.com>
Date: Wed, 6 Mar 2024 15:24:09 -0800
Subject: [PATCH] update error handle

---
 internal/agent/probe.go | 1 +
 web/agents.go           | 5 ++++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/internal/agent/probe.go b/internal/agent/probe.go
index a69b29b..614db7f 100644
--- 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
index 64bb3a8..3a188f1 100644
--- 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