From 0ec28b736786ac699a416ff0594626e4aa88f088 Mon Sep 17 00:00:00 2001
From: Xi Xu <i@xi-xu.me>
Date: Wed, 3 Jul 2024 01:52:43 +0800
Subject: [PATCH] Update Client.java

---
 src/client/Client.java | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/src/client/Client.java b/src/client/Client.java
index 0c31381..e77e2d9 100644
--- a/src/client/Client.java
+++ b/src/client/Client.java
@@ -82,6 +82,7 @@ public class Client {
 					export();
 					break;
 				default:
+					System.out.println("Invalid input!");
 					break;
 			}
 			choose = menu();
@@ -217,9 +218,12 @@ public class Client {
 					sql = "SELECT * FROM book WHERE oldprice = ?";
 					System.out.println("Please enter the price:");
 					break;
-				default:
+				case 5:
 					System.out.println("Returning to previous menu.");
 					break;
+				default:
+					System.out.println("Invalid input!");
+					break;
 			}
 			if (!sql.isEmpty()) {
 				ps = con.prepareStatement(sql);
@@ -287,9 +291,12 @@ public class Client {
 				case 3:
 					updatePrice(title);
 					break;
-				default:
+				case 4:
 					System.out.println("Returning to previous menu.");
 					break;
+				default:
+					System.out.println("Invalid input!");
+					break;
 			}
 		} catch (Exception e) {
 			e.printStackTrace();
@@ -346,9 +353,12 @@ public class Client {
 			case 2:
 				ExportToXLS.to("SELECT * FROM book", "output.xls");
 				break;
-			default:
+			case 3:
 				System.out.println("Returning to previous menu.");
 				break;
+			default:
+				System.out.println("Invalid input!");
+				break;
 		}
 	}
 }