wip

pgdog/src/backend/pool/connection/binding.rs

@@ -1,7 +1,7 @@
 //! Binding between frontend client and a connection on the backend.
 
 use crate::{
-    net::{parameter::Parameters, ProtocolMessage},
+    net::{parameter::Parameters, ProtocolMessage, Query},
     state::State,
 };
 
@@ -240,23 +240,8 @@ impl Binding {
     /// Execute a BEGIN on all servers
     /// TODO: Block mutli-shard BEGINs as transaction should not occur on multiple shards
     pub async fn begin(&mut self) -> Result<(), Error> {
-        let begin: &str = "BEGIN;";
-
-        match self {
-            Binding::Server(Some(ref mut server)) => {
-                server.execute(&begin).await?;
-            }
-
-            Binding::MultiShard(ref mut servers, _) => {
-                for server in servers {
-                    server.execute(&begin).await?;
-                }
-            }
-
-            _ => (),
-        }
-
-        Ok(())
+        let query = Query::new("BEGIN");
+        self.execute(query.query()).await
     }
 
     pub async fn link_client(&mut self, params: &Parameters) -> Result<usize, Error> {

pgdog/src/frontend/client/mod.rs

@@ -454,14 +454,12 @@ impl Client {
                 }
                 Some(Command::RollbackTransaction) => {
                     self.end_transaction(true).await?;
-                    self.logical_transaction.rollback()?;
 
                     inner.done(self.in_transaction());
                     return Ok(false);
                 }
                 Some(Command::CommitTransaction) => {
                     self.end_transaction(false).await?;
-                    self.logical_transaction.commit()?;
 
                     inner.done(self.in_transaction());
                     return Ok(false);

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

@@ -511,6 +511,13 @@ async fn test_transaction_state() {
 
     read!(conn, ['2', 'D', 'C', 'Z']);
 
+    println!("router.route {:?}", inner.router.route());
+    println!("router.route {:?}", inner.router.route());
+    println!("router.route {:?}", inner.router.route());
+    println!("router.route {:?}", inner.router.route());
+    println!("router.route {:?}", inner.router.route());
+    println!("router.route {:?}", inner.router.route());
+
     assert!(inner.router.routed());
     assert!(client.in_transaction());
     assert!(inner.router.route().is_write());