refactor: use impl future instead of async-trait in most traits

Protryon · Mar 31, 2024 · d609de5 · d609de5
1 parent db802c1
commit d609de5
Show file tree

Hide file tree

Showing 18 changed files with 269 additions and 251 deletions.
diff --git a/klickhouse/src/io.rs b/klickhouse/src/io.rs
@@ -1,20 +1,20 @@
-use crate::{KlickhouseError, Result};
+use futures::Future;
 use tokio::io::{AsyncRead, AsyncReadExt, AsyncWrite, AsyncWriteExt};
 
+use crate::{KlickhouseError, Result};
+
 use crate::protocol::MAX_STRING_SIZE;
 
-#[async_trait::async_trait]
 pub trait ClickhouseRead: AsyncRead + Unpin + Send + Sync {
-    async fn read_var_uint(&mut self) -> Result<u64>;
+    fn read_var_uint(&mut self) -> impl Future<Output = Result<u64>> + Send;
 
-    async fn read_string(&mut self) -> Result<Vec<u8>>;
+    fn read_string(&mut self) -> impl Future<Output = Result<Vec<u8>>> + Send;
 
-    async fn read_utf8_string(&mut self) -> Result<String> {
-        Ok(String::from_utf8(self.read_string().await?)?)
+    fn read_utf8_string(&mut self) -> impl Future<Output = Result<String>> + Send {
+        async { Ok(String::from_utf8(self.read_string().await?)?) }
     }
 }
 
-#[async_trait::async_trait]
 impl<T: AsyncRead + Unpin + Send + Sync> ClickhouseRead for T {
     async fn read_var_uint(&mut self) -> Result<u64> {
         let mut out = 0u64;
@@ -50,14 +50,15 @@ impl<T: AsyncRead + Unpin + Send + Sync> ClickhouseRead for T {
     }
 }
 
-#[async_trait::async_trait]
 pub trait ClickhouseWrite: AsyncWrite + Unpin + Send + Sync + 'static {
-    async fn write_var_uint(&mut self, value: u64) -> Result<()>;
+    fn write_var_uint(&mut self, value: u64) -> impl Future<Output = Result<()>> + Send;
 
-    async fn write_string(&mut self, value: impl AsRef<[u8]> + Send) -> Result<()>;
+    fn write_string(
+        &mut self,
+        value: impl AsRef<[u8]> + Send,
+    ) -> impl Future<Output = Result<()>> + Send;
 }
 
-#[async_trait::async_trait]
 impl<T: AsyncWrite + Unpin + Send + Sync + 'static> ClickhouseWrite for T {
     async fn write_var_uint(&mut self, mut value: u64) -> Result<()> {
         for _ in 0..9u64 {

diff --git a/klickhouse/src/types/deserialize/array.rs b/klickhouse/src/types/deserialize/array.rs
@@ -30,7 +30,6 @@ impl ArrayDeserializerGeneric for ArrayDeserializer {
     }
 }
 
-#[async_trait::async_trait]
 impl<T: ArrayDeserializerGeneric + 'static> Deserializer for T {
     async fn read_prefix<R: ClickhouseRead>(
         type_: &Type,

diff --git a/klickhouse/src/types/deserialize/geo.rs b/klickhouse/src/types/deserialize/geo.rs
@@ -6,7 +6,6 @@ use crate::values;
 
 pub struct PointDeserializer;
 
-#[async_trait::async_trait]
 impl Deserializer for PointDeserializer {
     async fn read_prefix<R: ClickhouseRead>(
         _type_: &Type,

diff --git a/klickhouse/src/types/deserialize/low_cardinality.rs b/klickhouse/src/types/deserialize/low_cardinality.rs
@@ -8,7 +8,6 @@ use crate::types::low_cardinality::*;
 
 pub struct LowCardinalityDeserializer;
 
-#[async_trait::async_trait]
 impl Deserializer for LowCardinalityDeserializer {
     async fn read_prefix<R: ClickhouseRead>(
         _type_: &Type,

diff --git a/klickhouse/src/types/deserialize/map.rs b/klickhouse/src/types/deserialize/map.rs
@@ -8,7 +8,6 @@ use super::{Deserializer, DeserializerState, Type};
 
 pub struct MapDeserializer;
 
-#[async_trait::async_trait]
 impl Deserializer for MapDeserializer {
     async fn read_prefix<R: ClickhouseRead>(
         type_: &Type,

diff --git a/klickhouse/src/types/deserialize/nullable.rs b/klickhouse/src/types/deserialize/nullable.rs
@@ -6,7 +6,6 @@ use super::{Deserializer, DeserializerState, Type};
 
 pub struct NullableDeserializer;
 
-#[async_trait::async_trait]
 impl Deserializer for NullableDeserializer {
     async fn read_prefix<R: ClickhouseRead>(
         type_: &Type,

diff --git a/klickhouse/src/types/deserialize/sized.rs b/klickhouse/src/types/deserialize/sized.rs
@@ -9,7 +9,6 @@ use super::{Deserializer, DeserializerState, Type};
 
 pub struct SizedDeserializer;
 
-#[async_trait::async_trait]
 impl Deserializer for SizedDeserializer {
     async fn read<R: ClickhouseRead>(
         type_: &Type,

diff --git a/klickhouse/src/types/deserialize/string.rs b/klickhouse/src/types/deserialize/string.rs
@@ -7,7 +7,6 @@ use super::{Deserializer, DeserializerState, Type};
 pub struct StringDeserializer;
 
 #[allow(clippy::uninit_vec)]
-#[async_trait::async_trait]
 impl Deserializer for StringDeserializer {
     async fn read<R: ClickhouseRead>(
         type_: &Type,

diff --git a/klickhouse/src/types/deserialize/tuple.rs b/klickhouse/src/types/deserialize/tuple.rs
@@ -4,7 +4,6 @@ use super::{Deserializer, DeserializerState, Type};
 
 pub struct TupleDeserializer;
 
-#[async_trait::async_trait]
 impl Deserializer for TupleDeserializer {
     async fn read_prefix<R: ClickhouseRead>(
         type_: &Type,