fix port forward

Author: diamonwiggins

applications/mlflow/Taskfile.yml

@@ -628,12 +628,18 @@ tasks:
         PORT_FORWARD_PID=$!
         echo "Port forwarding set up with PID: $PORT_FORWARD_PID"
         
+        # Verify the PID was captured properly
+        if [ -z "$PORT_FORWARD_PID" ] || [ "$PORT_FORWARD_PID" -eq "0" ]; then
+          echo "ERROR: Failed to capture port-forward process PID"
+          exit 1
+        fi
+        
         # Give port-forward more time to establish
         echo "Waiting for port-forward to establish..."
         sleep 5
         
         # Check if port-forward process is still running
-        if ! ps -p $PORT_FORWARD_PID > /dev/null; then
+        if ! ps -p $PORT_FORWARD_PID > /dev/null 2>&1; then
           echo "ERROR: Port forwarding process died. Check the logs:"
           cat $PORT_FORWARD_LOG
           exit 1
@@ -658,7 +664,7 @@ tasks:
             echo "Connection attempt $CONN_RETRY_COUNT failed, retrying in 5 seconds..."
             
             # Check if port-forward is still running
-            if ! ps -p $PORT_FORWARD_PID > /dev/null; then
+            if ! ps -p $PORT_FORWARD_PID > /dev/null 2>&1; then
               echo "ERROR: Port forwarding process died during connection attempts."
               echo "Port forwarding log:"
               cat $PORT_FORWARD_LOG
@@ -667,7 +673,15 @@ tasks:
               echo "Attempting to restart port forwarding..."
               kubectl port-forward -n {{.NAMESPACE}} $SERVICE_NAME {{.PORT}}:5000 > $PORT_FORWARD_LOG 2>&1 &
               PORT_FORWARD_PID=$!
+              
+              # Verify the PID was captured properly after restart
+              if [ -z "$PORT_FORWARD_PID" ] || [ "$PORT_FORWARD_PID" -eq "0" ]; then
+                echo "ERROR: Failed to capture restarted port-forward process PID"
+                exit 1
+              fi
+              
               echo "Restarted port forwarding with PID: $PORT_FORWARD_PID"
+              sleep 5 # Give the new port-forward time to establish
             fi
             
             sleep 5