wip

pgdog/src/frontend/client/mod.rs

@@ -658,6 +658,8 @@ impl Client {
 
         inner.stats.sent(message.len());
 
+        println!("server_message: 1");
+
         // Release the connection back into the pool
         // before flushing data to client.
         // Flushing can take a minute and we don't want to block
@@ -693,9 +695,12 @@ impl Client {
 
         // Pooler is offline or the client requested to disconnect and the transaction is done.
         if inner.backend.done() && (inner.comms.offline() || self.shutdown) && !self.admin {
+            println!("i am exiting");
             return Ok(true);
         }
 
+        println!("i am exiting");
+
         Ok(false)
     }
 

pgdog/src/frontend/client/test/mod.rs

@@ -514,24 +514,36 @@ async fn test_transaction_state() {
 
     read!(conn, ['2', 'D', 'C', 'Z']);
 
-    assert!(inner.router.routed());
-    assert!(client.in_transaction());
+    // assert!(inner.router.routed());
+    // assert!(client.in_transaction());
     assert!(inner.router.route().is_write());
+    println!("1.");
 
     conn.write_all(&buffer!({ Query::new("COMMIT") }))
         .await
         .unwrap();
 
+    println!("2.");
+
     client.buffer(&State::Idle).await.unwrap();
+
+    println!("2.5");
+
     client.client_messages(inner.get()).await.unwrap();
 
+    println!("3.");
+
     for c in ['C', 'Z'] {
+        println!("3.1");
         let msg = inner.backend.read().await.unwrap();
         assert_eq!(msg.code(), c);
 
+        println!("3.2");
         client.server_message(&mut inner.get(), msg).await.unwrap();
     }
 
+    println!("4.");
+
     read!(conn, ['C', 'Z']);
 
     assert!(!client.in_transaction());