stress test

kristina · kristina · commit 3d0bb9cf74cc · 2009-02-04T16:03:18.000-05:00
diff --git a/src/test/com/mongodb/framework/Stress1.java b/src/test/com/mongodb/framework/Stress1.java
@@ -0,0 +1,50 @@
+
+package com.mongodb.framework;
+
+import java.util.*;
+import java.net.*;
+
+import com.mongodb.*;
+import com.mongodb.util.*;
+
+public class Stress1 {
+
+    public static void doStuff( DBCollection c, int count ) {
+        DBObject obj = new BasicDBObject();
+        obj.put( "id", count );
+        DBObject x = c.findOne( obj );
+        x.put( "subarray", "foo" + count );
+        c.save( x );
+    }
+
+    public static void main(String[] args) 
+        throws UnknownHostException {
+
+        Mongo m = new Mongo( new DBAddress( "127.0.0.1:27017/driver_test_framework" ) );
+        DBCollection c = m.getCollection( "stress1" );
+
+        String blah = "lksjhasoh1298alshasoidiohaskjasiouashoasasiugoas" + 
+            "lksjhasoh1298alshasoidiohaskjasiouashoasasiugoas" + 
+            "lksjhasoh1298alshasoidiohaskjasiouashoasasiugoas" + 
+            "lksjhasoh1298alshasoidiohaskjasiouashoasasiugoas" + 
+            "lksjhasoh1298alshasoidiohaskjasiouashoasasiugoas" + 
+            "lksjhasoh1298alshasoidiohaskjasiouashoasasiugoas";
+
+        for( int i=0; i<50000; i++ ) {
+            DBObject foo = new BasicDBObject();
+            foo.put( "name", "asdf"+i );
+            foo.put( "date", new Date() );
+            foo.put( "id", i );
+            foo.put( "blah", blah );
+            c.save( foo );
+        }
+
+        for( int count=0; count<10000; count++ ) {
+            doStuff( c, count );
+        }
+
+        DBObject idx = new BasicDBObject();
+        idx.put( "date", 1 );
+        c.ensureIndex( idx );
+    }
+}
diff --git a/src/test/com/mongodb/framework/dispatch b/src/test/com/mongodb/framework/dispatch
@@ -18,6 +18,8 @@ elif [[ "$1" = "capped" ]]; then
 #    java com.mongodb.framework.Circular
 elif [[ "$1" = "dbs" ]]; then
     java com.mongodb.framework.Dbs > $2
+elif [[ "$1" = "stress1" ]]; then
+    java com.mongodb.framework.Stress1
 #elif [[ "$1" = "admin" ]]; then
 #    java com.mongodb.framework.Admin > $2
 else
