Setting different throughputs for dev/prod tables.

2 files changed, 31 insertions, 7 deletions

diff --git a/src/com/p4square/grow/backend/dynamo/DbTool.java b/src/com/p4square/grow/backend/dynamo/DbTool.java
index 4bccd8b..758dc0e 100644
--- a/src/com/p4square/grow/backend/dynamo/DbTool.java
+++ b/src/com/p4square/grow/backend/dynamo/DbTool.java
@@ -101,7 +101,11 @@ public class DbTool {
 
                 /* Bootstrap Commands */
                 } else if ("--bootstrap".equals(args[offset])) {
-                    offset = bootstrapTables(args, ++offset);
+                    if ("dev".equals(mConfig.getDomain())) {
+                        offset = bootstrapDevTables(args, ++offset);
+                    } else {
+                        offset = bootstrapTables(args, ++offset);
+                    }
                     offset = loadStrings(args, offset);
 
                 } else if ("--loadStrings".equals(args[offset])) {
@@ -202,12 +206,25 @@ public class DbTool {
     private static int bootstrapTables(String[] args, int offset) {
         DynamoDatabase db = getDatabase();
 
-        db.createTable("strings",      10,  1);
-        db.createTable("accounts",     10,  1);
-        db.createTable("assessments",  10,  5);
-        db.createTable("training",     10,  5);
-        db.createTable("feedthreads",  10,  1);
-        db.createTable("feedmessages", 10,  1);
+        db.createTable("strings",      5,  1);
+        db.createTable("accounts",     5,  1);
+        db.createTable("assessments",  5,  5);
+        db.createTable("training",     5,  5);
+        db.createTable("feedthreads",  5,  1);
+        db.createTable("feedmessages", 5,  1);
+
+        return offset;
+    }
+
+    private static int bootstrapDevTables(String[] args, int offset) {
+        DynamoDatabase db = getDatabase();
+
+        db.createTable("strings",      1,  1);
+        db.createTable("accounts",     1,  1);
+        db.createTable("assessments",  1,  1);
+        db.createTable("training",     1,  1);
+        db.createTable("feedthreads",  1,  1);
+        db.createTable("feedmessages", 1,  1);
 
         return offset;
     }
diff --git a/src/com/p4square/grow/config/Config.java b/src/com/p4square/grow/config/Config.java
index 7c9cc0d..e89990b 100644
--- a/src/com/p4square/grow/config/Config.java
+++ b/src/com/p4square/grow/config/Config.java
@@ -54,6 +54,13 @@ public class Config {
     }
 
     /**
+     * @return the current domain.
+     */
+    public String getDomain() {
+        return mDomain;
+    }
+
+    /**
      * Load properties from a file.
      * Any exceptions are logged and suppressed.
      */