浏览代码

verify Fixed undertow's db test

Mike Smith 11 年之前
父节点
当前提交
9f9f877043
共有 2 个文件被更改,包括 16 次插入2 次删除
  1. 8 1
      undertow/src/main/java/hello/DbMongoHandler.java
  2. 8 1
      undertow/src/main/java/hello/DbSqlHandler.java

+ 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));
+    }
   }
 }