statement: move retry policy to StatementConfig

As StatementConfig is no longer `pub`, we can move retry_policy field
there, instead of having it directly on Query, PreparedStatement and
Batch.

Author: wprzytula

scylla/src/statement/batch.rs

@@ -17,9 +17,6 @@ pub use crate::frame::request::batch::BatchType;
 pub struct Batch {
     pub(crate) config: StatementConfig,
 
-    // TODO: Move this after #701 is fixed
-    retry_policy: Option<Arc<dyn RetryPolicy>>,
-
     pub statements: Vec<BatchStatement>,
     batch_type: BatchType,
 }
@@ -116,13 +113,13 @@ impl Batch {
     /// 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>>) {
-        self.retry_policy = retry_policy;
+        self.config.retry_policy = retry_policy;
     }
 
     /// Get the retry policy set for the batch.
     #[inline]
     pub fn get_retry_policy(&self) -> Option<&Arc<dyn RetryPolicy>> {
-        self.retry_policy.as_ref()
+        self.config.retry_policy.as_ref()
     }
 
     /// Sets the listener capable of listening what happens during query execution.
@@ -151,7 +148,6 @@ impl Default for Batch {
     fn default() -> Self {
         Self {
             statements: Vec::new(),
-            retry_policy: None,
             batch_type: BatchType::Logged,
             config: Default::default(),
         }

scylla/src/statement/mod.rs

@@ -1,15 +1,15 @@
 use std::{sync::Arc, time::Duration};
 
-use crate::history::HistoryListener;
 use crate::transport::execution_profile::ExecutionProfileHandle;
+use crate::{history::HistoryListener, retry_policy::RetryPolicy};
 
 pub mod batch;
 pub mod prepared_statement;
 pub mod query;
 
 pub use crate::frame::types::{Consistency, SerialConsistency};
 
-#[derive(Debug)]
+#[derive(Debug, Clone, Default)]
 pub(crate) struct StatementConfig {
     pub(crate) consistency: Option<Consistency>,
     pub(crate) serial_consistency: Option<Option<SerialConsistency>>,
@@ -23,32 +23,7 @@ pub(crate) struct StatementConfig {
     pub(crate) history_listener: Option<Arc<dyn HistoryListener>>,
 
     pub(crate) execution_profile_handle: Option<ExecutionProfileHandle>,
-}
-
-#[allow(clippy::derivable_impls)]
-impl Default for StatementConfig {
-    fn default() -> Self {
-        Self {
-            consistency: Default::default(),
-            serial_consistency: None,
-            is_idempotent: false,
-            tracing: false,
-            timestamp: None,
-            request_timeout: None,
-            history_listener: None,
-            execution_profile_handle: None,
-        }
-    }
-}
-
-impl Clone for StatementConfig {
-    fn clone(&self) -> Self {
-        Self {
-            history_listener: self.history_listener.clone(),
-            execution_profile_handle: self.execution_profile_handle.clone(),
-            ..*self
-        }
-    }
+    pub(crate) retry_policy: Option<Arc<dyn RetryPolicy>>,
 }
 
 impl StatementConfig {

scylla/src/statement/prepared_statement.rs

@@ -25,9 +25,6 @@ pub struct PreparedStatement {
     pub(crate) config: StatementConfig,
     pub prepare_tracing_ids: Vec<Uuid>,
 
-    // TODO: Move this after #701 is fixed
-    retry_policy: Option<Arc<dyn RetryPolicy>>,
-
     id: Bytes,
     shared: Arc<PreparedStatementSharedData>,
     page_size: Option<i32>,
@@ -45,7 +42,6 @@ impl Clone for PreparedStatement {
     fn clone(&self) -> Self {
         Self {
             config: self.config.clone(),
-            retry_policy: self.retry_policy.clone(),
             prepare_tracing_ids: Vec::new(),
             id: self.id.clone(),
             shared: self.shared.clone(),
@@ -62,7 +58,6 @@ impl PreparedStatement {
         is_lwt: bool,
         metadata: PreparedMetadata,
         statement: String,
-        retry_policy: Option<Arc<dyn RetryPolicy>>,
         page_size: Option<i32>,
         config: StatementConfig,
     ) -> Self {
@@ -72,7 +67,6 @@ impl PreparedStatement {
                 metadata,
                 statement,
             }),
-            retry_policy,
             prepare_tracing_ids: Vec::new(),
             page_size,
             config,
@@ -303,13 +297,13 @@ impl PreparedStatement {
     /// Set the retry policy for this statement, overriding the one from execution profile if not None.
     #[inline]
     pub fn set_retry_policy(&mut self, retry_policy: Option<Arc<dyn RetryPolicy>>) {
-        self.retry_policy = retry_policy;
+        self.config.retry_policy = retry_policy;
     }
 
     /// Get the retry policy set for the statement.
     #[inline]
     pub fn get_retry_policy(&self) -> Option<&Arc<dyn RetryPolicy>> {
-        self.retry_policy.as_ref()
+        self.config.retry_policy.as_ref()
     }
 
     /// Sets the listener capable of listening what happens during query execution.

scylla/src/statement/query.rs

@@ -14,8 +14,6 @@ pub struct Query {
     pub(crate) config: StatementConfig,
 
     // TODO: Move this after #701 is fixed
-    retry_policy: Option<Arc<dyn RetryPolicy>>,
-
     pub contents: String,
     page_size: Option<i32>,
 }
@@ -25,7 +23,6 @@ impl Query {
     pub fn new(query_text: impl Into<String>) -> Self {
         Self {
             contents: query_text.into(),
-            retry_policy: None,
             page_size: None,
             config: Default::default(),
         }
@@ -131,13 +128,13 @@ impl Query {
     /// Set the retry policy for this statement, overriding the one from execution profile if not None.
     #[inline]
     pub fn set_retry_policy(&mut self, retry_policy: Option<Arc<dyn RetryPolicy>>) {
-        self.retry_policy = retry_policy;
+        self.config.retry_policy = retry_policy;
     }
 
     /// Get the retry policy set for the statement.
     #[inline]
     pub fn get_retry_policy(&self) -> Option<&Arc<dyn RetryPolicy>> {
-        self.retry_policy.as_ref()
+        self.config.retry_policy.as_ref()
     }
 
     /// Sets the listener capable of listening what happens during query execution.

scylla/src/transport/caching_session.rs

@@ -166,13 +166,11 @@ where
 
         if let Some(raw) = self.cache.get(&query.contents) {
             let page_size = query.get_page_size();
-            let retry_policy = query.get_retry_policy().cloned();
             let mut stmt = PreparedStatement::new(
                 raw.id.clone(),
                 raw.is_confirmed_lwt,
                 raw.metadata.clone(),
                 query.contents,
-                retry_policy,
                 page_size,
                 query.config,
             );

scylla/src/transport/connection.rs

@@ -548,7 +548,6 @@ impl Connection {
                     .prepared_flags_contain_lwt_mark(p.prepared_metadata.flags as u32),
                 p.prepared_metadata,
                 query.contents.clone(),
-                query.get_retry_policy().cloned(),
                 query.get_page_size(),
                 query.config.clone(),
             ),