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

(*
* 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