bound Client to right codec

Signed-off-by: iosmanthus <[email protected]>

Author: iosmanthus

src/raw/client.rs

@@ -13,10 +13,7 @@ use crate::{
     config::Config,
     pd::{PdClient, PdRpcClient},
     raw::lowering::*,
-    request::{
-        codec::{RawCodec, RequestCodec},
-        Collect, CollectSingle, Plan,
-    },
+    request::{codec::RawCodec, Collect, CollectSingle, Plan},
     Backoff, BoundRange, ColumnFamily, Key, KvPair, Result, Value,
 };
 
@@ -29,7 +26,7 @@ const MAX_RAW_KV_SCAN_LIMIT: u32 = 10240;
 ///
 /// The returned results of raw request methods are [`Future`](std::future::Future)s that must be
 /// awaited to execute.
-pub struct Client<C, PdC: PdClient = PdRpcClient<C>> {
+pub struct Client<C: RawCodec, PdC: PdClient = PdRpcClient<C>> {
     rpc: Arc<PdC>,
     cf: Option<ColumnFamily>,
     /// Whether to use the [`atomic mode`](Client::with_atomic_for_cas).
@@ -38,7 +35,7 @@ pub struct Client<C, PdC: PdClient = PdRpcClient<C>> {
     _phantom: PhantomData<C>,
 }
 
-impl<C: RequestCodec> Clone for Client<C> {
+impl<C: RawCodec> Clone for Client<C> {
     fn clone(&self) -> Self {
         Self {
             rpc: self.rpc.clone(),

src/request/codec.rs

@@ -359,3 +359,7 @@ impl<M: Mode> RequestCodec for ApiV2<M> {
         req.decode_response(self, resp)
     }
 }
+
+impl RawCodec for ApiV2<RawMode> {}
+
+impl TxnCodec for ApiV2<TxnMode> {}

src/transaction/client.rs

@@ -37,13 +37,13 @@ const SCAN_LOCK_BATCH_SIZE: u32 = 1024;
 ///
 /// The returned results of transactional requests are [`Future`](std::future::Future)s that must be
 /// awaited to execute.
-pub struct Client<C> {
+pub struct Client<C: TxnCodec> {
     pd: Arc<PdRpcClient<C>>,
     logger: Logger,
     _phantom: PhantomData<C>,
 }
 
-impl<C> Clone for Client<C> {
+impl<C: TxnCodec> Clone for Client<C> {
     fn clone(&self) -> Self {
         Self {
             pd: self.pd.clone(),