perf: stream_chunks_of_combined_source_map (#116)

* perf: stream_chunks_of_combined_source_map

* perf: rm SourceMapLineChunk

* perf: chunks can without WithIndices

src/helpers.rs

@@ -1,7 +1,6 @@
 use std::{
  borrow::{BorrowMut, Cow},
- cell::RefCell,
- rc::Rc,
+ cell::{OnceCell, RefCell},
 };
 
 use rustc_hash::FxHashMap as HashMap;
@@ -17,7 +16,7 @@ use crate::{
 
 // Adding this type because sourceContentLine not happy
 type InnerSourceContentLine<'a> =
- RefCell<HashMap<u32, Option<Rc<Vec<WithIndices<&'a str>>>>>>;
+ RefCell<LinearMap<OnceCell<Option<Vec<WithIndices<&'a str>>>>>>;
 
 pub fn get_map<'a, S: StreamChunks<'a>>(
  stream: &'a S,
@@ -600,7 +599,7 @@ fn stream_chunks_of_source_map_lines_full<'a>(
 #[derive(Debug)]
 struct SourceMapLineData<'a> {
  pub mappings_data: Vec<i64>,
- pub chunks: Vec<WithIndices<Cow<'a, str>>>,
+ pub chunks: Vec<Cow<'a, str>>,
 }
 
 type InnerSourceIndexValueMapping<'a> =
@@ -638,7 +637,7 @@ pub fn stream_chunks_of_combined_source_map<'a>(
  let inner_source_contents: RefCell<LinearMap<Option<&str>>> =
  RefCell::new(LinearMap::default());
  let inner_source_content_lines: InnerSourceContentLine =
- RefCell::new(HashMap::default());
+ RefCell::new(LinearMap::default());
  let inner_name_index_mapping: RefCell<LinearMap<i64>> =
  RefCell::new(LinearMap::default());
  let inner_name_index_value_mapping: RefCell<LinearMap<Cow<str>>> =
@@ -668,6 +667,7 @@ pub fn stream_chunks_of_combined_source_map<'a>(
  }
  Some(l as u32 - 1)
  };
+
  stream_chunks_of_source_map(
  source,
  source_map,
@@ -715,41 +715,38 @@ pub fn stream_chunks_of_combined_source_map<'a>(
  let inner_generated_column = mappings_data[mi];
  let location_in_chunk = original_column - inner_generated_column;
  if location_in_chunk > 0 {
- let mut inner_source_content_lines =
+ let inner_source_content_lines =
  inner_source_content_lines.borrow_mut();
- let mut original_source_lines = inner_source_content_lines
- .get(&inner_source_index)
- .cloned()
- .and_then(|id| id);
- if original_source_lines.is_none() {
- let inner_source_contents = inner_source_contents.borrow();
- original_source_lines = if let Some(Some(original_source)) =
- inner_source_contents.get(&inner_source_index)
- {
- Some(Rc::new(
- split_into_lines(original_source)
- .map(WithIndices::new)
- .collect(),
- ))
- } else {
- None
+ let original_source_lines =
+ match inner_source_content_lines.get(&inner_source_index) {
+ Some(once_cell) => once_cell.get_or_init(|| {
+ let inner_source_contents = inner_source_contents.borrow();
+ match inner_source_contents.get(&inner_source_index) {
+ Some(Some(source_content)) => Some(
+ split_into_lines(source_content)
+ .map(WithIndices::new)
+ .collect(),
+ ),
+ _ => None,
+ }
+ }),
+ None => &None,
  };
- inner_source_content_lines
- .insert(inner_source_index, original_source_lines.clone());
- }
  if let Some(original_source_lines) = original_source_lines {
  let original_chunk = original_source_lines
  .get(inner_original_line as usize - 1)
- .map_or("", |lines| {
+ .map(|lines| {
  let start = inner_original_column as usize;
  let end = start + location_in_chunk as usize;
  lines.substring(start, end)
  });
- if inner_chunk.substring(0, location_in_chunk as usize)
- == original_chunk
- {
- inner_original_column += location_in_chunk;
- inner_name_index = -1;
+ if let Some(original_chunk) = original_chunk {
+ if original_chunk.len() <= inner_chunk.len()
+ && &inner_chunk[..original_chunk.len()] == original_chunk
+ {
+ inner_original_column += location_in_chunk;
+ inner_name_index = -1;
+ }
  }
  }
  }
@@ -817,30 +814,23 @@ pub fn stream_chunks_of_combined_source_map<'a>(
  // when we don't have an inner name,
  // but we have an outer name
  // it can be used when inner original code equals to the name
- let mut inner_source_content_lines =
+ let inner_source_content_lines =
  inner_source_content_lines.borrow_mut();
- let mut original_source_lines = inner_source_content_lines
- .get(&inner_source_index)
- .cloned()
- .and_then(|id| id);
- if original_source_lines.is_none() {
- let inner_source_contents = inner_source_contents.borrow_mut();
- original_source_lines = inner_source_contents
- .get(&inner_source_index)
- .and_then(|original_source| {
- original_source.as_ref().map(|s| {
- let lines = split_into_lines(s);
- Rc::new(
- lines
- .into_iter()
+ let original_source_lines =
+ match inner_source_content_lines.get(&inner_source_index) {
+ Some(once_cell) => once_cell.get_or_init(|| {
+ let inner_source_contents = inner_source_contents.borrow();
+ match inner_source_contents.get(&inner_source_index) {
+ Some(Some(source_content)) => Some(
+ split_into_lines(source_content)
  .map(WithIndices::new)
- .collect::<Vec<_>>(),
- )
- })
- });
- inner_source_content_lines
- .insert(inner_source_index, original_source_lines.clone());
- }
+ .collect(),
+ ),
+  _ => None,
+  }
+  }),
+ None => &None,
+  };
  if let Some(original_source_lines) = original_source_lines {
  let name_index_value_mapping =
  name_index_value_mapping.borrow();
@@ -1049,14 +1039,13 @@ pub fn stream_chunks_of_combined_source_map<'a>(
  .unwrap_or(-1),
  );
  // SAFETY: final_source is false
- let chunk = WithIndices::new(chunk.unwrap());
- data.chunks.push(chunk);
+ data.chunks.push(chunk.unwrap());
  },
  &mut |i, source, source_content| {
- inner_source_contents
+ inner_source_contents.borrow_mut().insert(i, source_content);
+ inner_source_content_lines
  .borrow_mut()
- .insert(i, source_content.map(Into::into));
- inner_source_content_lines.borrow_mut().insert(i, None);
+ .insert(i, Default::default());
  inner_source_index_mapping.borrow_mut().insert(i, -2);
  inner_source_index_value_mapping
  .borrow_mut()