Explorar el Código

return 404 on pod not found for logs

Alexander Belanger hace 4 años
padre
commit
a50a50e2d7

+ 11 - 1
api/server/handlers/namespace/stream_pod_logs.go

@@ -1,6 +1,8 @@
 package namespace
 package namespace
 
 
 import (
 import (
+	"errors"
+	"fmt"
 	"net/http"
 	"net/http"
 
 
 	"github.com/porter-dev/porter/api/server/authz"
 	"github.com/porter-dev/porter/api/server/authz"
@@ -10,6 +12,7 @@ import (
 	"github.com/porter-dev/porter/api/server/shared/config"
 	"github.com/porter-dev/porter/api/server/shared/config"
 	"github.com/porter-dev/porter/api/server/shared/requestutils"
 	"github.com/porter-dev/porter/api/server/shared/requestutils"
 	"github.com/porter-dev/porter/api/types"
 	"github.com/porter-dev/porter/api/types"
+	"github.com/porter-dev/porter/internal/kubernetes"
 	"github.com/porter-dev/porter/internal/models"
 	"github.com/porter-dev/porter/internal/models"
 )
 )
 
 
@@ -51,7 +54,14 @@ func (c *StreamPodLogsHandler) ServeHTTP(w http.ResponseWriter, r *http.Request)
 
 
 	err = agent.GetPodLogs(namespace, name, conn)
 	err = agent.GetPodLogs(namespace, name, conn)
 
 
-	if err != nil {
+	if targetErr := kubernetes.IsNotFoundError; errors.Is(err, targetErr) {
+		c.HandleAPIError(w, r, apierrors.NewErrPassThroughToClient(
+			fmt.Errorf("pod %s/%s was not found", namespace, name),
+			http.StatusNotFound,
+		))
+
+		return
+	} else if err != nil {
 		c.HandleAPIError(w, r, apierrors.NewErrInternal(err))
 		c.HandleAPIError(w, r, apierrors.NewErrInternal(err))
 		return
 		return
 	}
 	}

+ 4 - 2
internal/kubernetes/agent.go

@@ -502,8 +502,10 @@ func (a *Agent) GetPodLogs(namespace string, name string, conn *websocket.Conn)
 		metav1.GetOptions{},
 		metav1.GetOptions{},
 	)
 	)
 
 
-	if err != nil {
-		return fmt.Errorf("Cannot get pod %s: %s", name, err.Error())
+	if err != nil && errors.IsNotFound(err) {
+		return IsNotFoundError
+	} else if err != nil {
+		return fmt.Errorf("Cannot get logs from pod %s: %s", name, err.Error())
 	}
 	}
 
 
 	container := pod.Spec.Containers[0].Name
 	container := pod.Spec.Containers[0].Name