Cleaner code using IndexMut instead of get_mut

yay nice
bvibber · Jul 28, 2020 · 8dd07e8 · 8dd07e8
1 parent bdebcac
commit 8dd07e8
Showing 1 changed file with 6 additions and 13 deletions.
diff --git a/src/encoder.rs b/src/encoder.rs
@@ -476,21 +476,14 @@ impl<T> ChunkMap<T> {
             panic!("Tried to land a future chunk");
         }
         self.running -= 1;
-        let len = self.chunks.len();
         let offset = index - self.cursor_out;
-        if len < offset {
-            let n = offset - len;
-            for _ in 0..n {
-                self.chunks.push_back(None);
-            }
+        while offset > self.chunks.len() {
+            self.chunks.push_back(None);
         }
-        match self.chunks.get_mut(offset) {
-            Some(mutref) => {
-                *mutref = Some(chunk);
-            },
-            None => {
-                self.chunks.push_back(Some(chunk));
-            },
+        if offset == self.chunks.len() {
+            self.chunks.push_back(Some(chunk));
+        } else {
+            self.chunks[offset] = Some(chunk);
         }
     }