cargo fmt

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

Author: iosmanthus

examples/pessimistic.rs

@@ -4,8 +4,7 @@ mod common;
 
 use crate::common::parse_args;
 use tikv_client::{
-    request::codec::TxnApiV1, Config, Key, TransactionClient as Client, TransactionOptions,
-    Value,
+    request::codec::TxnApiV1, Config, Key, TransactionClient as Client, TransactionOptions, Value,
 };
 
 #[tokio::main]

examples/raw.rs

@@ -6,8 +6,8 @@ mod common;
 
 use crate::common::parse_args;
 use tikv_client::{
-    request::codec::RawApiV1, Config, IntoOwnedRange, Key, KvPair, RawClient as Client,
-    Result, Value,
+    request::codec::RawApiV1, Config, IntoOwnedRange, Key, KvPair, RawClient as Client, Result,
+    Value,
 };
 
 const KEY: &str = "TiKV";

src/raw/requests.rs

@@ -11,15 +11,15 @@ use tikv_client_store::Request;
 
 use crate::{
     collect_first,
-    ColumnFamily,
-    Key,
-    KvPair,
     pd::PdClient,
     request::{
-        codec::RequestCodec, Collect, CollectSingle, DefaultProcessor, KvRequest,
-        Merge, plan::ResponseWithShard, Process, Shardable, SingleKey,
+        codec::RequestCodec, plan::ResponseWithShard, Collect, CollectSingle, DefaultProcessor,
+        KvRequest, Merge, Process, Shardable, SingleKey,
     },
-    Result, store::{RegionStore, store_stream_for_keys, store_stream_for_ranges}, transaction::HasLocks, util::iter::FlatMapOkIterExt, Value,
+    store::{store_stream_for_keys, store_stream_for_ranges, RegionStore},
+    transaction::HasLocks,
+    util::iter::FlatMapOkIterExt,
+    ColumnFamily, Key, KvPair, Result, Value,
 };
 
 use super::RawRpcRequest;
@@ -371,7 +371,7 @@ impl Process<kvrpcpb::RawCasResponse> for DefaultProcessor {
 }
 
 type RawCoprocessorRequestDataBuilder =
