diff --git a/scylla/benches/benchmark.rs b/scylla/benches/benchmark.rs index 20440ea0b7..61d35d8f97 100644 --- a/scylla/benches/benchmark.rs +++ b/scylla/benches/benchmark.rs @@ -1,11 +1,11 @@ use criterion::{criterion_group, criterion_main, Criterion}; use bytes::BytesMut; -use scylla::{ - frame::types, - transport::partitioner::{calculate_token_for_partition_key, Murmur3Partitioner}, +use scylla::transport::partitioner::{calculate_token_for_partition_key, Murmur3Partitioner}; +use scylla_cql::{ + frame::{response::result::ColumnType, types}, + types::serialize::row::SerializedValues, }; -use scylla_cql::{frame::response::result::ColumnType, types::serialize::row::SerializedValues}; fn types_benchmark(c: &mut Criterion) { let mut buf = BytesMut::with_capacity(64); diff --git a/scylla/src/lib.rs b/scylla/src/lib.rs index 381fad34d3..1b46559698 100644 --- a/scylla/src/lib.rs +++ b/scylla/src/lib.rs @@ -102,7 +102,30 @@ pub mod macros; #[doc(inline)] pub use macros::*; -pub use scylla_cql::frame; +pub mod frame { + pub use scylla_cql::frame::{frame_errors, value, Authenticator, Compression}; + pub(crate) use scylla_cql::frame::{ + parse_response_body_extensions, protocol_features, read_response_frame, request, + server_event_type, FrameParams, SerializedRequest, + }; + + pub mod types { + pub use scylla_cql::frame::types::{Consistency, SerialConsistency}; + } + + pub mod response { + pub use scylla_cql::frame::response::cql_to_rust; + pub(crate) use scylla_cql::frame::response::*; + + pub mod result { + pub(crate) use scylla_cql::frame::response::result::*; + pub use scylla_cql::frame::response::result::{ + ColumnSpec, ColumnType, CqlValue, PartitionKeyIndex, Row, TableSpec, + }; + } + } +} + pub use scylla_cql::types::serialize; pub mod authentication; diff --git a/scylla/tests/integration/lwt_optimisation.rs b/scylla/tests/integration/lwt_optimisation.rs index 004840e66a..a36d7367de 100644 --- a/scylla/tests/integration/lwt_optimisation.rs +++ b/scylla/tests/integration/lwt_optimisation.rs @@ -1,10 +1,10 @@ use crate::utils::{setup_tracing, test_with_3_node_cluster}; - -use scylla::frame::types; use scylla::retry_policy::FallthroughRetryPolicy; +use scylla::test_utils::unique_keyspace_name; use scylla::transport::session::Session; -use scylla::{frame::protocol_features::ProtocolFeatures, test_utils::unique_keyspace_name}; use scylla::{ExecutionProfile, SessionBuilder}; +use scylla_cql::frame::protocol_features::ProtocolFeatures; +use scylla_cql::frame::types; use std::sync::Arc; use tokio::sync::mpsc;