소스 검색

refactor FromBytes and ToBytes trait auto impl in flowy-sys

appflowy 3 년 전
부모
커밋
e5ca614ceb

+ 1 - 1
rust-lib/flowy-ast/src/ty_ext.rs

@@ -99,7 +99,7 @@ pub fn generate_hashmap_ty_info<'a>(
         ident: &path_segment.ident,
         ty,
         primitive_ty: PrimitiveTy::Map(MapInfo::new(key, value)),
-        bracket_ty_info,
+        bracket_ty_info: bracket_ty_info,
     });
 }
 

+ 10 - 0
rust-lib/flowy-derive/src/proto_buf/protobuf_trait.rs

@@ -0,0 +1,10 @@
+pub trait SerializeProtoBuf {
+    type ProtoBufType;
+    fn to_protobuf(&self) -> Self::ProtoBufType;
+}
+
+pub trait DeserializeProtoBuf {
+    type ProtoBufType;
+    type ObjectType;
+    fn from_protobuf(pb: &mut Self::ProtoBufType) -> Self::ObjectType;
+}

+ 17 - 0
rust-lib/flowy-derive/src/proto_buf/util.rs

@@ -0,0 +1,17 @@
+pub enum TypeCategory {
+    Array,
+    Map,
+    Str,
+    Protobuf,
+    Bytes,
+    Enum,
+    Opt,
+    Primitive,
+}
+
+fn category_from_str(type_str: &str) -> TypeCategory { TypeCategory::Protobuf }
+
+pub fn ident_category(ident: &syn::Ident) -> TypeCategory {
+    let ident_str: &str = &ident.to_string();
+    category_from_str(ident_str)
+}

+ 22 - 21
rust-lib/flowy-sdk/tests/sdk/helper.rs

