1
1
mirror of https://github.com/github/semantic.git synced 2024-12-11 08:45:48 +03:00

Index the elements, not the fold.

This commit is contained in:
Rob Rix 2019-10-16 18:15:37 -04:00
parent ce0866dc1b
commit 444f0b306c
No known key found for this signature in database
GPG Key ID: F188A01508EA1CF7

View File

@ -90,18 +90,18 @@ newtype DedupeKey = DedupeKey (T.Text, T.Text) deriving (Eq, Ord)
-- identifiers) are in the list.
-- Action: Combine them into a single Replaced entry.
dedupe :: [(Entry, Declaration)] -> [(Entry, Declaration)]
dedupe = map snd . sortOn fst . Map.elems . snd . foldl' go (0, Map.empty)
dedupe = map snd . sortOn fst . Map.elems . foldl' go Map.empty . zip [0..]
where
go :: (Int, Map.Map DedupeKey (Int, (Entry, Declaration)))
-> (Entry, Declaration)
-> (Int, Map.Map DedupeKey (Int, (Entry, Declaration)))
go (index, m) (entry, decl)
go :: Map.Map DedupeKey (Int, (Entry, Declaration))
-> (Int, (Entry, Declaration))
-> Map.Map DedupeKey (Int, (Entry, Declaration))
go m (index, (entry, decl))
| Just (_, (_, similar)) <- Map.lookup (dedupeKey decl) m
= (succ index,) $ if similar == decl then
= if similar == decl then
m
else
Map.insert (dedupeKey similar) (index, (Replaced, similar)) m
| otherwise = (succ index, Map.insert (dedupeKey decl) (index, (entry, decl)) m)
| otherwise = Map.insert (dedupeKey decl) (index, (entry, decl)) m
dedupeKey (Declaration kind ident _ _ _) = DedupeKey (toCategoryName kind, T.toLower ident)