fix: set new node's available = capacity (#2111)

Closes STOR-507

Author: pjenvey

tokenserver-db/src/tests.rs

@@ -432,7 +432,7 @@ async fn test_init_sync15_node() -> DbResult<()> {
 
             assert_eq!(node.node, "https://testo.example.gg");
             assert_eq!(node.capacity, 38383);
-            assert_eq!(node.available, 1);
+            assert_eq!(node.available, 38383);
             assert_eq!(node.current_load, 0);
 
             Ok(())
@@ -474,6 +474,7 @@ async fn test_init_sync15_node_with_default_capacity() -> DbResult<()> {
 
             assert_eq!(node.node, "https://testo.example.gg");
             assert_eq!(node.capacity, 100000);
+            assert_eq!(node.available, 100000);
 
             Ok(())
         },

tokenserver-mysql/src/db/db_impl.rs

@@ -383,7 +383,7 @@ impl Db for TokenserverDb {
             INSERT IGNORE INTO nodes (service, node, available, current_load, capacity, downed, backoff)
             VALUES (
                 (SELECT id FROM services WHERE service = '{}'),
-                ?, 1, 0, ?, 0, 0
+                ?, ?, 0, ?, 0, 0
             )
             "#,
             params::Sync15Node::SERVICE_NAME
@@ -392,6 +392,7 @@ impl Db for TokenserverDb {
         let affected_rows = diesel::sql_query(query)
             .bind::<Text, _>(&params.node)
             .bind::<Integer, _>(params.capacity)
+            .bind::<Integer, _>(params.capacity)
             .execute(&mut self.conn)
             .await?;
 

tokenserver-postgres/src/db/db_impl.rs

@@ -70,10 +70,10 @@ impl Db for TokenserverPgDb {
     async fn insert_sync15_node(&mut self, params: params::Sync15Node) -> DbResult<bool> {
         let query = format!(
             r#"
-              INSERT INTO nodes (service, node,  available, current_load, capacity, downed, backoff)
+              INSERT INTO nodes (service, node, available, current_load, capacity, downed, backoff)
               VALUES (
                   (SELECT id FROM services WHERE service = '{}'),
-                  $1, 1, 0, $2, 0, 0
+                  $1, $2, 0, $2, 0, 0
               )
               ON CONFLICT (service, node) DO NOTHING
               "#,