fix: add recover middleware

ZCube · Nov 9, 2024 · 0d6cd3b · 0d6cd3b
1 parent 12cab8e
commit 0d6cd3b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/cmd/local.go b/cmd/local.go
@@ -107,7 +107,7 @@ var localCmd = &cobra.Command{
 				cc, remote, rderr := conn.ReadFromUDP(b)
 				if rderr != nil {
 					sugar.Error("net.ReadFromUDP() error: %s", rderr)
-					return c.String(http.StatusInternalServerError, rderr.Error())
+					return c.String(http.StatusBadGateway, rderr.Error())
 				} else {
 					sugar.Debugw("Read from socket",
 						"Bytes", cc,
@@ -159,7 +159,7 @@ var localCmd = &cobra.Command{
 				cc, remote, rderr := conn.ReadFromUDP(b)
 				if rderr != nil {
 					sugar.Error("net.ReadFromUDP() error: %s", rderr)
-					return c.String(http.StatusInternalServerError, rderr.Error())
+					return c.String(http.StatusBadGateway, rderr.Error())
 				} else {
 					sugar.Debugw("Read from socket",
 						"Bytes", cc,

diff --git a/cmd/remote.go b/cmd/remote.go
@@ -106,7 +106,7 @@ var remoteCmd = &cobra.Command{
 				cc, remote, rderr := conn.ReadFromUDP(b)
 				if rderr != nil {
 					sugar.Error("net.ReadFromUDP() error: %s", rderr)
-					return c.String(http.StatusInternalServerError, rderr.Error())
+					return c.String(http.StatusBadGateway, rderr.Error())
 				} else {
 					sugar.Debugw("Read from socket",
 						"Bytes", cc,
@@ -158,7 +158,7 @@ var remoteCmd = &cobra.Command{
 				cc, remote, rderr := conn.ReadFromUDP(b)
 				if rderr != nil {
 					sugar.Error("net.ReadFromUDP() error: %s", rderr)
-					return c.String(http.StatusInternalServerError, rderr.Error())
+					return c.String(http.StatusBadGateway, rderr.Error())
 				} else {
 					sugar.Debugw("Read from socket",
 						"Bytes", cc,