Skip to content

Commit c530f2f

Browse files
committed
change use of usize to u64
1 parent 0b1bf3d commit c530f2f

File tree

3 files changed

+30
-36
lines changed

3 files changed

+30
-36
lines changed

ipld/amt/src/amt.rs

+4-4
Original file line numberDiff line numberDiff line change
@@ -441,7 +441,7 @@ where
441441
/// map.set(1, "One".to_owned()).unwrap();
442442
/// map.set(4, "Four".to_owned()).unwrap();
443443
///
444-
/// let mut values: Vec<(usize, String)> = Vec::new();
444+
/// let mut values: Vec<(u64, String)> = Vec::new();
445445
/// map.for_each(|i, v| {
446446
/// values.push((*i, v.clone()));
447447
/// Ok(())
@@ -452,11 +452,11 @@ where
452452
#[deprecated = "use `.iter()` instead"]
453453
pub fn for_each<F>(&self, mut f: F) -> Result<(), Error>
454454
where
455-
F: FnMut(&usize, &V) -> anyhow::Result<()>,
455+
F: FnMut(u64, &V) -> anyhow::Result<()>,
456456
{
457457
for res in self {
458458
let (k, v) = res?;
459-
(f)(&k, v)?;
459+
(f)(k, v)?;
460460
}
461461
Ok(())
462462
}
@@ -465,7 +465,7 @@ where
465465
/// function keeps returning `true`.
466466
pub fn for_each_while<F>(&self, mut f: F) -> Result<(), Error>
467467
where
468-
F: FnMut(usize, &V) -> anyhow::Result<bool>,
468+
F: FnMut(u64, &V) -> anyhow::Result<bool>,
469469
{
470470
for res in self.iter() {
471471
let (i, v) = res?;

ipld/amt/src/iter.rs

+24-30
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ where
2626
/// let store = MemoryBlockstore::default();
2727
///
2828
/// let mut amt = Amt::new(store);
29-
/// let kvs: Vec<usize> = (0..=5).collect();
29+
/// let kvs: Vec<u64> = (0..=5).collect();
3030
/// kvs
3131
/// .iter()
3232
/// .map(|k| amt.set(u64::try_from(*k).unwrap(), k.to_string()))
@@ -101,7 +101,7 @@ where
101101
///
102102
/// # anyhow::Ok(())
103103
/// ```
104-
pub fn keys(&self) -> impl Iterator<Item = usize> + '_ {
104+
pub fn keys(&self) -> impl Iterator<Item = u64> + '_ {
105105
self.iter().map(|res| {
106106
res.map(|(k, _)| k)
107107
.expect("Failed to generate iterator from AMT keys")
@@ -119,7 +119,7 @@ where
119119
///
120120
/// // Create an AMT with 5 keys.
121121
/// let mut amt = Amt::new(store);
122-
/// let kvs: Vec<usize> = (0..=5).collect();
122+
/// let kvs: Vec<u64> = (0..=5).collect();
123123
///
124124
/// let _ = kvs
125125
/// .iter()
@@ -152,14 +152,14 @@ where
152152
///
153153
/// # anyhow::Ok(())
154154
/// ```
155-
pub fn iter_from(&self, key: usize) -> Result<Iter<'_, V, &BS, Ver>, crate::Error> {
155+
pub fn iter_from(&self, key: u64) -> Result<Iter<'_, V, &BS, Ver>, crate::Error> {
156156
let mut iter = self.iter();
157157
while key > iter.key {
158158
let stack = iter.stack.last_mut().expect("Stack is empty");
159159
match stack.node {
160160
Some(Node::Leaf { vals }) => {
161-
while stack.idx < vals.len() {
162-
match vals[stack.idx] {
161+
while (stack.idx as usize) < vals.len() {
162+
match vals[stack.idx as usize] {
163163
Some(_) => {
164164
stack.idx += 1;
165165
iter.key += 1;
@@ -176,8 +176,8 @@ where
176176
iter.stack.pop();
177177
}
178178
Some(Node::Link { links }) => {
179-
if stack.idx < links.len() {
180-
let link = &links[stack.idx];
179+
if (stack.idx as usize) < links.len() {
180+
let link = &links[stack.idx as usize];
181181
match link {
182182
Some(Link::Cid { cid, cache }) => {
183183
match cache.get_or_try_init(|| {
@@ -206,7 +206,7 @@ where
206206
}
207207
None => {
208208
stack.idx += 1;
209-
iter.key += 2_usize.pow(iter.bit_width);
209+
iter.key += 2_u64.pow(iter.bit_width);
210210
}
211211
};
212212
} else {
@@ -235,7 +235,7 @@ where
235235
BS: Blockstore,
236236
{
237237
type IntoIter = Iter<'a, V, &'a BS, Ver>;
238-
type Item = Result<(usize, &'a V), crate::Error>;
238+
type Item = Result<(u64, &'a V), crate::Error>;
239239
fn into_iter(self) -> Self::IntoIter {
240240
self.iter()
241241
}
@@ -246,27 +246,27 @@ pub struct Iter<'a, V, BS, Ver> {
246246
blockstore: BS,
247247
bit_width: u32,
248248
ver: PhantomData<Ver>,
249-
key: usize,
249+
key: u64,
250250
}
251251

252252
pub struct IterStack<'a, V> {
253253
pub(crate) node: Option<&'a Node<V>>,
254-
pub(crate) idx: usize,
254+
pub(crate) idx: u64,
255255
}
256256

257257
impl<'a, V, BS, Ver> Iterator for Iter<'a, V, BS, Ver>
258258
where
259259
BS: Blockstore,
260260
V: Serialize + DeserializeOwned,
261261
{
262-
type Item = Result<(usize, &'a V), crate::Error>;
262+
type Item = Result<(u64, &'a V), crate::Error>;
263263
fn next(&mut self) -> Option<Self::Item> {
264264
loop {
265265
let stack = self.stack.last_mut()?;
266266
match stack.node {
267267
Some(Node::Leaf { vals }) => {
268-
while stack.idx < vals.len() {
269-
match vals[stack.idx] {
268+
while (stack.idx as usize) < vals.len() {
269+
match vals[stack.idx as usize] {
270270
Some(ref v) => {
271271
stack.idx += 1;
272272
self.key += 1;
@@ -281,8 +281,8 @@ where
281281
self.stack.pop();
282282
}
283283
Some(Node::Link { links }) => {
284-
if stack.idx < links.len() {
285-
let link = &links[stack.idx];
284+
if (stack.idx as usize) < links.len() {
285+
let link = &links[stack.idx as usize];
286286
match link {
287287
Some(Link::Cid { cid, cache }) => {
288288
match cache.get_or_try_init(|| {
@@ -311,7 +311,7 @@ where
311311
}
312312
None => {
313313
stack.idx += 1;
314-
self.key += 2_usize.pow(self.bit_width);
314+
self.key += 2_u64.pow(self.bit_width);
315315
}
316316
};
317317
} else {
@@ -393,14 +393,8 @@ mod tests {
393393
amt.set(8, "foo".to_owned()).unwrap();
394394
amt.set(64, "bar".to_owned()).unwrap();
395395
let mut amt_iter = amt.iter();
396-
assert_eq!(
397-
amt_iter.next().unwrap().unwrap(),
398-
(8_usize, &"foo".to_owned())
399-
);
400-
assert_eq!(
401-
amt_iter.next().unwrap().unwrap(),
402-
(64_usize, &"bar".to_owned())
403-
);
396+
assert_eq!(amt_iter.next().unwrap().unwrap(), (8, &"foo".to_owned()));
397+
assert_eq!(amt_iter.next().unwrap().unwrap(), (64, &"bar".to_owned()));
404398
}
405399

406400
#[test]
@@ -412,7 +406,7 @@ mod tests {
412406
let mut restored = Vec::new();
413407
#[allow(deprecated)]
414408
a.for_each(|k, v| {
415-
restored.push((*k, v.clone()));
409+
restored.push((k as usize, v.clone()));
416410
Ok(())
417411
})
418412
.unwrap();
@@ -451,7 +445,7 @@ mod tests {
451445
#[allow(deprecated)]
452446
new_amt
453447
.for_each(|k, _: &BytesDe| {
454-
if *k as u64 != indexes[x] {
448+
if k as u64 != indexes[x] {
455449
panic!(
456450
"for each found wrong index: expected {} got {}",
457451
indexes[x], k
@@ -494,7 +488,7 @@ mod tests {
494488
amt.set(idx, "foo".to_owned()).unwrap();
495489
assert_eq!(
496490
amt.iter().next().unwrap().unwrap(),
497-
(idx as usize, &"foo".to_owned())
491+
(idx, &"foo".to_owned())
498492
);
499493
}
500494

@@ -533,7 +527,7 @@ mod tests {
533527

534528
// Create an AMT with 5 keys.
535529
let mut amt = Amt::new(store);
536-
let kvs: Vec<usize> = (0..=5).collect();
530+
let kvs: Vec<u64> = (0..=5).collect();
537531
let _ = kvs
538532
.iter()
539533
.map(|k| amt.set(u64::try_from(*k).unwrap(), k.to_string()))

ipld/amt/tests/amt_tests.rs

+2-2
Original file line numberDiff line numberDiff line change
@@ -370,7 +370,7 @@ fn for_each() {
370370
#[allow(deprecated)]
371371
new_amt
372372
.for_each(|k, _: &BytesDe| {
373-
if *k as u64 != indexes[x] {
373+
if k as u64 != indexes[x] {
374374
panic!(
375375
"for each found wrong index: expected {} got {}",
376376
indexes[x], k
@@ -627,7 +627,7 @@ fn new_from_iter() {
627627
let mut restored = Vec::new();
628628
#[allow(deprecated)]
629629
a.for_each(|k, v| {
630-
restored.push((*k, v.clone()));
630+
restored.push((k as usize, v.clone()));
631631
Ok(())
632632
})
633633
.unwrap();

0 commit comments

Comments
 (0)