Browse Source

verify Fixed undertow's db test

Mike Smith 11 years ago
parent
commit
9f9f877043

+ 8 - 1
undertow/src/main/java/hello/DbMongoHandler.java

@@ -46,6 +46,13 @@ final class DbMongoHandler implements HttpHandler {
     }
     exchange.getResponseHeaders().put(
         Headers.CONTENT_TYPE, JSON_UTF8);
-    exchange.getResponseSender().send(objectMapper.writeValueAsString(worlds));
+    if (queries == 1)
+    {
+      exchange.getResponseSender().send(objectMapper.writeValueAsString(worlds[0]));
+    }
+    else
+    {
+      exchange.getResponseSender().send(objectMapper.writeValueAsString(worlds));
+    }
   }
 }

+ 8 - 1
undertow/src/main/java/hello/DbSqlHandler.java

@@ -51,6 +51,13 @@ final class DbSqlHandler implements HttpHandler {
     }
     exchange.getResponseHeaders().put(
         Headers.CONTENT_TYPE, JSON_UTF8);
-    exchange.getResponseSender().send(objectMapper.writeValueAsString(worlds));
+    if (queries == 1)
+    {
+      exchange.getResponseSender().send(objectMapper.writeValueAsString(world[0]));
+    }
+    else
+    {
+      exchange.getResponseSender().send(objectMapper.writeValueAsString(worlds));
+    }
   }
 }