rust - 如何在 Rust 中正确使用 `peek()`?
问题描述
我正在尝试做一些简单的事情。在切片中u8
,我想找到两个字符的出现"\r\n"
。但是,我无法将该切片转换为String
使用from_utf8
,因为切片之后的部分"\r\n"
可能不是 utf-8 并且我尽可能不想使用from_utf8_unchecked
. 所以我尝试了类似以下的方法。
fn find_crlf(text: &[u8]) -> Option<usize> {
let mut textiter = text.iter().peekable();
for (idx, &elem) in textiter.enumerate() {
if Some(&elem) == Some(&b'\r') {
if textiter.peek() == Some(&&b'\n') {
return Some(idx);
}
}
}
None
}
我得到以下编译错误,这是可以理解的。但是我不太确定如何去做。如果是a str
,那简直就是.find("\r\n")
。
编译错误->
error[E0382]: borrow of moved value: `textiter`
--> src/lib.rs:6:16
|
2 | let mut textiter = text.iter().peekable();
| ------------ move occurs because `textiter` has type `std::iter::Peekable<std::slice::Iter<'_, u8>>`, which does not implement the `Copy` trait
3 |
4 | for (idx, &elem) in textiter.enumerate() {
| -------- value moved here
5 | if Some(&elem) == Some(&b'\r') {
6 | if textiter.peek() == Some(&&b'\n') {
| ^^^^^^^^ value borrowed here after move
可能是我错过了一些非常简单的东西,但现在坚持了很长一段时间。
解决方案
通常,编写此类代码的最佳方法是根本不使用Peekable
。这是一个使用起来很棘手的 API,因为您经常想peek
在迭代过程中调用它,这通常意味着您已经可变地借用了迭代器,因此您不能再次借用它。
但是,由于您特别询问Peekable
,您可以重写代码以next
在循环中显式调用,这通常是使用的唯一方法peek
:
fn find_crlf(text: &[u8]) -> Option<usize> {
let mut textiter = text.iter().enumerate().peekable();
while let Some((idx, &elem)) = textiter.next() {
if Some(&elem) == Some(&b'\r') {
if let Some((_, &b'\n')) = textiter.peek() {
return Some(idx);
}
}
}
None
}
通常,更好的前瞻方法是使用slice::windows
or tuple_windows
from itertools
。
鉴于您的输入是一个切片,您可以使用slice::windows
:
fn find_crlf(text: &[u8]) -> Option<usize> {
for (idx, window) in text.windows(2).enumerate() {
if window[0] == b'\r' && window[1] == b'\n' {
return Some(idx);
}
}
None
}
不过总的来说,我更喜欢该itertools
方法的语法,因为您可以在元组上进行模式匹配,这比索引切片更干净:
use itertools::Itertools; // 0.9.0
fn find_crlf(text: &[u8]) -> Option<usize> {
for (idx, (&elem, &next)) in text.iter().tuple_windows().enumerate() {
if elem == b'\r' && next == b'\n' {
return Some(idx);
}
}
None
}
或者,甚至更好:
use itertools::Itertools; // 0.9.0
fn find_crlf(text: &[u8]) -> Option<usize> {
text.iter()
.tuple_windows()
.position(|(elem, next)| elem == &b'\r' && next == &b'\n')
}
推荐阅读
- python - Scipy错误不等式约束不兼容(退出模式4)
- vue.js - Vuex - 在 created() 中调度的操作在重新加载窗口之前不会运行?
- prometheus - Istio - 如何添加自定义的基于标签的指标?
- go - 如何使用 golang 客户端从 Kubernetes 中驱逐或删除 Pod
- javascript - 修复 FlatList 的位置
- javafx - 最大化时 JavaFX 节点的边界未更新
- javascript - Discord.js v12 怎么拿大家提到的?
- java - java - 如何在java中创建一个文件,而不会在文件名中添加随机数并在使用后删除?
- php - 如何解决 SQLSTATE [42S22]: Column not found: 1054 Unknown column 'id' in 'where 子句'
- hystrix - Hystrix 断路器不打开电路