mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-16 18:42:30 +03:00
e43d0273e0
Co-authored-by: Antoine Leblanc <1618949+nicuveo@users.noreply.github.com> GitOrigin-RevId: 6905d5914c8a698445c0ef03d6a8303747701e1c
37 lines
1.2 KiB
Haskell
37 lines
1.2 KiB
Haskell
module Data.List.Extended
|
|
( duplicates
|
|
, uniques
|
|
, getDifference
|
|
, getDifferenceOn
|
|
, getOverlapWith
|
|
, module L
|
|
) where
|
|
|
|
import Data.Function (on)
|
|
import Data.Hashable (Hashable)
|
|
import Prelude
|
|
|
|
import qualified Data.HashMap.Strict.Extended as Map
|
|
import qualified Data.HashSet as Set
|
|
import qualified Data.List as L
|
|
import qualified Data.List.NonEmpty as NE
|
|
|
|
duplicates :: (Eq a, Hashable a) => [a] -> Set.HashSet a
|
|
duplicates =
|
|
Set.fromList . Map.keys . Map.filter (> 1) . Map.fromListWith (+) . map (,1::Int)
|
|
|
|
uniques :: Eq a => [a] -> [a]
|
|
uniques = map NE.head . NE.group
|
|
|
|
getDifference :: (Eq a, Hashable a) => [a] -> [a] -> Set.HashSet a
|
|
getDifference = Set.difference `on` Set.fromList
|
|
|
|
getDifferenceOn :: (Eq k, Hashable k) => (v -> k) -> [v] -> [v] -> [v]
|
|
getDifferenceOn f l = Map.elems . Map.differenceOn f l
|
|
|
|
getOverlapWith :: (Eq k, Hashable k) => (v -> k) -> [v] -> [v] -> [(v, v)]
|
|
getOverlapWith getKey left right =
|
|
Map.elems $ Map.intersectionWith (,) (mkMap left) (mkMap right)
|
|
where
|
|
mkMap = Map.fromList . map (\v -> (getKey v, v))
|