mirrors/jj
1
0
Fork 1
mirror of https://github.com/martinvonz/jj.git synced 2025-01-12 23:23:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

files: replace Vec + index access with iterator in DiffLineIterator

Browse source
This commit is contained in:
Yuya Nishihara 2024-08-14 14:12:56 +09:00
parent 54f5c01eae
commit f85792288f
1 changed files with 8 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
lib/src/files.rs
View file

@ -16,6 +16,7 @@
use std::collections::VecDeque;
use std::fmt::{Debug, Error, Formatter};
use std::{iter, vec};
use crate::diff;
use crate::diff::{Diff, DiffHunk};
@ -50,8 +51,7 @@ pub fn diff<'a>(left: &'a [u8], right: &'a [u8]) -> DiffLineIterator<'a> {
}
pub struct DiffLineIterator<'a> {
diff_hunks: Vec<DiffHunk<'a>>,
current_pos: usize,
diff_hunks: iter::Fuse<vec::IntoIter<DiffHunk<'a>>>,
current_line: DiffLine<'a>,
queued_lines: VecDeque<DiffLine<'a>>,
}
@ -66,8 +66,7 @@ impl<'a> DiffLineIterator<'a> {
hunks: vec![],
};
DiffLineIterator {
diff_hunks,
current_pos: 0,
diff_hunks: diff_hunks.into_iter().fuse(),
current_line,
queued_lines: VecDeque::new(),
}
@ -80,10 +79,11 @@ impl<'a> Iterator for DiffLineIterator<'a> {
fn next(&mut self) -> Option<Self::Item> {
// TODO: Should we attempt to interpret as utf-8 and otherwise break only at
// newlines?
while self.current_pos < self.diff_hunks.len() && self.queued_lines.is_empty() {
let hunk = &self.diff_hunks[self.current_pos];
self.current_pos += 1;
match hunk {
while self.queued_lines.is_empty() {
let Some(hunk) = self.diff_hunks.next() else {
break;
};
match &hunk {
DiffHunk::Matching(text) => {
let lines = text.split_inclusive(|b| *b == b'\n');
for line in lines {