Set db pool sizes

* set primary db pool size in middleware
* set db pool sizes in test environment

Author: justahero

src/middleware.rs

@@ -83,7 +83,7 @@ pub fn build_middleware(app: Arc<App>, endpoints: RouteBuilder) -> MiddlewareBui
     // In production we currently have 2 equally sized pools (primary and a read-only replica).
     // Because such a large portion of production traffic is for download requests (which update
     // download counts), we consider only the primary pool here.
-    if let Ok(capacity) = env::var("DB_POOL_SIZE") {
+    if let Ok(capacity) = env::var("DB_PRIMARY_POOL_SIZE") {
         if let Ok(capacity) = capacity.parse() {
             if capacity >= 10 {
                 println!(
@@ -92,7 +92,7 @@ pub fn build_middleware(app: Arc<App>, endpoints: RouteBuilder) -> MiddlewareBui
                 );
                 m.around(balance_capacity::BalanceCapacity::new(capacity))
             } else {
-                println!("BalanceCapacity middleware not enabled. DB_POOL_SIZE is too low.");
+                println!("BalanceCapacity middleware not enabled. DB_PRIMARY_POOL_SIZE is too low.");
             }
         }
     }

src/tests/server_binary.rs

@@ -88,7 +88,8 @@ impl ServerBin {
         // Bind a random port every time the server is started.
         env.insert("PORT".into(), "0".into());
         // Avoid creating too many database connections.
-        env.insert("DB_POOL_SIZE".into(), "2".into());
+        env.insert("DB_PRIMARY_POOL_SIZE".into(), "2".into());
+        env.insert("DB_REPLICA_POOL_SIZE".into(), "1".into());
         env.remove("DB_MIN_SIZE");
         // Other configuration variables needed for the application to boot.
         env.insert("WEB_ALLOWED_ORIGINS".into(), "http://localhost:8888".into());