docs and remove Option (#58)

* docs and remove Option

* lint

Author: levkk

src/client.rs

@@ -407,12 +407,12 @@ impl Client {
                         }
 
                         // Report query executed statistics.
-                        self.stats.query(address.id);
+                        self.stats.query(self.process_id, address.id);
 
                         // The transaction is over, we can release the connection back to the pool.
                         if !server.in_transaction() {
                             // Report transaction executed statistics.
-                            self.stats.transaction(address.id);
+                            self.stats.transaction(self.process_id, address.id);
 
                             // Release server back to the pool if we are in transaction mode.
                             // If we are in session mode, we keep the server until the client disconnects.
@@ -493,12 +493,12 @@ impl Client {
                         }
 
                         // Report query executed statistics.
-                        self.stats.query(address.id);
+                        self.stats.query(self.process_id, address.id);
 
                         // Release server back to the pool if we are in transaction mode.
                         // If we are in session mode, we keep the server until the client disconnects.
                         if !server.in_transaction() {
-                            self.stats.transaction(address.id);
+                            self.stats.transaction(self.process_id, address.id);
 
                             if self.transaction_mode {
                                 self.stats.server_idle(server.process_id(), address.id);
@@ -532,7 +532,7 @@ impl Client {
                         // Release server back to the pool if we are in transaction mode.
                         // If we are in session mode, we keep the server until the client disconnects.
                         if !server.in_transaction() {
-                            self.stats.transaction(address.id);
+                            self.stats.transaction(self.process_id, address.id);
 
                             if self.transaction_mode {
                                 self.stats.server_idle(server.process_id(), address.id);

src/pool.rs

@@ -234,7 +234,7 @@ impl ConnectionPool {
                     self.ban(address, shard);
                     self.stats.client_disconnecting(process_id, address.id);
                     self.stats
-                        .checkout_time(now.elapsed().as_micros(), address.id);
+                        .checkout_time(now.elapsed().as_micros(), process_id, address.id);
                     continue;
                 }
             };
@@ -255,7 +255,7 @@ impl ConnectionPool {
                 Ok(res) => match res {
                     Ok(_) => {
                         self.stats
-                            .checkout_time(now.elapsed().as_micros(), address.id);
+                            .checkout_time(now.elapsed().as_micros(), process_id, address.id);
                         self.stats.server_idle(conn.process_id(), address.id);
                         return Ok((conn, address.clone()));
                     }
@@ -267,7 +267,7 @@ impl ConnectionPool {
                         self.ban(address, shard);
                         self.stats.client_disconnecting(process_id, address.id);
                         self.stats
-                            .checkout_time(now.elapsed().as_micros(), address.id);
+                            .checkout_time(now.elapsed().as_micros(), process_id, address.id);
                         continue;
                     }
                 },
@@ -280,7 +280,7 @@ impl ConnectionPool {
                     self.ban(address, shard);
                     self.stats.client_disconnecting(process_id, address.id);
                     self.stats
-                        .checkout_time(now.elapsed().as_micros(), address.id);
+                        .checkout_time(now.elapsed().as_micros(), process_id, address.id);
                     continue;
                 }
             }

src/server.rs

@@ -268,7 +268,8 @@ impl Server {
 
     /// Send messages to the server from the client.
     pub async fn send(&mut self, messages: BytesMut) -> Result<(), Error> {
-        self.stats.data_sent(messages.len(), self.address.id);
+        self.stats
+            .data_sent(messages.len(), self.process_id, self.address.id);
 
         match write_all_half(&mut self.write, messages).await {
             Ok(_) => Ok(()),
@@ -374,7 +375,8 @@ impl Server {
         let bytes = self.buffer.clone();
 
         // Keep track of how much data we got from the server for stats.
-        self.stats.data_received(bytes.len(), self.address.id);
+        self.stats
+            .data_received(bytes.len(), self.process_id, self.address.id);
 
         // Clear the buffer for next query.
         self.buffer.clear();