Skip to content

Commit b578f63

Browse files
committed
Fixup privacy in raw::requests
Signed-off-by: Nick Cameron <nrc@ncameron.org>
1 parent 69aed77 commit b578f63

File tree

3 files changed

+41
-99
lines changed

3 files changed

+41
-99
lines changed

src/raw/client.rs

Lines changed: 11 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,7 @@
11
// Copyright 2019 TiKV Project Authors. Licensed under Apache-2.0.
22

3-
use super::requests::{
4-
BatchDeleteInner, BatchGetInner, BatchPutInner, BatchScanInner, DeleteInner, DeleteRangeInner,
5-
GetInner, PutInner, ScanInner,
6-
};
73
use super::{BatchDelete, BatchGet, BatchPut, BatchScan, Delete, DeleteRange, Get, Put, Scan};
8-
use crate::{rpc::RpcClient, Config, Key, BoundRange, KvPair, Result, Value};
4+
use crate::{rpc::RpcClient, BoundRange, Config, Key, KvPair, Result, Value};
95

106
use futures::prelude::*;
117
use futures::task::{Context, Poll};
@@ -55,7 +51,7 @@ impl Client {
5551
/// # });
5652
/// ```
5753
pub fn get(&self, key: impl Into<Key>) -> Get {
58-
Get::new(self.rpc(), GetInner::new(key.into()))
54+
Get::new(self.rpc(), key.into())
5955
}
6056

6157
/// Create a new [`BatchGet`](BatchGet) request.
@@ -76,10 +72,7 @@ impl Client {
7672
/// # });
7773
/// ```
7874
pub fn batch_get(&self, keys: impl IntoIterator<Item = impl Into<Key>>) -> BatchGet {
79-
BatchGet::new(
80-
self.rpc(),
81-
BatchGetInner::new(keys.into_iter().map(Into::into).collect()),
82-
)
75+
BatchGet::new(self.rpc(), keys.into_iter().map(Into::into).collect())
8376
}
8477

8578
/// Create a new [`Put`](Put) request.
@@ -100,7 +93,7 @@ impl Client {
10093
/// # });
10194
/// ```
10295
pub fn put(&self, key: impl Into<Key>, value: impl Into<Value>) -> Put {
103-
Put::new(self.rpc(), PutInner::new(key.into(), value.into()))
96+
Put::new(self.rpc(), key.into(), value.into())
10497
}
10598

10699
/// Create a new [`BatchPut`](BatchPut) request.
@@ -122,10 +115,7 @@ impl Client {
122115
/// # });
123116
/// ```
124117
pub fn batch_put(&self, pairs: impl IntoIterator<Item = impl Into<KvPair>>) -> BatchPut {
125-
BatchPut::new(
126-
self.rpc(),
127-
BatchPutInner::new(pairs.into_iter().map(Into::into).collect()),
128-
)
118+
BatchPut::new(self.rpc(), pairs.into_iter().map(Into::into).collect())
129119
}
130120

131121
/// Create a new [`Delete`](Delete) request.
@@ -145,7 +135,7 @@ impl Client {
145135
/// # });
146136
/// ```
147137
pub fn delete(&self, key: impl Into<Key>) -> Delete {
148-
Delete::new(self.rpc(), DeleteInner::new(key.into()))
138+
Delete::new(self.rpc(), key.into())
149139
}
150140

151141
/// Create a new [`BatchDelete`](BatchDelete) request.
@@ -165,10 +155,7 @@ impl Client {
165155
/// # });
166156
/// ```
167157
pub fn batch_delete(&self, keys: impl IntoIterator<Item = impl Into<Key>>) -> BatchDelete {
168-
BatchDelete::new(
169-
self.rpc(),
170-
BatchDeleteInner::new(keys.into_iter().map(Into::into).collect()),
171-
)
158+
BatchDelete::new(self.rpc(), keys.into_iter().map(Into::into).collect())
172159
}
173160

174161
/// Create a new [`Scan`](Scan) request.
@@ -188,7 +175,7 @@ impl Client {
188175
/// # });
189176
/// ```
190177
pub fn scan(&self, range: impl Into<BoundRange>, limit: u32) -> Scan {
191-
Scan::new(self.rpc(), ScanInner::new(range.into(), limit))
178+
Scan::new(self.rpc(), range.into(), limit)
192179
}
193180

194181
/// Create a new [`BatchScan`](BatchScan) request.
@@ -216,10 +203,8 @@ impl Client {
216203
) -> BatchScan {
217204
BatchScan::new(
218205
self.rpc(),
219-
BatchScanInner::new(
220-
ranges.into_iter().map(KeyRange::into).collect(),
221-
each_limit,
222-
),
206+
ranges.into_iter().map(Into::into).collect(),
207+
each_limit,
223208
)
224209
}
225210

@@ -240,7 +225,7 @@ impl Client {
240225
/// # });
241226
/// ```
242227
pub fn delete_range(&self, range: impl Into<BoundRange>) -> DeleteRange {
243-
DeleteRange::new(self.rpc(), DeleteRangeInner::new(range.into()))
228+
DeleteRange::new(self.rpc(), range.into())
244229
}
245230
}
246231

src/raw/mod.rs

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@ pub use self::requests::{
1515
BatchDelete, BatchGet, BatchPut, BatchScan, Delete, DeleteRange, Get, Put, Scan,
1616
};
1717

18-
1918
use std::fmt;
2019

2120
mod client;

src/raw/requests.rs

Lines changed: 30 additions & 72 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
11
// Copyright 2019 TiKV Project Authors. Licensed under Apache-2.0.
22

33
use super::ColumnFamily;
4-
use crate::{rpc::RpcClient, Error, Key, BoundRange, KvPair, Result, Value};
4+
use crate::{rpc::RpcClient, BoundRange, Error, Key, KvPair, Result, Value};
55
use futures::prelude::*;
66
use futures::task::{Context, Poll};
7-
use std::{ops::Bound, pin::Pin, sync::Arc, u32};
7+
use std::{pin::Pin, sync::Arc, u32};
88

99
const MAX_RAW_KV_SCAN_LIMIT: u32 = 10240;
1010

@@ -82,9 +82,9 @@ pub struct Get {
8282
}
8383

8484
impl Get {
85-
pub fn new(client: Arc<RpcClient>, inner: GetInner) -> Self {
85+
pub fn new(client: Arc<RpcClient>, key: Key) -> Self {
8686
Self {
87-
state: RequestState::new(client, inner),
87+
state: RequestState::new(client, GetInner { key }),
8888
}
8989
}
9090

@@ -103,16 +103,10 @@ impl Future for Get {
103103
}
104104
}
105105

106-
pub struct GetInner {
106+
struct GetInner {
107107
key: Key,
108108
}
109109

110-
impl GetInner {
111-
pub fn new(key: Key) -> Self {
112-
GetInner { key }
113-
}
114-
}
115-
116110
impl RequestInner for GetInner {
117111
type Resp = Option<Value>;
118112

@@ -134,9 +128,9 @@ pub struct BatchGet {
134128
}
135129

136130
impl BatchGet {
137-
pub fn new(client: Arc<RpcClient>, inner: BatchGetInner) -> Self {
131+
pub fn new(client: Arc<RpcClient>, keys: Vec<Key>) -> Self {
138132
Self {
139-
state: RequestState::new(client, inner),
133+
state: RequestState::new(client, BatchGetInner { keys }),
140134
}
141135
}
142136

@@ -155,7 +149,7 @@ impl Future for BatchGet {
155149
}
156150
}
157151

158-
pub struct BatchGetInner {
152+
struct BatchGetInner {
159153
keys: Vec<Key>,
160154
}
161155

@@ -171,12 +165,6 @@ impl RequestInner for BatchGetInner {
171165
}
172166
}
173167

174-
impl BatchGetInner {
175-
pub fn new(keys: Vec<Key>) -> Self {
176-
BatchGetInner { keys }
177-
}
178-
}
179-
180168
/// An unresolved [`Client::put`](Client::put) request.
181169
///
182170
/// Once resolved this request will result in the putting of the value associated with the given
@@ -186,9 +174,9 @@ pub struct Put {
186174
}
187175

188176
impl Put {
189-
pub fn new(client: Arc<RpcClient>, inner: PutInner) -> Self {
177+
pub fn new(client: Arc<RpcClient>, key: Key, value: Value) -> Self {
190178
Self {
191-
state: RequestState::new(client, inner),
179+
state: RequestState::new(client, PutInner { key, value }),
192180
}
193181
}
194182

@@ -207,17 +195,11 @@ impl Future for Put {
207195
}
208196
}
209197

210-
pub struct PutInner {
198+
struct PutInner {
211199
key: Key,
212200
value: Value,
213201
}
214202

215-
impl PutInner {
216-
pub fn new(key: Key, value: Value) -> Self {
217-
PutInner { key, value }
218-
}
219-
}
220-
221203
impl RequestInner for PutInner {
222204
type Resp = ();
223205

@@ -235,9 +217,9 @@ pub struct BatchPut {
235217
}
236218

237219
impl BatchPut {
238-
pub fn new(client: Arc<RpcClient>, inner: BatchPutInner) -> Self {
220+
pub fn new(client: Arc<RpcClient>, pairs: Vec<KvPair>) -> Self {
239221
Self {
240-
state: RequestState::new(client, inner),
222+
state: RequestState::new(client, BatchPutInner { pairs }),
241223
}
242224
}
243225

@@ -256,16 +238,10 @@ impl Future for BatchPut {
256238
}
257239
}
258240

259-
pub struct BatchPutInner {
241+
struct BatchPutInner {
260242
pairs: Vec<KvPair>,
261243
}
262244

263-
impl BatchPutInner {
264-
pub fn new(pairs: Vec<KvPair>) -> Self {
265-
BatchPutInner { pairs }
266-
}
267-
}
268-
269245
impl RequestInner for BatchPutInner {
270246
type Resp = ();
271247

@@ -282,9 +258,9 @@ pub struct Delete {
282258
}
283259

284260
impl Delete {
285-
pub fn new(client: Arc<RpcClient>, inner: DeleteInner) -> Self {
261+
pub fn new(client: Arc<RpcClient>, key: Key) -> Self {
286262
Self {
287-
state: RequestState::new(client, inner),
263+
state: RequestState::new(client, DeleteInner { key }),
288264
}
289265
}
290266

@@ -303,16 +279,10 @@ impl Future for Delete {
303279
}
304280
}
305281

306-
pub struct DeleteInner {
282+
struct DeleteInner {
307283
key: Key,
308284
}
309285

310-
impl DeleteInner {
311-
pub fn new(key: Key) -> Self {
312-
DeleteInner { key }
313-
}
314-
}
315-
316286
impl RequestInner for DeleteInner {
317287
type Resp = ();
318288

@@ -329,9 +299,9 @@ pub struct BatchDelete {
329299
}
330300

331301
impl BatchDelete {
332-
pub fn new(client: Arc<RpcClient>, inner: BatchDeleteInner) -> Self {
302+
pub fn new(client: Arc<RpcClient>, keys: Vec<Key>) -> Self {
333303
Self {
334-
state: RequestState::new(client, inner),
304+
state: RequestState::new(client, BatchDeleteInner { keys }),
335305
}
336306
}
337307

@@ -350,16 +320,10 @@ impl Future for BatchDelete {
350320
}
351321
}
352322

353-
pub struct BatchDeleteInner {
323+
struct BatchDeleteInner {
354324
keys: Vec<Key>,
355325
}
356326

357-
impl BatchDeleteInner {
358-
pub fn new(keys: Vec<Key>) -> Self {
359-
BatchDeleteInner { keys }
360-
}
361-
}
362-
363327
impl RequestInner for BatchDeleteInner {
364328
type Resp = ();
365329

@@ -376,9 +340,9 @@ pub struct Scan {
376340
}
377341

378342
impl Scan {
379-
pub fn new(client: Arc<RpcClient>, inner: ScanInner) -> Self {
343+
pub fn new(client: Arc<RpcClient>, range: BoundRange, limit: u32) -> Self {
380344
Scan {
381-
state: RequestState::new(client, inner),
345+
state: RequestState::new(client, ScanInner::new(range, limit)),
382346
}
383347
}
384348

@@ -404,14 +368,14 @@ impl Future for Scan {
404368
}
405369
}
406370

407-
pub struct ScanInner {
371+
struct ScanInner {
408372
range: BoundRange,
409373
limit: u32,
410374
key_only: bool,
411375
}
412376

413377
impl ScanInner {
414-
pub fn new(range: BoundRange, limit: u32) -> Self {
378+
fn new(range: BoundRange, limit: u32) -> Self {
415379
ScanInner {
416380
range,
417381
limit,
@@ -447,9 +411,9 @@ pub struct BatchScan {
447411
}
448412

449413
impl BatchScan {
450-
pub fn new(client: Arc<RpcClient>, inner: BatchScanInner) -> Self {
414+
pub fn new(client: Arc<RpcClient>, ranges: Vec<BoundRange>, each_limit: u32) -> Self {
451415
BatchScan {
452-
state: RequestState::new(client, inner),
416+
state: RequestState::new(client, BatchScanInner::new(ranges, each_limit)),
453417
}
454418
}
455419

@@ -475,14 +439,14 @@ impl Future for BatchScan {
475439
}
476440
}
477441

478-
pub struct BatchScanInner {
442+
struct BatchScanInner {
479443
ranges: Vec<BoundRange>,
480444
each_limit: u32,
481445
key_only: bool,
482446
}
483447

484448
impl BatchScanInner {
485-
pub fn new(ranges: Vec<BoundRange>, each_limit: u32) -> Self {
449+
fn new(ranges: Vec<BoundRange>, each_limit: u32) -> Self {
486450
BatchScanInner {
487451
ranges,
488452
each_limit,
@@ -519,9 +483,9 @@ pub struct DeleteRange {
519483
}
520484

521485
impl DeleteRange {
522-
pub fn new(client: Arc<RpcClient>, inner: DeleteRangeInner) -> Self {
486+
pub fn new(client: Arc<RpcClient>, range: BoundRange) -> Self {
523487
Self {
524-
state: RequestState::new(client, inner),
488+
state: RequestState::new(client, DeleteRangeInner { range }),
525489
}
526490
}
527491

@@ -544,12 +508,6 @@ pub struct DeleteRangeInner {
544508
range: BoundRange,
545509
}
546510

547-
impl DeleteRangeInner {
548-
pub fn new(range: BoundRange) -> Self {
549-
DeleteRangeInner { range }
550-
}
551-
}
552-
553511
impl RequestInner for DeleteRangeInner {
554512
type Resp = ();
555513

0 commit comments

Comments
 (0)