Merge pull request #1391 from wprzytula/batch-set-timeout

batch: add request timeout setter/getter

Author: wprzytula

scylla/src/statement/batch.rs

@@ -1,5 +1,6 @@
 use std::borrow::Cow;
 use std::sync::Arc;
+use std::time::Duration;
 
 use crate::client::execution_profile::ExecutionProfileHandle;
 use crate::observability::history::HistoryListener;
@@ -137,6 +138,19 @@ impl Batch {
         self.config.timestamp
     }
 
+    /// Sets the client-side timeout for this batch.
+    /// If not None, the driver will stop waiting for the request
+    /// to finish after `timeout` passed.
+    /// Otherwise, execution profile timeout will be applied.
+    pub fn set_request_timeout(&mut self, timeout: Option<Duration>) {
+        self.config.request_timeout = timeout
+    }
+
+    /// Gets client timeout associated with this batch.
+    pub fn get_request_timeout(&self) -> Option<Duration> {
+        self.config.request_timeout
+    }
+
     /// Set the retry policy for this batch, overriding the one from execution profile if not None.
     #[inline]
     pub fn set_retry_policy(&mut self, retry_policy: Option<Arc<dyn RetryPolicy>>) {

scylla/src/statement/prepared.rs

@@ -393,7 +393,7 @@ impl PreparedStatement {
     /// Sets the client-side timeout for this statement.
     /// If not None, the driver will stop waiting for the request
     /// to finish after `timeout` passed.
-    /// Otherwise, default session client timeout will be applied.
+    /// Otherwise, execution profile timeout will be applied.
     pub fn set_request_timeout(&mut self, timeout: Option<Duration>) {
         self.config.request_timeout = timeout
     }

scylla/src/statement/unprepared.rs

@@ -134,7 +134,7 @@ impl Statement {
     /// Sets the client-side timeout for this statement.
     /// If not None, the driver will stop waiting for the request
     /// to finish after `timeout` passed.
-    /// Otherwise, default session client timeout will be applied.
+    /// Otherwise, execution profile timeout will be applied.
     pub fn set_request_timeout(&mut self, timeout: Option<Duration>) {
         self.config.request_timeout = timeout
     }