From f28195e5e952dfda9b584f223dd5ca373fb15fe3 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Wed, 2 Mar 2016 18:28:22 +0000
Subject: Use lookup instead of Map.get

---
 src/main/java/org/zwobble/mammoth/internal/documents/Notes.java | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'src/main/java/org/zwobble/mammoth/internal/documents/Notes.java')

diff --git a/src/main/java/org/zwobble/mammoth/internal/documents/Notes.java b/src/main/java/org/zwobble/mammoth/internal/documents/Notes.java
index d8ca28d..d7de8a6 100644
--- a/src/main/java/org/zwobble/mammoth/internal/documents/Notes.java
+++ b/src/main/java/org/zwobble/mammoth/internal/documents/Notes.java
@@ -8,6 +8,7 @@ import java.util.Map;
 import java.util.Optional;
 
 import static org.zwobble.mammoth.internal.util.MammothLists.list;
+import static org.zwobble.mammoth.internal.util.MammothMaps.lookup;
 
 public class Notes {
     public final static Notes EMPTY = new Notes(list());
@@ -21,7 +22,7 @@ public class Notes {
     }
 
     public Optional<Note> findNote(NoteType noteType, String noteId) {
-        return Optional.ofNullable(notes.get(noteType))
-            .flatMap(notesOfType -> Optional.ofNullable(notesOfType.get(noteId)));
+        return lookup(notes, noteType)
+            .flatMap(notesOfType -> lookup(notesOfType, noteId));
     }
 }
-- 
cgit v1.2.3