更新 setup.sh

setup.sh

@@ -21,11 +21,28 @@ for choice in $choices; do
         sysctl -p
         ;;
     2)
-        echo "输入Swap内存大小(MB) [默认: 2G]: "
+      while true; do
-        read -t 5 -p "> " swap_size
+          echo "输入Swap内存大小(MB) [默认: 2048, 5秒后自动使用默认值]: "
-        if [ $? -ne 0 ] || [ -z "$swap_size" ]; then
+          if read -t 5 -p "> " swap_size; then
-            echo -e "\n使用默认值: ${swap_size}MB"
+              if [ -z "$swap_size" ]; then
+                  swap_size=2048
+                  echo "使用默认值: ${swap_size}MB"
+                  break
+              elif [[ "$swap_size" =~ ^[0-9]+$ ]]; then
+                  echo "使用输入的大小: ${swap_size}MB"
+                  break
+              else
+                  echo "错误: 请输入有效的数字！"
+                  continue
               fi
+          else
+              swap_size=2048
+              echo -e "\n超时, 使用默认值: ${swap_size}MB"
+              break
+          fi
+      done
+
+      echo "执行设置Swap内存大小的脚本..."
       dd if=/dev/zero of=/swapfile bs=1M count=$swap_size
       mkswap /swapfile
       swapon /swapfile