scylla benchmark: Use calculate_token_for_partition_key behind feature flag

This will allow us to unpub the original calculate_token_for_partition_key,
which is the last step towards eliminating SerializedValues from public API.

Author: Lorak-mmk

scylla/benches/benchmark.rs

@@ -1,7 +1,8 @@
 use criterion::{criterion_group, criterion_main, Criterion};
 
 use bytes::BytesMut;
-use scylla::routing::partitioner::{calculate_token_for_partition_key, PartitionerName};
+use scylla::internal_testing::calculate_token_for_partition_key;
+use scylla::routing::partitioner::PartitionerName;
 use scylla_cql::frame::response::result::{ColumnType, NativeType};
 use scylla_cql::frame::types;
 use scylla_cql::serialize::row::SerializedValues;

scylla/src/lib.rs

@@ -257,3 +257,22 @@ pub(crate) mod utils;
 
 #[cfg(test)]
 pub(crate) use utils::test_utils;
+
+#[cfg(feature = "unstable-testing")]
+pub mod internal_testing {
+    use scylla_cql::serialize::row::SerializedValues;
+
+    use crate::routing::partitioner::PartitionerName;
+    use crate::routing::Token;
+    use crate::statement::prepared::TokenCalculationError;
+
+    pub fn calculate_token_for_partition_key(
+        serialized_partition_key_values: &SerializedValues,
+        partitioner: &PartitionerName,
+    ) -> Result<Token, TokenCalculationError> {
+        crate::routing::partitioner::calculate_token_for_partition_key(
+            serialized_partition_key_values,
+            partitioner,
+        )
+    }
+}