You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
30 lines
666 B
30 lines
666 B
3 years ago
|
(*
|
||
|
|
* Copyright (c) Facebook, Inc. and its affiliates.
|
||
|
|
*
|
||
|
|
* This source code is licensed under the MIT license found in the
|
||
|
|
* LICENSE file in the root directory of this source tree.
|
||
|
|
*)
|
||
|
|
|
||
|
|
include CCHashSet
|
||
|
|
|
||
|
|
module Make (E : ELEMENT) = struct
|
||
|
|
include CCHashSet.Make [@inlined] (E)
|
||
|
|
|
||
|
|
let update s e ~f =
|
||
|
|
let eo = find s e in
|
||
|
|
match (eo, f eo) with
|
||
|
|
| None, None -> ()
|
||
|
|
| Some e, Some e' when e == e' -> ()
|
||
|
|
| Some _, None -> remove s e
|
||
|
|
| _, Some e' -> insert s e'
|
||
|
|
|
||
|
|
let add s e =
|
||
|
|
let change = ref false in
|
||
|
|
update s e ~f:(function
|
||
|
|
| None ->
|
||
|
|
change := true ;
|
||
|
|
Some e
|
||
|
|
| Some e -> Some e ) ;
|
||
|
|
!change
|
||
|
|
end
|