update client side parameters when changed

src/client.rs

@@ -1452,7 +1452,7 @@ where
  timeout => tokio::time::Duration::from_millis(timeout),
  };
 
- match tokio::time::timeout(statement_timeout_duration, server.recv()).await {
+ match tokio::time::timeout(statement_timeout_duration, server.recv(Some(&mut self.server_parameters))).await {
  Ok(result) => match result {
  Ok(message) => Ok(message),
  Err(err) => {

src/mirrors.rs

@@ -78,7 +78,7 @@ impl MirroredClient {
  }
 
  // Incoming data from server (we read to clear the socket buffer and discard the data)
- recv_result = server.recv() => {
+ recv_result = server.recv(None) => {
  match recv_result {
  Ok(message) => trace!("Received from mirror: {} {:?}", String::from_utf8_lossy(&message[..]), address.clone()),
  Err(err) => {

src/server.rs

@@ -879,7 +879,10 @@ impl Server {
  /// Receive data from the server in response to a client request.
  /// This method must be called multiple times while `self.is_data_available()` is true
  /// in order to receive all data the server has to offer.
- pub async fn recv(&mut self) -> Result<BytesMut, Error> {
+ pub async fn recv(
+ &mut self,
+ mut client_server_parameters: Option<&mut ServerParameters>,
+ ) -> Result<BytesMut, Error> {
  loop {
  let mut message = match read_message(&mut self.stream).await {
  Ok(message) => message,
@@ -974,6 +977,10 @@ impl Server {
  let key = message.read_string().unwrap();
  let value = message.read_string().unwrap();
 
+ if let Some(client_server_parameters) = client_server_parameters.as_mut() {
+ client_server_parameters.set_param(key.clone(), value.clone(), false);
+ }
+
  self.server_parameters.set_param(key, value, false);
  }
 
@@ -1117,7 +1124,7 @@ impl Server {
  self.send(&query).await?;
 
  loop {
- let _ = self.recv().await?;
+ let _ = self.recv(None).await?;
 
  if !self.data_available {
  break;
@@ -1211,7 +1218,7 @@ impl Server {
  .await?;
  debug!("Connected!, sending query.");
  server.send(&simple_query(query)).await?;
- let mut message = server.recv().await?;
+ let mut message = server.recv(None).await?;
 
  Ok(parse_query_message(&mut message).await?)
  }