feat: add protocol implementation for shmipc

cloudwego · Sep 24, 2024 · fa78415 · fa78415
1 parent bf5c817
commit fa78415
Show file tree

Hide file tree

Showing 6 changed files with 1,549 additions and 428 deletions.
diff --git a/pilota/src/thrift/binary.rs b/pilota/src/thrift/binary.rs
@@ -559,189 +559,6 @@ impl TOutputProtocol for TBinaryProtocol<&mut LinkedBytes> {
     }
 }
 
-impl<'a> TOutputProtocol for TBinaryProtocol<&'a mut [u8]> {
-    type BufMut = &'a mut [u8];
-
-    #[inline]
-    fn write_message_begin(
-        &mut self,
-        identifier: &TMessageIdentifier,
-    ) -> Result<(), ThriftException> {
-        let msg_type_u8: u8 = identifier.message_type.into();
-        let version = (VERSION_1 | msg_type_u8 as u32) as i32;
-        self.write_i32(version)?;
-        self.write_faststr(identifier.name.clone())?;
-        self.write_i32(identifier.sequence_number)?;
-        Ok(())
-    }
-
-    #[inline]
-    fn write_message_end(&mut self) -> Result<(), ThriftException> {
-        Ok(())
-    }
-
-    #[inline]
-    fn write_struct_begin(&mut self, _: &TStructIdentifier) -> Result<(), ThriftException> {
-        Ok(())
-    }
-
-    #[inline]
-    fn write_struct_end(&mut self) -> Result<(), ThriftException> {
-        Ok(())
-    }
-
-    #[inline]
-    fn write_field_begin(&mut self, field_type: TType, id: i16) -> Result<(), ThriftException> {
-        let mut data: [u8; 3] = [0; 3];
-        data[0] = field_type as u8;
-        let id = id.to_be_bytes();
-        data[1] = id[0];
-        data[2] = id[1];
-        self.trans.write_slice(&data);
-        Ok(())
-    }
-
-    #[inline]
-    fn write_field_end(&mut self) -> Result<(), ThriftException> {
-        Ok(())
-    }
-
-    #[inline]
-    fn write_field_stop(&mut self) -> Result<(), ThriftException> {
-        self.write_byte(TType::Stop as u8)
-    }
-
-    #[inline]
-    fn write_bool(&mut self, b: bool) -> Result<(), ThriftException> {
-        if b {
-            self.write_i8(1)
-        } else {
-            self.write_i8(0)
-        }
-    }
-
-    #[inline]
-    fn write_bytes(&mut self, b: Bytes) -> Result<(), ThriftException> {
-        self.write_i32(b.len() as i32)?;
-        self.write_bytes_without_len(b)
-    }
-
-    #[inline]
-    fn write_bytes_without_len(&mut self, b: Bytes) -> Result<(), ThriftException> {
-        self.trans.write_slice(&b);
-        Ok(())
-    }
-
-    #[inline]
-    fn write_byte(&mut self, b: u8) -> Result<(), ThriftException> {
-        self.trans.write_u8(b);
-        Ok(())
-    }
-
-    #[inline]
-    fn write_uuid(&mut self, u: [u8; 16]) -> Result<(), ThriftException> {
-        self.trans.write_slice(&u);
-        Ok(())
-    }
-
-    #[inline]
-    fn write_i8(&mut self, i: i8) -> Result<(), ThriftException> {
-        self.trans.write_i8(i);
-        Ok(())
-    }
-
-    #[inline]
-    fn write_i16(&mut self, i: i16) -> Result<(), ThriftException> {
-        self.trans.write_i16(i);
-        Ok(())
-    }
-
-    #[inline]
-    fn write_i32(&mut self, i: i32) -> Result<(), ThriftException> {
-        self.trans.write_i32(i);
-        Ok(())
-    }
-
-    #[inline]
-    fn write_i64(&mut self, i: i64) -> Result<(), ThriftException> {
-        self.trans.write_i64(i);
-        Ok(())
-    }
-
-    #[inline]
-    fn write_double(&mut self, d: f64) -> Result<(), ThriftException> {
-        self.trans.write_f64(d);
-        Ok(())
-    }
-
-    #[inline]
-    fn write_string(&mut self, s: &str) -> Result<(), ThriftException> {
-        self.write_i32(s.len() as i32)?;
-        self.trans.write_slice(s.as_bytes());
-        Ok(())
-    }
-
-    #[inline]
-    fn write_faststr(&mut self, s: FastStr) -> Result<(), ThriftException> {
-        self.write_i32(s.len() as i32)?;
-        self.trans.write_slice(s.as_ref());
-        Ok(())
-    }
-
-    #[inline]
-    fn write_list_begin(&mut self, identifier: TListIdentifier) -> Result<(), ThriftException> {
-        self.write_byte(identifier.element_type.into())?;
-        self.write_i32(identifier.size as i32)
-    }
-
-    #[inline]
-    fn write_list_end(&mut self) -> Result<(), ThriftException> {
-        Ok(())
-    }
-
-    #[inline]
-    fn write_set_begin(&mut self, identifier: TSetIdentifier) -> Result<(), ThriftException> {
-        self.write_byte(identifier.element_type.into())?;
-        self.write_i32(identifier.size as i32)
-    }
-
-    #[inline]
-    fn write_set_end(&mut self) -> Result<(), ThriftException> {
-        Ok(())
-    }
-
-    #[inline]
-    fn write_map_begin(&mut self, identifier: TMapIdentifier) -> Result<(), ThriftException> {
-        let key_type = identifier.key_type;
-        self.write_byte(key_type.into())?;
-        let val_type = identifier.value_type;
-        self.write_byte(val_type.into())?;
-        self.write_i32(identifier.size as i32)
-    }
-
-    #[inline]
-    fn write_map_end(&mut self) -> Result<(), ThriftException> {
-        Ok(())
-    }
-
-    #[inline]
-    fn flush(&mut self) -> Result<(), ThriftException> {
-        Ok(())
-    }
-
-    #[inline]
-    fn write_bytes_vec(&mut self, b: &[u8]) -> Result<(), ThriftException> {
-        self.write_i32(b.len() as i32)?;
-        self.trans.write_slice(b);
-        Ok(())
-    }
-
-    #[inline]
-    fn buf_mut(&mut self) -> &mut Self::BufMut {
-        &mut self.trans
-    }
-}
-
 impl TInputProtocol for TBinaryProtocol<&mut Bytes> {
     type Buf = Bytes;