-Arc<dyn Fn(metapb::Region, Vec<kvrpcpb::KeyRange>) -> Vec<u8> + Send + Sync>;
+    Arc<dyn Fn(metapb::Region, Vec<kvrpcpb::KeyRange>) -> Vec<u8> + Send + Sync>;
 
 pub fn new_raw_coprocessor_request(
     copr_name: String,
@@ -441,8 +441,8 @@ impl Shardable for RawCoprocessorRequest {
 
 #[allow(clippy::type_complexity)]
 impl
-Process<Vec<Result<ResponseWithShard<kvrpcpb::RawCoprocessorResponse, Vec<kvrpcpb::KeyRange>>>>>
-for DefaultProcessor
+    Process<Vec<Result<ResponseWithShard<kvrpcpb::RawCoprocessorResponse, Vec<kvrpcpb::KeyRange>>>>>
+    for DefaultProcessor
 {
     type Out = Vec<(Vec<u8>, Vec<Range<Key>>)>;
 
@@ -522,9 +522,9 @@ mod test {
 
     use crate::{
         backoff::{DEFAULT_REGION_BACKOFF, OPTIMISTIC_BACKOFF},
-        Key,
         mock::{MockKvClient, MockPdClient},
         request::Plan,
+        Key,
     };
 
     use super::*;

src/request/codec.rs

@@ -1,8 +1,7 @@
 use tikv_client_common::Error;
-use tikv_client_proto::kvrpcpb;
-use tikv_client_proto::metapb::Region;
+use tikv_client_proto::{kvrpcpb, metapb::Region};
 
-use crate::{Key, kv::codec::decode_bytes_in_place, Result};
+use crate::{kv::codec::decode_bytes_in_place, Key, Result};
 
 const RAW_MODE_PREFIX: u8 = b'r';
 const TXN_MODE_PREFIX: u8 = b'x';
@@ -26,7 +25,9 @@ pub trait RequestCodec: Sized + Clone + Sync + Send + 'static {
         keys
     }
 
-    fn encode_pairs(&self, pairs: Vec<kvrpcpb::KvPair>) -> Vec<kvrpcpb::KvPair> { pairs }
+    fn encode_pairs(&self, pairs: Vec<kvrpcpb::KvPair>) -> Vec<kvrpcpb::KvPair> {
+        pairs
+    }
 
     fn decode_key(&self, key: Vec<u8>) -> Result<Vec<u8>> {
         Ok(key)
@@ -40,7 +41,9 @@ pub trait RequestCodec: Sized + Clone + Sync + Send + 'static {
         (start, end)
     }
 
-    fn encode_ranges(&self, ranges: Vec<kvrpcpb::KeyRange>) -> Vec<kvrpcpb::KeyRange> { ranges }
+    fn encode_ranges(&self, ranges: Vec<kvrpcpb::KeyRange>) -> Vec<kvrpcpb::KeyRange> {
+        ranges
+    }
 
     fn encode_pd_query(&self, key: Vec<u8>) -> Vec<u8> {
         key
@@ -204,16 +207,14 @@ impl RequestCodec for KeySpaceCodec {
         if region.get_start_key() < self.mode.min_key().as_slice() {
             *region.mut_start_key() = vec![];
         } else {
-            *region.mut_start_key() = self
-                .decode_key(region.get_start_key().to_vec())?;
+            *region.mut_start_key() = self.decode_key(region.get_start_key().to_vec())?;
         }
 
         // Map the region's end key to the keyspace's end key.
         if region.get_end_key() > self.mode.max_key().as_slice() {
             *region.mut_end_key() = vec![];
         } else {
-            *region.mut_end_key() = self
-                .decode_key(region.get_end_key().to_vec())?;
+            *region.mut_end_key() = self.decode_key(region.get_end_key().to_vec())?;
         }
 
         Ok(region)

src/request/mod.rs

@@ -11,7 +11,6 @@ use crate::{
     backoff::{Backoff, DEFAULT_REGION_BACKOFF, OPTIMISTIC_BACKOFF, PESSIMISTIC_BACKOFF},
     transaction::HasLocks,
 };
-use crate::request::codec::RequestCodec;
 
 pub use self::{
     plan::{
@@ -56,6 +55,7 @@ macro_rules! impl_kv_request_for_single_key_op {
                     return Cow::Borrowed(self);
                 }
                 let mut req = self.clone();
+
                 *req.mut_key() = codec.encode_key(req.take_key());
 
                 Cow::Owned(req)

src/transaction/requests.rs

@@ -14,8 +14,8 @@ use crate::{
     collect_first,
     pd::PdClient,
     request::{
-        codec::RequestCodec, Collect, CollectSingle, CollectWithShard, DefaultProcessor,
-        KvRequest, Merge, Process, ResponseWithShard, Shardable, SingleKey,
+        codec::RequestCodec, Collect, CollectSingle, CollectWithShard, DefaultProcessor, KvRequest,
+        Merge, Process, ResponseWithShard, Shardable, SingleKey,
     },
     store::{store_stream_for_keys, store_stream_for_range_by_start_key, RegionStore},
     timestamp::TimestampExt,
@@ -278,7 +278,10 @@ pub fn new_batch_rollback_request(
     req
 }
 
-impl_kv_request_for_single_key_op!(kvrpcpb::BatchRollbackRequest, kvrpcpb::BatchRollbackResponse);
+impl_kv_request_for_single_key_op!(
+    kvrpcpb::BatchRollbackRequest,
+    kvrpcpb::BatchRollbackResponse
+);
 shardable_keys!(kvrpcpb::BatchRollbackRequest);
 
 pub fn new_pessimistic_rollback_request(
@@ -294,7 +297,10 @@ pub fn new_pessimistic_rollback_request(
     req
 }
 
-impl_kv_request_for_single_key_op!(kvrpcpb::PessimisticRollbackRequest, kvrpcpb::PessimisticRollbackResponse);
+impl_kv_request_for_single_key_op!(
+    kvrpcpb::PessimisticRollbackRequest,
+    kvrpcpb::PessimisticRollbackResponse
+);
 shardable_keys!(kvrpcpb::PessimisticRollbackRequest);
 
 pub fn new_pessimistic_lock_request(
@@ -322,7 +328,10 @@ pub fn new_pessimistic_lock_request(
     req
 }
 
-impl_kv_request_for_single_key_op!(kvrpcpb::PessimisticLockRequest, kvrpcpb::PessimisticLockResponse);
+impl_kv_request_for_single_key_op!(
+    kvrpcpb::PessimisticLockRequest,
+    kvrpcpb::PessimisticLockResponse
+);
 
 impl Shardable for kvrpcpb::PessimisticLockRequest {
     type Shard = Vec<kvrpcpb::Mutation>;
@@ -496,7 +505,10 @@ impl Process<kvrpcpb::TxnHeartBeatResponse> for DefaultProcessor {
     }
 }
 
-impl_kv_request_for_single_key_op!(kvrpcpb::CheckTxnStatusRequest, kvrpcpb::CheckTxnStatusResponse);
+impl_kv_request_for_single_key_op!(
+    kvrpcpb::CheckTxnStatusRequest,
+    kvrpcpb::CheckTxnStatusResponse
+);
 
 impl Shardable for kvrpcpb::CheckTxnStatusRequest {
     type Shard = Vec<Vec<u8>>;
@@ -563,7 +575,10 @@ impl From<(u64, u64, Option<kvrpcpb::LockInfo>)> for TransactionStatusKind {
     }
 }
 
-impl_kv_request_for_single_key_op!(kvrpcpb::CheckSecondaryLocksRequest, kvrpcpb::CheckSecondaryLocksResponse);
+impl_kv_request_for_single_key_op!(
+    kvrpcpb::CheckSecondaryLocksRequest,
+    kvrpcpb::CheckSecondaryLocksResponse
+);
 shardable_keys!(kvrpcpb::CheckSecondaryLocksRequest);
 
 impl Merge<kvrpcpb::CheckSecondaryLocksResponse> for Collect {