Browse Source

verify Fixed restexpress' db test

Mike Smith 11 years ago
parent
commit
07009827f2

+ 6 - 1
restexpress/src/main/java/hello/controller/MongodbController.java

@@ -22,7 +22,7 @@ public class MongodbController
 		this.worldRepo = worldsRepository;
 	}
 
-	public World[] read(Request request, Response response)
+	public Object read(Request request, Response response)
 	{
 		// Get the count of queries to run.
 		int count = 1;
@@ -52,6 +52,11 @@ public class MongodbController
 			worlds[i] = worldRepo.find(random.nextInt(DB_ROWS) + 1);
 		}
 
+		if (count == 1)
+		{
+			return worlds[0];
+		}
+
 		return worlds;
 	}
 }

+ 6 - 1
restexpress/src/main/java/hello/controller/MysqlController.java

@@ -28,7 +28,7 @@ public class MysqlController
 		this.mysqlDataSource = dataSource;
 	}
 
-	public World[] read(Request request, Response response)
+	public Object read(Request request, Response response)
 	throws SQLException
 	{
 		final DataSource source = mysqlDataSource;
@@ -76,6 +76,11 @@ public class MysqlController
 			}
 		}
 
+		if (count == 1)
+		{
+			return worlds[0];
+		}
+
 		return worlds;
 	}
 }