@@ -33,35 +33,36 @@ pub struct EventTester {
 }
 
 impl EventTester {
-    pub fn new<E>(event: E, payload: Payload) -> Self
+    pub fn new<E, P>(event: E, payload: P) -> Self
     where
         E: Eq + Hash + Debug + Clone + Display,
+        P: std::convert::Into<Payload>,
     {
         init_sdk();
         Self {
-            request: DispatchRequest::new(event, payload),
+            request: DispatchRequest::new(event, payload.into()),
         }
     }
 
-    #[allow(dead_code)]
-    pub fn bytes_payload<T>(mut self, payload: T) -> Self
-    where
-        T: serde::Serialize,
-    {
-        let bytes: Vec<u8> = bincode::serialize(&payload).unwrap();
-        self.request = self.request.payload(Payload::Bytes(bytes));
-        self
-    }
-
-    #[allow(dead_code)]
-    pub fn protobuf_payload<T>(mut self, payload: T) -> Self
-    where
-        T: ::protobuf::Message,
-    {
-        let bytes: Vec<u8> = payload.write_to_bytes().unwrap();
-        self.request = self.request.payload(Payload::Bytes(bytes));
-        self
-    }
+    // #[allow(dead_code)]
+    // pub fn bytes_payload<T>(mut self, payload: T) -> Self
+    // where
+    //     T: serde::Serialize,
+    // {
+    //     let bytes: Vec<u8> = bincode::serialize(&payload).unwrap();
+    //     self.request = self.request.payload(Payload::Bytes(bytes));
+    //     self
+    // }
+    //
+    // #[allow(dead_code)]
+    // pub fn protobuf_payload<T>(mut self, payload: T) -> Self
+    // where
+    //     T: ::protobuf::Message,
+    // {
+    //     let bytes: Vec<u8> = payload.write_to_bytes().unwrap();
+    //     self.request = self.request.payload(Payload::Bytes(bytes));
+    //     self
+    // }
 
     #[allow(dead_code)]
     pub async fn async_send(self) -> EventResponse {

+ 8 - 6
rust-lib/flowy-sdk/tests/sdk/user_check.rs

@@ -6,15 +6,17 @@ use tokio::time::{sleep, Duration};
 #[test]
 #[should_panic]
 fn auth_check_no_payload() {
-    let resp = EventTester::new(AuthCheck).sync_send();
+    let resp = EventTester::new(AuthCheck, Payload::None).sync_send();
     assert_eq!(resp.status_code, StatusCode::Ok);
 }
 
 #[tokio::test]
 async fn auth_check_with_user_name_email_payload() {
-    let user_data = UserData::new("jack".to_owned(), "[email protected]".to_owned());
-
-    EventTester::new(AuthCheck)
-        .bytes_payload(user_data)
-        .sync_send();
+    // let user_data = UserData::new("jack".to_owned(),
+    // "[email protected]".to_owned());
+    //
+    //
+    // EventTester::new(AuthCheck)
+    //     .bytes_payload(user_data)
+    //     .sync_send();
 }

+ 45 - 23
rust-lib/flowy-sys/src/request/request.rs

@@ -116,34 +116,56 @@ impl<T> ops::DerefMut for Data<T> {
     fn deref_mut(&mut self) -> &mut T { &mut self.0 }
 }
 
+// #[cfg(feature = "use_serde")]
+// impl<T> FromRequest for Data<T>
+// where
+//     T: serde::de::DeserializeOwned + 'static,
+// {
+//     type Error = SystemError;
+//     type Future = Ready<Result<Self, SystemError>>;
+//
+//     #[inline]
+//     fn from_request(req: &EventRequest, payload: &mut Payload) ->
+// Self::Future {         match payload {
+//             Payload::None => ready(Err(unexpected_none_payload(req))),
+//             Payload::Bytes(bytes) => {
+//                 let s = String::from_utf8_lossy(bytes);
+//                 match serde_json::from_str(s.as_ref()) {
+//                     Ok(data) => ready(Ok(Data(data))),
+//                     Err(e) => ready(Err(InternalError::new(format!("{:?}",
+// e)).into())),                 }
+//             },
+//         }
+//     }
+// }
+
+pub trait FromBytes: Sized {
+    fn parse_from_bytes(bytes: &Vec<u8>) -> Result<Self, SystemError>;
+}
+
+#[cfg(not(feature = "use_serde"))]
+impl<T> FromBytes for T
+where
+    // https://stackoverflow.com/questions/62871045/tryfromu8-trait-bound-in-trait
+    T: for<'a> std::convert::TryFrom<&'a Vec<u8>, Error = SystemError>,
+{
+    fn parse_from_bytes(bytes: &Vec<u8>) -> Result<Self, SystemError> { T::try_from(bytes) }
+}
+
 #[cfg(feature = "use_serde")]
-impl<T> FromRequest for Data<T>
+impl<T> FromBytes for T
 where
     T: serde::de::DeserializeOwned + 'static,
 {
-    type Error = SystemError;
-    type Future = Ready<Result<Self, SystemError>>;
-
-    #[inline]
-    fn from_request(req: &EventRequest, payload: &mut Payload) -> Self::Future {
-        match payload {
-            Payload::None => ready(Err(unexpected_none_payload(req))),
-            Payload::Bytes(bytes) => {
-                let s = String::from_utf8_lossy(bytes);
-                match serde_json::from_str(s.as_ref()) {
-                    Ok(data) => ready(Ok(Data(data))),
-                    Err(e) => ready(Err(InternalError::new(format!("{:?}", e)).into())),
-                }
-            },
+    fn parse_from_bytes(bytes: &Vec<u8>) -> Result<Self, SystemError> {
+        let s = String::from_utf8_lossy(bytes);
+        match serde_json::from_str::<T>(s.as_ref()) {
+            Ok(data) => Ok(data),
+            Err(e) => InternalError::new(format!("{:?}", e)).into(),
         }
     }
 }
 
-pub trait FromBytes: Sized {
-    fn parse_from_bytes(bytes: &Vec<u8>) -> Result<Self, SystemError>;
-}
-
-#[cfg(not(feature = "use_serde"))]
 impl<T> FromRequest for Data<T>
 where
     T: FromBytes + 'static,
@@ -155,9 +177,9 @@ where
     fn from_request(req: &EventRequest, payload: &mut Payload) -> Self::Future {
         match payload {
             Payload::None => ready(Err(unexpected_none_payload(req))),
-            Payload::Bytes(bytes) => {
-                let data = T::parse_from_bytes(bytes).unwrap();
-                ready(Ok(Data(data)))
+            Payload::Bytes(bytes) => match T::parse_from_bytes(bytes) {
+                Ok(data) => ready(Ok(Data(data))),
+                Err(e) => ready(Err(InternalError::new(format!("{:?}", e)).into())),
             },
         }
     }

+ 13 - 13
rust-lib/flowy-sys/src/response/responder.rs

@@ -1,11 +1,11 @@
+#[allow(unused_imports)]
+use crate::error::{InternalError, SystemError};
 use crate::{
-    error::SystemError,
     request::{Data, EventRequest},
     response::{EventResponse, ResponseBuilder},
 };
 use bytes::Bytes;
 
-
 pub trait Responder {
     fn respond_to(self, req: &EventRequest) -> EventResponse;
 }
@@ -42,26 +42,27 @@ pub trait ToBytes {
     fn into_bytes(self) -> Result<Vec<u8>, SystemError>;
 }
 
-#[cfg(feature = "use_serde")]
-impl<T> Responder for Data<T>
+#[cfg(not(feature = "use_serde"))]
+impl<T> ToBytes for T
 where
-    T: serde::Serialize,
+    T: std::convert::TryInto<Vec<u8>, Error = SystemError>,
 {
-    fn respond_to(self, _request: &EventRequest) -> EventResponse {
-        let bytes: Vec<u8> = bincode::serialize(&self.0).unwrap();
-        ResponseBuilder::Ok().data(bytes).build()
-    }
+    fn into_bytes(self) -> Result<Vec<u8>, SystemError> { self.try_into() }
 }
 
 #[cfg(feature = "use_serde")]
-impl<T> std::convert::From<T> for Data<T>
+impl<T> ToBytes for T
 where
     T: serde::Serialize,
 {
-    fn from(val: T) -> Self { Data(val) }
+    fn into_bytes(self) -> Result<Vec<u8>, SystemError> {
+        match serde_json::to_string(&self.0) {
+            Ok(s) => Ok(s.into_bytes()),
+            Err(e) => InternalError::new(format!("{:?}", e)).into(),
+        }
+    }
 }
 
-#[cfg(not(feature = "use_serde"))]
 impl<T> Responder for Data<T>
 where
     T: ToBytes,
@@ -74,7 +75,6 @@ where
     }
 }
 
-#[cfg(not(feature = "use_serde"))]
 impl<T> std::convert::From<T> for Data<T>
 where
     T: ToBytes,