From 367ea013d905057fa272417f514b4dcd89086453 Mon Sep 17 00:00:00 2001
From: kevin <1141706615@qq.com>
Date: Wed, 10 May 2017 23:52:46 +0800
Subject: [PATCH] from my laptop

---
 src/step5/QueryRecordTest.java | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/src/step5/QueryRecordTest.java b/src/step5/QueryRecordTest.java
index 3fe3331..518c27a 100644
--- a/src/step5/QueryRecordTest.java
+++ b/src/step5/QueryRecordTest.java
@@ -13,9 +13,16 @@ public class QueryRecordTest {
 
 	public static void main(String[] args) throws SQLException {
 	        String tableName = "book";
-	        
+	        String bookName = "";
 	        Scanner scanner = new Scanner(System.in);
-	        String bookName = scanner.nextLine();
+	        String input = scanner.nextLine();
+			String[] queryInfo = input.split(" ");
+			if(queryInfo.length != 1)
+			{
+				logger.error("参数输入错误");
+				return；
+			}
+			bookName = queryInfo[0];
 	        QueryRecord queryRecord = new QueryRecord();
 	        String sqlFile=System.getProperty("user.dir")+File.separator+"challenge4.sql";
 	        SQLExec sqlExec = new SQLExec();