Browse Source

PR #648 - add checks for queries

Martin Tzvetanov Grigorov 11 years ago
parent
commit
97e07d0f34
1 changed files with 10 additions and 1 deletions
  1. 10 1
      wicket/src/main/java/hellowicket/HelloDbResponse.java

+ 10 - 1
wicket/src/main/java/hellowicket/HelloDbResponse.java

@@ -20,7 +20,16 @@ public class HelloDbResponse extends AbstractResource
 
 
   protected ResourceResponse newResourceResponse(Attributes attributes)
   protected ResourceResponse newResourceResponse(Attributes attributes)
   {
   {
-    final int queries = attributes.getRequest().getQueryParameters().getParameterValue("queries").toInt(1);
+    int qs = attributes.getRequest().getQueryParameters().getParameterValue("queries").toInt(1);
+    if (qs < 1)
+    {
+      qs = 1;
+    } 
+    else if (qs > 500)
+    {
+      qs = 500;
+    }
+    final int queries = qs;
     final World[] worlds = new World[queries];
     final World[] worlds = new World[queries];
     final ThreadLocalRandom random = ThreadLocalRandom.current();
     final ThreadLocalRandom random = ThreadLocalRandom.current();